clerk: internal/linter/rule_multiple_omitted

clerk/internal/linter/rule_multiple_omitted_amounts.go (view raw)

Oleksandr Smirnov ast: store postings as values instead of pointers, 1 month ago

package linter

import (
	"olexsmir.xyz/clerk/internal/analyzer"
	"olexsmir.xyz/clerk/journal/ast"
)

const MultipleOmittedAmountsID RuleID = "multiple-omitted-amounts"

// MultipleOmittedAmounts flags entries where more than one posting has an ommited amount.
type MultipleOmittedAmounts struct{}

func (MultipleOmittedAmounts) ID() RuleID { return MultipleOmittedAmountsID }
func (m *MultipleOmittedAmounts) CheckJournal(an *analyzer.Analysis) []Find {
	var finds []Find
	for _, txn := range an.Transactions {
		finds = append(finds, m.check(txn.Postings)...)
	}
	for _, ptx := range an.PeriodicTransactions {
		finds = append(finds, m.check(ptx.Postings)...)
	}
	for _, atx := range an.AutomatedTransactions {
		finds = append(finds, m.check(atx.Postings)...)
	}
	return finds
}

func (m *MultipleOmittedAmounts) check(postings []ast.Posting) []Find {
	var finds []Find
	for _, p := range postings {
		if p.Amount == nil && p.Balance == nil {
			finds = append(finds, Find{
				Code:    m.ID(),
				Message: "more than one posting has omitted amount",
				Span:    p.Span,
			})
		}
	}
	if len(finds) > 1 {
		return finds
	}
	return nil
}

1	package linter
2
3	import (
4	"olexsmir.xyz/clerk/internal/analyzer"
5	"olexsmir.xyz/clerk/journal/ast"
6	)
7
8	const MultipleOmittedAmountsID RuleID = "multiple-omitted-amounts"
9
10	// MultipleOmittedAmounts flags entries where more than one posting has an ommited amount.
11	type MultipleOmittedAmounts struct{}
12
13	func (MultipleOmittedAmounts) ID() RuleID { return MultipleOmittedAmountsID }
14	func (m MultipleOmittedAmounts) CheckJournal(an analyzer.Analysis) []Find {
15	var finds []Find
16	for _, txn := range an.Transactions {
17	finds = append(finds, m.check(txn.Postings)...)
18	}
19	for _, ptx := range an.PeriodicTransactions {
20	finds = append(finds, m.check(ptx.Postings)...)
21	}
22	for _, atx := range an.AutomatedTransactions {
23	finds = append(finds, m.check(atx.Postings)...)
24	}
25	return finds
26	}
27
28	func (m *MultipleOmittedAmounts) check(postings []ast.Posting) []Find {
29	var finds []Find
30	for _, p := range postings {
31	if p.Amount == nil && p.Balance == nil {
32	finds = append(finds, Find{
33	Code: m.ID(),
34	Message: "more than one posting has omitted amount",
35	Span: p.Span,
36	})
37	}
38	}
39	if len(finds) > 1 {
40	return finds
41	}
42	return nil
43	}