@timurproko/a1 0.1.8-dev.472 → 0.1.8-dev.508

package/dist/integrations/pi/engine/workflow-contexts.js

@@ -0,0 +1,316 @@
+import { realpathSync } from "node:fs";
+import { dirname, resolve } from "node:path";
+import { dynamicObject, isRecord, requireCapability, stringProperty } from "./message-values.js";
+import { ProjectTrustStore, SessionManager } from "../startup-public.js";
+import { resolveConfiguredModelIds, scopedModelRecords, scopedModelReference, workflowOptions } from "./workflow-support.js";
+/**
+ * The selector contexts and option lists the owned UI's workflow dialogs read from pinned Pi:
+ * model, scoped-model, session, tree, fork, project-trust, and provider-authentication state,
+ * shaped for the owned controllers. Everything here is a read of the current session and runtime
+ * plus the few writes the selectors commit directly (scoped models, project trust); running a
+ * workflow belongs to the runner.
+ */
+export class PiWorkflowContexts {
+    #agentDir;
+    #ports;
+    constructor(options, ports) {
+        this.#agentDir = options.agentDir;
+        this.#ports = ports;
+    }
+    #requireSession() {
+        const session = this.#ports.session();
+        if (this.#ports.disposed() || !this.#ports.runtime() || !session)
+            throw new Error("engine adapter is not running");
+        return session;
+    }
+    pinnedModelSelectorContext() {
+        const session = this.#requireSession();
+        const runtime = this.#ports.runtime();
+        if (!runtime)
+            throw new Error("engine runtime is unavailable");
+        const scoped = session.scopedModels;
+        const modelRuntime = runtime.services.modelRuntime;
+        const selectorRuntime = typeof modelRuntime.getAvailableSnapshot === "function"
+            ? modelRuntime
+            : {
+                getAvailableSnapshot: () => session.model === undefined ? [] : [session.model],
+                getModel: (providerId, modelId) => modelRuntime.getModel(providerId, modelId),
+                getError: () => undefined,
+                refresh: async () => undefined,
+            };
+        const settingsManager = runtime.services.settingsManager;
+        const defaultProvider = settingsManager?.getDefaultProvider?.();
+        const defaultModelId = settingsManager?.getDefaultModel?.();
+        return {
+            currentModel: this.#ports.activeModel() === null ? undefined : session.model,
+            ...(defaultProvider && defaultModelId ? { defaultModel: { provider: defaultProvider, id: defaultModelId } } : {}),
+            modelRuntime: selectorRuntime,
+            scopedModels: Array.isArray(scoped) ? scoped : [],
+        };
+    }
+    pinnedProjectTrustContext() {
+        const runtime = this.#ports.runtime();
+        if (!runtime)
+            throw new Error("engine runtime is unavailable");
+        const cwd = resolve(this.#ports.cwd());
+        let trustPath = cwd;
+        try {
+            trustPath = realpathSync(cwd);
+        }
+        catch {
+            // Compatibility: pinned Pi retains the resolved path when real-path lookup fails.
+        }
+        // Security: trust the target's parent, not the parent containing a directory alias.
+        const parent = dirname(trustPath);
+        const trustStore = new ProjectTrustStore(this.#agentDir);
+        const trustOptions = [
+            { label: "Trust", trusted: true, updates: [{ path: trustPath, decision: true }], savedPath: trustPath },
+            ...(parent === trustPath ? [] : [{
+                    label: `Trust parent folder (${parent})`,
+                    trusted: true,
+                    updates: [{ path: parent, decision: true }, { path: trustPath, decision: null }],
+                    savedPath: parent,
+                }]),
+            { label: "Do not trust", trusted: false, updates: [{ path: trustPath, decision: false }], savedPath: trustPath },
+        ];
+        return {
+            cwd,
+            savedDecision: trustStore.getEntry(cwd),
+            projectTrusted: runtime.services.settingsManager?.isProjectTrusted?.() === true,
+            trustOptions,
+        };
+    }
+    persistProjectTrust(updates) {
+        new ProjectTrustStore(this.#agentDir).setMany([...updates]);
+    }
+    pinnedSessionSelectorContext() {
+        const session = this.#requireSession();
+        const manager = session.sessionManager;
+        const cwd = manager?.getCwd?.();
+        const sessionDir = manager?.getSessionDir?.();
+        const currentSessionFilePath = manager?.getSessionFile?.();
+        const usesDefaultSessionDir = manager?.usesDefaultSessionDir?.() === true;
+        const resolvedCwd = typeof cwd === "string" ? cwd : this.#ports.cwd();
+        const resolvedSessionDir = typeof sessionDir === "string" ? sessionDir : undefined;
+        return {
+            currentSessionFilePath: typeof currentSessionFilePath === "string" ? currentSessionFilePath : undefined,
+            loadCurrentSessions: onProgress => SessionManager.list(resolvedCwd, resolvedSessionDir, onProgress),
+            loadAllSessions: onProgress => usesDefaultSessionDir
+                ? SessionManager.listAll(onProgress)
+                : resolvedSessionDir === undefined ? SessionManager.listAll(onProgress) : SessionManager.listAll(resolvedSessionDir, onProgress),
+            renameSession: async (sessionFilePath, nextName) => {
+                const next = (nextName ?? "").trim();
+                if (!next)
+                    return;
+                SessionManager.open(sessionFilePath).appendSessionInfo(next);
+            },
+        };
+    }
+    pinnedScopedModelsContext() {
+        const session = this.#requireSession();
+        const runtime = this.#ports.runtime();
+        if (!runtime)
+            throw new Error("engine runtime is unavailable");
+        const models = scopedModelRecords(runtime.services.modelRuntime);
+        const scoped = session.scopedModels;
+        if (Array.isArray(scoped) && scoped.length > 0) {
+            return {
+                models: models.map(item => item.descriptor),
+                enabledModelIds: scoped.map(scopedModelReference).filter((id) => id !== undefined),
+            };
+        }
+        const patterns = runtime.services.settingsManager?.getEnabledModels?.();
+        return {
+            models: models.map(item => item.descriptor),
+            enabledModelIds: Array.isArray(patterns)
+                ? resolveConfiguredModelIds(patterns.filter((value) => typeof value === "string"), models)
+                : null,
+        };
+    }
+    updateScopedModels(enabledModelIds) {
+        const session = this.#requireSession();
+        const runtime = this.#ports.runtime();
+        if (!runtime)
+            throw new Error("engine runtime is unavailable");
+        const models = scopedModelRecords(runtime.services.modelRuntime);
+        const availableIds = new Set(models.map(item => `${item.descriptor.provider}/${item.descriptor.id}`));
+        const selected = enabledModelIds?.filter(id => availableIds.has(id)) ?? [];
+        const allAvailableEnabled = enabledModelIds !== null && availableIds.size > 0 && selected.length === availableIds.size;
+        const scoped = enabledModelIds !== null && selected.length > 0 && !allAvailableEnabled
+            ? selected.flatMap(id => {
+                const item = models.find(candidate => `${candidate.descriptor.provider}/${candidate.descriptor.id}` === id);
+                return item === undefined ? [] : [{ model: item.model }];
+            })
+            : [];
+        requireCapability(session.setScopedModels, "setScopedModels").call(session, scoped);
+        this.#ports.emitView();
+    }
+    persistScopedModels(enabledModelIds) {
+        const runtime = this.#ports.runtime();
+        if (!runtime)
+            throw new Error("engine runtime is unavailable");
+        const availableCount = scopedModelRecords(runtime.services.modelRuntime).length;
+        const patterns = enabledModelIds === null || enabledModelIds.length === availableCount
+            ? undefined
+            : [...enabledModelIds];
+        requireCapability(runtime.services.settingsManager?.setEnabledModels, "setEnabledModels").call(runtime.services.settingsManager, patterns === undefined ? undefined : [...patterns]);
+    }
+    async refreshScopedModels(signal) {
+        const runtime = this.#ports.runtime();
+        if (!runtime)
+            throw new Error("engine runtime is unavailable");
+        const result = await runtime.services.modelRuntime.refresh?.({ signal });
+        const context = this.pinnedScopedModelsContext();
+        if (isRecord(result) && result.aborted === true) {
+            return { ...context, status: "Model refresh timed out; showing cached models.", statusKind: "warning" };
+        }
+        const errors = isRecord(result) ? result.errors : undefined;
+        if (errors instanceof Map && errors.size > 0) {
+            return {
+                ...context,
+                status: `Could not refresh ${[...errors.keys()].join(", ")}; showing cached models.`,
+                statusKind: "warning",
+            };
+        }
+        return { ...context, status: "Model catalogs refreshed.", statusKind: "success" };
+    }
+    pinnedLoginMethodOptions(providerReference) {
+        const runtime = this.#ports.runtime();
+        if (!runtime)
+            throw new Error("engine runtime is unavailable");
+        const modelRuntime = runtime.services.modelRuntime;
+        const normalized = providerReference.trim().toLowerCase();
+        const providers = modelRuntime.getProviders?.();
+        const matches = Array.isArray(providers) ? providers.filter(isRecord).filter(candidate => stringProperty(candidate, "id")?.toLowerCase() === normalized
+            || stringProperty(candidate, "name")?.toLowerCase() === normalized) : [];
+        const providerId = matches.length === 1 ? stringProperty(matches[0], "id") ?? providerReference : providerReference;
+        const provider = modelRuntime.getProvider?.(providerId);
+        const providerName = stringProperty(provider, "name") ?? providerId;
+        const oauth = dynamicObject(dynamicObject(provider, "auth"), "oauth");
+        const loginLabel = stringProperty(oauth, "loginLabel") ?? "Sign in with an account";
+        const options = this.loginOptions().filter(option => option.id.endsWith(`:${providerId}`)).map(option => ({
+            id: option.id,
+            label: option.id.startsWith("api_key:") ? "Sign in with an API key" : loginLabel,
+            ...(option.description === undefined ? {} : { description: option.description }),
+        }));
+        return { title: `Select authentication method for ${providerName}:`, options };
+    }
+    pinnedAmbientAuthentication(selection) {
+        if (!selection.startsWith("api_key:"))
+            return null;
+        const providerId = selection.slice("api_key:".length);
+        const provider = this.#ports.runtime()?.services.modelRuntime.getProvider?.(providerId);
+        const method = dynamicObject(dynamicObject(provider, "auth"), "apiKey");
+        if (!method || typeof method.login === "function")
+            return null;
+        const providerName = stringProperty(provider, "name") ?? providerId;
+        return { providerId, providerName, title: `${providerName} setup`, message: `${stringProperty(method, "name") ?? "Authentication"} is configured outside pi.` };
+    }
+    pinnedForkOptions() {
+        const session = this.#requireSession();
+        const messages = requireCapability(session.getUserMessagesForForking, "getUserMessagesForForking").call(session);
+        return workflowOptions(messages, "entryId", "text");
+    }
+    pinnedTreeSelectorContext() {
+        const manager = this.#requireSession().sessionManager;
+        const settings = this.#ports.runtime()?.services.settingsManager;
+        const tree = manager?.getTree?.();
+        const leaf = manager?.getLeafId?.();
+        const configuredFilter = settings?.getTreeFilterMode?.();
+        const filterMode = configuredFilter === "no-tools" || configuredFilter === "user-only" || configuredFilter === "labeled-only" || configuredFilter === "all"
+            ? configuredFilter
+            : "default";
+        return {
+            tree: Array.isArray(tree) ? tree : [],
+            currentLeafId: typeof leaf === "string" ? leaf : null,
+            filterMode,
+            skipSummaryPrompt: settings?.getBranchSummarySkipPrompt?.() === true,
+            appendLabelChange: (entryId, label) => {
+                requireCapability(manager?.appendLabelChange, "appendLabelChange").call(manager, entryId, label);
+            },
+        };
+    }
+    modelOptions() {
+        const runtime = this.#ports.runtime();
+        if (!runtime)
+            return [];
+        const models = runtime.services.modelRuntime.getAvailableSnapshot?.();
+        if (!Array.isArray(models)) {
+            const active = this.#ports.activeModel();
+            return active ? [{ id: `${active.providerId}/${active.modelId}`, label: active.displayName, description: `${active.providerId}/${active.modelId}` }] : [];
+        }
+        return models.filter(isRecord).flatMap(model => {
+            const provider = stringProperty(model, "provider");
+            const id = stringProperty(model, "id");
+            if (!provider || !id)
+                return [];
+            return [{ id: `${provider}/${id}`, label: stringProperty(model, "name") ?? id, description: `${provider}/${id}` }];
+        });
+    }
+    loginOptions(authType) {
+        const runtime = this.#ports.runtime();
+        const modelRuntime = runtime?.services.modelRuntime;
+        if (!modelRuntime)
+            return [];
+        const providers = modelRuntime.getProviders?.();
+        if (!Array.isArray(providers))
+            return [];
+        return providers.filter(isRecord).flatMap(provider => {
+            const id = stringProperty(provider, "id");
+            if (!id)
+                return [];
+            const name = stringProperty(provider, "name") ?? id;
+            const auth = isRecord(provider.auth) ? provider.auth : {};
+            const authStatus = modelRuntime.getProviderAuthStatus?.(id);
+            const source = authStatus?.label ?? authStatus?.source;
+            const status = authStatus?.configured === true
+                ? {
+                    type: modelRuntime.isUsingOAuth?.(id) === true ? "oauth" : "api_key",
+                    ...(source === undefined ? {} : { source }),
+                }
+                : undefined;
+            return [
+                ...(authType !== "api_key" && auth.oauth ? [{
+                        id: `oauth:${id}`,
+                        providerId: id,
+                        label: name,
+                        description: "Account / OAuth",
+                        authType: "oauth",
+                        ...(status === undefined ? {} : { status }),
+                    }] : []),
+                ...(authType !== "oauth" && auth.apiKey ? [{
+                        id: `api_key:${id}`,
+                        providerId: id,
+                        label: name,
+                        description: "API key",
+                        authType: "api_key",
+                        ...(status === undefined ? {} : { status }),
+                    }] : []),
+            ];
+        }).sort((left, right) => left.label.localeCompare(right.label));
+    }
+    async logoutOptions() {
+        const runtime = this.#ports.runtime();
+        if (!runtime)
+            return [];
+        const modelRuntime = runtime.services.modelRuntime;
+        const credentials = await requireCapability(modelRuntime.listCredentials, "listCredentials").call(modelRuntime, { signal: AbortSignal.timeout(15_000) });
+        if (!Array.isArray(credentials))
+            return [];
+        return credentials.filter(isRecord).flatMap(credential => {
+            const providerId = stringProperty(credential, "providerId");
+            if (!providerId)
+                return [];
+            const credentialType = stringProperty(credential, "type") === "api_key" ? "api_key" : "oauth";
+            const provider = modelRuntime.getProvider?.(providerId);
+            return [{
+                    id: `${credentialType}:${providerId}`,
+                    providerId,
+                    label: stringProperty(provider, "name") ?? providerId,
+                    description: credentialType,
+                    authType: credentialType,
+                    status: { type: credentialType, source: "stored credential" },
+                }];
+        });
+    }
+}

package/dist/integrations/pi/engine/workflow-runner.d.ts

@@ -0,0 +1,60 @@
+import type { AgentSession, AgentSessionRuntime } from "../startup-public.js";
+import type { PiWorkflowContexts } from "./workflow-contexts.js";
+import type { OwnedUiModelInfo, OwnedUiSnapshot, OwnedUiThinkingLevel } from "../../../contracts/owned-ui/index.js";
+import type { PiBashWorkflowResult, PiWorkflowHost, PiWorkflowInteractionHost, PiWorkflowRequest, PiWorkflowResult } from "./workflows.js";
+type PiSessionApi = AgentSession;
+type PiRuntimeApi = AgentSessionRuntime;
+export interface PiWorkflowRunnerOptions {
+    readonly agentDir: string;
+    readonly host: PiWorkflowHost;
+    readonly contexts: PiWorkflowContexts;
+}
+export interface PiWorkflowRunnerPorts {
+    session(): PiSessionApi | undefined;
+    runtime(): PiRuntimeApi | undefined;
+    disposed(): boolean;
+    sessionGeneration(): number;
+    /** The owned UI's prompt/notify host for login flows; the adapter swaps it when the shell binds. */
+    interaction(): PiWorkflowInteractionHost;
+    /** True while the adapter refuses new work: delivery overload, stopped admission, or disposal. */
+    admissionStopped(): boolean;
+    /** Engine commands currently admitted; they share the admission budget with workflows. */
+    pendingCommandCount(): number;
+    beginRunning(): void;
+    endRunning(): void;
+    activeModel(): OwnedUiModelInfo | null;
+    setActiveModel(model: OwnedUiModelInfo): void;
+    thinkingLevel(): OwnedUiThinkingLevel;
+    setThinkingLevel(level: OwnedUiThinkingLevel): void;
+    emitView(): void;
+    reconcileActiveModelAvailability(): void;
+    bindExtensionUiToSession(): Promise<void>;
+    applyPinnedSetting(selection: string): Promise<PiWorkflowResult>;
+    snapshot(): OwnedUiSnapshot;
+    dispose(): Promise<void>;
+}
+/**
+ * Pinned Pi's slash-command workflows, run against the adapter's current session and runtime
+ * through ports. The runner owns workflow admission (the pending set and its cancellation), the
+ * clipboard writer the owned UI binds, provider authentication, and the per-command result
+ * wording; the adapter decides what the session is, how the view is published, and when work is
+ * refused, and the contexts collaborator supplies selector state.
+ */
+export declare class PiWorkflowRunner {
+    #private;
+    constructor(options: PiWorkflowRunnerOptions, ports: PiWorkflowRunnerPorts);
+    /** Workflows admitted and not yet settled. */
+    get pendingCount(): number;
+    /** Resolve every admitted workflow as cancelled, except the named command. */
+    cancelPending(except?: PiWorkflowRequest["command"]): void;
+    cycleModelWorkflow(direction: "forward" | "backward"): Promise<PiWorkflowResult>;
+    /** Bind the owned UI's clipboard lifecycle without changing the comparison host. True means acknowledged delivery. */
+    bindClipboardWriter(writer: (text: string) => Promise<boolean>): () => void;
+    /** Workflow and tree copying share the active owner's write fence and delivery acknowledgment. */
+    copyWorkflowText(text: string): Promise<boolean>;
+    abortBashWorkflow(): void;
+    executeBashWorkflow(command: string, excludeFromContext: boolean): Promise<PiBashWorkflowResult>;
+    reloadBlockedResult(): PiWorkflowResult | null;
+    executeWorkflow(request: PiWorkflowRequest): Promise<PiWorkflowResult>;
+}
+export {};