Add error handling for failed connection when pulling initial token.
This commit is contained in:
parent
446ed2bc74
commit
bc77b71659
16
Grabazine.go
16
Grabazine.go
|
|
@ -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{
|
||||||
|
|
|
||||||
BIN
built/zinigo
BIN
built/zinigo
Binary file not shown.
BIN
built/zinigo.exe
BIN
built/zinigo.exe
Binary file not shown.
Loading…
Reference in New Issue