diff options
author | John Wiegley <johnw@newartisans.com> | 2009-02-05 04:20:49 -0400 |
---|---|---|
committer | John Wiegley <johnw@newartisans.com> | 2009-02-05 04:20:49 -0400 |
commit | edb9cff4a2c803bde01f75221aed0b7ecfc3b3bf (patch) | |
tree | edda6471ea8bd3f4d8b3d211dba1642c607d88ea /src | |
parent | 87cfef221a189deebc79a606800585f75ab05395 (diff) | |
download | fork-ledger-edb9cff4a2c803bde01f75221aed0b7ecfc3b3bf.tar.gz fork-ledger-edb9cff4a2c803bde01f75221aed0b7ecfc3b3bf.tar.bz2 fork-ledger-edb9cff4a2c803bde01f75221aed0b7ecfc3b3bf.zip |
Moved work.cc and global_scope_t into a new global.cc file.
Diffstat (limited to 'src')
-rw-r--r-- | src/global.cc | 410 | ||||
-rw-r--r-- | src/global.h (renamed from src/work.h) | 75 | ||||
-rw-r--r-- | src/main.cc | 274 | ||||
-rw-r--r-- | src/work.cc | 187 |
4 files changed, 480 insertions, 466 deletions
diff --git a/src/global.cc b/src/global.cc new file mode 100644 index 00000000..9dff9aac --- /dev/null +++ b/src/global.cc @@ -0,0 +1,410 @@ +/* + * Copyright (c) 2003-2009, John Wiegley. All rights reserved. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions are + * met: + * + * - Redistributions of source code must retain the above copyright + * notice, this list of conditions and the following disclaimer. + * + * - Redistributions in binary form must reproduce the above copyright + * notice, this list of conditions and the following disclaimer in the + * documentation and/or other materials provided with the distribution. + * + * - Neither the name of New Artisans LLC nor the names of its + * contributors may be used to endorse or promote products derived from + * this software without specific prior written permission. + * + * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS + * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT + * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR + * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT + * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, + * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT + * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, + * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY + * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT + * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE + * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + */ + +#include <ledger.h> + +#include "global.h" + +namespace ledger { + +global_scope_t::global_scope_t(char ** envp) +{ + TRACE_CTOR(global_scope_t, ""); + + session_ptr.reset(new LEDGER_SESSION_T); + + set_session_context(session_ptr.get()); + + // Create the report object, which maintains state relating to each + // command invocation. Because we're running from main(), the + // distinction between session and report doesn't really matter, but if + // a GUI were calling into Ledger it would have one session object per + // open document, with a separate report_t object for each report it + // generated. + report_stack.push_front(new report_t(session())); + + // Read the user's options, in the following order: + // + // 1. environment variables (LEDGER_<option>) + // 2. initialization file (~/.ledgerrc) + // 3. command-line (--option or -o) + // + // Before processing command-line options, we must notify the session object + // that such options are beginning, since options like -f cause a complete + // override of files found anywhere else. + session().now_at_command_line(false); + read_environment_settings(report(), envp); + session().read_init(); +} + +global_scope_t::~global_scope_t() +{ + TRACE_DTOR(global_scope_t); + + // If memory verification is being performed (which can be very slow), + // clean up everything by closing the session and deleting the session + // object, and then shutting down the memory tracing subsystem. + // Otherwise, let it all leak because we're about to exit anyway. + IF_VERIFY() set_session_context(NULL); +} + +void global_scope_t::read_journal_files() +{ + INFO_START(journal, "Read journal file"); + + std::size_t count = session().read_data(*session().create_journal(), + report().account); + if (count == 0) + throw_(parse_error, "Failed to locate any journal entries; " + "did you specify a valid file with -f?"); + + INFO_FINISH(journal); + + INFO("Found " << count << " entries"); +} + +char * global_scope_t::prompt_string() +{ + static char prompt[32]; + std::size_t i; + for (i = 0; i < report_stack.size(); i++) + prompt[i] = ']'; + prompt[i++] = ' '; + prompt[i] = '\0'; + return prompt; +} + +void global_scope_t::report_error(const std::exception& err) +{ + std::cout.flush(); // first display anything that was pending + + if (caught_signal == NONE_CAUGHT) { + // Display any pending error context information + string context = error_context(); + if (! context.empty()) + std::cerr << context << std::endl; + + std::cerr << "Error: " << err.what() << std::endl; + } else { + caught_signal = NONE_CAUGHT; + } +} + +void global_scope_t::execute_command(strings_list args, bool at_repl) +{ + // Process the command verb, arguments and options + args = read_command_arguments(report(), args); + if (args.empty()) + return; + + strings_list::iterator arg = args.begin(); + string verb = *arg++; + + // Look for a precommand first, which is defined as any defined function + // whose name starts with "ledger_precmd_". The difference between a + // precommand and a regular command is that precommands ignore the journal + // data file completely, nor is the user's init file read. + // + // Here are some examples of pre-commands: + // + // parse STRING ; show how a value expression is parsed + // eval STRING ; simply evaluate a value expression + // format STRING ; show how a format string is parsed + // + // If such a command is found, create the output stream for the result and + // then invoke the command. + + function_t command; + bool is_precommand = false; + bind_scope_t bound_scope(*this, report()); + + if (bool(command = look_for_precommand(bound_scope, verb))) + is_precommand = true; + else if (! bool(command = look_for_command(bound_scope, verb))) + throw_(std::logic_error, "Unrecognized command '" << verb << "'"); + + // If it is not a pre-command, then parse the user's ledger data at this + // time if not done alreday (i.e., if not at a REPL). Then patch up the + // report options based on the command verb. + + if (! is_precommand) { + if (! at_repl) + read_journal_files(); + + // jww (2009-02-02): This is a complete hack, and a leftover from long, + // long ago. The question is, how best to remove its necessity... + normalize_report_options(report(), verb); + } + + // Create the output stream (it might be a file, the console or a PAGER + // subprocess) and invoke the report command. The output stream is closed + // by the caller of this function. + + report().output_stream.initialize(report().output_file, + session().pager_path); + + // Create an argument scope containing the report command's arguments, and + // then invoke the command. The bound scope causes lookups to happen + // first in the global scope, and then in the report scope. + + call_scope_t command_args(bound_scope); + for (strings_list::iterator i = arg; i != args.end(); i++) + command_args.push_back(string_value(*i)); + + INFO_START(command, "Finished executing command"); + command(command_args); + INFO_FINISH(command); +} + +int global_scope_t::execute_command_wrapper(strings_list args, bool at_repl) +{ + int status = 1; + + try { + push_report(); + execute_command(args, at_repl); + pop_report(); + + // If we've reached this point, everything succeeded fine. Ledger uses + // exceptions to notify of error conditions, so if you're using gdb, + // just type "catch throw" to find the source point of any error. + status = 0; + } + catch (const std::exception& err) { + pop_report(); + report_error(err); + } + + return status; +} + +expr_t::ptr_op_t global_scope_t::lookup(const string& name) +{ + const char * p = name.c_str(); + switch (*p) { + case 'l': + if (std::strncmp(p, "ledger_precmd_", 14) == 0) { + p = p + 14; + switch (*p) { + case 'p': + if (std::strcmp(p, "push") == 0) + return MAKE_FUNCTOR(global_scope_t::push_report_cmd); + else if (std::strcmp(p, "pop") == 0) + return MAKE_FUNCTOR(global_scope_t::pop_report_cmd); + break; + } + } + break; + + case 'o': + if (std::strncmp(p, "opt_", 4) == 0) { + p = p + 4; + switch (*p) { + case 'd': + if (std::strcmp(p, "debug_") == 0) + return MAKE_FUNCTOR(global_scope_t::ignore); + break; + + case 's': + if (std::strcmp(p, "script_") == 0) + return MAKE_FUNCTOR(global_scope_t::option_script_); + break; + + case 't': + if (std::strcmp(p, "trace_") == 0) + return MAKE_FUNCTOR(global_scope_t::ignore); + break; + + case 'v': + if (! *(p + 1) || std::strcmp(p, "verbose") == 0) + return MAKE_FUNCTOR(global_scope_t::ignore); + else if (std::strcmp(p, "verify") == 0) + return MAKE_FUNCTOR(global_scope_t::ignore); + break; + } + } + } + + // If you're wondering how symbols from report() will be found, it's + // because of the bind_scope_t object in execute_command() below. + return expr_t::ptr_op_t(); +} + +void handle_debug_options(int argc, char * argv[]) +{ + for (int i = 1; i < argc; i++) { + if (argv[i][0] == '-') { + if (std::strcmp(argv[i], "--verify") == 0) { +#if defined(VERIFY_ON) + verify_enabled = true; // global in utils.h +#endif + } + else if (std::strcmp(argv[i], "--verbose") == 0 || + std::strcmp(argv[i], "-v") == 0) { +#if defined(LOGGING_ON) + _log_level = LOG_INFO; // global in utils.h +#endif + } + else if (i + 1 < argc && std::strcmp(argv[i], "--debug") == 0) { +#if defined(DEBUG_ON) + _log_level = LOG_DEBUG; // global in utils.h + _log_category = argv[i + 1]; // global in utils.h + i++; +#endif + } + else if (i + 1 < argc && std::strcmp(argv[i], "--trace") == 0) { +#if defined(TRACING_ON) + _log_level = LOG_TRACE; // global in utils.h + try { + // global in utils.h + _trace_level = boost::lexical_cast<int>(argv[i + 1]); + } + catch (const boost::bad_lexical_cast& e) { + throw std::logic_error("Argument to --trace must be an integer"); + } + i++; +#endif + } + } + } +} + +void read_environment_settings(report_t& report, char * envp[]) +{ + TRACE_START(environment, 1, "Processed environment variables"); + + process_environment(const_cast<const char **>(envp), "LEDGER_", + report); + +#if 1 + // These are here for backwards compatability, but are deprecated. + + if (const char * p = std::getenv("LEDGER")) + process_option("file", report, p, "LEDGER"); + if (const char * p = std::getenv("LEDGER_INIT")) + process_option("init-file", report, p, "LEDGER_INIT"); + if (const char * p = std::getenv("PRICE_HIST")) + process_option("price-db", report, p, "PRICE_HIST"); + if (const char * p = std::getenv("PRICE_EXP")) + process_option("price-exp", report, p, "PRICE_EXP"); +#endif + + TRACE_FINISH(environment, 1); +} + +strings_list read_command_arguments(scope_t& scope, strings_list args) +{ + TRACE_START(arguments, 1, "Processed command-line arguments"); + + strings_list remaining = process_arguments(args, scope); + + TRACE_FINISH(arguments, 1); + + return remaining; +} + +void normalize_session_options(session_t& session) +{ + INFO("Initialization file is " << session.init_file->string()); + INFO("Price database is " << session.price_db->string()); + + foreach (const path& pathname, session.data_files) + INFO("Journal file is " << pathname.string()); +} + +function_t look_for_precommand(scope_t& scope, const string& verb) +{ + if (expr_t::ptr_op_t def = scope.lookup(string("ledger_precmd_") + verb)) + return def->as_function(); + else + return function_t(); +} + +function_t look_for_command(scope_t& scope, const string& verb) +{ + if (expr_t::ptr_op_t def = scope.lookup(string("ledger_cmd_") + verb)) + return def->as_function(); + else + return function_t(); +} + +void normalize_report_options(report_t& report, const string& verb) +{ + // Patch up some of the reporting options based on what kind of + // command it was. + + // jww (2008-08-14): This code really needs to be rationalized away + // for 3.0. + + if (verb == "print" || verb == "entry" || verb == "dump") { + report.show_related = true; + report.show_all_related = true; + } + else if (verb == "equity") { + report.show_subtotal = true; + } + else if (report.show_related) { + if (verb[0] == 'r') { + report.show_inverted = true; + } else { + report.show_subtotal = true; + report.show_all_related = true; + } + } + + if (verb[0] != 'b' && verb[0] != 'r') + report.what_to_keep.keep_base = true; + + // Setup the default value for the display predicate + + if (report.display_predicate.empty()) { + if (verb[0] == 'b') { + if (! report.show_empty) + report.display_predicate = "total"; + if (! report.show_subtotal) { + if (! report.display_predicate.empty()) + report.display_predicate += "&"; + report.display_predicate += "depth<=1"; + } + } + else if (verb == "equity") { + report.display_predicate = "amount_expr"; // jww (2008-08-14): ??? + } + else if (verb[0] == 'r' && ! report.show_empty) { + report.display_predicate = "amount"; + } + } + + if (! report.report_period.empty() && ! report.sort_all) + report.entry_sort = true; +} + +} // namespace ledger diff --git a/src/work.h b/src/global.h index e52093ef..f47967fb 100644 --- a/src/work.h +++ b/src/global.h @@ -30,18 +30,81 @@ */ /** - * @file work.h + * @file global.h * @author John Wiegley * * @brief Contains the top-level functions used by main.cc */ -#ifndef _WORK_H -#define _WORK_H +#ifndef _GLOBAL_H +#define _GLOBAL_H + +#include "report.h" namespace ledger { -typedef strings_list::iterator string_iterator; -typedef std::pair<string_iterator, string_iterator> string_iterator_pair; +class global_scope_t : public noncopyable, public scope_t +{ + scoped_ptr<session_t> session_ptr; + ptr_list<report_t> report_stack; + +public: + path script_file; + + global_scope_t(char ** envp); + ~global_scope_t(); + + void read_journal_files(); + + char * prompt_string(); + + session_t& session() { + return *session_ptr.get(); + } + report_t& report() { + return report_stack.front(); + } + + void push_report() { + report_stack.push_front(new report_t(report_stack.front())); + } + void pop_report() { + if (! report_stack.empty()) + report_stack.pop_front(); + } + + void report_error(const std::exception& err); + + /** + * @return \c true if a command was actually executed; otherwise, it probably + * just resulted in setting some options. + */ + void execute_command(strings_list args, bool at_repl); + int execute_command_wrapper(strings_list args, bool at_repl); + + value_t push_report_cmd(call_scope_t&) { + // Make a copy at position 2, because the topmost report object has an + // open output stream at this point. We want it to get popped off as + // soon as this command terminate so that the stream is closed cleanly. + report_stack.insert(++report_stack.begin(), + new report_t(report_stack.front())); + return true; + } + value_t pop_report_cmd(call_scope_t&) { + pop_report(); + return true; + } + + value_t option_script_(call_scope_t& args) { + script_file = args[0].as_string(); + return true; + } + + value_t ignore(call_scope_t&) { + return true; + } + + virtual expr_t::ptr_op_t lookup(const string& name); +}; void handle_debug_options(int argc, char * argv[]); void read_environment_settings(report_t& report, char * envp[]); @@ -53,4 +116,4 @@ void normalize_report_options(report_t& report, const string& verb); } // namespace ledger -#endif // _WORK_H +#endif // _GLOBAL_H diff --git a/src/main.cc b/src/main.cc index 061d6b2d..c8192082 100644 --- a/src/main.cc +++ b/src/main.cc @@ -31,283 +31,11 @@ #include <ledger.h> // Read this file for a top-level overview -#include "work.h" // This is where the meat of main() is, which +#include "global.h" // This is where the meat of main() is, which // was moved there for the sake of clarity using namespace ledger; namespace { - class global_scope_t : public noncopyable, public scope_t - { - scoped_ptr<session_t> session_ptr; - ptr_list<report_t> report_stack; - - public: - path script_file; - - global_scope_t(char ** envp) - { - TRACE_CTOR(global_scope_t, ""); - - session_ptr.reset(new LEDGER_SESSION_T); - - set_session_context(session_ptr.get()); - - // Create the report object, which maintains state relating to each - // command invocation. Because we're running from main(), the - // distinction between session and report doesn't really matter, but if - // a GUI were calling into Ledger it would have one session object per - // open document, with a separate report_t object for each report it - // generated. - report_stack.push_front(new report_t(session())); - - // Read the user's options, in the following order: - // - // 1. environment variables (LEDGER_<option>) - // 2. initialization file (~/.ledgerrc) - // 3. command-line (--option or -o) - // - // Before processing command-line options, we must notify the session object - // that such options are beginning, since options like -f cause a complete - // override of files found anywhere else. - session().now_at_command_line(false); - read_environment_settings(report(), envp); - session().read_init(); - } - ~global_scope_t() - { - TRACE_DTOR(global_scope_t); - - // If memory verification is being performed (which can be very slow), - // clean up everything by closing the session and deleting the session - // object, and then shutting down the memory tracing subsystem. - // Otherwise, let it all leak because we're about to exit anyway. - IF_VERIFY() set_session_context(NULL); - } - - void read_journal_files() - { - INFO_START(journal, "Read journal file"); - - std::size_t count = session().read_data(*session().create_journal(), - report().account); - if (count == 0) - throw_(parse_error, "Failed to locate any journal entries; " - "did you specify a valid file with -f?"); - - INFO_FINISH(journal); - - INFO("Found " << count << " entries"); - } - - char * prompt_string() - { - static char prompt[32]; - std::size_t i; - for (i = 0; i < report_stack.size(); i++) - prompt[i] = ']'; - prompt[i++] = ' '; - prompt[i] = '\0'; - return prompt; - } - - session_t& session() { - return *session_ptr.get(); - } - report_t& report() { - return report_stack.front(); - } - - void push_report() { - report_stack.push_front(new report_t(report_stack.front())); - } - void pop_report() { - if (! report_stack.empty()) - report_stack.pop_front(); - } - - void report_error(const std::exception& err) - { - std::cout.flush(); // first display anything that was pending - - if (caught_signal == NONE_CAUGHT) { - // Display any pending error context information - string context = error_context(); - if (! context.empty()) - std::cerr << context << std::endl; - - std::cerr << "Error: " << err.what() << std::endl; - } else { - caught_signal = NONE_CAUGHT; - } - } - - /** - * @return \c true if a command was actually executed; otherwise, it probably - * just resulted in setting some options. - */ - void execute_command(strings_list args, bool at_repl) - { - // Process the command verb, arguments and options - args = read_command_arguments(report(), args); - if (args.empty()) - return; - - string_iterator arg = args.begin(); - string verb = *arg++; - - // Look for a precommand first, which is defined as any defined function - // whose name starts with "ledger_precmd_". The difference between a - // precommand and a regular command is that precommands ignore the journal - // data file completely, nor is the user's init file read. - // - // Here are some examples of pre-commands: - // - // parse STRING ; show how a value expression is parsed - // eval STRING ; simply evaluate a value expression - // format STRING ; show how a format string is parsed - // - // If such a command is found, create the output stream for the result and - // then invoke the command. - - function_t command; - bool is_precommand = false; - bind_scope_t bound_scope(*this, report()); - - if (bool(command = look_for_precommand(bound_scope, verb))) - is_precommand = true; - else if (! bool(command = look_for_command(bound_scope, verb))) - throw_(std::logic_error, "Unrecognized command '" << verb << "'"); - - // If it is not a pre-command, then parse the user's ledger data at this - // time if not done alreday (i.e., if not at a REPL). Then patch up the - // report options based on the command verb. - - if (! is_precommand) { - if (! at_repl) - read_journal_files(); - - // jww (2009-02-02): This is a complete hack, and a leftover from long, - // long ago. The question is, how best to remove its necessity... - normalize_report_options(report(), verb); - } - - // Create the output stream (it might be a file, the console or a PAGER - // subprocess) and invoke the report command. The output stream is closed - // by the caller of this function. - - report().output_stream.initialize(report().output_file, - session().pager_path); - - // Create an argument scope containing the report command's arguments, and - // then invoke the command. The bound scope causes lookups to happen - // first in the global scope, and then in the report scope. - - call_scope_t command_args(bound_scope); - for (string_iterator i = arg; i != args.end(); i++) - command_args.push_back(string_value(*i)); - - INFO_START(command, "Finished executing command"); - command(command_args); - INFO_FINISH(command); - } - - int execute_command_wrapper(strings_list args, bool at_repl) - { - int status = 1; - - try { - push_report(); - execute_command(args, at_repl); - pop_report(); - - // If we've reached this point, everything succeeded fine. Ledger uses - // exceptions to notify of error conditions, so if you're using gdb, - // just type "catch throw" to find the source point of any error. - status = 0; - } - catch (const std::exception& err) { - pop_report(); - report_error(err); - } - - return status; - } - - value_t push_report_cmd(call_scope_t&) { - // Make a copy at position 2, because the topmost report object has an - // open output stream at this point. We want it to get popped off as - // soon as this command terminate so that the stream is closed cleanly. - report_stack.insert(++report_stack.begin(), - new report_t(report_stack.front())); - return true; - } - value_t pop_report_cmd(call_scope_t&) { - pop_report(); - return true; - } - - value_t option_script_(call_scope_t& args) { - script_file = args[0].as_string(); - return true; - } - - value_t ignore(call_scope_t&) { - return true; - } - - virtual expr_t::ptr_op_t lookup(const string& name) - { - const char * p = name.c_str(); - switch (*p) { - case 'l': - if (std::strncmp(p, "ledger_precmd_", 14) == 0) { - p = p + 14; - switch (*p) { - case 'p': - if (std::strcmp(p, "push") == 0) - return MAKE_FUNCTOR(global_scope_t::push_report_cmd); - else if (std::strcmp(p, "pop") == 0) - return MAKE_FUNCTOR(global_scope_t::pop_report_cmd); - break; - } - } - break; - - case 'o': - if (std::strncmp(p, "opt_", 4) == 0) { - p = p + 4; - switch (*p) { - case 'd': - if (std::strcmp(p, "debug_") == 0) - return MAKE_FUNCTOR(global_scope_t::ignore); - break; - - case 's': - if (std::strcmp(p, "script_") == 0) - return MAKE_FUNCTOR(global_scope_t::option_script_); - break; - - case 't': - if (std::strcmp(p, "trace_") == 0) - return MAKE_FUNCTOR(global_scope_t::ignore); - break; - - case 'v': - if (! *(p + 1) || std::strcmp(p, "verbose") == 0) - return MAKE_FUNCTOR(global_scope_t::ignore); - else if (std::strcmp(p, "verify") == 0) - return MAKE_FUNCTOR(global_scope_t::ignore); - break; - } - } - } - - // If you're wondering how symbols from report() will be found, it's - // because of the bind_scope_t object in execute_command() below. - return expr_t::ptr_op_t(); - } - }; - strings_list split_arguments(char * line) { strings_list args; diff --git a/src/work.cc b/src/work.cc deleted file mode 100644 index d6a7bdac..00000000 --- a/src/work.cc +++ /dev/null @@ -1,187 +0,0 @@ -/* - * Copyright (c) 2003-2009, John Wiegley. All rights reserved. - * - * Redistribution and use in source and binary forms, with or without - * modification, are permitted provided that the following conditions are - * met: - * - * - Redistributions of source code must retain the above copyright - * notice, this list of conditions and the following disclaimer. - * - * - Redistributions in binary form must reproduce the above copyright - * notice, this list of conditions and the following disclaimer in the - * documentation and/or other materials provided with the distribution. - * - * - Neither the name of New Artisans LLC nor the names of its - * contributors may be used to endorse or promote products derived from - * this software without specific prior written permission. - * - * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS - * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT - * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR - * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT - * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, - * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT - * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, - * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY - * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT - * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE - * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. - */ - -#include <ledger.h> - -#include "work.h" - -namespace ledger { - -void handle_debug_options(int argc, char * argv[]) -{ - for (int i = 1; i < argc; i++) { - if (argv[i][0] == '-') { - if (std::strcmp(argv[i], "--verify") == 0) { -#if defined(VERIFY_ON) - verify_enabled = true; // global in utils.h -#endif - } - else if (std::strcmp(argv[i], "--verbose") == 0 || - std::strcmp(argv[i], "-v") == 0) { -#if defined(LOGGING_ON) - _log_level = LOG_INFO; // global in utils.h -#endif - } - else if (i + 1 < argc && std::strcmp(argv[i], "--debug") == 0) { -#if defined(DEBUG_ON) - _log_level = LOG_DEBUG; // global in utils.h - _log_category = argv[i + 1]; // global in utils.h - i++; -#endif - } - else if (i + 1 < argc && std::strcmp(argv[i], "--trace") == 0) { -#if defined(TRACING_ON) - _log_level = LOG_TRACE; // global in utils.h - try { - // global in utils.h - _trace_level = boost::lexical_cast<int>(argv[i + 1]); - } - catch (const boost::bad_lexical_cast& e) { - throw std::logic_error("Argument to --trace must be an integer"); - } - i++; -#endif - } - } - } -} - -void read_environment_settings(report_t& report, char * envp[]) -{ - TRACE_START(environment, 1, "Processed environment variables"); - - process_environment(const_cast<const char **>(envp), "LEDGER_", - report); - -#if 1 - // These are here for backwards compatability, but are deprecated. - - if (const char * p = std::getenv("LEDGER")) - process_option("file", report, p, "LEDGER"); - if (const char * p = std::getenv("LEDGER_INIT")) - process_option("init-file", report, p, "LEDGER_INIT"); - if (const char * p = std::getenv("PRICE_HIST")) - process_option("price-db", report, p, "PRICE_HIST"); - if (const char * p = std::getenv("PRICE_EXP")) - process_option("price-exp", report, p, "PRICE_EXP"); -#endif - - TRACE_FINISH(environment, 1); -} - -strings_list read_command_arguments(scope_t& scope, strings_list args) -{ - TRACE_START(arguments, 1, "Processed command-line arguments"); - - strings_list remaining = process_arguments(args, scope); - - TRACE_FINISH(arguments, 1); - - return remaining; -} - -void normalize_session_options(session_t& session) -{ - INFO("Initialization file is " << session.init_file->string()); - INFO("Price database is " << session.price_db->string()); - - foreach (const path& pathname, session.data_files) - INFO("Journal file is " << pathname.string()); -} - -function_t look_for_precommand(scope_t& scope, const string& verb) -{ - if (expr_t::ptr_op_t def = scope.lookup(string("ledger_precmd_") + verb)) - return def->as_function(); - else - return function_t(); -} - -function_t look_for_command(scope_t& scope, const string& verb) -{ - if (expr_t::ptr_op_t def = scope.lookup(string("ledger_cmd_") + verb)) - return def->as_function(); - else - return function_t(); -} - -void normalize_report_options(report_t& report, const string& verb) -{ - // Patch up some of the reporting options based on what kind of - // command it was. - - // jww (2008-08-14): This code really needs to be rationalized away - // for 3.0. - - if (verb == "print" || verb == "entry" || verb == "dump") { - report.show_related = true; - report.show_all_related = true; - } - else if (verb == "equity") { - report.show_subtotal = true; - } - else if (report.show_related) { - if (verb[0] == 'r') { - report.show_inverted = true; - } else { - report.show_subtotal = true; - report.show_all_related = true; - } - } - - if (verb[0] != 'b' && verb[0] != 'r') - report.what_to_keep.keep_base = true; - - // Setup the default value for the display predicate - - if (report.display_predicate.empty()) { - if (verb[0] == 'b') { - if (! report.show_empty) - report.display_predicate = "total"; - if (! report.show_subtotal) { - if (! report.display_predicate.empty()) - report.display_predicate += "&"; - report.display_predicate += "depth<=1"; - } - } - else if (verb == "equity") { - report.display_predicate = "amount_expr"; // jww (2008-08-14): ??? - } - else if (verb[0] == 'r' && ! report.show_empty) { - report.display_predicate = "amount"; - } - } - - if (! report.report_period.empty() && ! report.sort_all) - report.entry_sort = true; -} - -} // namespace ledger |