summaryrefslogtreecommitdiff
path: root/reconcile.cc
blob: 555f3d601b15b24c8aec9432ba8f5f9c5c9482ab (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
#include "reconcile.h"
#include "walk.h"

namespace ledger {

#define xact_next(x)     ((transaction_t *)(x)->data)
#define xact_next_ptr(x) ((transaction_t **)&(x)->data)

bool search_for_balance(amount_t& amount,
			transaction_t ** prev, transaction_t * next)
{
  for (; next; next = xact_next(next)) {
    transaction_t * temp = *prev;
    *prev = next;

    amount -= next->amount;
    if (! amount ||
	search_for_balance(amount, xact_next_ptr(next), xact_next(next)))
      return true;
    amount += next->amount;

    *prev = temp;
  }
  return false;
}

static void push_chain_to_list(transaction_t * first,
			       transactions_list& xact_list)
{
  while (first) {
    transaction_t * curr = first;
    xact_list.push_back(curr);
    first = xact_next(first);
    curr->data = 0;
  }
}

void reconcile_transactions(transactions_list& xact_list,
			    value_t&	       balance,
			    const time_t       cutoff,
			    const bool         all_pending)
{
  value_t cleared_balance;
  value_t pending_balance;

  transaction_t *  first    = NULL;
  transaction_t ** last_ptr = &first;

  clear_transactions_xdata();

  bool found_pending = false;
  for (transactions_list::iterator x = xact_list.begin();
       x != xact_list.end();
       x++)
    if (! cutoff || std::difftime((*x)->entry->date, cutoff) < 0)
      switch ((*x)->entry->state) {
      case entry_t::CLEARED:
	cleared_balance += (*x)->amount;
	if (! found_pending)
	  break;
	// fall through...
      case entry_t::UNCLEARED:
      case entry_t::PENDING:
	pending_balance += (*x)->amount;
	if (all_pending)
	  found_pending = true;
	*last_ptr = *x;
	last_ptr = (transaction_t **) &(*x)->data;
	break;
      }

  if (all_pending) {
    xact_list.clear();
    push_chain_to_list(first, xact_list);
    return;
  }

  balance -= cleared_balance;
  if (balance.type >= value_t::BALANCE)
    throw error("Cannot reconcile accounts with multiple commodities");
  balance.cast(value_t::AMOUNT);

  // If the amount to reconcile is the same as the pending balance,
  // then assume an exact match and return the results right away.
  amount_t to_reconcile = *((amount_t *) balance.data);
  pending_balance.cast(value_t::AMOUNT);
  if (to_reconcile == *((amount_t *) pending_balance.data) ||
      search_for_balance(to_reconcile, &first, first)) {
    xact_list.clear();
    push_chain_to_list(first, xact_list);
  } else {
    throw error("Could not reconcile account!");
  }
}

} // namespace ledger