mirror of
https://github.com/cgzirim/seek-tune.git
synced 2025-12-17 00:44:19 +00:00
9886871070
Add selective deletion to erase command: - 'db' (default): clear only database - 'all': clear database and song files Updated README and help messages with new usage examples.
110 lines
3 KiB
Go
110 lines
3 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"flag"
|
|
"fmt"
|
|
"log/slog"
|
|
"os"
|
|
"song-recognition/utils"
|
|
|
|
"github.com/joho/godotenv"
|
|
"github.com/mdobak/go-xerrors"
|
|
)
|
|
|
|
func main() {
|
|
err := utils.CreateFolder("tmp")
|
|
if err != nil {
|
|
logger := utils.GetLogger()
|
|
err := xerrors.New(err)
|
|
ctx := context.Background()
|
|
logger.ErrorContext(ctx, "Failed create tmp dir.", slog.Any("error", err))
|
|
}
|
|
|
|
err = utils.CreateFolder(SONGS_DIR)
|
|
if err != nil {
|
|
err := xerrors.New(err)
|
|
logger := utils.GetLogger()
|
|
ctx := context.Background()
|
|
logMsg := fmt.Sprintf("failed to create directory %v", SONGS_DIR)
|
|
logger.ErrorContext(ctx, logMsg, slog.Any("error", err))
|
|
}
|
|
|
|
if len(os.Args) < 2 {
|
|
fmt.Println("Expected 'find', 'download', 'erase', 'save', or 'serve' subcommands")
|
|
fmt.Println("\nUsage examples:")
|
|
fmt.Println(" find <path_to_wav_file>")
|
|
fmt.Println(" download <spotify_url>")
|
|
fmt.Println(" erase [db | all] (default: db)")
|
|
fmt.Println(" save [-f|--force] <path_to_file_or_dir>")
|
|
fmt.Println(" serve [-proto <http|https>] [-p <port>]")
|
|
os.Exit(1)
|
|
}
|
|
_ = godotenv.Load()
|
|
|
|
switch os.Args[1] {
|
|
case "find":
|
|
if len(os.Args) < 3 {
|
|
fmt.Println("Usage: main.go find <path_to_wav_file>")
|
|
os.Exit(1)
|
|
}
|
|
filePath := os.Args[2]
|
|
find(filePath)
|
|
case "download":
|
|
if len(os.Args) < 3 {
|
|
fmt.Println("Usage: main.go download <spotify_url>")
|
|
os.Exit(1)
|
|
}
|
|
url := os.Args[2]
|
|
download(url)
|
|
case "serve":
|
|
serveCmd := flag.NewFlagSet("serve", flag.ExitOnError)
|
|
protocol := serveCmd.String("proto", "http", "Protocol to use (http or https)")
|
|
port := serveCmd.String("p", "5000", "Port to use")
|
|
serveCmd.Parse(os.Args[2:])
|
|
serve(*protocol, *port)
|
|
case "erase":
|
|
// Default is to clear only database (db mode)
|
|
dbOnly := true
|
|
all := false
|
|
|
|
if len(os.Args) > 2 {
|
|
subCmd := os.Args[2]
|
|
switch subCmd {
|
|
case "db":
|
|
dbOnly = true
|
|
all = false
|
|
case "all":
|
|
dbOnly = false
|
|
all = true
|
|
default:
|
|
fmt.Println("Usage: main.go erase [db | all]")
|
|
fmt.Println(" db : only clear the database (default)")
|
|
fmt.Println(" all : clear database and songs folder")
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
erase(SONGS_DIR, dbOnly, all)
|
|
case "save":
|
|
indexCmd := flag.NewFlagSet("save", flag.ExitOnError)
|
|
force := indexCmd.Bool("force", false, "save song with or without YouTube ID")
|
|
indexCmd.BoolVar(force, "f", false, "save song with or without YouTube ID (shorthand)")
|
|
indexCmd.Parse(os.Args[2:])
|
|
if indexCmd.NArg() < 1 {
|
|
fmt.Println("Usage: main.go save [-f|--force] <path_to_wav_file_or_dir>")
|
|
os.Exit(1)
|
|
}
|
|
filePath := indexCmd.Arg(0)
|
|
save(filePath, *force)
|
|
default:
|
|
fmt.Println("Expected 'find', 'download', 'erase', 'save', or 'serve' subcommands")
|
|
fmt.Println("\nUsage examples:")
|
|
fmt.Println(" find <path_to_wav_file>")
|
|
fmt.Println(" download <spotify_url>")
|
|
fmt.Println(" erase [db | all] (default: db)")
|
|
fmt.Println(" save [-f|--force] <path_to_file_or_dir>")
|
|
fmt.Println(" serve [-proto <http|https>] [-p <port>]")
|
|
os.Exit(1)
|
|
}
|
|
}
|