@timurproko/a1 0.1.8-dev.527 → 0.1.8-dev.536

package/dist/integrations/pi/engine/session-runtime.d.ts

@@ -4,6 +4,7 @@ import type { PiProjectTrustPreflightPrompt } from "./project-trust-preflight.js
 import type { PiSessionForkPrompt, PiSessionSelection } from "./session-selection.js";
 import type { OwnedUiDiagnostics } from "../../../contracts/owned-ui/index.js";
 import type { PiSessionResumeMetadata, PiWorkflowHost } from "./workflows.js";
+import { type PiPullRequestIdentity, type PiPullRequestProbe } from "./repository-pr.js";
 export interface PiEngineRuntimeFactoryInput {
     readonly cwd: string;
     readonly agentDir: string;
@@ -25,6 +26,9 @@ export interface PiEngineRuntimeOptions {
     readonly projectTrustPrompt: PiProjectTrustPreflightPrompt | undefined;
     readonly createRuntime: PiEngineRuntimeFactory | undefined;
     readonly checkPackageUpdates: PiEnginePackageUpdateProbe | undefined;
+    readonly pullRequestProbe?: PiPullRequestProbe;
+    readonly pullRequestRefreshMs?: number;
+    readonly gitBranchReader?: (cwd: string) => Promise<string | null>;
     readonly host: PiWorkflowHost;
 }
 export interface PiEngineRuntimePorts {
@@ -62,6 +66,7 @@ export declare class PiEngineRuntime {
     get bindingGeneration(): number;
     get cwd(): string;
     get gitBranch(): string | null;
+    get pullRequest(): PiPullRequestIdentity | null;
     get sessionCommands(): PiSessionCommandIntegration | undefined;
     get started(): boolean;
     /** Create the runtime, bind its first session, and announce the changelog; startup failures propagate. */

package/dist/integrations/pi/engine/session-runtime.js

@@ -6,7 +6,9 @@ import { observeCompactionProgress } from "./compaction-progress.js";
 import { createPiRuntimeIntegration } from "./runtime-integration.js";
 import { PiSessionCommandIntegration } from "./session-integration.js";
 import { DefaultPackageManager, VERSION } from "../startup-public.js";
+import { readOpenPullRequest, } from "./repository-pr.js";
 const execFileAsync = promisify(execFile);
+const PULL_REQUEST_REFRESH_MS = 60_000;
 /**
  * The pinned Pi runtime and the session bound to it: creation through the runtime factory, the
  * session generation that stamps and invalidates events, the subscription that forwards session
@@ -18,6 +20,9 @@ export class PiEngineRuntime {
     #ports;
     #runtimeFactory;
     #checkPackageUpdates;
+    #pullRequestProbe;
+    #pullRequestRefreshMs;
+    #gitBranchReader;
     #cwd;
     #runtime;
     #session;
@@ -27,6 +32,10 @@ export class PiEngineRuntime {
     #sessionCommands;
     #compactionProgress = null;
     #gitBranch = null;
+    #pullRequest = null;
+    #pullRequestTimer;
+    #pullRequestAbort;
+    #disposed = false;
     constructor(options, ports) {
         this.#options = options;
         this.#ports = ports;
@@ -36,6 +45,9 @@ export class PiEngineRuntime {
             ?? (options.createRuntime
                 ? async () => []
                 : settingsManager => checkDefaultPiPackageUpdates(this.#cwd, options.agentDir, settingsManager));
+        this.#pullRequestProbe = options.pullRequestProbe ?? readOpenPullRequest;
+        this.#pullRequestRefreshMs = options.pullRequestRefreshMs ?? PULL_REQUEST_REFRESH_MS;
+        this.#gitBranchReader = options.gitBranchReader ?? readGitBranch;
     }
     get runtime() {
         return this.#runtime;
@@ -57,6 +69,9 @@ export class PiEngineRuntime {
     get gitBranch() {
         return this.#gitBranch;
     }
+    get pullRequest() {
+        return this.#pullRequest;
+    }
     get sessionCommands() {
         return this.#sessionCommands;
     }
@@ -76,7 +91,7 @@ export class PiEngineRuntime {
         });
         this.#runtime = runtime;
         this.#cwd = runtime.cwd ?? this.#cwd;
-        this.#gitBranch = await readGitBranch(this.#cwd);
+        this.#gitBranch = await this.#gitBranchReader(this.#cwd);
         this.#ports.started(runtime);
         runtime.setRebindSession(async (session) => {
             if (this.#ports.rebindBlocked())
@@ -89,6 +104,7 @@ export class PiEngineRuntime {
         }
         this.bindSession(runtime.session);
         await this.#announceChangelog(runtime.services.settingsManager);
+        this.#startPullRequestRefresh();
         return runtime;
     }
     /** Probe extension packages for updates after startup and report them as an informational diagnostic. */
@@ -184,12 +200,52 @@ export class PiEngineRuntime {
     }
     /** Unsubscribe, stop observing compaction, and dispose the runtime; the session reference is kept for final reads. */
     async dispose() {
+        this.#disposed = true;
+        if (this.#pullRequestTimer !== undefined)
+            clearTimeout(this.#pullRequestTimer);
+        this.#pullRequestTimer = undefined;
+        this.#pullRequestAbort?.abort();
+        this.#pullRequestAbort = undefined;
         this.#unsubscribe?.();
         this.#unsubscribe = undefined;
         this.#compactionProgress?.dispose();
         this.#compactionProgress = null;
         await this.#runtime?.dispose();
     }
+    #startPullRequestRefresh() {
+        if (this.#gitBranch === null || this.#disposed || this.#ports.disposed())
+            return;
+        void this.#refreshPullRequest();
+    }
+    async #refreshPullRequest() {
+        const branch = this.#gitBranch;
+        if (branch === null || this.#pullRequestAbort !== undefined || this.#disposed || this.#ports.disposed())
+            return;
+        const controller = new AbortController();
+        this.#pullRequestAbort = controller;
+        let next = null;
+        try {
+            next = await this.#pullRequestProbe(this.#cwd, branch, controller.signal);
+        }
+        catch {
+            next = null;
+        }
+        finally {
+            if (this.#pullRequestAbort === controller)
+                this.#pullRequestAbort = undefined;
+        }
+        if (controller.signal.aborted || this.#disposed || this.#ports.disposed())
+            return;
+        if (!samePullRequest(this.#pullRequest, next)) {
+            this.#pullRequest = next;
+            this.#ports.emitView();
+        }
+        this.#pullRequestTimer = setTimeout(() => {
+            this.#pullRequestTimer = undefined;
+            void this.#refreshPullRequest();
+        }, this.#pullRequestRefreshMs);
+        this.#pullRequestTimer.unref?.();
+    }
     #subscribe() {
         const session = this.#session;
         if (session === undefined)
@@ -237,6 +293,9 @@ async function checkDefaultPiPackageUpdates(cwd, agentDir, settingsManager) {
     const updates = await packageManager.checkForAvailableUpdates();
     return updates.map(update => update.displayName);
 }
+function samePullRequest(left, right) {
+    return left?.number === right?.number && left?.url === right?.url;
+}
 async function readGitBranch(cwd) {
     try {
         const { stdout } = await execFileAsync("git", ["branch", "--show-current"], { cwd, windowsHide: true });

package/dist/integrations/pi/engine/workflows.d.ts

@@ -14,7 +14,7 @@ export type PiHiddenWorkflowCommandName = typeof PINNED_PI_HIDDEN_COMMAND_NAMES[
 export declare const OWNED_MODELS_COMMAND_NAME: "models";
 export type OwnedModelsCommandName = typeof OWNED_MODELS_COMMAND_NAME;
 export type OwnedWorkflowCommandName = Exclude<PiWorkflowCommandName, "model" | "scoped-models"> | OwnedModelsCommandName;
-/** Bare A1's advertised command catalog: the pinned catalog with `models` in place of `model` and `scoped-models`. */
+/** Bare A1's advertised catalog: `models` replaces the pinned pair and is followed by `thinking`. */
 export declare const OWNED_WORKFLOW_COMMAND_NAMES: readonly OwnedWorkflowCommandName[];
 export type PiProductMode = "bare" | "comparison";
 /** The advertised command catalog for a product mode; hidden routes are shared by both. */

package/dist/integrations/pi/engine/workflows.js

@@ -33,8 +33,10 @@ export const PINNED_PI_HIDDEN_COMMAND_NAMES = ["debug", "arminsayshi", "demented
 export const PINNED_PI_DECLINED_COMMAND_NAMES = ["bug"];
 /** The one bare-A1 model command; it replaces the pinned `model` and `scoped-models` routes there. */
 export const OWNED_MODELS_COMMAND_NAME = "models";
-/** Bare A1's advertised command catalog: the pinned catalog with `models` in place of `model` and `scoped-models`. */
-export const OWNED_WORKFLOW_COMMAND_NAMES = Object.freeze(PINNED_PI_WORKFLOW_COMMAND_NAMES.flatMap((name) => name === "model" ? [OWNED_MODELS_COMMAND_NAME] : name === "scoped-models" ? [] : [name]));
+/** Bare A1's advertised catalog: `models` replaces the pinned pair and is followed by `thinking`. */
+export const OWNED_WORKFLOW_COMMAND_NAMES = Object.freeze(PINNED_PI_WORKFLOW_COMMAND_NAMES.flatMap((name) => name === "model"
+    ? [OWNED_MODELS_COMMAND_NAME, "thinking"]
+    : name === "thinking" || name === "scoped-models" ? [] : [name]));
 /** The advertised command catalog for a product mode; hidden routes are shared by both. */
 export function workflowCommandNames(productMode) {
     return productMode === "bare" ? OWNED_WORKFLOW_COMMAND_NAMES : PINNED_PI_WORKFLOW_COMMAND_NAMES;