diff options
Diffstat (limited to 'src/session.cc')
| -rw-r--r-- | src/session.cc | 220 |
1 files changed, 220 insertions, 0 deletions
diff --git a/src/session.cc b/src/session.cc new file mode 100644 index 00000000..78fc2596 --- /dev/null +++ b/src/session.cc @@ -0,0 +1,220 @@ +#include "session.h" + +namespace ledger { + +unsigned int session_t::read_journal(std::istream& in, + journal_t * journal, + account_t * master, + const string * original_file) +{ + if (! master) + master = journal->master; + + for (std::list<parser_t *>::iterator i = parsers.begin(); + i != parsers.end(); + i++) + if ((*i)->test(in)) + return (*i)->parse(in, journal, master, original_file); + + return 0; +} + +unsigned int session_t::read_journal(const string& path, + journal_t * journal, + account_t * master, + const string * original_file) +{ + journal->sources.push_back(path); + + if (access(path.c_str(), R_OK) == -1) + throw_(exception, "Cannot read file '" << path << "'"); + + if (! original_file) + original_file = &path; + + std::ifstream stream(path.c_str()); + return read_journal(stream, journal, master, original_file); +} + +void session_t::read_init() +{ + if (init_file.empty()) + return; + + if (access(init_file.c_str(), R_OK) == -1) + throw_(exception, "Cannot read init file '" << init_file << "'"); + + std::ifstream init(init_file.c_str()); + + // jww (2006-09-15): Read initialization options here! +} + +journal_t * session_t::read_data(const string& master_account) +{ + TRACE_START(parser, 1, "Parsing journal file"); + + journal_t * journal = new_journal(); + journal->document = new xml::document_t; + journal->document->set_top(xml::wrap_node(journal->document, journal)); + + unsigned int entry_count = 0; + + DEBUG_("ledger.cache", "3. use_cache = " << use_cache); + + if (use_cache && ! cache_file.empty() && + ! data_file.empty()) { + DEBUG_("ledger.cache", "using_cache " << cache_file); + cache_dirty = true; + if (access(cache_file.c_str(), R_OK) != -1) { + std::ifstream stream(cache_file.c_str()); + + string price_db_orig = journal->price_db; + journal->price_db = price_db; + entry_count += read_journal(stream, journal, NULL, + &data_file); + if (entry_count > 0) + cache_dirty = false; + else + journal->price_db = price_db_orig; + } + } + + if (entry_count == 0 && ! data_file.empty()) { + account_t * acct = NULL; + if (! master_account.empty()) + acct = journal->find_account(master_account); + + journal->price_db = price_db; + if (! journal->price_db.empty() && + access(journal->price_db.c_str(), R_OK) != -1) { + if (read_journal(journal->price_db, journal)) { + throw_(exception, "Entries not allowed in price history file"); + } else { + DEBUG_("ledger.cache", "read price database " << journal->price_db); + journal->sources.pop_back(); + } + } + + DEBUG_("ledger.cache", "rejected cache, parsing " << data_file); + if (data_file == "-") { + use_cache = false; + journal->sources.push_back("<stdin>"); + entry_count += read_journal(std::cin, journal, acct); + } + else if (access(data_file.c_str(), R_OK) != -1) { + entry_count += read_journal(data_file, journal, acct); + if (! journal->price_db.empty()) + journal->sources.push_back(journal->price_db); + } + } + + VERIFY(journal->valid()); + + if (entry_count == 0) + throw_(exception, "Failed to locate any journal entries; " + "did you specify a valid file with -f?"); + + TRACE_STOP(parser, 1); + + return journal; +} + +bool session_t::resolve(const string& name, value_t& result, + xml::xpath_t::scope_t * locals) +{ + const char * p = name.c_str(); + switch (*p) { + case 'd': + if (name == "date_format") { + // jww (2007-04-18): What to do here? +#if 0 + result.set_string(moment_t::output_format); +#endif + return true; + } + break; + + case 'n': + switch (*++p) { + case 'o': + if (name == "now") { + result = now; + return true; + } + break; + } + break; + + case 'r': + if (name == "register_format") { + result = register_format; + return true; + } + break; + } + + return xml::xpath_t::scope_t::resolve(name, result, locals); +} + +xml::xpath_t::op_t * session_t::lookup(const string& name) +{ + const char * p = name.c_str(); + switch (*p) { + case 'o': + if (std::strncmp(p, "option_", 7) == 0) { + p = p + 7; + switch (*p) { + case 'd': + if (std::strcmp(p, "debug") == 0) + return MAKE_FUNCTOR(session_t, option_debug); + break; + + case 'f': + if (! *(p + 1) || std::strcmp(p, "file") == 0) + return MAKE_FUNCTOR(session_t, option_file); + break; + + case 't': + if (std::strcmp(p, "trace") == 0) + return MAKE_FUNCTOR(session_t, option_trace); + break; + + case 'v': + if (! *(p + 1) || std::strcmp(p, "verbose") == 0) + return MAKE_FUNCTOR(session_t, option_verbose); + else if (std::strcmp(p, "verify") == 0) + return MAKE_FUNCTOR(session_t, option_verify); + break; + } + } + break; + } + + return xml::xpath_t::scope_t::lookup(name); +} + +// jww (2007-04-26): All of Ledger should be accessed through a +// session_t object +void initialize() +{ + IF_VERIFY() + initialize_memory_tracing(); + + amount_t::initialize(); + xml::xpath_t::initialize(); +} + +void shutdown() +{ + xml::xpath_t::shutdown(); + amount_t::shutdown(); + + IF_VERIFY() { + INFO("Ledger shutdown (Boost/libstdc++ may still hold memory)"); + shutdown_memory_tracing(); + } else { + INFO("Ledger shutdown"); + } +} + +} // namespace ledger |