thar we GO

main
Bel LaPointe 2023-10-25 07:52:05 -06:00
parent f4ca327e86
commit 68fba6b746
3 changed files with 41 additions and 7 deletions

View File

@ -11,7 +11,10 @@ import (
func main() { func main() {
foo := flag.String("foo", "bal", "bal or reg") foo := flag.String("foo", "bal", "bal or reg")
likePattern := flag.String("like", ".", "regexp to match") likeName := flag.String("like", ".", "regexp to match")
likeBefore := flag.String("like-before", "9", "date str to compare")
likeAfter := flag.String("like-after", "0", "date str to compare")
likeLedger := flag.Bool("like-ledger", false, "limit data to these -like-* rather than zoom to these -like-*")
flag.Parse() flag.Parse()
if flag.NArg() < 1 { if flag.NArg() < 1 {
@ -26,8 +29,16 @@ func main() {
if err != nil { if err != nil {
panic(err) panic(err)
} }
like := ledger.LikeName(*likePattern) like := []ledger.Like{ledger.LikeName(*likeName)}
deltas = deltas.Like(like) if *likeLedger {
like = append(like, ledger.LikeBefore(*likeBefore))
like = append(like, ledger.LikeAfter(*likeAfter))
deltas = deltas.Like(like...)
} else {
deltas = deltas.Like(like...)
like = append(like, ledger.LikeBefore(*likeBefore))
like = append(like, ledger.LikeAfter(*likeAfter))
}
switch *foo { switch *foo {
case "reg": case "reg":
@ -36,8 +47,14 @@ func main() {
}) })
register := deltas.Register() register := deltas.Register()
for i := range deltas { for i := range deltas {
all := true
for _, like := range like {
all = all && like(deltas[i])
}
if all {
fmt.Printf("%s (%+v)\n", deltas[i].Debug(), register[deltas[i].Date][deltas[i].Name].Debug()) fmt.Printf("%s (%+v)\n", deltas[i].Debug(), register[deltas[i].Date][deltas[i].Name].Debug())
} }
}
case "bal": case "bal":
for k, v := range deltas.Balances() { for k, v := range deltas.Balances() {
results := []string{} results := []string{}

View File

@ -1,6 +1,8 @@
package ledger package ledger
import "regexp" import (
"regexp"
)
type Like func(Delta) bool type Like func(Delta) bool
@ -8,13 +10,13 @@ type likes []Like
func LikeBefore(date string) Like { func LikeBefore(date string) Like {
return func(d Delta) bool { return func(d Delta) bool {
return date <= d.Date return date >= d.Date
} }
} }
func LikeAfter(date string) Like { func LikeAfter(date string) Like {
return func(d Delta) bool { return func(d Delta) bool {
return date >= d.Date return date <= d.Date
} }
} }

View File

@ -2,6 +2,21 @@ package ledger
import "testing" import "testing"
func TestLikeBeforeAfter(t *testing.T) {
if got := LikeBefore("9")(Delta{Date: "2021"}); !got {
t.Error("got 2021 is NOT before 9")
}
if got := LikeBefore("1")(Delta{Date: "2021"}); got {
t.Error("got 2021 IS before 1")
}
if got := LikeAfter("9")(Delta{Date: "2021"}); got {
t.Error("got 2021 IS after 9")
}
if got := LikeAfter("1")(Delta{Date: "2021"}); !got {
t.Error("got 2021 is NOT after 1")
}
}
func TestLikeName(t *testing.T) { func TestLikeName(t *testing.T) {
delta := Delta{Name: "x"} delta := Delta{Name: "x"}
if got := LikeName("^x$")(delta); !got { if got := LikeName("^x$")(delta); !got {