diff --git a/auth.go b/auth.go index 5c3e101..e66f0bb 100644 --- a/auth.go +++ b/auth.go @@ -28,19 +28,21 @@ func (c *Client) LogIn(username, pass string) (*AuthUser, error) { } status := env.Code - if status == http.StatusOK { - c.SetToken(u.AccessToken) - return u, nil - } else if status == http.StatusBadRequest { - return nil, fmt.Errorf("Bad request: %s", env.ErrorMessage) - } else if status == http.StatusUnauthorized { - return nil, fmt.Errorf("Incorrect password.") - } else if status == http.StatusNotFound { - return nil, fmt.Errorf("User does not exist.") - } else if status == http.StatusTooManyRequests { - return nil, fmt.Errorf("Stop repeatedly trying to log in.") + if status != http.StatusOK { + if status == http.StatusBadRequest { + return nil, fmt.Errorf("Bad request: %s", env.ErrorMessage) + } else if status == http.StatusUnauthorized { + return nil, fmt.Errorf("Incorrect password.") + } else if status == http.StatusNotFound { + return nil, fmt.Errorf("User does not exist.") + } else if status == http.StatusTooManyRequests { + return nil, fmt.Errorf("Stop repeatedly trying to log in.") + } + return nil, fmt.Errorf("Problem authenticating: %d. %v\n", status, err) } - return nil, fmt.Errorf("Problem authenticating: %d. %v\n", status, err) + + c.SetToken(u.AccessToken) + return u, nil } // LogOut logs the current user out, making the Client's current access token