feat: add local DB to save downloads

This commit is contained in:
Mathis Maquenne
2025-04-26 13:09:59 +02:00
parent 6675834842
commit f931cad746
8 changed files with 215 additions and 64 deletions
+58
View File
@@ -0,0 +1,58 @@
package db
import (
"encoding/json"
"fmt"
"time"
"go.etcd.io/bbolt"
)
type DownloadInfo struct {
SongID string `json:"song_id"`
Quality string `json:"quality"`
Path string `json:"path"`
Downloaded time.Time `json:"downloaded_at"`
}
func Get(songID string) (*DownloadInfo, error) {
var info DownloadInfo
if err := db.View(func(tx *bbolt.Tx) error {
b := tx.Bucket(trackBucket)
if b == nil {
return fmt.Errorf("bucket not found")
}
data := b.Get([]byte(songID))
if data == nil {
return fmt.Errorf("not found")
}
return json.Unmarshal(data, &info)
}); err != nil {
return nil, err
}
return &info, nil
}
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")
}
data, err := json.Marshal(d)
if err != nil {
return err
}
err = b.Put([]byte(d.SongID), data)
if err != nil {
return err
}
return nil
})
}
+44
View File
@@ -0,0 +1,44 @@
package db
import (
"fmt"
"os"
"path"
"github.com/mathismqn/godeez/internal/utils"
bolt "go.etcd.io/bbolt"
)
var (
db *bolt.DB
trackBucket = []byte("tracks")
)
func Init() {
homeDir, err := os.UserHomeDir()
if err != nil {
fmt.Fprintf(os.Stderr, "Error: could not get home directory: %v\n", err)
os.Exit(1)
}
appDir := path.Join(homeDir, ".godeez")
if err := utils.EnsureDir(appDir); err != nil {
fmt.Fprintf(os.Stderr, "Error: could not create app directory: %v\n", err)
os.Exit(1)
}
dbPath := path.Join(appDir, "tracks.db")
db, err = bolt.Open(dbPath, 0600, nil)
if err != nil {
fmt.Fprintf(os.Stderr, "Error: could not open database: %v\n", err)
os.Exit(1)
}
if err := db.Update(func(tx *bolt.Tx) error {
_, err := tx.CreateBucketIfNotExists(trackBucket)
return err
}); err != nil {
fmt.Fprintf(os.Stderr, "Error: could not create bucket: %v\n", err)
os.Exit(1)
}
}