refactor: open track store explicitly and remove global state

internal/store/download_info.go

@@ -18,10 +18,10 @@ type DownloadInfo struct {
 
 var trackBucket = []byte("tracks")
 
-func GetDownloadInfo(trackID string) (*DownloadInfo, error) {
+func (s *Store) DownloadInfo(trackID string) (*DownloadInfo, error) {
 	var info DownloadInfo
 
-	if err := db.View(func(tx *bbolt.Tx) error {
+	if err := s.db.View(func(tx *bbolt.Tx) error {
 		b := tx.Bucket(trackBucket)
 		if b == nil {
 			return fmt.Errorf("bucket not found")
@@ -39,8 +39,8 @@ func GetDownloadInfo(trackID string) (*DownloadInfo, error) {
 	return &info, nil
 }
 
-func (d *DownloadInfo) Save() error {
-	return db.Update(func(tx *bbolt.Tx) error {
+func (s *Store) PutDownloadInfo(d *DownloadInfo) error {
+	return s.db.Update(func(tx *bbolt.Tx) error {
 		b, err := tx.CreateBucketIfNotExists(trackBucket)
 		if err != nil {
 			return fmt.Errorf("failed to create bucket: %w", err)

internal/store/store.go

@@ -7,13 +7,19 @@ import (
 	bolt "go.etcd.io/bbolt"
 )
 
-var db *bolt.DB
+type Store struct {
+	db *bolt.DB
+}
 
-func OpenDB(cfgDir string) error {
-	var err error
-	db, err = bolt.Open(path.Join(cfgDir, ".tracks.db"), 0600, nil)
+func Open(dir string) (*Store, error) {
+	db, err := bolt.Open(path.Join(dir, ".tracks.db"), 0600, nil)
 	if err != nil {
-		return fmt.Errorf("failed to open database: %w", err)
+		return nil, fmt.Errorf("failed to open database: %w", err)
 	}
-	return nil
+
+	return &Store{db: db}, nil
+}
+
+func (s *Store) Close() error {
+	return s.db.Close()
 }