package database

import (
	"encoding/json"
	"git.jojodev.com/Minecraft/canopeas/discord/track"
	"go.etcd.io/bbolt"
)

// LoadTracker loads the tracks from the database
func (db *Database) LoadTracker() (track.Tracker, error) {
	t := track.New()
	return t, db.db.View(func(tx *bbolt.Tx) error {
		return tx.Bucket(trackBucket).ForEach(func(k, v []byte) error {
			var users map[string]struct{}
			if err := json.Unmarshal(v, &users); err != nil {
				return err
			}
			t[string(k)] = users
			return nil
		})
	})
}

// SetTrack sets a list of trackers for a word
func (db *Database) SetTrack(word string, trackers map[string]struct{}) error {
	return db.db.Update(func(tx *bbolt.Tx) error {
		v, err := json.Marshal(trackers)
		if err != nil {
			return err
		}
		return tx.Bucket(trackBucket).Put([]byte(word), v)
	})
}