canopeas/database/database.go

package database

import (
	"go.etcd.io/bbolt"
)

var (
	firedBucket = []byte("fired")
	unbanBucket = []byte("unban")

	buckets = [][]byte{
		firedBucket,
		unbanBucket,
	}
)

type Database struct {
	db *bbolt.DB
}

func Load(dbPath string) (*Database, error) {
	db, err := bbolt.Open(dbPath, 0600, bbolt.DefaultOptions)
	if err != nil {
		return nil, err
	}
	for _, b := range buckets {
		if err := db.Update(func(tx *bbolt.Tx) error {
			_, err := tx.CreateBucketIfNotExists(b)
			return err
		}); err != nil {
			return nil, err
		}
	}
	return &Database{
		db: db,
	}, nil
}