From ae3dfc95b15a4e8006ad9004e38f343c1adb0509 Mon Sep 17 00:00:00 2001
From: itouakirai <85016486+itouakirai@users.noreply.github.com>
Date: Sun, 28 Apr 2024 09:51:09 +0800
Subject: [PATCH] add config.yaml support

---
 main_atmos.go | 12 +++++-------
 1 file changed, 5 insertions(+), 7 deletions(-)

diff --git a/main_atmos.go b/main_atmos.go
index cc280ac..09de4c4 100644
--- a/main_atmos.go
+++ b/main_atmos.go
@@ -1249,12 +1249,10 @@ func rip(albumId string, token string, storefront string, userToken string) erro
 }
 
 func main() {
-	var mediaUserToken string
-	if _, err := os.Stat("media-user-token.txt"); err == nil {
-		file, err := os.ReadFile("media-user-token.txt")
-		if err == nil && file != nil {
-			mediaUserToken = string(file)
-		}
+	err := loadConfig()
+	if err != nil {
+		fmt.Printf("load config failed: %v", err)
+		return
 	}
 	token, err := getToken()
 	if err != nil {
@@ -1274,7 +1272,7 @@ func main() {
 			fmt.Printf("Invalid URL: %s\n", url)
 			continue
 		}
-		err := rip(albumId, token, storefront, mediaUserToken)
+		err := rip(albumId, token, storefront, config.MediaUserToken)
 		if err != nil {
 			fmt.Println("Album failed.")
 			fmt.Println(err)