@getpaseo/server 0.1.71 → 0.1.72

package/dist/server/server/session.js

@@ -29,15 +29,16 @@ import { deriveProjectSlug } from "./workspace-git-metadata.js";
 import { spawnWorkspaceScript } from "./worktree-bootstrap.js";
 import { applyMutableProviderConfigToOverrides } from "./daemon-config-store.js";
 import { getErrorMessage, getErrorMessageOr } from "../shared/error-utils.js";
+import { getAgentStatusPriority } from "../shared/agent-state-bucket.js";
 import { buildProviderRegistry } from "./agent/provider-registry.js";
 import { resolveSnapshotCwd } from "./agent/provider-snapshot-manager.js";
 import { scheduleAgentMetadataGeneration } from "./agent/agent-metadata-generator.js";
+import { resolveCreateAgentTitles } from "./agent/create-agent-title.js";
 import { buildStoredAgentPayload, resolveEffectiveThinkingOptionId, resolveStoredAgentPayloadUpdatedAt, toAgentPayload, } from "./agent/agent-projections.js";
-import { MAX_EXPLICIT_AGENT_TITLE_CHARS } from "./agent/agent-title-limits.js";
 import { appendTimelineItemIfAgentKnown, emitLiveTimelineItemIfAgentKnown, } from "./agent/timeline-append.js";
 import { projectTimelineRows, selectTimelineWindowByProjectedLimit, } from "./agent/timeline-projection.js";
 import { DEFAULT_STRUCTURED_GENERATION_PROVIDERS, StructuredAgentFallbackError, StructuredAgentResponseError, generateStructuredAgentResponseWithFallback, } from "./agent/agent-response-loop.js";
-import { ImportSessionsRequestError, listImportableProviderSessions, normalizeImportAgentRequest, } from "./agent/import-sessions.js";
+import { ImportSessionsRequestError, importProviderSession, listImportableProviderSessions, normalizeImportAgentRequest, } from "./agent/import-sessions.js";
 import { checkoutLiteFromGitSnapshot, normalizeWorkspaceId as normalizePersistedWorkspaceId, deriveProjectGroupingName, classifyDirectoryForProjectMembership, deriveWorkspaceDisplayName, } from "./workspace-registry-model.js";
 import { createPersistedProjectRecord, createPersistedWorkspaceRecord, } from "./workspace-registry.js";
 import { buildVoiceModeSystemPrompt, stripVoiceModeSystemPrompt, wrapSpokenInput, } from "./voice-config.js";
@@ -52,6 +53,7 @@ import { getProjectIcon } from "../utils/project-icon.js";
 import { expandTilde } from "../utils/path.js";
 import { searchHomeDirectories, searchWorkspaceEntries } from "../utils/directory-suggestions.js";
 import { toCheckoutError } from "./checkout-git-utils.js";
+import { buildCheckoutPrStatusPayloadFromSnapshot, buildCheckoutStatusPayloadFromSnapshot, } from "./checkout/status-projection.js";
 import { toResolver } from "./speech/provider-resolver.js";
 import { resolveClientMessageId } from "./client-message-id.js";
 import { ChatServiceError, parseMentionAgentIds, } from "./chat/chat-service.js";
@@ -63,7 +65,6 @@ import { attemptFirstAgentBranchAutoName, createPaseoWorktree, } from "./paseo-w
 import { generateBranchNameFromFirstAgentContext } from "./worktree-branch-name-generator.js";
 import { assertSafeGitRef as assertWorktreeSafeGitRef, buildAgentSessionConfig as buildWorktreeAgentSessionConfig, createPaseoWorktreeWorkflow as createWorktreeWorkflow, handleCreatePaseoWorktreeRequest as handleCreateWorktreeRequest, handlePaseoWorktreeArchiveRequest as handleWorktreeArchiveRequest, handlePaseoWorktreeListRequest as handleWorktreeListRequest, handleWorkspaceSetupStatusRequest as handleWorkspaceSetupStatusRequestMessage, } from "./worktree-session.js";
 import { toWorktreeWireError } from "./worktree-errors.js";
-const MAX_INITIAL_AGENT_TITLE_CHARS = Math.min(60, MAX_EXPLICIT_AGENT_TITLE_CHARS);
 const WORKSPACE_GIT_WATCH_REMOVED_STATE_KEY = "__removed__";
 async function resolveKnownProjectRootForConfig(input) {
     const requestedRoot = canonicalizeConfigRoot(input.repoRoot);
@@ -184,44 +185,6 @@ function beginAgentDeleteIfSupported(agentStorage, agentId) {
         agentStorage.beginDelete(agentId);
     }
 }
-function deriveInitialAgentTitle(prompt) {
-    const firstContentLine = prompt
-        .split(/\r?\n/)
-        .map((line) => line.trim())
-        .find((line) => line.length > 0);
-    if (!firstContentLine) {
-        return null;
-    }
-    const normalized = firstContentLine.replace(/\s+/g, " ").trim();
-    if (!normalized) {
-        return null;
-    }
-    const clamped = normalized.slice(0, MAX_INITIAL_AGENT_TITLE_CHARS).trim();
-    return clamped.length > 0 ? clamped : null;
-}
-export function resolveCreateAgentTitles(options) {
-    const explicitTitle = typeof options.configTitle === "string" && options.configTitle.trim().length > 0
-        ? options.configTitle.trim()
-        : null;
-    const trimmedPrompt = options.initialPrompt?.trim();
-    const provisionalTitle = explicitTitle ?? (trimmedPrompt ? deriveInitialAgentTitle(trimmedPrompt) : null);
-    return {
-        explicitTitle,
-        provisionalTitle,
-    };
-}
-function getFirstUserMessageText(timeline) {
-    for (const item of timeline) {
-        if (item.type !== "user_message") {
-            continue;
-        }
-        const text = item.text.trim();
-        if (text) {
-            return text;
-        }
-    }
-    return null;
-}
 const FETCH_AGENTS_SORT_KEYS = ["status_priority", "created_at", "updated_at", "title"];
 export function resolveWaitForFinishError(options) {
     if (options.status !== "error") {
@@ -255,31 +218,6 @@ class VoiceFeatureUnavailableError extends Error {
         this.missingModelIds = [...context.missingModelIds];
     }
 }
-function buildImportPersistenceHandle(input) {
-    const cwd = input.cwd ?? process.cwd();
-    return {
-        provider: input.provider,
-        sessionId: input.providerHandleId,
-        nativeHandle: input.providerHandleId,
-        metadata: {
-            provider: input.provider,
-            cwd,
-        },
-    };
-}
-function applyImportCwdOverride(handle, cwd) {
-    if (!cwd) {
-        return handle;
-    }
-    return {
-        ...handle,
-        metadata: {
-            ...handle.metadata,
-            provider: handle.provider,
-            cwd,
-        },
-    };
-}
 function convertPCMToWavBuffer(pcmBuffer, sampleRate, channels, bitsPerSample) {
     const headerSize = 44;
     const wavBuffer = Buffer.alloc(headerSize + pcmBuffer.length);
@@ -369,7 +307,12 @@ export class Session {
             getSortValue: (agent, key) => {
                 switch (key) {
                     case "status_priority":
-                        return this.getStatusPriority(agent);
+                        return getAgentStatusPriority({
+                            status: agent.status,
+                            pendingPermissionCount: agent.pendingPermissions?.length ?? 0,
+                            requiresAttention: agent.requiresAttention,
+                            attentionReason: agent.attentionReason ?? null,
+                        });
                     case "created_at":
                         return Date.parse(agent.createdAt);
                     case "updated_at":
@@ -2195,26 +2138,18 @@ export class Session {
             });
             return;
         }
-        const { provider, providerHandleId, cwd, labels, requestId } = normalized;
+        const { provider, providerHandleId, requestId } = normalized;
         this.sessionLogger.info({ providerHandleId, provider }, `Importing agent ${providerHandleId} (${provider})`);
         try {
-            const descriptor = await this.agentManager.findPersistedAgent(provider, providerHandleId);
-            if (!descriptor && provider === "opencode" && !cwd) {
-                throw new Error("OpenCode sessions require --cwd when the session cannot be found in persisted agents");
-            }
-            const handle = descriptor
-                ? applyImportCwdOverride(descriptor.persistence, cwd)
-                : buildImportPersistenceHandle({ provider, providerHandleId, cwd });
-            const overrides = cwd ? { cwd } : undefined;
-            await this.unarchiveAgentByHandle(handle);
-            const snapshot = await this.agentManager.resumeAgentFromPersistence(handle, overrides, undefined, {
-                labels,
+            const { snapshot, timelineSize } = await importProviderSession({
+                request: normalized,
+                agentManager: this.agentManager,
+                agentStorage: this.agentStorage,
+                workspaceGitService: this.workspaceGitService,
+                paseoHome: this.paseoHome,
+                logger: this.sessionLogger,
             });
-            await unarchiveAgentState(this.agentStorage, this.agentManager, snapshot.id);
-            await this.agentManager.hydrateTimelineFromProvider(snapshot.id);
-            await this.applyImportedAgentTitle(snapshot);
             await this.forwardAgentUpdate(snapshot);
-            const timelineSize = this.agentManager.getTimeline(snapshot.id).length;
             const agentPayload = await this.buildAgentPayload(snapshot);
             this.emit({
                 type: "status",
@@ -2249,29 +2184,6 @@ export class Session {
             });
         }
     }
-    async applyImportedAgentTitle(snapshot) {
-        const initialPrompt = getFirstUserMessageText(this.agentManager.getTimeline(snapshot.id));
-        if (!initialPrompt) {
-            return;
-        }
-        const { explicitTitle, provisionalTitle } = resolveCreateAgentTitles({
-            configTitle: snapshot.config.title,
-            initialPrompt,
-        });
-        if (!explicitTitle && provisionalTitle) {
-            await this.agentManager.setTitle(snapshot.id, provisionalTitle);
-        }
-        scheduleAgentMetadataGeneration({
-            agentManager: this.agentManager,
-            agentId: snapshot.id,
-            cwd: snapshot.cwd,
-            workspaceGitService: this.workspaceGitService,
-            initialPrompt,
-            explicitTitle,
-            paseoHome: this.paseoHome,
-            logger: this.sessionLogger,
-        });
-    }
     async handleRefreshAgentRequest(msg) {
         const { agentId, requestId } = msg;
         this.sessionLogger.info({ agentId }, `Refreshing agent ${agentId} from persistence`);
@@ -3227,7 +3139,7 @@ export class Session {
             const snapshot = await this.workspaceGitService.getSnapshot(resolvedCwd);
             this.emit({
                 type: "checkout_status_response",
-                payload: this.buildCheckoutStatusPayloadFromSnapshot({
+                payload: buildCheckoutStatusPayloadFromSnapshot({
                     cwd,
                     requestId,
                     snapshot,
@@ -3537,94 +3449,18 @@ export class Session {
         this.checkoutDiffSubscriptions.get(msg.subscriptionId)?.();
         this.checkoutDiffSubscriptions.delete(msg.subscriptionId);
     }
-    buildCheckoutStatusPayloadFromSnapshot({ cwd, requestId, snapshot, }) {
-        if (!snapshot.git.isGit) {
-            return {
-                cwd,
-                isGit: false,
-                repoRoot: null,
-                currentBranch: null,
-                isDirty: null,
-                baseRef: null,
-                aheadBehind: null,
-                aheadOfOrigin: null,
-                behindOfOrigin: null,
-                hasRemote: false,
-                remoteUrl: null,
-                isPaseoOwnedWorktree: false,
-                error: null,
-                requestId,
-            };
-        }
-        if (snapshot.git.repoRoot === null || snapshot.git.isDirty === null) {
-            throw new Error("Workspace git snapshot is missing required checkout status fields");
-        }
-        if (snapshot.git.isPaseoOwnedWorktree) {
-            if (snapshot.git.mainRepoRoot === null || snapshot.git.baseRef === null) {
-                throw new Error("Workspace git snapshot is missing required worktree status fields");
-            }
-            return {
-                cwd,
-                isGit: true,
-                repoRoot: snapshot.git.repoRoot,
-                mainRepoRoot: snapshot.git.mainRepoRoot,
-                currentBranch: snapshot.git.currentBranch ?? null,
-                isDirty: snapshot.git.isDirty,
-                baseRef: snapshot.git.baseRef,
-                aheadBehind: snapshot.git.aheadBehind ?? null,
-                aheadOfOrigin: snapshot.git.aheadOfOrigin ?? null,
-                behindOfOrigin: snapshot.git.behindOfOrigin ?? null,
-                hasRemote: snapshot.git.hasRemote,
-                remoteUrl: snapshot.git.remoteUrl,
-                isPaseoOwnedWorktree: true,
-                error: null,
-                requestId,
-            };
-        }
-        return {
-            cwd,
-            isGit: true,
-            repoRoot: snapshot.git.repoRoot,
-            mainRepoRoot: snapshot.git.mainRepoRoot,
-            currentBranch: snapshot.git.currentBranch ?? null,
-            isDirty: snapshot.git.isDirty,
-            baseRef: snapshot.git.baseRef ?? null,
-            aheadBehind: snapshot.git.aheadBehind ?? null,
-            aheadOfOrigin: snapshot.git.aheadOfOrigin ?? null,
-            behindOfOrigin: snapshot.git.behindOfOrigin ?? null,
-            hasRemote: snapshot.git.hasRemote,
-            remoteUrl: snapshot.git.remoteUrl,
-            isPaseoOwnedWorktree: false,
-            error: null,
-            requestId,
-        };
-    }
-    buildCheckoutPrStatusPayloadFromSnapshot({ cwd, requestId, snapshot, }) {
-        return {
-            cwd,
-            status: normalizeCheckoutPrStatusPayload(snapshot.github.pullRequest),
-            githubFeaturesEnabled: snapshot.github.featuresEnabled,
-            error: snapshot.github.error
-                ? {
-                    code: "UNKNOWN",
-                    message: snapshot.github.error.message,
-                }
-                : null,
-            requestId,
-        };
-    }
     emitCheckoutStatusUpdate(cwd, snapshot) {
         try {
             const requestId = `subscription:${cwd}`;
             this.emit({
                 type: "checkout_status_update",
                 payload: {
-                    ...this.buildCheckoutStatusPayloadFromSnapshot({
+                    ...buildCheckoutStatusPayloadFromSnapshot({
                         cwd,
                         requestId,
                         snapshot,
                     }),
-                    prStatus: this.buildCheckoutPrStatusPayloadFromSnapshot({
+                    prStatus: buildCheckoutPrStatusPayloadFromSnapshot({
                         cwd,
                         requestId,
                         snapshot,
@@ -3994,18 +3830,11 @@ export class Session {
             const snapshot = await this.workspaceGitService.getSnapshot(cwd);
             this.emit({
                 type: "checkout_pr_status_response",
-                payload: {
+                payload: buildCheckoutPrStatusPayloadFromSnapshot({
                     cwd,
-                    status: normalizeCheckoutPrStatusPayload(snapshot.github.pullRequest),
-                    githubFeaturesEnabled: snapshot.github.featuresEnabled,
-                    error: snapshot.github.error
-                        ? {
-                            code: "UNKNOWN",
-                            message: snapshot.github.error.message,
-                        }
-                        : null,
                     requestId,
-                },
+                    snapshot,
+                }),
             });
         }
         catch (error) {
@@ -4403,23 +4232,6 @@ export class Session {
         const payload = this.buildStoredAgentPayload(record);
         return this.isProviderVisibleToClient(payload.provider) ? payload : null;
     }
-    getStatusPriority(agent) {
-        const attentionReason = agent.attentionReason ?? null;
-        const hasPendingPermission = (agent.pendingPermissions?.length ?? 0) > 0;
-        if (hasPendingPermission || attentionReason === "permission") {
-            return 0;
-        }
-        if (agent.status === "error" || attentionReason === "error") {
-            return 1;
-        }
-        if (agent.status === "running") {
-            return 2;
-        }
-        if (agent.status === "initializing") {
-            return 3;
-        }
-        return 4;
-    }
     async buildActiveProjectPlacementsByWorkspaceCwd() {
         const [persistedWorkspaces, persistedProjects] = await Promise.all([
             this.workspaceRegistry.list(),
@@ -6851,27 +6663,6 @@ export class Session {
 // Stash handlers
 // ---------------------------------------------------------------------------
 Session.PASEO_STASH_PREFIX = "paseo-auto-stash:";
-export function normalizeCheckoutPrStatusPayload(status) {
-    if (!status) {
-        return null;
-    }
-    return {
-        number: status.number,
-        url: status.url,
-        title: status.title,
-        state: status.state,
-        repoOwner: status.repoOwner,
-        repoName: status.repoName,
-        baseRefName: status.baseRefName,
-        headRefName: status.headRefName,
-        isMerged: status.isMerged,
-        isDraft: status.isDraft ?? false,
-        mergeable: status.mergeable ?? "UNKNOWN",
-        checks: status.checks ?? [],
-        checksStatus: status.checksStatus,
-        reviewDecision: status.reviewDecision,
-    };
-}
 function isValidPullRequestTimelineIdentity(options) {
     if (!Number.isInteger(options.prNumber) || options.prNumber <= 0) {
         return false;