From 6e4e2fa779b641711977ab921615ce3c803276c0 Mon Sep 17 00:00:00 2001 From: tv Date: Wed, 11 Sep 2013 19:43:44 +0200 Subject: ledger balance: initial commit --- ledger/lib/balance | 73 ++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 73 insertions(+) create mode 100755 ledger/lib/balance (limited to 'ledger/lib/balance') diff --git a/ledger/lib/balance b/ledger/lib/balance new file mode 100755 index 00000000..8d84dd7b --- /dev/null +++ b/ledger/lib/balance @@ -0,0 +1,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 + } + } +} -- cgit v1.2.3 From a48c908949884f1e550abbd5bda5b30131fe71c4 Mon Sep 17 00:00:00 2001 From: tv Date: Wed, 11 Sep 2013 19:48:44 +0200 Subject: ledger balance: fix usage description --- ledger/lib/balance | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'ledger/lib/balance') diff --git a/ledger/lib/balance b/ledger/lib/balance index 8d84dd7b..1e9e2db2 100755 --- a/ledger/lib/balance +++ b/ledger/lib/balance @@ -1,6 +1,6 @@ #! /bin/awk -f # -# usage: [colorize=false] [scale=N] //ledger/lib/balance LEDGER_FILE +# usage: [colorize=false] [scale=N] //ledger/lib/balance LEDGER_FILE... # see also: //cholerab/ledger-spec.markdown # -- cgit v1.2.3 From 87b90098679c9daccd558d6df0fef508cf7a4a24 Mon Sep 17 00:00:00 2001 From: tv Date: Wed, 11 Sep 2013 19:59:53 +0200 Subject: ledger balance: clarify max_balance_len computation --- ledger/lib/balance | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) (limited to 'ledger/lib/balance') diff --git a/ledger/lib/balance b/ledger/lib/balance index 1e9e2db2..f75410c3 100755 --- a/ledger/lib/balance +++ b/ledger/lib/balance @@ -42,15 +42,14 @@ function display_accounts() { 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) + n = length(int(accounts[name][u])) + if (n > max_balance_len) { + max_balance_len = n } } } if (scale > 0) { - max_balance_len += 1 + scale + max_balance_len += length(".") + scale } for (name in accounts) { -- cgit v1.2.3 From 74961ef05afc5d06d0af192752912466021882c4 Mon Sep 17 00:00:00 2001 From: tv Date: Wed, 11 Sep 2013 20:14:06 +0200 Subject: ledger balance: add description and pipe-usage --- ledger/lib/balance | 17 +++++++++++++++-- 1 file changed, 15 insertions(+), 2 deletions(-) (limited to 'ledger/lib/balance') diff --git a/ledger/lib/balance b/ledger/lib/balance index f75410c3..77cb7f6d 100755 --- a/ledger/lib/balance +++ b/ledger/lib/balance @@ -1,7 +1,20 @@ #! /bin/awk -f # -# usage: [colorize=false] [scale=N] //ledger/lib/balance LEDGER_FILE... -# see also: //cholerab/ledger-spec.markdown +# ledger balance calculator +# +# usage: +# [colorize=false] [scale=N] //ledger/lib/balance LEDGER_FILE... +# [colorize=false] [scale=N] //ledger/lib/balance < LEDGER_FILE +# +# description: +# The ledger balance calculator computes the balance of each account it +# encounters in the provided ledger files. +# +# example: +# //ledger/lib/balance < //cholerab/ledger-spec.markdown +# +# see also: +# //cholerab/ledger-spec.markdown (ledger file format) # BEGIN { -- cgit v1.2.3