@averyyy/pi-coding-agent 0.99.1-piclient.1 → 1.0.0-piclient.2

package/dist/modes/interactive/interactive-mode.js

@@ -26,6 +26,7 @@ import { defaultModelPerProvider, findExactModelReferenceMatch, resolveModelScop
 import { CredentialSynchronizationError } from "../../core/model-runtime.js";
 import { DefaultPackageManager } from "../../core/package-manager.js";
 import { getCliBinName } from "../../core/pi-client-cli-adapter.js";
+import { RADIUS_MCP_URL, RADIUS_PROVIDER_ID } from "../../core/radius.js";
 import { formatMissingSessionCwdPrompt, MissingSessionCwdError } from "../../core/session-cwd.js";
 import { SessionManager, sessionEntryToContextMessages, } from "../../core/session-manager.js";
 import { BUILTIN_SLASH_COMMANDS } from "../../core/slash-commands.js";
@@ -33,6 +34,7 @@ import { isInstallTelemetryEnabled } from "../../core/telemetry.js";
 import { withBuiltInRenderers } from "../../core/tools/renderers/index.js";
 import { hasTrustRequiringProjectResources, ProjectTrustStore } from "../../core/trust-manager.js";
 import { getUsageCostBreakdown } from "../../core/usage-totals.js";
+import { addMcpServerConfig, loadMcpConfig } from "../../extensions/mcp/config.js";
 import { getChangelogPath, getNewEntries, normalizeChangelogLinks, parseChangelog } from "../../utils/changelog.js";
 import { copyToClipboard, readClipboardFilePaths, readClipboardText } from "../../utils/clipboard.js";
 import { extensionForImageMimeType, readClipboardImage } from "../../utils/clipboard-image.js";
@@ -64,8 +66,10 @@ import { formatKeyText, keyDisplayText, keyHint, keyText, rawKeyHint } from "./c
 import { LoginDialogComponent } from "./components/login-dialog.js";
 import { createMermaidMarkdownTransformer } from "./components/mermaid.js";
 import { ModelSelectorComponent } from "./components/model-selector.js";
-import { formatAuthSelectorProviderType, OAuthSelectorComponent, } from "./components/oauth-selector.js";
-import { piLogoLines } from "./components/pi-logo.js";
+import { formatAuthSelectorProviderStatus, formatAuthSelectorProviderType, OAuthSelectorComponent, } from "./components/oauth-selector.js";
+import { piLogoLines, piWordmark, supportsPiLogo } from "./components/pi-logo.js";
+import { playPiLogoAnimation } from "./components/pi-logo-animation.lazy.js";
+import { createLoginMenuSelector } from "./components/radius-login-selector.js";
 import { ScopedModelsSelectorComponent } from "./components/scoped-models-selector.js";
 import { SessionSelectorComponent } from "./components/session-selector.js";
 import { SettingsSelectorComponent } from "./components/settings-selector.js";
@@ -107,6 +111,16 @@ class ExpandableText extends ThemedText {
         this.invalidate();
     }
 }
+/** The built-in header. Clicking its logo (the first two lines, after one column of padding) plays an easter egg. */
+class BuiltInHeader extends ExpandableText {
+    onLogoClick;
+    handleMouse(event) {
+        if (event.type !== "click" || event.y > 1 || event.x < 1 || event.x > 4 || !this.onLogoClick)
+            return undefined;
+        this.onLogoClick(event.screenX - event.x + 1, event.screenY - event.y);
+        return { handled: true };
+    }
+}
 function isCustomSessionEntry(item) {
     return "type" in item && item.type === "custom";
 }
@@ -176,6 +190,7 @@ function llamaCppPostLoginGuidance(actionLabel, loadedModelCount) {
         ? `${actionLabel}. No llama.cpp models are loaded. Use /llama to load a model, then /model to select it.`
         : `${actionLabel}. Use /model to select a loaded llama.cpp model, or /llama to manage models.`;
 }
+const RADIUS_LOGIN_INTRO = "Radius is a service crafted for Pi by the builders of Pi, Earendil Works";
 const AUTH_TYPE_ORDER = { oauth: 0, api_key: 1 };
 function createFuzzyAutocompleteItems(items, prefix, getSearchText, toAutocompleteItem) {
     const filtered = fuzzyFilter(items, prefix, getSearchText);
@@ -198,18 +213,21 @@ function getLoginProviderCompletionOptions(providerOptions) {
             id: provider.id,
             name: provider.name,
             authTypes: [provider.authType],
+            subscription: provider.subscription,
         });
     }
     return Array.from(byId.values()).sort((a, b) => a.name.localeCompare(b.name));
 }
 function getLoginProviderSearchText(provider) {
     const authTypes = provider.authTypes
-        .map((authType) => `${authType} ${formatAuthSelectorProviderType(authType)}`)
+        .map((authType) => `${authType} ${formatAuthSelectorProviderType(authType, provider.subscription)}`)
         .join(" ");
     return `${provider.id} ${provider.name} ${authTypes}`;
 }
 function formatLoginProviderCompletionDescription(provider) {
-    const authTypes = provider.authTypes.map(formatAuthSelectorProviderType).join("/");
+    const authTypes = provider.authTypes
+        .map((authType) => formatAuthSelectorProviderType(authType, provider.subscription))
+        .join("/");
     return provider.name === provider.id ? authTypes : `${provider.name} · ${authTypes}`;
 }
 export class InteractiveMode {
@@ -642,7 +660,7 @@ export class InteractiveMode {
         this.registerSignalHandlers();
         // Load changelog (only show new entries, skip for resumed sessions)
         this.changelogMarkdown = this.getChangelogForDisplay();
-        if (this.session.scopedModels.length > 0 && (this.options.verbose || !this.settingsManager.getQuietStartup())) {
+        if (this.session.scopedModels.length > 0 && this.shouldShowStartupDetails()) {
             const modelList = this.session.scopedModels
                 .map((sm) => {
                 const thinkingStr = sm.thinkingLevel ? `:${sm.thinkingLevel}` : "";
@@ -694,10 +712,15 @@ export class InteractiveMode {
         // reported its colors. This ends at the terminal's DA1 reply, or after 100 ms if it answers nothing.
         await this.themeController.waitForTerminalColors();
         // Add header with keybindings from config (unless silenced)
-        if (this.options.verbose || !this.settingsManager.getQuietStartup()) {
+        if (this.shouldShowStartupHeader()) {
+            const showDetails = this.shouldShowStartupDetails();
             // Built on demand so the header follows theme changes. The logo's first line carries the version,
-            // its second line the first line of key hints.
+            // its second line the first line of key hints. Terminals that cannot render the logo get a
+            // "Pi vX" line instead, with the key hints below it.
+            const showLogo = supportsPiLogo();
             const withLogo = (hints) => {
+                if (!showLogo)
+                    return `${piWordmark()} ${theme.fg("dim", `v${this.version}`)}\n${hints}`;
                 const [top, bottom] = piLogoLines();
                 return `${top} ${theme.fg("dim", `v${this.version}`)}\n${bottom} ${hints}`;
             };
@@ -731,9 +754,12 @@ export class InteractiveMode {
                 rawKeyHint("!", "bash"),
                 hint("app.tools.expand", "more"),
             ].join(theme.fg("muted", " · "));
-            const compactOnboarding = () => theme.fg("dim", `Press ${keyText("app.tools.expand")} to show full startup help and loaded resources.`);
+            const compactOnboarding = () => theme.fg("dim", `Press ${keyText("app.tools.expand")} to show full startup help${showDetails ? " and loaded resources" : ""}.`);
             const onboarding = () => theme.fg("dim", `Pi can explain its own features and look up its docs. Ask it how to use or extend Pi.`);
-            this.builtInHeader = new ExpandableText(() => `${withLogo(compactInstructions())}\n${compactOnboarding()}\n\n${onboarding()}`, () => `${withLogo(expandedInstructions())}\n\n${onboarding()}`, this.getStartupExpansionState(), 1, 0);
+            const header = new BuiltInHeader(() => `${withLogo(compactInstructions())}\n${compactOnboarding()}\n\n${onboarding()}`, () => `${withLogo(expandedInstructions())}\n\n${onboarding()}`, this.getStartupExpansionState(), 1, 0);
+            if (showLogo)
+                header.onLogoClick = (column, row) => playPiLogoAnimation(this.renderer, column, row);
+            this.builtInHeader = header;
             // Setup UI layout
             this.headerContainer.addChild(new Spacer(1));
             this.headerContainer.addChild(this.builtInHeader);
@@ -1046,6 +1072,14 @@ export class InteractiveMode {
     getStartupExpansionState() {
         return this.options.verbose || this.toolOutputExpanded;
     }
+    /** Startup header (logo, version, key hints). Hidden only by quietStartup: true. */
+    shouldShowStartupHeader() {
+        return this.options.verbose === true || this.settingsManager.getQuietStartup() !== true;
+    }
+    /** Startup details (model scope, loaded resources). Hidden by quietStartup: true or "header". */
+    shouldShowStartupDetails() {
+        return this.options.verbose === true || this.settingsManager.getQuietStartup() === false;
+    }
     /**
      * Get a short path relative to the package root for display.
      */
@@ -1307,7 +1341,7 @@ export class InteractiveMode {
     showLoadedResources(options) {
         // Resource rendering is idempotent; chat clears no longer clear this separate container.
         this.loadedResourcesContainer.clear();
-        const showListing = options?.force || this.options.verbose || !this.settingsManager.getQuietStartup();
+        const showListing = options?.force || this.shouldShowStartupDetails();
         const showDiagnostics = showListing || options?.showDiagnosticsWhenQuiet === true;
         if (!showListing && !showDiagnostics) {
             return;
@@ -4299,8 +4333,8 @@ export class InteractiveMode {
                 onEnableInstallTelemetryChange: (enabled) => {
                     this.settingsManager.setEnableInstallTelemetry(enabled);
                 },
-                onQuietStartupChange: (enabled) => {
-                    this.settingsManager.setQuietStartup(enabled);
+                onQuietStartupChange: (quiet) => {
+                    this.settingsManager.setQuietStartup(quiet);
                 },
                 onDefaultProjectTrustChange: (defaultProjectTrust) => {
                     this.settingsManager.setDefaultProjectTrust(defaultProjectTrust);
@@ -4952,6 +4986,7 @@ export class InteractiveMode {
                     source: authStatus.label ?? authStatus.source,
                 }
                 : undefined;
+            const subscription = provider.auth.oauth?.isSubscription === true;
             if ((!authType || authType === "oauth") && provider.auth.oauth) {
                 options.push({
                     id: provider.id,
@@ -4959,6 +4994,7 @@ export class InteractiveMode {
                     authType: "oauth",
                     method: provider.auth.oauth,
                     status,
+                    subscription,
                 });
             }
             if ((!authType || authType === "api_key") && provider.auth.apiKey) {
@@ -4968,6 +5004,7 @@ export class InteractiveMode {
                     authType: "api_key",
                     method: provider.auth.apiKey,
                     status,
+                    subscription,
                 });
             }
         }
@@ -4975,12 +5012,16 @@ export class InteractiveMode {
     }
     async getLogoutProviderOptions() {
         return (await this.session.modelRuntime.listCredentials({ signal: AbortSignal.timeout(15_000) }))
-            .map(({ providerId, type }) => ({
-            id: providerId,
-            name: this.session.modelRuntime.getProvider(providerId)?.name ?? providerId,
-            authType: type,
-            status: { type, source: "stored credential" },
-        }))
+            .map(({ providerId, type }) => {
+            const provider = this.session.modelRuntime.getProvider(providerId);
+            return {
+                id: providerId,
+                name: provider?.name ?? providerId,
+                authType: type,
+                status: { type, source: "stored credential" },
+                subscription: provider?.auth.oauth?.isSubscription === true,
+            };
+        })
             .sort((a, b) => a.name.localeCompare(b.name));
     }
     findLoginProviderOptions(providerRef) {
@@ -5010,18 +5051,25 @@ export class InteractiveMode {
         }
         this.showLoginProviderSelector(undefined, providerRef);
     }
-    async startProviderLogin(providerOption) {
+    /** `onBack` reopens the selector the login was started from when the user cancels it. */
+    async startProviderLogin(providerOption, onBack) {
         if (providerOption.authType === "oauth") {
-            await this.showLoginDialog(providerOption.id, providerOption.name);
+            await this.showLoginDialog(providerOption.id, providerOption.name, onBack);
         }
         else if (providerOption.method?.login) {
-            await this.showApiKeyLoginDialog(providerOption.id, providerOption.name);
+            await this.showApiKeyLoginDialog(providerOption.id, providerOption.name, onBack);
         }
         else {
-            this.showAmbientAuthDialog(providerOption);
+            this.showAmbientAuthDialog(providerOption, onBack);
         }
     }
     showLoginAuthTypeSelector(providerOptions) {
+        // The top-level selector offers Radius directly, as its last option.
+        const radiusOption = providerOptions
+            ? undefined
+            : this.getLoginProviderOptions("oauth").find((provider) => provider.id === RADIUS_PROVIDER_ID);
+        const radiusText = radiusOption ? `Sign in with ${radiusOption.name}` : undefined;
+        const radiusLabel = radiusOption ? `${radiusText}${formatAuthSelectorProviderStatus(radiusOption)}` : undefined;
         const oauthProvider = providerOptions?.find((provider) => provider.authType === "oauth");
         const oauthLoginLabel = oauthProvider?.method && "loginLabel" in oauthProvider.method ? oauthProvider.method.loginLabel : undefined;
         const subscriptionLabel = oauthLoginLabel ?? "Sign in with an account";
@@ -5036,6 +5084,8 @@ export class InteractiveMode {
         if (availableAuthTypes.has("api_key")) {
             options.push(apiKeyLabel);
         }
+        if (radiusLabel)
+            options.push(radiusLabel);
         if (options.length === 0) {
             this.showStatus("No login methods available.");
             return;
@@ -5051,29 +5101,37 @@ export class InteractiveMode {
             ? `Select authentication method for ${providerOptions[0].name}:`
             : "Select authentication method:";
         this.showSelector((done) => {
-            const selector = new ExtensionSelectorComponent(title, options, (option) => {
+            const onSelect = (option) => {
                 done();
+                if (radiusOption && option === radiusLabel) {
+                    void this.startProviderLogin(radiusOption, () => this.showLoginAuthTypeSelector());
+                    return;
+                }
                 const authType = option === subscriptionLabel ? "oauth" : "api_key";
                 if (providerOptions) {
                     const providerOption = providerOptions.find((provider) => provider.authType === authType);
                     if (providerOption) {
-                        void this.startProviderLogin(providerOption);
+                        void this.startProviderLogin(providerOption, () => this.showLoginAuthTypeSelector(providerOptions));
                     }
                     return;
                 }
                 this.showLoginProviderSelector(authType);
-            }, () => {
+            };
+            const onCancel = () => {
                 done();
                 this.ui.requestRender();
-            });
-            return { component: selector, focus: selector };
+            };
+            const selector = radiusLabel && radiusText
+                ? createLoginMenuSelector(this.ui, title, options, { label: radiusLabel, text: radiusText }, onSelect, onCancel)
+                : new ExtensionSelectorComponent(title, options, onSelect, onCancel);
+            return { component: selector, focus: selector, dispose: () => selector.dispose() };
         });
     }
     showLoginProviderSelector(authType, initialSearchInput) {
         const providerOptions = this.getLoginProviderOptions(authType);
         if (providerOptions.length === 0) {
             const message = authType === "oauth"
-                ? "No subscription providers available."
+                ? "No account providers available."
                 : authType === "api_key"
                     ? "No API key providers available."
                     : "No login providers available.";
@@ -5087,7 +5145,7 @@ export class InteractiveMode {
                 if (!providerOption) {
                     return;
                 }
-                await this.startProviderLogin(providerOption);
+                await this.startProviderLogin(providerOption, () => this.showLoginProviderSelector(authType, initialSearchInput));
             }, () => {
                 done();
                 if (authType) {
@@ -5241,21 +5299,24 @@ export class InteractiveMode {
         })
             .finally(() => clearTimeout(timeout));
     }
-    showAmbientAuthDialog(providerOption) {
+    showAmbientAuthDialog(providerOption, onBack) {
         const restoreEditor = () => {
             this.editorContainer.clear();
             this.editorContainer.addChild(this.editor);
             this.ui.setFocus(this.editor);
             this.ui.requestRender();
         };
-        const dialog = new LoginDialogComponent(this.ui, providerOption.id, () => restoreEditor(), providerOption.name, `${providerOption.name} setup`);
+        const dialog = new LoginDialogComponent(this.ui, providerOption.id, () => {
+            restoreEditor();
+            onBack?.();
+        }, providerOption.name, `${providerOption.name} setup`);
         dialog.showInfo(`${providerOption.method?.name ?? "Authentication"} is configured outside ${APP_NAME}.`, [], true);
         this.editorContainer.clear();
         this.editorContainer.addChild(dialog);
         this.ui.setFocus(dialog);
         this.ui.requestRender();
     }
-    async showApiKeyLoginDialog(providerId, providerName) {
+    async showApiKeyLoginDialog(providerId, providerName, onBack) {
         const previousModel = this.session.model;
         const dialog = new LoginDialogComponent(this.ui, providerId, (_success, _message) => {
             // Completion handled below
@@ -5288,12 +5349,15 @@ export class InteractiveMode {
             if (error instanceof CredentialSynchronizationError) {
                 this.showError(`Saved API key for ${providerName}, but local model state could not be synchronized: ${errorMsg}`);
             }
-            else if (errorMsg !== "Login cancelled") {
+            else if (errorMsg === "Login cancelled") {
+                onBack?.();
+            }
+            else {
                 this.showError(`Failed to save API key for ${providerName}: ${errorMsg}`);
             }
         }
     }
-    showAuthSelect(dialog, prompt) {
+    showAuthSelect(dialog, prompt, providerId) {
         return new Promise((resolve, reject) => {
             const restoreDialog = () => {
                 this.editorContainer.clear();
@@ -5312,17 +5376,17 @@ export class InteractiveMode {
             }, () => {
                 restoreDialog();
                 reject(new Error("Login cancelled"));
-            });
+            }, { description: providerId === RADIUS_PROVIDER_ID ? RADIUS_LOGIN_INTRO : undefined });
             this.editorContainer.clear();
             this.editorContainer.addChild(selector);
             this.ui.setFocus(selector);
             this.ui.requestRender();
         });
     }
-    async showAuthPrompt(dialog, prompt) {
+    async showAuthPrompt(dialog, prompt, providerId) {
         let response;
         if (prompt.type === "select") {
-            response = this.showAuthSelect(dialog, prompt);
+            response = this.showAuthSelect(dialog, prompt, providerId);
         }
         else if (prompt.type === "manual_code") {
             response = dialog.showManualInput(prompt.message);
@@ -5366,11 +5430,11 @@ export class InteractiveMode {
     async loginProvider(dialog, providerId, method) {
         await this.session.modelRuntime.login(providerId, method, {
             signal: dialog.signal,
-            prompt: (prompt) => this.showAuthPrompt(dialog, prompt),
+            prompt: (prompt) => this.showAuthPrompt(dialog, prompt, providerId),
             notify: (event) => this.notifyAuthDialog(dialog, event),
         }, { getDeviceId: () => this.settingsManager.getOrCreateDeviceId() });
     }
-    async showLoginDialog(providerId, providerName) {
+    async showLoginDialog(providerId, providerName, onBack) {
         const previousModel = this.session.model;
         const dialog = new LoginDialogComponent(this.ui, providerId, (_success, _message) => { }, providerName);
         this.editorContainer.clear();
@@ -5387,6 +5451,8 @@ export class InteractiveMode {
             await this.loginProvider(dialog, providerId, "oauth");
             restoreEditor();
             await this.completeProviderAuthentication(providerId, providerName, "oauth", previousModel);
+            if (providerId === RADIUS_PROVIDER_ID)
+                this.offerRadiusMcpServer(providerId, providerName);
         }
         catch (error) {
             restoreEditor();
@@ -5394,11 +5460,58 @@ export class InteractiveMode {
             if (error instanceof CredentialSynchronizationError) {
                 this.showError(`Logged in to ${providerName}, but local model state could not be synchronized: ${errorMsg}`);
             }
-            else if (errorMsg !== "Login cancelled") {
+            else if (errorMsg === "Login cancelled") {
+                onBack?.();
+            }
+            else {
                 this.showError(`Failed to login to ${providerName}: ${errorMsg}`);
             }
         }
     }
+    /**
+     * Offer to point the Radius MCP server in the global mcp.json at the Radius login, adding the server
+     * when missing. Nothing is asked when a global server already uses this login.
+     */
+    offerRadiusMcpServer(providerId, providerName) {
+        const mcpPath = path.join(getAgentDir(), "mcp.json");
+        const normalizeUrl = (url) => url.replace(/\/+$/u, "");
+        const { servers } = loadMcpConfig({
+            agentDir: getAgentDir(),
+            cwd: this.sessionManager.getCwd(),
+            projectTrusted: false,
+        });
+        const existing = servers.find((server) => "url" in server.config && normalizeUrl(server.config.url) === normalizeUrl(RADIUS_MCP_URL));
+        if (existing && "url" in existing.config && existing.config.auth?.provider === providerId)
+            return;
+        let name = existing?.name ?? "radius";
+        if (!existing && servers.some((server) => server.name === name))
+            name = "radius-mcp";
+        const config = existing && "url" in existing.config
+            ? { ...existing.config, auth: { provider: providerId } }
+            : { url: RADIUS_MCP_URL, auth: { provider: providerId } };
+        // `auth` replaces the MCP OAuth sign-in.
+        delete config.oauth;
+        this.showSelector((done) => {
+            const selector = new ExtensionSelectorComponent(`Configure ${providerName} MCP in ${mcpPath}?`, ["Yes", "No"], (option) => {
+                done();
+                if (option !== "Yes")
+                    return;
+                try {
+                    addMcpServerConfig(mcpPath, name, config);
+                }
+                catch (error) {
+                    this.showError(`Could not update ${mcpPath}: ${error instanceof Error ? error.message : String(error)}`);
+                    return;
+                }
+                // The MCP extension reads mcp.json when the session starts.
+                void this.handleReloadCommand();
+            }, () => {
+                done();
+                this.ui.requestRender();
+            });
+            return { component: selector, focus: selector };
+        });
+    }
     // =========================================================================
     // Command handlers
     // =========================================================================