Skip to content

Commit

Permalink
token renewal added
Browse files Browse the repository at this point in the history
  • Loading branch information
drunkleen committed Jul 2, 2024
1 parent 99b1035 commit aa619c9
Show file tree
Hide file tree
Showing 2 changed files with 10 additions and 9 deletions.
10 changes: 9 additions & 1 deletion main.go
Original file line number Diff line number Diff line change
Expand Up @@ -77,7 +77,15 @@ func mainLoop(queryList []string) {
// fetching user info
userInfo, err := requests.GetUserInfo(token, queryID)
if err != nil {
log.Printf(red("Error: %v\n"), err)
if err.Error() == "token is invalid" {
newToken, err := requests.GetNewToken(queryID)
if err != nil {
log.Printf(red("Error: %v\n"), err)
}
tokenMap[queryID] = newToken
} else {
log.Printf(red("Error: %v\n"), err)
}
}
username := userInfo["username"]
printText += fmt.Sprintf("---\n"+"["+bold(cyan("User"))+"] "+"%s\n", bold(green(username)))
Expand Down
9 changes: 1 addition & 8 deletions requests/requests.go
Original file line number Diff line number Diff line change
Expand Up @@ -127,14 +127,7 @@ func GetUserInfo(token string, queryID string) (map[string]any, error) {
}

if tokenResponse.Message == "Token is invalid" {
fmt.Println("Token is invalid, getting a new token...")
newToken, err := GetNewToken(queryID)
if err != nil {
fmt.Println("Failed to get new token.")
return nil, err
}
fmt.Println("New token obtained, try again...")
return GetUserInfo(newToken, queryID)
return nil, fmt.Errorf("token is invalid")
} else {
fmt.Println("Failed to get user information.")
return nil, fmt.Errorf("unexpected status code: %d", resp.StatusCode)
Expand Down

0 comments on commit aa619c9

Please sign in to comment.