klinsmaya/CLIProxyAPI
1
0
Fork 0
mirror of https://gh-proxy.org/https://github.com/klinsmaya/CLIProxyAPI.git synced 2026-11-08 01:16:47 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(oauth): hide completed sessions from legacy getter

Browse Source
This commit is contained in:
Supra4E8C
2026-07-10 16:17:01 +08:00
parent f081b91e57
commit d1ef06cb5e
2 changed files with 18 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/api/handlers/management/oauth_sessions.go
+1 -1
View File
@@ -259,7 +259,7 @@ func CompletePluginOAuthSessionsByProvider(provider string) int {
func GetOAuthSession(state string) (provider string, status string, ok bool) {
session, ok := oauthSessions.Get(state)
if !ok {
if !ok || session.Completed {
return "", "", false
}
return session.Provider, session.Status, true
internal/api/handlers/management/oauth_sessions_test.go
+17
View File
@@ -26,6 +26,23 @@ func TestOAuthSessionStoreCompleteKeepsShortLivedSession(t *testing.T) {
}
}
func TestGetOAuthSessionHidesCompletedSession(t *testing.T) {
store := newOAuthSessionStore(time.Minute)
replaceOAuthSessionStoreForTest(t, store)
store.Register("completed-state", "codex")
store.Complete("completed-state")
provider, status, ok := GetOAuthSession("completed-state")
if ok {
t.Fatalf("GetOAuthSession() = (%q, %q, true), want completed session hidden", provider, status)
}
_, _, _, _, completed, detailsOK := GetOAuthSessionDetails("completed-state")
if !detailsOK || !completed {
t.Fatalf("GetOAuthSessionDetails() completed/ok = %t/%t, want true/true", completed, detailsOK)
}
}
func TestGetAuthStatusRejectsUnknownStateAndAcceptsCompletedState(t *testing.T) {
store := newOAuthSessionStore(time.Minute)
replaceOAuthSessionStoreForTest(t, store)