From bb52f9d45b2524633b944a6c6117f4802d379791 Mon Sep 17 00:00:00 2001 From: Franck Cuny Date: Sun, 10 Oct 2021 16:10:32 -0700 Subject: mpd-scrobbler: proper default arguments The program needs two arguments: the mpd host and port, which can be passed as flags (default is to use the local instance of mpd). We store the database in `XDG_CONFIG_HOME/mpd-scrobbler`, and we create the path if needed. --- tools/mpd-stats/cmd/mpd-scrobbler/main.go | 33 +++++++++++++++++++++++++++++-- 1 file changed, 31 insertions(+), 2 deletions(-) (limited to 'tools/mpd-stats') diff --git a/tools/mpd-stats/cmd/mpd-scrobbler/main.go b/tools/mpd-stats/cmd/mpd-scrobbler/main.go index fdd9e4d..c2693a4 100644 --- a/tools/mpd-stats/cmd/mpd-scrobbler/main.go +++ b/tools/mpd-stats/cmd/mpd-scrobbler/main.go @@ -1,17 +1,29 @@ package main import ( + "flag" "fmt" "log" "os" + "path/filepath" "golang.fcuny.net/mpd-stats/internal/scrobbler" ) func main() { + var ( + mpdHost = flag.String("host", "localhost", "The MPD server to connect to (default: localhost)") + mpdPort = flag.Int("port", 6600, "The TCP port of the MPD server to connect to (default: 6600)") + ) + flag.Parse() + net := "tcp" - addr := "localhost:6600" - dbpath := fmt.Sprintf("%s/.config/scrobbler.sql", os.Getenv("HOME")) + addr := fmt.Sprintf("%s:%d", *mpdHost, *mpdPort) + + dbpath, err := getDbPath() + if err != nil { + log.Fatalf("failed to get the path to the database: %v", err) + } s, err := scrobbler.NewScrobbler(net, addr, dbpath) if err != nil { @@ -26,3 +38,20 @@ func main() { s.Run() } + +func getDbPath() (string, error) { + xch := os.Getenv("XDG_CONFIG_HOME") + if xch == "" { + home := os.Getenv("HOME") + xch = filepath.Join(home, ".config") + } + + scrobblerHome := filepath.Join(xch, "mpd-scrobbler") + if _, err := os.Stat(scrobblerHome); os.IsNotExist(err) { + if err := os.Mkdir(scrobblerHome, 0755); err != nil { + return "", err + } + } + + return filepath.Join(scrobblerHome, "scrobbler.sql"), nil +} -- cgit 1.4.1