Salastil/libkiwi
1
0
Fork 0
mirror of https://gitgud.io/yats/libkiwi.git synced 2026-06-20 01:55:23 -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 {
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) {
@@ -142,8 +143,8 @@ func (kf *KF) RefreshSession(ctx context.Context) (string, error) {
resp.Body.Close()
session := kf.Cookies.GetCookie(kf.domain, COOKIE_NAME)
if session == nil {
return "", errors.New("Failed to get new xf_session cookie.")
if session == nil || session.Value == "" {
return "", fmt.Errorf("Failed to get new %s cookie", COOKIE_NAME)
}
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("Checking RefreshSession()")
session, err := kf.RefreshSession(ctx)
if err != nil {
t.Error(err)
return
}
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) {