summaryrefslogtreecommitdiff
path: root/src/derive.cc
diff options
context:
space:
mode:
authorJohn Wiegley <johnw@newartisans.com>2008-08-05 13:17:04 -0400
committerJohn Wiegley <johnw@newartisans.com>2008-08-05 18:05:49 -0400
commitf6f4a46cf5b14f9a2170cd6475958efbf320caec (patch)
tree05bc1defcdebc201de3dd10477483d906a842821 /src/derive.cc
parentb7970b29855563e4c67f85af8b31233eda80c22a (diff)
downloadfork-ledger-f6f4a46cf5b14f9a2170cd6475958efbf320caec.tar.gz
fork-ledger-f6f4a46cf5b14f9a2170cd6475958efbf320caec.tar.bz2
fork-ledger-f6f4a46cf5b14f9a2170cd6475958efbf320caec.zip
Moved around most of the files so that source code is in src/, documentation
is in doc/, etc.
Diffstat (limited to 'src/derive.cc')
-rw-r--r--src/derive.cc228
1 files changed, 228 insertions, 0 deletions
diff --git a/src/derive.cc b/src/derive.cc
new file mode 100644
index 00000000..dd9dc4ca
--- /dev/null
+++ b/src/derive.cc
@@ -0,0 +1,228 @@
+/*
+ * Copyright (c) 2003-2008, 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 "derive.h"
+#include "session.h"
+#include "iterators.h"
+
+namespace ledger {
+
+entry_t * derive_new_entry(report_t& report,
+ strings_list::iterator i,
+ strings_list::iterator end)
+{
+ session_t& session(report.session);
+
+ std::auto_ptr<entry_t> added(new entry_t);
+
+ entry_t * matching = NULL;
+
+ added->_date = parse_date(*i++);
+ if (i == end)
+ throw std::runtime_error("Too few arguments to 'entry'");
+
+ mask_t regexp(*i++);
+
+ journals_iterator iter(session);
+ entries_list::reverse_iterator j;
+
+ for (journal_t * journal = iter(); journal; journal = iter()) {
+ for (j = journal->entries.rbegin();
+ j != journal->entries.rend();
+ j++) {
+ if (regexp.match((*j)->payee)) {
+ matching = *j;
+ break;
+ }
+ }
+ if (matching) break;
+ }
+
+ added->payee = matching ? matching->payee : regexp.expr.str();
+
+ if (! matching) {
+ account_t * acct;
+ if (i == end || ((*i)[0] == '-' || std::isdigit((*i)[0]))) {
+ acct = session.find_account("Expenses");
+ }
+ else if (i != end) {
+ acct = session.find_account_re(*i);
+ if (! acct)
+ acct = session.find_account(*i);
+ assert(acct);
+ i++;
+ }
+
+ if (i == end) {
+ added->add_xact(new xact_t(acct));
+ } else {
+ xact_t * xact = new xact_t(acct, amount_t(*i++));
+ added->add_xact(xact);
+
+ if (! xact->amount.commodity()) {
+ // If the amount has no commodity, we can determine it given
+ // the account by creating a final for the account and then
+ // checking if it contains only a single commodity. An
+ // account to which only dollars are applied would imply that
+ // dollars are wanted now too.
+
+ report.sum_all_accounts();
+
+ value_t total = acct->xdata().total;
+ if (total.is_type(value_t::AMOUNT))
+ xact->amount.set_commodity(total.as_amount().commodity());
+ }
+ }
+
+ acct = NULL;
+
+ if (i != end) {
+ if (! acct)
+ acct = session.find_account_re(*i);
+ if (! acct)
+ acct = session.find_account(*i);
+ }
+
+ if (! acct) {
+ if (matching && matching->journal->basket)
+ acct = matching->journal->basket;
+ else
+ acct = session.find_account("Equity");
+ }
+
+ added->add_xact(new xact_t(acct));
+ }
+ else if (i == end) {
+ // If no argument were given but the payee, assume the user wants
+ // to see the same xact as last time.
+ added->code = matching->code;
+
+ foreach (xact_t * xact, matching->xacts)
+ added->add_xact(new xact_t(*xact));
+ }
+ else if ((*i)[0] == '-' || std::isdigit((*i)[0])) {
+ xact_t * m_xact, * xact, * first;
+ m_xact = matching->xacts.front();
+
+ first = xact = new xact_t(m_xact->account, amount_t(*i++));
+ added->add_xact(xact);
+
+ if (! xact->amount.commodity())
+ xact->amount.set_commodity(m_xact->amount.commodity());
+
+ m_xact = matching->xacts.back();
+
+ xact = new xact_t(m_xact->account, - first->amount);
+ added->add_xact(xact);
+
+ if (i != end) {
+ account_t * acct = session.find_account_re(*i);
+ if (! acct)
+ acct = session.find_account(*i);
+ assert(acct);
+ added->xacts.back()->account = acct;
+ }
+ }
+ else {
+ account_t * draw_acct = NULL;
+
+ while (i != end) {
+ string& re_pat(*i++);
+ account_t * acct = NULL;
+ amount_t * amt = NULL;
+
+ mask_t acct_regex(re_pat);
+
+ for (; j != matching->journal->entries.rend(); j++)
+ if (regexp.match((*j)->payee)) {
+ entry_t * entry = *j;
+ foreach (xact_t * xact, entry->xacts)
+ if (acct_regex.match(xact->account->fullname())) {
+ acct = xact->account;
+ amt = &xact->amount;
+ matching = entry;
+ goto found;
+ }
+ }
+
+ found:
+ xact_t * xact;
+ if (i == end) {
+ if (amt)
+ xact = new xact_t(acct, *amt);
+ else
+ xact = new xact_t(acct);
+ } else {
+ amount_t amount(*i++);
+
+ strings_list::iterator x = i;
+ if (i != end && ++x == end) {
+ draw_acct = session.find_account_re(*i);
+ if (! draw_acct)
+ draw_acct = session.find_account(*i);
+ i++;
+ }
+
+ if (! acct)
+ acct = session.find_account_re(re_pat);
+ if (! acct)
+ acct = session.find_account(re_pat);
+
+ xact = new xact_t(acct, amount);
+ if (! xact->amount.commodity()) {
+ if (amt)
+ xact->amount.set_commodity(amt->commodity());
+ else if (amount_t::current_pool->default_commodity)
+ xact->amount.set_commodity(*amount_t::current_pool->default_commodity);
+ }
+ }
+ added->add_xact(xact);
+ }
+
+ if (! draw_acct) {
+ assert(matching->xacts.back()->account);
+ draw_acct = matching->xacts.back()->account;
+ }
+ if (draw_acct)
+ added->add_xact(new xact_t(draw_acct));
+ }
+
+ if ((matching &&
+ ! matching->journal->entry_finalize_hooks.run_hooks(*added, false)) ||
+ ! added->finalize() ||
+ (matching &&
+ ! matching->journal->entry_finalize_hooks.run_hooks(*added, true)))
+ throw std::runtime_error("Failed to finalize derived entry (check commodities)");
+
+ return added.release();
+}
+
+} // namespace ledger