Add error handling for failed connection when pulling initial token.

This commit is contained in:
Barold 2020-02-24 19:57:51 +02:00
parent 446ed2bc74
commit bc77b71659
3 changed files with 12 additions and 4 deletions

View File

@ -36,7 +36,10 @@ func main() {
flag.Parse() flag.Parse()
fmt.Println("Starting the application...") fmt.Println("Starting the application...")
initialToken := GetInitialToken() initialToken, err := GetInitialToken()
if err != nil {
os.Exit(1)
}
loginToken := GetLoginToken(initialToken, *usernamePtr, *passwordPtr) loginToken := GetLoginToken(initialToken, *usernamePtr, *passwordPtr)
issues := GetLibrary(loginToken) issues := GetLibrary(loginToken)
fmt.Println("Found " + strconv.Itoa(len(issues.Data))+ " issues in library.") fmt.Println("Found " + strconv.Itoa(len(issues.Data))+ " issues in library.")
@ -121,15 +124,20 @@ func GetPages(userToken LoginDto.Response, issue LibraryDto.Data) Response {
return responseType return responseType
} }
func GetInitialToken() string { func GetInitialToken() (token string, err error) {
page, _ := http.Get("https://www.zinio.com/za/sign-in") page, err := http.Get("https://www.zinio.com/za/sign-in")
if err != nil {
fmt.Println("Unable to get initial token: " + err.Error())
return "", err
}
data, _ := ioutil.ReadAll(page.Body) data, _ := ioutil.ReadAll(page.Body)
re := regexp.MustCompile(`"(jwt)":"((\\"|[^"])*)"`) re := regexp.MustCompile(`"(jwt)":"((\\"|[^"])*)"`)
found := re.FindSubmatch(data) found := re.FindSubmatch(data)
return string(found[2]) return string(found[2]), nil
} }
func GetLoginToken(initialToken string, username string, password string) LoginDto.Response{ func GetLoginToken(initialToken string, username string, password string) LoginDto.Response{

Binary file not shown.

Binary file not shown.