MaysWind
94 lines
3.0 KiB
Go
94 lines
3.0 KiB
Go
package beancount
|
|
|
|
import "strings"
|
|
|
|
const beancountEquityAccountNameOpeningBalance = "Opening-Balances"
|
|
|
|
// beancountDirective represents the Beancount directive
|
|
type beancountDirective string
|
|
|
|
// Beancount directives
|
|
const (
|
|
beancountDirectiveOpen beancountDirective = "open"
|
|
beancountDirectiveClose beancountDirective = "close"
|
|
beancountDirectiveTransaction beancountDirective = "txn"
|
|
beancountDirectiveCompletedTransaction beancountDirective = "*"
|
|
beancountDirectiveInCompleteTransaction beancountDirective = "!"
|
|
beancountDirectivePaddingTransaction beancountDirective = "P"
|
|
beancountDirectiveCommodity beancountDirective = "commodity"
|
|
beancountDirectivePrice beancountDirective = "price"
|
|
beancountDirectiveNote beancountDirective = "note"
|
|
beancountDirectiveDocument beancountDirective = "document"
|
|
beancountDirectiveEvent beancountDirective = "event"
|
|
beancountDirectiveBalance beancountDirective = "balance"
|
|
beancountDirectivePad beancountDirective = "pad"
|
|
beancountDirectiveQuery beancountDirective = "query"
|
|
beancountDirectiveCustom beancountDirective = "custom"
|
|
)
|
|
|
|
// beancountAccountType represents the Beancount account type
|
|
type beancountAccountType byte
|
|
|
|
// Beancount account types
|
|
const (
|
|
beancountUnknownAccountType beancountAccountType = 0
|
|
beancountAssetsAccountType beancountAccountType = 1
|
|
beancountLiabilitiesAccountType beancountAccountType = 2
|
|
beancountEquityAccountType beancountAccountType = 3
|
|
beancountIncomeAccountType beancountAccountType = 4
|
|
beancountExpensesAccountType beancountAccountType = 5
|
|
)
|
|
|
|
// beancountData defines the structure of beancount data
|
|
type beancountData struct {
|
|
Accounts map[string]*beancountAccount
|
|
Transactions []*beancountTransactionEntry
|
|
}
|
|
|
|
// beancountAccount defines the structure of beancount account
|
|
type beancountAccount struct {
|
|
Name string
|
|
AccountType beancountAccountType
|
|
OpenDate string
|
|
CloseDate string
|
|
}
|
|
|
|
// beancountTransactionEntry defines the structure of beancount transaction entry
|
|
type beancountTransactionEntry struct {
|
|
Date string
|
|
Directive beancountDirective
|
|
Payee string
|
|
Narration string
|
|
Postings []*beancountPosting
|
|
Tags []string
|
|
Links []string
|
|
Metadata map[string]string
|
|
}
|
|
|
|
// beancountPosting defines the structure of beancount transaction posting
|
|
type beancountPosting struct {
|
|
Account string
|
|
Amount string
|
|
OriginalAmount string
|
|
Commodity string
|
|
TotalCost string
|
|
TotalCostCommodity string
|
|
Price string
|
|
PriceCommodity string
|
|
Metadata map[string]string
|
|
}
|
|
|
|
func (a *beancountAccount) isOpeningBalanceEquityAccount() bool {
|
|
if a.AccountType != beancountEquityAccountType {
|
|
return false
|
|
}
|
|
|
|
nameItems := strings.Split(a.Name, string(beancountMetadataKeySuffix))
|
|
|
|
if len(nameItems) != 2 {
|
|
return false
|
|
}
|
|
|
|
return nameItems[1] == beancountEquityAccountNameOpeningBalance
|
|
}
|