db.go

package ttdb

import (
	"encoding/binary"
	"os"
	"path"
	"path/filepath"
	"time"

	"github.com/alanxoc3/ttrack/internal/types"

	bolt "go.etcd.io/bbolt"
)

func open(dir string) (*bolt.DB, error) {
	dbpath := path.Join(dir, "db")

	err := os.MkdirAll(filepath.Dir(dbpath), 0755)
	if err != nil {
		return nil, err
	}

	db, err := bolt.Open(dbpath, 0666, &bolt.Options{Timeout: 1 * time.Second})
	if err != nil {
		return nil, err
	}

	return db, nil
}

func GetSeconds(b *bolt.Bucket, key string) types.DaySeconds {
	return types.CreateSecondsFromBytes(b.Get([]byte(key)))
}

func SetSeconds(b *bolt.Bucket, key string, secs types.DaySeconds) {
	timeout_bytes := make([]byte, 4, 4)
	binary.BigEndian.PutUint32(timeout_bytes[:], uint32(secs.GetAsUint32()))

	if secs.IsZero() {
		b.Delete([]byte(key))
	} else {
		b.Put([]byte(key), timeout_bytes) // TODO: Error handling.
	}
}

// For adding seconds, do this:
// SetSeconds(b, date_key, GetSeconds(b, date_key).Add(secs))

func GetTimestamp(b *bolt.Bucket, key string) time.Time {
	t := time.Time{}
	b_val := b.Get([]byte(key))
	if b_val != nil {
		t.UnmarshalBinary(b_val) // TODO: Error handling.
	}
	return t
}

func SetTimestamp(b *bolt.Bucket, key string, t time.Time) {
	raw, _ := t.MarshalBinary() // TODO: Error handling.
	b.Put([]byte(key), raw)     // TODO: Error handling.
}

func ViewCmd(cacheDir string, f func(*bolt.Tx) error) {
	db, err := open(cacheDir)
	if err != nil {
		panic(err)
	}
	defer db.Close()

	err = db.View(func(tx *bolt.Tx) error {
		return f(tx)
	})

	if err != nil {
		panic(err)
	}
}

func UpdateCmd(cacheDir string, f func(*bolt.Tx) error) {
	db, err := open(cacheDir)
	if err != nil {
		panic(err)
	}
	defer db.Close()

	err = db.Update(func(tx *bolt.Tx) error {
		return f(tx)
	})

	if err != nil {
		panic(err)
	}
}