diff options
Diffstat (limited to 'main.cc')
-rw-r--r-- | main.cc | 424 |
1 files changed, 424 insertions, 0 deletions
diff --git a/main.cc b/main.cc new file mode 100644 index 00000000..c70fcff9 --- /dev/null +++ b/main.cc @@ -0,0 +1,424 @@ +#include <ledger.h> +#include "acconf.h" +#include "debug.h" +#ifdef USE_BOOST_PYTHON +#include "py_eval.h" +#endif + +using namespace ledger; + +#include <iostream> +#include <fstream> +#include <sstream> +#include <memory> +#include <algorithm> +#include <exception> +#include <iterator> +#include <string> +#include <cstdlib> +#include <cstring> +#include <ctime> + +#if !defined(DEBUG_LEVEL) || DEBUG_LEVEL <= RELEASE + +#define auto_ptr bogus_auto_ptr + +// This version of auto_ptr does not delete on deconstruction. +namespace std { + template <typename T> + struct bogus_auto_ptr { + T * ptr; + bogus_auto_ptr() : ptr(NULL) {} + explicit bogus_auto_ptr(T * _ptr) : ptr(_ptr) {} + T& operator*() const throw() { + return *ptr; + } + T * operator->() const throw() { + return ptr; + } + T * get() const throw() { return ptr; } + T * release() throw() { + T * tmp = ptr; + ptr = 0; + return tmp; + } + void reset(T * p = 0) throw() { + if (p != ptr) { + delete ptr; + ptr = p; + } + } + }; +} + +#endif + +item_handler<transaction_t> * +chain_xact_handlers(const std::string& command, + item_handler<transaction_t> * base_formatter, + journal_t * journal, + account_t * master, + std::list<item_handler<transaction_t> *>& ptrs) +{ + item_handler<transaction_t> * formatter = NULL; + + ptrs.push_back(formatter = base_formatter); + + // format_transactions write each transaction received to the + // output stream. + if (! (command == "b" || command == "E")) { + // filter_transactions will only pass through transactions + // matching the `display_predicate'. + if (! config.display_predicate.empty()) + ptrs.push_back(formatter = + new filter_transactions(formatter, + config.display_predicate)); + + // calc_transactions computes the running total. When this + // appears will determine, for example, whether filtered + // transactions are included or excluded from the running total. + ptrs.push_back(formatter = new calc_transactions(formatter)); + + // sort_transactions will sort all the transactions it sees, based + // on the `sort_order' value expression. + if (! config.sort_string.empty()) + ptrs.push_back(formatter = + new sort_transactions(formatter, config.sort_string)); + + // changed_value_transactions adds virtual transactions to the + // list to account for changes in market value of commodities, + // which otherwise would affect the running total unpredictably. + if (config.show_revalued) + ptrs.push_back(formatter = + new changed_value_transactions(formatter, + config.show_revalued_only)); + + // collapse_transactions causes entries with multiple transactions + // to appear as entries with a subtotaled transaction for each + // commodity used. + if (config.show_collapsed) + ptrs.push_back(formatter = new collapse_transactions(formatter)); + } + + // subtotal_transactions combines all the transactions it receives + // into one subtotal entry, which has one transaction for each + // commodity in each account. + // + // period_transactions is like subtotal_transactions, but it + // subtotals according to time periods rather than totalling + // everything. + // + // dow_transactions is like period_transactions, except that it + // reports all the transactions that fall on each subsequent day + // of the week. + if (config.show_subtotal && ! (command == "b" || command == "E")) + ptrs.push_back(formatter = new subtotal_transactions(formatter)); + + if (config.days_of_the_week) + ptrs.push_back(formatter = new dow_transactions(formatter)); + else if (config.by_payee) + ptrs.push_back(formatter = new by_payee_transactions(formatter)); + + if (! config.report_period.empty()) { + ptrs.push_back(formatter = + new interval_transactions(formatter, + config.report_period, + config.report_period_sort)); + ptrs.push_back(formatter = new sort_transactions(formatter, "d")); + } + + // invert_transactions inverts the value of the transactions it + // receives. + if (config.show_inverted) + ptrs.push_back(formatter = new invert_transactions(formatter)); + + // related_transactions will pass along all transactions related + // to the transaction received. If `show_all_related' is true, + // then all the entry's transactions are passed; meaning that if + // one transaction of an entry is to be printed, all the + // transaction for that entry will be printed. + if (config.show_related) + ptrs.push_back(formatter = + new related_transactions(formatter, + config.show_all_related)); + + // This filter_transactions will only pass through transactions + // matching the `predicate'. + if (! config.predicate.empty()) + ptrs.push_back(formatter = new filter_transactions(formatter, + config.predicate)); + + // budget_transactions takes a set of transactions from a data + // file and uses them to generate "budget transactions" which + // balance against the reported transactions. + // + // forecast_transactions is a lot like budget_transactions, except + // that it adds entries only for the future, and does not balance + // them against anything but the future balance. + + if (config.budget_flags) { + budget_transactions * handler + = new budget_transactions(formatter, config.budget_flags); + handler->add_period_entries(journal->period_entries); + ptrs.push_back(formatter = handler); + } + else if (! config.forecast_limit.empty()) { + forecast_transactions * handler + = new forecast_transactions(formatter, config.forecast_limit); + handler->add_period_entries(journal->period_entries); + ptrs.push_back(formatter = handler); + } + + if (config.comm_as_payee) + ptrs.push_back(formatter = new set_comm_as_payee(formatter)); + + return formatter; +} + +int parse_and_report(int argc, char * argv[], char * envp[]) +{ + std::auto_ptr<journal_t> journal(new journal_t); + + // Parse command-line arguments, and those set in the environment + + std::list<std::string> args; + process_arguments(config_options, argc - 1, argv + 1, false, args); + + if (args.empty()) { + option_help(std::cerr); + return 1; + } + strings_list::iterator arg = args.begin(); + + config.use_cache = config.data_file.empty() && config.price_db.empty(); + DEBUG_PRINT("ledger.config.cache", "1. use_cache = " << config.use_cache); + + process_environment(config_options, envp, "LEDGER_"); + +#if 1 + // These are here for backwards compatability, but are deprecated. + + if (const char * p = std::getenv("LEDGER")) + process_option(config_options, "file", p); + if (const char * p = std::getenv("LEDGER_INIT")) + process_option(config_options, "init-file", p); + if (const char * p = std::getenv("PRICE_HIST")) + process_option(config_options, "price-db", p); + if (const char * p = std::getenv("PRICE_EXP")) + process_option(config_options, "price-exp", p); +#endif + + const char * p = std::getenv("HOME"); + std::string home = p ? p : ""; + + if (config.init_file.empty()) + config.init_file = home + "/.ledgerrc"; + if (config.cache_file.empty()) + config.cache_file = home + "/.ledger-cache"; + if (config.price_db.empty()) + config.price_db = home + "/.pricedb"; + + if (config.data_file == config.cache_file) + config.use_cache = false; + DEBUG_PRINT("ledger.config.cache", "2. use_cache = " << config.use_cache); + + // Read the command word, canonicalize it to its one letter form, + // then configure the system based on the kind of report to be + // generated + + std::string command = *arg++; + + if (command == "balance" || command == "bal" || command == "b") + command = "b"; + else if (command == "register" || command == "reg" || command == "r") + command = "r"; + else if (command == "print" || command == "p") + command = "p"; + else if (command == "xml") + command = "X"; + else if (command == "entry") + command = "e"; + else if (command == "equity") + command = "E"; + else if (command == "prices") + command = "P"; + else + throw error(std::string("Unrecognized command '") + command + "'"); + + // Parse initialization files, ledger data, price database, etc. + + std::auto_ptr<binary_parser_t> bin_parser(new binary_parser_t); +#ifdef HAVE_XMLPARSE + std::auto_ptr<xml_parser_t> xml_parser(new xml_parser_t); + std::auto_ptr<gnucash_parser_t> gnucash_parser(new gnucash_parser_t); +#endif + std::auto_ptr<qif_parser_t> qif_parser(new qif_parser_t); + std::auto_ptr<textual_parser_t> text_parser(new textual_parser_t); + + register_parser(bin_parser.get()); +#ifdef HAVE_XMLPARSE + register_parser(xml_parser.get()); + register_parser(gnucash_parser.get()); +#endif + register_parser(qif_parser.get()); + register_parser(text_parser.get()); + + parse_ledger_data(journal.get(), bin_parser.get(), text_parser.get() +#ifdef HAVE_XMLPARSE + , xml_parser.get() +#endif + ); + + // process the command word and its following arguments + + config.process_options(command, arg, args.end()); + + std::auto_ptr<entry_t> new_entry; + if (command == "e") { + new_entry.reset(derive_new_entry(*journal, arg, args.end())); + if (! new_entry.get()) + return 1; + } + + // Configure the output stream + + std::ostream * out = &std::cout; + if (! config.output_file.empty()) + out = new std::ofstream(config.output_file.c_str()); + + // Compile the format strings + + const std::string * format; + if (! config.format_string.empty()) + format = &config.format_string; + else if (command == "b") + format = &config.balance_format; + else if (command == "r") + format = &config.register_format; + else if (command == "E") + format = &config.equity_format; + else if (command == "P") + format = &config.prices_format; + else + format = &config.print_format; + +#ifdef USE_BOOST_PYTHON + + // If Python support is compiled, we can easily report minimum and + // maximum values for each commodity. There is a line in config.cc + // which configures the prices report to call these two functions, + // if Python is available. + + if (command == "P") + python_eval("\ +min_val = 0\n\ +def vmin(d, val):\n\ + global min_val\n\ + if not min_val or val < min_val:\n\ + min_val = val\n\ + return val\n\ + return min_val\n\ +\n\ +max_val = 0\n\ +def vmax(d, val):\n\ + global max_val\n\ + if not max_val or val > max_val:\n\ + max_val = val\n\ + return val\n\ + return max_val\n", PY_EVAL_MULTI); + +#endif // USE_BOOST_PYTHON + + // Walk the entries based on the report type and the options + + item_handler<transaction_t> * formatter; + std::list<item_handler<transaction_t> *> formatter_ptrs; + + if (command == "b" || command == "E") + formatter = new set_account_value; + else if (command == "p" || command == "e") + formatter = new format_entries(*out, *format); + else if (command == "X") + formatter = new format_xml_entries(*out, config.show_totals); + else + formatter = new format_transactions(*out, *format); + + formatter = chain_xact_handlers(command, formatter, journal.get(), + journal->master, formatter_ptrs); + + if (command == "e") + walk_transactions(new_entry->transactions, *formatter); + else if (command == "P") + walk_commodities(commodity_t::commodities, *formatter); + else + walk_entries(journal->entries, *formatter); + + if (command != "P") + formatter->flush(); + + // For the balance and equity reports, output the sum totals. + + if (command == "b") { + format_account acct_formatter(*out, *format, config.display_predicate); + sum_accounts(*journal->master); + walk_accounts(*journal->master, acct_formatter, config.sort_string); + acct_formatter.flush(); + + if (account_has_xdata(*journal->master)) { + account_xdata_t& xdata = account_xdata(*journal->master); + if (! config.show_collapsed && xdata.total) { + *out << "--------------------\n"; + xdata.value = xdata.total; + acct_formatter.format.format(*out, details_t(*journal->master)); + } + } + } + else if (command == "E") { + format_equity acct_formatter(*out, *format, config.display_predicate); + sum_accounts(*journal->master); + walk_accounts(*journal->master, acct_formatter, config.sort_string); + acct_formatter.flush(); + } + +#if DEBUG_LEVEL >= BETA + + clear_transactions_xdata(); + clear_accounts_xdata(); + + if (! config.output_file.empty()) + delete out; + + for (std::list<item_handler<transaction_t> *>::iterator i + = formatter_ptrs.begin(); + i != formatter_ptrs.end(); + i++) + delete *i; + +#endif // DEBUG_LEVEL >= BETA + + // Write out the binary cache, if need be + + if (config.use_cache && config.cache_dirty && + ! config.cache_file.empty()) { + std::ofstream stream(config.cache_file.c_str()); + write_binary_journal(stream, journal.get(), &journal->sources); + } + + return 0; +} + +int main(int argc, char * argv[], char * envp[]) +{ + try { + return parse_and_report(argc, argv, envp); + } + catch (const std::exception& err) { + std::cerr << "Error: " << err.what() << std::endl; + return 1; + } + catch (int& val) { + return val; // this acts like a std::setjmp + } +} + +// main.cc ends here. |