Salastil/libkiwi
1
0
Fork 0
mirror of https://gitgud.io/yats/libkiwi.git synced 2026-06-28 23:12:04 -04:00
Code Issues Packages Projects Releases Wiki Activity

Fix IsLoggedIn() and some other error checking

Browse Source
This commit is contained in:
y a t s
2026-06-19 10:21:21 -04:00
parent 92438691f1
commit 74b923f06f
2 changed files with 11 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
auth.go
+4 -3
View File
@@ -123,7 +123,8 @@ func (kf *KF) TwoFactorAuth(ctx context.Context, resp *http.Response, code uint3
} }
func (kf *KF) IsLoggedIn() bool { func (kf *KF) IsLoggedIn() bool {
return kf.Cookies.GetCookie(kf.domain, "xf_user") != nil cookie := kf.Cookies.GetCookie(kf.domain, "xf_user")
return !(cookie == nil || cookie.Value == "")
} }
func (kf *KF) RefreshSession(ctx context.Context) (string, error) { func (kf *KF) RefreshSession(ctx context.Context) (string, error) {
@@ -142,8 +143,8 @@ func (kf *KF) RefreshSession(ctx context.Context) (string, error) {
resp.Body.Close() resp.Body.Close()
session := kf.Cookies.GetCookie(kf.domain, COOKIE_NAME) session := kf.Cookies.GetCookie(kf.domain, COOKIE_NAME)
if session == nil { if session == nil || session.Value == "" {
return "", errors.New("Failed to get new xf_session cookie.") return "", fmt.Errorf("Failed to get new %s cookie", COOKIE_NAME)
} }
return session.Value, nil return session.Value, nil
auth_test.go
+7
View File
@@ -64,12 +64,19 @@ func TestLogin(t *testing.T) {
t.Logf("2FA response: %+v\n", lr) t.Logf("2FA response: %+v\n", lr)
} }
t.Logf("Checking RefreshSession()")
session, err := kf.RefreshSession(ctx) session, err := kf.RefreshSession(ctx)
if err != nil { if err != nil {
t.Error(err) t.Error(err)
return return
} }
t.Logf("xf_session: %s\n", session) t.Logf("xf_session: %s\n", session)
t.Logf("Checking IsLoggedIn()\n")
if !kf.IsLoggedIn() {
t.Error("IsLoggedIn() test returned false.")
return
}
} }
func genTest2FACode() (uint32, error) { func genTest2FACode() (uint32, error) {