summaryrefslogtreecommitdiffstats
path: root/ledger/lib/balance
blob: 8d84dd7bccd33a2ac759c7d3d81187284a3bd2c1 (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
#! /bin/awk -f
#
# usage: [colorize=false] [scale=N] //ledger/lib/balance LEDGER_FILE
# see also: //cholerab/ledger-spec.markdown
#

BEGIN {
  colorize = ENVIRON["colorize"] == "" || ENVIRON["colorize"] == "true"
  # TODO use bc for arbitrary precision arithmetic
  scale = ENVIRON["scale"]
}

/^[[:space:]]*[0-9]+-[0-9][0-9]-[0-9][0-9]/{
  tx($2, $3, $4, $5)
}

END {
  display_accounts()
}

function tx (dst, src, amt, u) {
  withdraw(src, amt, u)
  deposit(dst, amt, u)
}

function deposit (name, amt, u) {
  accounts[name][u] += amt
}

function withdraw (name, amt, u) {
  accounts[name][u] -= amt
}

function display_accounts() {
  max_name_len = 0
  for (name in accounts) {
    if (length(name) > max_name_len) {
      max_name_len = length(name)
    }
  }

  max_balance_len = 0
  for (name in accounts) {
    for (u in accounts[name]) {
      s = accounts[name][u]
      sub(/\..*/, "", s)
      if (length(s) > max_balance_len) {
        max_balance_len = length(s)
      }
    }
  }
  if (scale > 0) {
    max_balance_len += 1 + scale
  }

  for (name in accounts) {
    for (u in accounts[name]) {

      fmt = "NAME BALANCE UNIT\n"
      balance = accounts[name][u]

      if (colorize) {
        sub("BALANCE", "[" (balance < 0 ? 31 : 32) "m&", fmt)
      }

      sub("NAME", "%-" max_name_len "s", fmt)
      sub("BALANCE", "%" max_balance_len "." scale "f", fmt)
      sub("UNIT", "%s", fmt)

      printf fmt, name, balance, u
    }
  }
}