feat: add local DB to save downloads

cmd/download.go

@@ -5,9 +5,11 @@ import (
 	"os"
 	"path"
 	"strings"
+	"time"
 
 	"github.com/flytam/filenamify"
 	"github.com/mathismqn/godeez/internal/config"
+	"github.com/mathismqn/godeez/internal/db"
 	"github.com/mathismqn/godeez/internal/deezer"
 	"github.com/mathismqn/godeez/internal/tags"
 	"github.com/mathismqn/godeez/internal/utils"
@@ -126,11 +128,9 @@ func downloadContent(contentType string, args []string) {
 				songTitle = fmt.Sprintf("%s %s", song.Title, song.Version)
 			}
 
-			fmt.Printf("    Downloading %s...", songTitle)
-
 			media, err := song.GetMediaData(session.LicenseToken, quality)
 			if err != nil {
-				fmt.Printf("\r    Downloading %s... FAILED\n", songTitle)
+				fmt.Printf("    Downloading %s... FAILED\n", songTitle)
 				fmt.Fprintf(os.Stderr, "Error: could not get media data: %v\n", err)
 				if err.Error() == "invalid license token" {
 					os.Exit(1)
@@ -139,7 +139,7 @@ func downloadContent(contentType string, args []string) {
 			}
 
 			if len(media.Data) == 0 || len(media.Data[0].Media) == 0 || len(media.Data[0].Media[0].Sources) == 0 {
-				fmt.Printf("\r    Downloading %s... FAILED\n", songTitle)
+				fmt.Printf("    Downloading %s... FAILED\n", songTitle)
 				fmt.Fprintf(os.Stderr, "Error: could not get media sources\n")
 				continue
 			}
@@ -165,6 +165,15 @@ func downloadContent(contentType string, args []string) {
 			fileName, _ = filenamify.Filenamify(fileName, filenamify.Options{})
 			filePath := path.Join(output, fileName)
 
+			if existing, err := db.Get(song.ID); err == nil {
+				if existing.Quality == media.Data[0].Media[0].Format && utils.FileExists(existing.Path) {
+					fmt.Printf("    Skipping %s (already downloaded at %s)\n", songTitle, existing.Path)
+					continue
+				}
+			}
+
+			fmt.Printf("    Downloading %s...", songTitle)
+
 			err = media.Download(url, filePath, song.ID)
 			if err != nil {
 				fmt.Printf("\r    Downloading %s... FAILED\n", songTitle)
@@ -184,6 +193,16 @@ func downloadContent(contentType string, args []string) {
 			if err := tags.AddTags(resource, song, filePath, tempo, key); err != nil {
 				fmt.Fprintf(os.Stderr, "Warning: could not add tags to song: %v\n", err)
 			}
+
+			info := &db.DownloadInfo{
+				SongID:     song.ID,
+				Quality:    media.Data[0].Media[0].Format,
+				Path:       filePath,
+				Downloaded: time.Now(),
+			}
+			if err := info.Save(); err != nil {
+				fmt.Fprintf(os.Stderr, "Warning: could not save download info: %v\n", err)
+			}
 		}
 	}
 

cmd/root.go

@@ -1,13 +1,9 @@
 package cmd
 
 import (
-	"fmt"
-	"os"
-	"path"
-
 	"github.com/mathismqn/godeez/internal/config"
+	"github.com/mathismqn/godeez/internal/db"
 	"github.com/spf13/cobra"
-	"github.com/spf13/viper"
 )
 
 var cfgFile string
@@ -22,53 +18,8 @@ var RootCmd = &cobra.Command{
 
 func init() {
 	RootCmd.PersistentFlags().StringVar(&cfgFile, "config", "", "config file (default is $HOME/.godeez)")
-	cobra.OnInitialize(initConfig)
-}
-
-func initConfig() {
-	if cfgFile != "" {
-		viper.SetConfigFile(cfgFile)
-	} else {
-		homedir, err := os.UserHomeDir()
-		if err != nil {
-			fmt.Fprintf(os.Stderr, "Error: could not get home directory: %v\n", err)
-			os.Exit(1)
-		}
-
-		path := path.Join(homedir, ".godeez")
-		if _, err := os.Stat(path); os.IsNotExist(err) {
-			fmt.Printf("Config file not found, creating one at %s\n", path)
-
-			content := []byte("arl_cookie = ''\nsecret_key = ''\niv = '0001020304050607'\n")
-			if err := os.WriteFile(path, content, 0644); err != nil {
-				fmt.Fprintf(os.Stderr, "Error: could not create config file: %v\n", err)
-				os.Exit(1)
-			}
-		}
-
-		viper.AddConfigPath(homedir)
-		viper.SetConfigName(".godeez")
-	}
-
-	viper.SetConfigType("toml")
-	viper.AutomaticEnv()
-	if err := viper.ReadInConfig(); err != nil {
-		fmt.Fprintf(os.Stderr, "Error: could not read config file: %v\n", err)
-		os.Exit(1)
-	}
-
-	cfg := &config.Cfg
-	if err := viper.Unmarshal(&cfg); err != nil {
-		fmt.Fprintf(os.Stderr, "Error: could not unmarshal config file: %v\n", err)
-		os.Exit(1)
-	}
-
-	if cfg.SecretKey == "" {
-		fmt.Fprintln(os.Stderr, "Error: secret_key is not set in config file")
-		os.Exit(1)
-	}
-	if cfg.IV == "" {
-		fmt.Fprintln(os.Stderr, "Error: iv is not set in config file")
-		os.Exit(1)
-	}
+	cobra.OnInitialize(func() {
+		config.Init(cfgFile)
+		db.Init()
+	})
 }