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

//go:build import

package main

import (
	"context"
	"database/sql"
	"encoding/json"
	"errors"
	"flag"
	"fmt"
	"os"
	"time"

	"github.com/golang-migrate/migrate/v4"
	"go.jolheiser.com/mint/database"
	"go.jolheiser.com/mint/database/sqlc/migrations"
)

type budget struct {
	Title  string
	Amount float64
	Date   time.Time
}

func main() {
	b := flag.String("budget", "budget.json", "Path to existing budget")
	d := flag.String("database", "mint.sqlite3", "Path to database file")
	flag.Parse()

	data, err := os.ReadFile(*b)
	if err != nil {
		panic(err)
	}

	dsn := fmt.Sprintf("file:%s", *d)
	sdb, err := sql.Open("sqlite", dsn)
	if err != nil {
		panic(err)
	}

	migrator, err := migrations.New(sdb)
	if err != nil {
		panic(err)
	}

	if err := migrator.Up(); err != nil && !errors.Is(err, migrate.ErrNoChange) {
		panic(err)
	}

	db := database.New(sdb)

	var transactions []budget
	if err := json.Unmarshal(data, &transactions); err != nil {
		panic(err)
	}

	for _, t := range transactions {
		if _, err := db.CreateTransaction(context.Background(), database.CreateTransactionParams{
			Title:      t.Title,
			Amount:     int64(t.Amount * 100),
			Date:       t.Date,
			Recurrance: "month",
		}); err != nil {
			panic(err)
		}
	}
	fmt.Println("Import complete")
}