/
renderer.go
126 lines (115 loc) · 3.09 KB
/
renderer.go
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
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
// Copyright 2021 Silvio Böhler
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package report
import (
"time"
"github.com/sboehler/knut/lib/common/mapper"
"github.com/sboehler/knut/lib/common/table"
"github.com/sboehler/knut/lib/journal"
"github.com/shopspring/decimal"
)
// Renderer renders a report.
type Renderer struct {
ShowCommodities bool
SortAlphabetically bool
Diff bool
dates []time.Time
}
// Render renders a report.
func (rn *Renderer) Render(r *Report) *table.Table {
rn.dates = r.dates
if !rn.SortAlphabetically {
r.ComputeWeights()
}
var tbl *table.Table
if rn.ShowCommodities {
tbl = table.New(1, 1, len(rn.dates))
} else {
tbl = table.New(1, len(rn.dates))
}
tbl.AddSeparatorRow()
header := tbl.AddRow().AddText("Account", table.Center)
if rn.ShowCommodities {
header.AddText("Comm", table.Center)
}
for _, d := range rn.dates {
header.AddText(d.Format("2006-01-02"), table.Center)
}
tbl.AddSeparatorRow()
totalAL, totalEIE := r.Totals(journal.KeyMapper{
Date: mapper.Identity[time.Time],
Commodity: journal.MapCommodity(rn.ShowCommodities),
}.Build())
for _, n := range r.AL.Children() {
rn.renderNode(tbl, 0, n)
tbl.AddEmptyRow()
}
rn.render(tbl, 0, "Total (A+L)", false, totalAL)
tbl.AddSeparatorRow()
for _, n := range r.EIE.Children() {
rn.renderNode(tbl, 0, n)
tbl.AddEmptyRow()
}
rn.render(tbl, 0, "Total (E+I+E)", true, totalEIE)
tbl.AddSeparatorRow()
totalAL.Plus(totalEIE)
rn.render(tbl, 0, "Delta", false, totalAL)
tbl.AddSeparatorRow()
return tbl
}
func (rn *Renderer) renderNode(t *table.Table, indent int, n *Node) {
if n.Account != nil {
vals := n.Amounts.SumBy(nil, journal.KeyMapper{
Date: mapper.Identity[time.Time],
Commodity: journal.MapCommodity(rn.ShowCommodities),
}.Build())
rn.render(t, indent, n.Account.Segment(), !n.Account.IsAL(), vals)
}
for _, ch := range n.Children() {
rn.renderNode(t, indent+2, ch)
}
}
func (rn *Renderer) render(t *table.Table, indent int, name string, neg bool, vals journal.Amounts) {
if len(vals) == 0 {
t.AddRow().AddIndented(name, indent).FillEmpty()
return
}
for i, c := range vals.CommoditiesSorted() {
row := t.AddRow()
if i == 0 {
row.AddIndented(name, indent)
} else {
row.AddEmpty()
}
if rn.ShowCommodities {
row.AddText(c.Name(), table.Left)
}
var total decimal.Decimal
for _, d := range rn.dates {
v := vals[journal.DateCommodityKey(d, c)]
if !rn.Diff {
total = total.Add(v)
v = total
}
if neg {
v = v.Neg()
}
if v.IsZero() {
row.AddEmpty()
} else {
row.AddNumber(v)
}
}
}
}