110 lines
2.0 KiB
Go
110 lines
2.0 KiB
Go
package http
|
|
|
|
import (
|
|
"context"
|
|
"encoding/csv"
|
|
"io"
|
|
"time"
|
|
|
|
"git.javil.eu/jacob1123/budgeteer/postgres"
|
|
"github.com/google/uuid"
|
|
)
|
|
|
|
type YNABImport struct {
|
|
Context context.Context
|
|
accounts []postgres.Account
|
|
payees []postgres.Payee
|
|
queries *postgres.Queries
|
|
}
|
|
|
|
func NewYNABImport(q *postgres.Queries) (*YNABImport, error) {
|
|
accounts, err := q.GetAccounts(context.Background(), uuid.UUID{})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
payees, err := q.GetPayees(context.Background(), uuid.UUID{})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &YNABImport{
|
|
Context: context.Background(),
|
|
accounts: accounts,
|
|
payees: payees,
|
|
queries: q,
|
|
}, nil
|
|
|
|
}
|
|
|
|
func (ynab *YNABImport) Import(q *postgres.Queries, r io.Reader) error {
|
|
csv := csv.NewReader(r)
|
|
csv.Comma = '\t'
|
|
|
|
csvData, err := csv.ReadAll()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for _, record := range csvData {
|
|
accountName := record[0]
|
|
account, err := ynab.GetAccount(accountName)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
//flag := record[1]
|
|
|
|
dateString := record[2]
|
|
date, err := time.Parse("02.01.2006", dateString)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
payee := record[3]
|
|
category := record[4] //also in 5 + 6 split by group/category
|
|
memo := record[7]
|
|
outflow := record[8]
|
|
inflow := record[9]
|
|
cleared := record[10]
|
|
|
|
transaction := postgres.Transaction{
|
|
Date: date,
|
|
Memo: memo,
|
|
AccountID: account.ID,
|
|
}
|
|
}
|
|
}
|
|
|
|
func (ynab *YNABImport) GetAccount(name string) (*postgres.Account, error) {
|
|
for _, acc := range ynab.accounts {
|
|
if acc.Name == name {
|
|
return &acc, nil
|
|
}
|
|
}
|
|
|
|
account, err := ynab.queries.CreateAccount(ynab.Context, name)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
ynab.accounts = append(ynab.accounts, account)
|
|
return &account, nil
|
|
}
|
|
|
|
func (ynab *YNABImport) GetPayee(name string) (*postgres.Payee, error) {
|
|
for _, pay := range ynab.payees {
|
|
if pay.Name == name {
|
|
return &pay, nil
|
|
}
|
|
}
|
|
|
|
payee, err := ynab.queries.CreatePayee(ynab.Context, name)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
ynab.payees = append(ynab.payees, payee)
|
|
return &payee, nil
|
|
}
|