Moved work.cc and global_scope_t into a new global.cc file.
This commit is contained in:
parent
87cfef221a
commit
edb9cff4a2
5 changed files with 482 additions and 468 deletions
|
|
@ -136,7 +136,7 @@ pkginclude_HEADERS = \
|
||||||
src/reconcile.h \
|
src/reconcile.h \
|
||||||
src/quotes.h \
|
src/quotes.h \
|
||||||
\
|
\
|
||||||
src/work.h \
|
src/global.h \
|
||||||
src/ledger.h \
|
src/ledger.h \
|
||||||
\
|
\
|
||||||
python/pyledger.h \
|
python/pyledger.h \
|
||||||
|
|
@ -172,7 +172,7 @@ ledger_CPPFLAGS = $(lib_cppflags)
|
||||||
if HAVE_BOOST_PYTHON
|
if HAVE_BOOST_PYTHON
|
||||||
ledger_CPPFLAGS += -I$(srcdir)/python
|
ledger_CPPFLAGS += -I$(srcdir)/python
|
||||||
endif
|
endif
|
||||||
ledger_SOURCES = src/main.cc src/work.cc
|
ledger_SOURCES = src/main.cc src/global.cc
|
||||||
ledger_LDADD = $(LIBOBJS) $(lib_LTLIBRARIES)
|
ledger_LDADD = $(LIBOBJS) $(lib_LTLIBRARIES)
|
||||||
if HAVE_BOOST_PYTHON
|
if HAVE_BOOST_PYTHON
|
||||||
ledger_LDADD += libledger_python.la
|
ledger_LDADD += libledger_python.la
|
||||||
|
|
|
||||||
410
src/global.cc
Normal file
410
src/global.cc
Normal file
|
|
@ -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
|
||||||
|
|
@ -30,18 +30,81 @@
|
||||||
*/
|
*/
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @file work.h
|
* @file global.h
|
||||||
* @author John Wiegley
|
* @author John Wiegley
|
||||||
*
|
*
|
||||||
* @brief Contains the top-level functions used by main.cc
|
* @brief Contains the top-level functions used by main.cc
|
||||||
*/
|
*/
|
||||||
#ifndef _WORK_H
|
#ifndef _GLOBAL_H
|
||||||
#define _WORK_H
|
#define _GLOBAL_H
|
||||||
|
|
||||||
|
#include "report.h"
|
||||||
|
|
||||||
namespace ledger {
|
namespace ledger {
|
||||||
|
|
||||||
typedef strings_list::iterator string_iterator;
|
class global_scope_t : public noncopyable, public scope_t
|
||||||
typedef std::pair<string_iterator, string_iterator> string_iterator_pair;
|
{
|
||||||
|
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 handle_debug_options(int argc, char * argv[]);
|
||||||
void read_environment_settings(report_t& report, char * envp[]);
|
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
|
} // namespace ledger
|
||||||
|
|
||||||
#endif // _WORK_H
|
#endif // _GLOBAL_H
|
||||||
274
src/main.cc
274
src/main.cc
|
|
@ -31,283 +31,11 @@
|
||||||
|
|
||||||
#include <ledger.h> // Read this file for a top-level overview
|
#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
|
// was moved there for the sake of clarity
|
||||||
using namespace ledger;
|
using namespace ledger;
|
||||||
|
|
||||||
namespace {
|
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 split_arguments(char * line)
|
||||||
{
|
{
|
||||||
strings_list args;
|
strings_list args;
|
||||||
|
|
|
||||||
187
src/work.cc
187
src/work.cc
|
|
@ -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
|
|
||||||
Loading…
Add table
Reference in a new issue