feat: watch playlists and download newly added tracks

internal/store/download_info.go

@@ -16,6 +16,8 @@ type DownloadInfo struct {
 	Downloaded time.Time `json:"downloaded_at"`
 }
 
+var trackBucket = []byte("tracks")
+
 func GetDownloadInfo(songID string) (*DownloadInfo, error) {
 	var info DownloadInfo
 
@@ -39,9 +41,9 @@ func GetDownloadInfo(songID string) (*DownloadInfo, error) {
 
 func (d *DownloadInfo) Save() error {
 	return db.Update(func(tx *bbolt.Tx) error {
-		b := tx.Bucket(trackBucket)
-		if b == nil {
-			return fmt.Errorf("bucket not found")
+		b, err := tx.CreateBucketIfNotExists(trackBucket)
+		if err != nil {
+			return fmt.Errorf("failed to create bucket: %w", err)
 		}
 
 		data, err := json.Marshal(d)
@@ -49,11 +51,6 @@ func (d *DownloadInfo) Save() error {
 			return err
 		}
 
-		err = b.Put([]byte(d.SongID), data)
-		if err != nil {
-			return err
-		}
-
-		return nil
+		return b.Put([]byte(d.SongID), data)
 	})
 }

internal/store/store.go

@@ -7,10 +7,7 @@ import (
 	bolt "go.etcd.io/bbolt"
 )
 
-var (
-	db          *bolt.DB
-	trackBucket = []byte("tracks")
-)
+var db *bolt.DB
 
 func OpenDB(cfgDir string) error {
 	var err error
@@ -21,12 +18,5 @@ func OpenDB(cfgDir string) error {
 		return fmt.Errorf("failed to open database: %w", err)
 	}
 
-	if err := db.Update(func(tx *bolt.Tx) error {
-		_, err := tx.CreateBucketIfNotExists(trackBucket)
-		return err
-	}); err != nil {
-		return fmt.Errorf("failed to create bucket: %w", err)
-	}
-
 	return nil
 }

internal/store/watched.go

@@ -0,0 +1,84 @@
+package store
+
+import (
+	"encoding/json"
+	"fmt"
+	"time"
+
+	bolt "go.etcd.io/bbolt"
+)
+
+type WatchedPlaylist struct {
+	ID      string        `json:"id"`
+	Quality string        `json:"quality"`
+	BPM     bool          `json:"bpm"`
+	Timeout time.Duration `json:"timeout"`
+}
+
+var watchedBucket = []byte("watched")
+
+func ListWatchedPlaylists() ([]*WatchedPlaylist, error) {
+	var playlists []*WatchedPlaylist
+	if err := db.View(func(tx *bolt.Tx) error {
+		b := tx.Bucket(watchedBucket)
+		if b == nil {
+			return nil
+		}
+
+		return b.ForEach(func(k, v []byte) error {
+			var p WatchedPlaylist
+			if err := json.Unmarshal(v, &p); err != nil {
+				return err
+			}
+			playlists = append(playlists, &p)
+
+			return nil
+		})
+	}); err != nil {
+		return nil, err
+	}
+
+	return playlists, nil
+}
+
+func (p *WatchedPlaylist) Save() error {
+	return db.Update(func(tx *bolt.Tx) error {
+		b, err := tx.CreateBucketIfNotExists(watchedBucket)
+		if err != nil {
+			return err
+		}
+
+		data, err := json.Marshal(p)
+		if err != nil {
+			return err
+		}
+
+		return b.Put([]byte(p.ID), data)
+	})
+}
+
+func RemoveWatchedPlaylist(playlistID string) error {
+	return db.Update(func(tx *bolt.Tx) error {
+		b := tx.Bucket(watchedBucket)
+		if b == nil {
+			return fmt.Errorf("bucket not found")
+		}
+
+		return b.Delete([]byte(playlistID))
+	})
+}
+
+func IsWatched(playlistID string) (bool, error) {
+	var found bool
+	err := db.View(func(tx *bolt.Tx) error {
+		b := tx.Bucket(watchedBucket)
+		if b == nil {
+			return nil
+		}
+		found = b.Get([]byte(playlistID)) != nil
+
+		return nil
+	})
+
+	return found, err
+}