fix(coding-agent): show models.json auth in login selector
This commit is contained in:
@@ -1,44 +0,0 @@
|
||||
import type { AuthCredential, AuthStatus } from "../../../core/auth-storage.js";
|
||||
|
||||
export type SelectorAuthType = "oauth" | "api_key";
|
||||
|
||||
export type AuthSelectorIndicator =
|
||||
| { kind: "configured"; label: string }
|
||||
| { kind: "configured-other"; label: string }
|
||||
| { kind: "runtime" }
|
||||
| { kind: "environment"; label: string }
|
||||
| { kind: "fallback" }
|
||||
| { kind: "unconfigured" };
|
||||
|
||||
function getConfiguredLabel(authType: SelectorAuthType): string {
|
||||
return authType === "oauth" ? "subscription configured" : "api key configured";
|
||||
}
|
||||
|
||||
export function getAuthSelectorIndicator(
|
||||
authType: SelectorAuthType,
|
||||
credential: AuthCredential | undefined,
|
||||
authStatus?: AuthStatus,
|
||||
): AuthSelectorIndicator {
|
||||
if (credential) {
|
||||
const label = getConfiguredLabel(credential.type);
|
||||
return credential.type === authType ? { kind: "configured", label } : { kind: "configured-other", label };
|
||||
}
|
||||
|
||||
if (authType === "oauth") {
|
||||
return { kind: "unconfigured" };
|
||||
}
|
||||
|
||||
if (authStatus?.source === "runtime") {
|
||||
return { kind: "runtime" };
|
||||
}
|
||||
|
||||
if (authStatus?.source === "environment") {
|
||||
return { kind: "environment", label: authStatus.label ?? "API key" };
|
||||
}
|
||||
|
||||
if (authStatus?.source === "fallback") {
|
||||
return { kind: "fallback" };
|
||||
}
|
||||
|
||||
return { kind: "unconfigured" };
|
||||
}
|
||||
@@ -7,9 +7,8 @@ import {
|
||||
Spacer,
|
||||
TruncatedText,
|
||||
} from "@mariozechner/pi-tui";
|
||||
import type { AuthStorage } from "../../../core/auth-storage.js";
|
||||
import type { AuthStatus, AuthStorage } from "../../../core/auth-storage.js";
|
||||
import { theme } from "../theme/theme.js";
|
||||
import { getAuthSelectorIndicator } from "./auth-selector-status.js";
|
||||
import { DynamicBorder } from "./dynamic-border.js";
|
||||
|
||||
export type AuthSelectorProvider = {
|
||||
@@ -40,6 +39,7 @@ export class OAuthSelectorComponent extends Container implements Focusable {
|
||||
private selectedIndex: number = 0;
|
||||
private mode: "login" | "logout";
|
||||
private authStorage: AuthStorage;
|
||||
private getAuthStatus: (providerId: string) => AuthStatus;
|
||||
private onSelectCallback: (providerId: string) => void;
|
||||
private onCancelCallback: () => void;
|
||||
|
||||
@@ -49,11 +49,13 @@ export class OAuthSelectorComponent extends Container implements Focusable {
|
||||
providers: AuthSelectorProvider[],
|
||||
onSelect: (providerId: string) => void,
|
||||
onCancel: () => void,
|
||||
getAuthStatus?: (providerId: string) => AuthStatus,
|
||||
) {
|
||||
super();
|
||||
|
||||
this.mode = mode;
|
||||
this.authStorage = authStorage;
|
||||
this.getAuthStatus = getAuthStatus ?? ((providerId) => this.authStorage.getAuthStatus(providerId));
|
||||
this.allProviders = providers;
|
||||
this.filteredProviders = providers;
|
||||
this.onSelectCallback = onSelect;
|
||||
@@ -147,29 +149,29 @@ export class OAuthSelectorComponent extends Container implements Focusable {
|
||||
}
|
||||
|
||||
private formatStatusIndicator(provider: AuthSelectorProvider): string {
|
||||
const status = provider.authType === "api_key" ? this.authStorage.getAuthStatus(provider.id) : undefined;
|
||||
const indicator = getAuthSelectorIndicator(provider.authType, this.authStorage.get(provider.id), status);
|
||||
const credential = this.authStorage.get(provider.id);
|
||||
if (credential?.type === provider.authType) return theme.fg("success", " ✓ configured");
|
||||
if (credential) {
|
||||
const label = credential.type === "oauth" ? "subscription configured" : "API key configured";
|
||||
return theme.fg("muted", " • ") + theme.fg("warning", label);
|
||||
}
|
||||
if (provider.authType !== "api_key") return theme.fg("muted", " • unconfigured");
|
||||
|
||||
if (indicator.kind === "configured") {
|
||||
return theme.fg("success", ` ✓ ${indicator.label}`);
|
||||
const status = this.getAuthStatus(provider.id);
|
||||
switch (status.source) {
|
||||
case "environment":
|
||||
return theme.fg("success", ` ✓ env: ${status.label ?? "API key"}`);
|
||||
case "runtime":
|
||||
return theme.fg("success", " ✓ runtime API key");
|
||||
case "fallback":
|
||||
return theme.fg("success", " ✓ custom API key");
|
||||
case "models_json_key":
|
||||
return theme.fg("success", " ✓ key in models.json");
|
||||
case "models_json_command":
|
||||
return theme.fg("success", " ✓ command in models.json");
|
||||
default:
|
||||
return theme.fg("muted", " • unconfigured");
|
||||
}
|
||||
|
||||
if (indicator.kind === "configured-other") {
|
||||
return theme.fg("muted", " • ") + theme.fg("warning", indicator.label);
|
||||
}
|
||||
|
||||
const base = theme.fg("muted", " • unconfigured");
|
||||
if (indicator.kind === "environment") {
|
||||
return base + theme.fg("success", ` · env: ${indicator.label}`);
|
||||
}
|
||||
if (indicator.kind === "runtime") {
|
||||
return base + theme.fg("success", " · runtime API key");
|
||||
}
|
||||
if (indicator.kind === "fallback") {
|
||||
return base + theme.fg("success", " · custom API key");
|
||||
}
|
||||
|
||||
return base;
|
||||
}
|
||||
|
||||
handleInput(keyData: string): void {
|
||||
|
||||
Reference in New Issue
Block a user