gnc-import-tx.hpp

Go to the documentation of this file.

/********************************************************************\
 * gnc-import-tx.hpp - import transactions from csv files           *
 *                                                                  *
 * This program is free software; you can redistribute it and/or    *
 * modify it under the terms of the GNU General Public License as   *
 * published by the Free Software Foundation; either version 2 of   *
 * the License, or (at your option) any later version.              *
 *                                                                  *
 * This program is distributed in the hope that it will be useful,  *
 * but WITHOUT ANY WARRANTY; without even the implied warranty of   *
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the    *
 * GNU General Public License for more details.                     *
 *                                                                  *
 * You should have received a copy of the GNU General Public License*
 * along with this program; if not, contact:                        *
 *                                                                  *
 * Free Software Foundation           Voice:  +1-617-542-5942       *
 * 51 Franklin Street, Fifth Floor    Fax:    +1-617-542-2652       *
 * Boston, MA  02110-1301,  USA       gnu@gnu.org                   *
\********************************************************************/
 
#ifndef GNC_TX_IMPORT_HPP
#define GNC_TX_IMPORT_HPP
 
#include <config.h>
 
#include "Account.h"
#include "Transaction.h"
 
#include <vector>
#include <set>
#include <map>
#include <memory>
#include <optional>
#include <cstdint>
 
#include "gnc-tokenizer.hpp"
#include "gnc-imp-props-tx.hpp"
#include "gnc-imp-settings-csv-tx.hpp"
 
 
/* A set of currency formats that the user sees. */
extern const int num_currency_formats;
extern const gchar* currency_format_user[];
 
enum parse_line_cols {
    PL_INPUT,
    PL_ERROR,
    PL_PRETRANS,
    PL_PRESPLIT,
    PL_SKIP
};
 
using StrVec = std::vector<std::string>;
 
using parse_line_t = std::tuple<StrVec,
                                ErrMap,
                                std::shared_ptr<GncPreTrans>,
                                std::shared_ptr<GncPreSplit>,
                                bool>;
 
struct ErrorList;
 
struct GncCsvImpParseError : public std::runtime_error
{
    GncCsvImpParseError(const std::string& err, ErrMap err_vec) : std::runtime_error(err), m_errors{err_vec} {}
    ErrMap errors() const {return m_errors;}
 
private:
    ErrMap m_errors;
};
 
class GncTxImport
{
public:
    // Constructor - Destructor
    GncTxImport(GncImpFileFormat format = GncImpFileFormat::UNKNOWN);
    ~GncTxImport();
 
    void file_format(GncImpFileFormat format);
    GncImpFileFormat file_format();
 
    void multi_split (bool multi_split);
    bool multi_split ();
 
    void base_account (Account *base_account);
    Account *base_account ();
 
    void currency_format (int currency_format);
    int currency_format ();
 
    void date_format (int date_format);
    int date_format ();
 
    void encoding (const std::string& encoding);
    std::string encoding ();
 
    void update_skipped_lines (std::optional<uint32_t> start, std::optional<uint32_t> end,
                               std::optional<bool> alt, std::optional<bool> errors);
    uint32_t skip_start_lines ();
    uint32_t skip_end_lines ();
    bool skip_alt_lines ();
    bool skip_err_lines ();
 
    void separators (std::string separators);
    std::string separators ();
 
    void enable_escape (bool enable);
    bool enable_escape ();
 
    void settings (const CsvTransImpSettings& settings);
    bool save_settings ();
 
    void settings_name (std::string name);
    std::string settings_name ();
 
 
    void load_file (const std::string& filename);
 
    void tokenize (bool guessColTypes);
 
    std::string verify(bool with_acct_errors);
 
    void create_transactions ();
    bool check_for_column_type (GncTransPropType type);
    void set_column_type (uint32_t position, GncTransPropType type, bool force = false);
    std::vector<GncTransPropType> column_types ();
 
    std::set<std::string> accounts ();
 
    std::unique_ptr<GncTokenizer> m_tokenizer;    
    std::vector<parse_line_t> m_parsed_lines;     
    std::multimap <time64, std::shared_ptr<DraftTransaction>> m_transactions;  
private:
    void create_transaction (std::vector<parse_line_t>::iterator& parsed_line);
 
    void verify_column_selections (ErrorList& error_msg);
 
    /* Internal helper function to force reparsing of columns subject to format changes */
    void reset_formatted_column (std::vector<GncTransPropType>& col_types);
 
    /* Internal helper function that does the actual conversion from property lists
     * to real (possibly unbalanced) transaction with splits.
     */
    std::shared_ptr<DraftTransaction> trans_properties_to_trans (std::vector<parse_line_t>::iterator& parsed_line);
 
    /* Internal helper functions that should only be called from within
     * set_column_type for consistency (otherwise error messages may not be (re)set)
     */
    void update_pre_split_multi_col_prop (parse_line_t& parsed_line, GncTransPropType col_type);
    void update_pre_trans_props (parse_line_t& parsed_line, uint32_t col, GncTransPropType old_type, GncTransPropType new_type);
    void update_pre_split_props (parse_line_t& parsed_line, uint32_t col, GncTransPropType old_type, GncTransPropType new_type);
 
    CsvTransImpSettings m_settings;
    bool m_skip_errors;
    /* Field used internally to track whether some transactions are multi-currency */
    bool m_multi_currency;
 
    /* The parameters below are only used while creating
     * transactions. They keep state information while processing multi-split
     * transactions.
     */
    std::shared_ptr<GncPreTrans> m_parent = nullptr;
    std::shared_ptr<DraftTransaction> m_current_draft = nullptr;
};
 
 
#endif