add config.yaml support

pull/14/head
itouakirai 5 months ago committed by GitHub
parent 112dfff400
commit ae3dfc95b1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
  1. 12
      main_atmos.go

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

Loading…
Cancel
Save