diff --git a/main.go b/main.go index cbf102e..bd2f157 100644 --- a/main.go +++ b/main.go @@ -60,6 +60,11 @@ type DownloadItem struct { var downloads = map[int]*DownloadItem{} func main() { + musicDir := os.Getenv("MUSIC_DIR") + if musicDir == "" { + musicDir = "./Music" + } + mClient := monochrome.NewClient(monochrome.ClientConfig{ ApiURL: "https://api.monochrome.tf", }) @@ -167,7 +172,7 @@ func main() { log.Printf("failed to download album cover: %v\n", err) } - if err := os.Mkdir("./Music/"+download.Album.Title, 0777); err != nil { + if err := os.Mkdir(musicDir+download.Album.Title, 0777); err != nil { return } @@ -209,7 +214,7 @@ func main() { return } - fPath := path.Join("./Music/", download.Album.Title, track.Item.Title+".flac") + fPath := path.Join(musicDir, download.Album.Title, track.Item.Title+".flac") if err := os.WriteFile(fPath, data, 0644); err != nil { log.Printf("failed to save track file: %v\n", err) return