mirror of
https://github.com/harivansh-afk/sandbox-agent.git
synced 2026-04-15 04:03:31 +00:00
a171956298
Revert actor communication from direct action calls to queue/workflow-based patterns for better observability (workflow history in RivetKit inspector), replay/recovery semantics, and idiomatic RivetKit usage. - Add queue/workflow infrastructure to all actors: organization, task, user, github-data, sandbox, and audit-log - Mutations route through named queues processed by workflow command loops with ctx.step() wrapping for c.state/c.db access and observability - Remove command action wrappers (~460 lines) — callers use .send() directly to queue names with expectQueueResponse() for wait:true results - Keep sendPrompt and runProcess as direct sandbox actions (long-running / large responses that would block the workflow loop or exceed 128KB limit) - Fix workspace fire-and-forget calls (enqueueWorkspaceEnsureSession, enqueueWorkspaceRefresh) to self-send to task queue instead of calling directly outside workflow step context Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
1613 lines
53 KiB
TypeScript
1613 lines
53 KiB
TypeScript
// @ts-nocheck
|
|
import { randomUUID } from "node:crypto";
|
|
import { basename, dirname } from "node:path";
|
|
import { asc, eq } from "drizzle-orm";
|
|
import {
|
|
DEFAULT_WORKSPACE_MODEL_GROUPS,
|
|
DEFAULT_WORKSPACE_MODEL_ID,
|
|
workspaceAgentForModel,
|
|
workspaceSandboxAgentIdForModel,
|
|
} from "@sandbox-agent/foundry-shared";
|
|
import { getActorRuntimeContext } from "../context.js";
|
|
import { getOrCreateOrganization, getOrCreateTaskSandbox, getOrCreateUser, getTaskSandbox, selfTask } from "../handles.js";
|
|
import { logActorWarning, resolveErrorMessage } from "../logging.js";
|
|
import { SANDBOX_REPO_CWD } from "../sandbox/index.js";
|
|
import { resolveSandboxProviderId } from "../../sandbox-config.js";
|
|
import { getBetterAuthService } from "../../services/better-auth.js";
|
|
import { resolveOrganizationGithubAuth } from "../../services/github-auth.js";
|
|
import { githubRepoFullNameFromRemote } from "../../services/repo.js";
|
|
import { taskWorkflowQueueName } from "./workflow/queue.js";
|
|
import { expectQueueResponse } from "../../services/queue.js";
|
|
import { userWorkflowQueueName } from "../user/workflow.js";
|
|
import { organizationWorkflowQueueName } from "../organization/queues.js";
|
|
|
|
import { task as taskTable, taskOwner, taskRuntime, taskSandboxes, taskWorkspaceSessions } from "./db/schema.js";
|
|
import { getCurrentRecord } from "./workflow/common.js";
|
|
|
|
function emptyGitState() {
|
|
return {
|
|
fileChanges: [],
|
|
diffs: {},
|
|
fileTree: [],
|
|
updatedAt: null as number | null,
|
|
};
|
|
}
|
|
|
|
const FALLBACK_MODEL = DEFAULT_WORKSPACE_MODEL_ID;
|
|
|
|
function agentKindForModel(model: string) {
|
|
return workspaceAgentForModel(model);
|
|
}
|
|
|
|
export function sandboxAgentIdForModel(model: string) {
|
|
return workspaceSandboxAgentIdForModel(model);
|
|
}
|
|
|
|
async function resolveWorkspaceModelGroups(c: any): Promise<any[]> {
|
|
try {
|
|
const sandbox = await getOrCreateTaskSandbox(c, c.state.organizationId, stableSandboxId(c));
|
|
const groups = await sandbox.listWorkspaceModelGroups();
|
|
return Array.isArray(groups) && groups.length > 0 ? groups : DEFAULT_WORKSPACE_MODEL_GROUPS;
|
|
} catch {
|
|
return DEFAULT_WORKSPACE_MODEL_GROUPS;
|
|
}
|
|
}
|
|
|
|
async function resolveSandboxAgentForModel(c: any, model: string): Promise<string> {
|
|
const groups = await resolveWorkspaceModelGroups(c);
|
|
return workspaceSandboxAgentIdForModel(model, groups);
|
|
}
|
|
|
|
function repoLabelFromRemote(remoteUrl: string): string {
|
|
const trimmed = remoteUrl.trim();
|
|
try {
|
|
const url = new URL(trimmed.startsWith("http") ? trimmed : `https://${trimmed}`);
|
|
const parts = url.pathname.replace(/\/+$/, "").split("/").filter(Boolean);
|
|
if (parts.length >= 2) {
|
|
return `${parts[0]}/${(parts[1] ?? "").replace(/\.git$/, "")}`;
|
|
}
|
|
} catch {
|
|
// ignore
|
|
}
|
|
|
|
return basename(trimmed.replace(/\.git$/, ""));
|
|
}
|
|
|
|
async function getRepositoryMetadata(c: any): Promise<{ defaultBranch: string | null; fullName: string | null; remoteUrl: string }> {
|
|
const organization = await getOrCreateOrganization(c, c.state.organizationId);
|
|
return await organization.getRepositoryMetadata({ repoId: c.state.repoId });
|
|
}
|
|
|
|
function parseDraftAttachments(value: string | null | undefined): Array<any> {
|
|
if (!value) {
|
|
return [];
|
|
}
|
|
|
|
try {
|
|
const parsed = JSON.parse(value) as unknown;
|
|
return Array.isArray(parsed) ? parsed : [];
|
|
} catch {
|
|
return [];
|
|
}
|
|
}
|
|
|
|
function parseTranscript(value: string | null | undefined): Array<any> {
|
|
if (!value) {
|
|
return [];
|
|
}
|
|
|
|
try {
|
|
const parsed = JSON.parse(value) as unknown;
|
|
return Array.isArray(parsed) ? parsed : [];
|
|
} catch {
|
|
return [];
|
|
}
|
|
}
|
|
|
|
function parseGitState(value: string | null | undefined): { fileChanges: Array<any>; diffs: Record<string, string>; fileTree: Array<any> } {
|
|
if (!value) {
|
|
return emptyGitState();
|
|
}
|
|
|
|
try {
|
|
const parsed = JSON.parse(value) as {
|
|
fileChanges?: unknown;
|
|
diffs?: unknown;
|
|
fileTree?: unknown;
|
|
};
|
|
return {
|
|
fileChanges: Array.isArray(parsed.fileChanges) ? parsed.fileChanges : [],
|
|
diffs: parsed.diffs && typeof parsed.diffs === "object" ? (parsed.diffs as Record<string, string>) : {},
|
|
fileTree: Array.isArray(parsed.fileTree) ? parsed.fileTree : [],
|
|
};
|
|
} catch {
|
|
return emptyGitState();
|
|
}
|
|
}
|
|
|
|
async function readTaskOwner(c: any): Promise<{
|
|
primaryUserId: string | null;
|
|
primaryGithubLogin: string | null;
|
|
primaryGithubEmail: string | null;
|
|
primaryGithubAvatarUrl: string | null;
|
|
} | null> {
|
|
const row = await c.db.select().from(taskOwner).where(eq(taskOwner.id, 1)).get();
|
|
if (!row) {
|
|
return null;
|
|
}
|
|
return {
|
|
primaryUserId: row.primaryUserId ?? null,
|
|
primaryGithubLogin: row.primaryGithubLogin ?? null,
|
|
primaryGithubEmail: row.primaryGithubEmail ?? null,
|
|
primaryGithubAvatarUrl: row.primaryGithubAvatarUrl ?? null,
|
|
};
|
|
}
|
|
|
|
async function upsertTaskOwner(
|
|
c: any,
|
|
owner: { primaryUserId: string; primaryGithubLogin: string; primaryGithubEmail: string; primaryGithubAvatarUrl: string | null },
|
|
): Promise<void> {
|
|
const now = Date.now();
|
|
await c.db
|
|
.insert(taskOwner)
|
|
.values({
|
|
id: 1,
|
|
primaryUserId: owner.primaryUserId,
|
|
primaryGithubLogin: owner.primaryGithubLogin,
|
|
primaryGithubEmail: owner.primaryGithubEmail,
|
|
primaryGithubAvatarUrl: owner.primaryGithubAvatarUrl,
|
|
updatedAt: now,
|
|
})
|
|
.onConflictDoUpdate({
|
|
target: taskOwner.id,
|
|
set: {
|
|
primaryUserId: owner.primaryUserId,
|
|
primaryGithubLogin: owner.primaryGithubLogin,
|
|
primaryGithubEmail: owner.primaryGithubEmail,
|
|
primaryGithubAvatarUrl: owner.primaryGithubAvatarUrl,
|
|
updatedAt: now,
|
|
},
|
|
})
|
|
.run();
|
|
}
|
|
|
|
/**
|
|
* Inject the user's GitHub OAuth token into the sandbox as a git credential store file.
|
|
* Also configures git user.name and user.email so commits are attributed correctly.
|
|
* The credential file is overwritten on each owner swap.
|
|
*
|
|
* Race condition note: If User A sends a message and the agent starts a long git operation,
|
|
* then User B triggers an owner swap, the in-flight git process still has User A's credentials
|
|
* (already read from the credential store). The next git operation uses User B's credentials.
|
|
*/
|
|
async function injectGitCredentials(sandbox: any, login: string, email: string, token: string): Promise<void> {
|
|
const script = [
|
|
"set -euo pipefail",
|
|
`git config --global user.name ${JSON.stringify(login)}`,
|
|
`git config --global user.email ${JSON.stringify(email)}`,
|
|
`git config --global credential.helper 'store --file=/home/user/.git-token'`,
|
|
`printf '%s\\n' ${JSON.stringify(`https://${login}:${token}@github.com`)} > /home/user/.git-token`,
|
|
`chmod 600 /home/user/.git-token`,
|
|
];
|
|
const result = await sandbox.runProcess({
|
|
command: "bash",
|
|
args: ["-lc", script.join("; ")],
|
|
cwd: "/",
|
|
timeoutMs: 30_000,
|
|
});
|
|
if ((result.exitCode ?? 0) !== 0) {
|
|
logActorWarning("task", "git credential injection failed", {
|
|
exitCode: result.exitCode,
|
|
output: [result.stdout, result.stderr].filter(Boolean).join(""),
|
|
});
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Resolves the current user's GitHub identity from their auth session.
|
|
* Returns null if the session is invalid or the user has no GitHub account.
|
|
*/
|
|
async function resolveGithubIdentity(authSessionId: string): Promise<{
|
|
userId: string;
|
|
login: string;
|
|
email: string;
|
|
avatarUrl: string | null;
|
|
accessToken: string;
|
|
} | null> {
|
|
const authService = getBetterAuthService();
|
|
const authState = await authService.getAuthState(authSessionId);
|
|
if (!authState?.user?.id) {
|
|
return null;
|
|
}
|
|
|
|
const tokenResult = await authService.getAccessTokenForSession(authSessionId);
|
|
if (!tokenResult?.accessToken) {
|
|
return null;
|
|
}
|
|
|
|
const githubAccount = authState.accounts?.find((account: any) => account.providerId === "github");
|
|
if (!githubAccount) {
|
|
return null;
|
|
}
|
|
|
|
// Resolve the GitHub login from the API since Better Auth only stores the
|
|
// numeric account ID, not the login username.
|
|
let login = authState.user.name ?? "unknown";
|
|
let avatarUrl = authState.user.image ?? null;
|
|
try {
|
|
const resp = await fetch("https://api.github.com/user", {
|
|
headers: {
|
|
Authorization: `Bearer ${tokenResult.accessToken}`,
|
|
Accept: "application/vnd.github+json",
|
|
},
|
|
});
|
|
if (resp.ok) {
|
|
const ghUser = (await resp.json()) as { login?: string; avatar_url?: string };
|
|
if (ghUser.login) {
|
|
login = ghUser.login;
|
|
}
|
|
if (ghUser.avatar_url) {
|
|
avatarUrl = ghUser.avatar_url;
|
|
}
|
|
}
|
|
} catch (error) {
|
|
console.warn("resolveGithubIdentity: failed to fetch GitHub user", error);
|
|
}
|
|
|
|
return {
|
|
userId: authState.user.id,
|
|
login,
|
|
email: authState.user.email ?? `${githubAccount.accountId}@users.noreply.github.com`,
|
|
avatarUrl,
|
|
accessToken: tokenResult.accessToken,
|
|
};
|
|
}
|
|
|
|
/**
|
|
* Check if the task owner needs to swap, and if so, update the owner record
|
|
* and inject new git credentials into the sandbox.
|
|
* Returns true if an owner swap occurred.
|
|
*/
|
|
async function maybeSwapTaskOwner(c: any, authSessionId: string | null | undefined, sandbox: any | null): Promise<boolean> {
|
|
if (!authSessionId) {
|
|
return false;
|
|
}
|
|
|
|
const identity = await resolveGithubIdentity(authSessionId);
|
|
if (!identity) {
|
|
return false;
|
|
}
|
|
|
|
const currentOwner = await readTaskOwner(c);
|
|
if (currentOwner?.primaryUserId === identity.userId) {
|
|
return false;
|
|
}
|
|
|
|
await upsertTaskOwner(c, {
|
|
primaryUserId: identity.userId,
|
|
primaryGithubLogin: identity.login,
|
|
primaryGithubEmail: identity.email,
|
|
primaryGithubAvatarUrl: identity.avatarUrl,
|
|
});
|
|
|
|
if (sandbox) {
|
|
await injectGitCredentials(sandbox, identity.login, identity.email, identity.accessToken);
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
/**
|
|
* Manually change the task owner. Updates the owner record and broadcasts the
|
|
* change to subscribers. Git credentials are NOT injected here — they will be
|
|
* injected the next time the target user sends a message (auto-swap path).
|
|
*/
|
|
export async function changeTaskOwnerManually(
|
|
c: any,
|
|
input: { primaryUserId: string; primaryGithubLogin: string; primaryGithubEmail: string; primaryGithubAvatarUrl: string | null },
|
|
): Promise<void> {
|
|
await upsertTaskOwner(c, input);
|
|
await broadcastTaskUpdate(c);
|
|
}
|
|
|
|
export function shouldMarkSessionUnreadForStatus(meta: { thinkingSinceMs?: number | null }, status: "running" | "idle" | "error"): boolean {
|
|
if (status === "running") {
|
|
return false;
|
|
}
|
|
|
|
// Only mark unread when we observe the transition out of an active thinking state.
|
|
// Repeated idle polls for an already-finished session must not flip unread back on.
|
|
return Boolean(meta.thinkingSinceMs);
|
|
}
|
|
|
|
export function shouldRecreateSessionForModelChange(meta: {
|
|
status: "pending_provision" | "pending_session_create" | "ready" | "error";
|
|
sandboxSessionId?: string | null;
|
|
created?: boolean;
|
|
transcript?: Array<any>;
|
|
}): boolean {
|
|
if (meta.status !== "ready" || !meta.sandboxSessionId) {
|
|
return false;
|
|
}
|
|
|
|
if (meta.created) {
|
|
return false;
|
|
}
|
|
|
|
return !Array.isArray(meta.transcript) || meta.transcript.length === 0;
|
|
}
|
|
|
|
async function listSessionMetaRows(c: any, options?: { includeClosed?: boolean }): Promise<Array<any>> {
|
|
const rows = await c.db.select().from(taskWorkspaceSessions).orderBy(asc(taskWorkspaceSessions.createdAt)).all();
|
|
const mapped = rows.map((row: any) => ({
|
|
...row,
|
|
id: row.sessionId,
|
|
sessionId: row.sessionId,
|
|
sandboxSessionId: row.sandboxSessionId ?? null,
|
|
status: row.status ?? "ready",
|
|
errorMessage: row.errorMessage ?? null,
|
|
transcript: parseTranscript(row.transcriptJson),
|
|
transcriptUpdatedAt: row.transcriptUpdatedAt ?? null,
|
|
created: row.created === 1,
|
|
closed: row.closed === 1,
|
|
}));
|
|
|
|
if (options?.includeClosed === true) {
|
|
return mapped;
|
|
}
|
|
|
|
return mapped.filter((row: any) => row.closed !== true);
|
|
}
|
|
|
|
async function nextSessionName(c: any): Promise<string> {
|
|
const rows = await listSessionMetaRows(c, { includeClosed: true });
|
|
return `Session ${rows.length + 1}`;
|
|
}
|
|
|
|
async function readSessionMeta(c: any, sessionId: string): Promise<any | null> {
|
|
const row = await c.db.select().from(taskWorkspaceSessions).where(eq(taskWorkspaceSessions.sessionId, sessionId)).get();
|
|
|
|
if (!row) {
|
|
return null;
|
|
}
|
|
|
|
return {
|
|
...row,
|
|
id: row.sessionId,
|
|
sessionId: row.sessionId,
|
|
sandboxSessionId: row.sandboxSessionId ?? null,
|
|
status: row.status ?? "ready",
|
|
errorMessage: row.errorMessage ?? null,
|
|
transcript: parseTranscript(row.transcriptJson),
|
|
transcriptUpdatedAt: row.transcriptUpdatedAt ?? null,
|
|
created: row.created === 1,
|
|
closed: row.closed === 1,
|
|
};
|
|
}
|
|
|
|
async function getUserTaskState(c: any, authSessionId?: string | null): Promise<{ activeSessionId: string | null; bySessionId: Map<string, any> }> {
|
|
if (!authSessionId) {
|
|
return { activeSessionId: null, bySessionId: new Map() };
|
|
}
|
|
|
|
const authState = await getBetterAuthService().getAuthState(authSessionId);
|
|
const userId = authState?.user?.id;
|
|
if (typeof userId !== "string" || userId.length === 0) {
|
|
return { activeSessionId: null, bySessionId: new Map() };
|
|
}
|
|
|
|
const user = await getOrCreateUser(c, userId);
|
|
const state = await user.getTaskState({ taskId: c.state.taskId });
|
|
const bySessionId = new Map(
|
|
(state?.sessions ?? []).map((row: any) => [
|
|
row.sessionId,
|
|
{
|
|
unread: Boolean(row.unread),
|
|
draftText: row.draftText ?? "",
|
|
draftAttachments: parseDraftAttachments(row.draftAttachmentsJson),
|
|
draftUpdatedAtMs: row.draftUpdatedAt ?? null,
|
|
},
|
|
]),
|
|
);
|
|
return {
|
|
activeSessionId: state?.activeSessionId ?? null,
|
|
bySessionId,
|
|
};
|
|
}
|
|
|
|
async function upsertUserTaskState(c: any, authSessionId: string | null | undefined, sessionId: string, patch: Record<string, unknown>): Promise<void> {
|
|
if (!authSessionId) {
|
|
return;
|
|
}
|
|
|
|
const authState = await getBetterAuthService().getAuthState(authSessionId);
|
|
const userId = authState?.user?.id;
|
|
if (typeof userId !== "string" || userId.length === 0) {
|
|
return;
|
|
}
|
|
|
|
const user = await getOrCreateUser(c, userId);
|
|
expectQueueResponse(
|
|
await user.send(
|
|
userWorkflowQueueName("user.command.task_state.upsert"),
|
|
{
|
|
taskId: c.state.taskId,
|
|
sessionId,
|
|
patch,
|
|
},
|
|
{ wait: true, timeout: 10_000 },
|
|
),
|
|
);
|
|
}
|
|
|
|
async function deleteUserTaskState(c: any, authSessionId: string | null | undefined, sessionId: string): Promise<void> {
|
|
if (!authSessionId) {
|
|
return;
|
|
}
|
|
|
|
const authState = await getBetterAuthService().getAuthState(authSessionId);
|
|
const userId = authState?.user?.id;
|
|
if (typeof userId !== "string" || userId.length === 0) {
|
|
return;
|
|
}
|
|
|
|
const user = await getOrCreateUser(c, userId);
|
|
await user.send(
|
|
userWorkflowQueueName("user.command.task_state.delete"),
|
|
{
|
|
taskId: c.state.taskId,
|
|
sessionId,
|
|
},
|
|
{ wait: true, timeout: 10_000 },
|
|
);
|
|
}
|
|
|
|
async function resolveDefaultModel(c: any, authSessionId?: string | null): Promise<string> {
|
|
if (!authSessionId) {
|
|
return FALLBACK_MODEL;
|
|
}
|
|
|
|
const authState = await getBetterAuthService().getAuthState(authSessionId);
|
|
const userId = authState?.user?.id;
|
|
if (typeof userId !== "string" || userId.length === 0) {
|
|
return FALLBACK_MODEL;
|
|
}
|
|
|
|
const user = await getOrCreateUser(c, userId);
|
|
const userState = await user.getAppAuthState({ sessionId: authSessionId });
|
|
return userState?.profile?.defaultModel ?? FALLBACK_MODEL;
|
|
}
|
|
|
|
async function ensureSessionMeta(
|
|
c: any,
|
|
params: {
|
|
sessionId: string;
|
|
sandboxSessionId?: string | null;
|
|
model?: string;
|
|
authSessionId?: string | null;
|
|
sessionName?: string;
|
|
created?: boolean;
|
|
status?: "pending_provision" | "pending_session_create" | "ready" | "error";
|
|
errorMessage?: string | null;
|
|
},
|
|
): Promise<any> {
|
|
const existing = await readSessionMeta(c, params.sessionId);
|
|
if (existing) {
|
|
return existing;
|
|
}
|
|
|
|
const now = Date.now();
|
|
const sessionName = params.sessionName ?? (await nextSessionName(c));
|
|
const model = params.model ?? (await resolveDefaultModel(c, params.authSessionId));
|
|
|
|
await c.db
|
|
.insert(taskWorkspaceSessions)
|
|
.values({
|
|
sessionId: params.sessionId,
|
|
sandboxSessionId: params.sandboxSessionId ?? null,
|
|
sessionName,
|
|
model,
|
|
status: params.status ?? "ready",
|
|
errorMessage: params.errorMessage ?? null,
|
|
transcriptJson: "[]",
|
|
transcriptUpdatedAt: null,
|
|
created: params.created === false ? 0 : 1,
|
|
closed: 0,
|
|
thinkingSinceMs: null,
|
|
createdAt: now,
|
|
updatedAt: now,
|
|
})
|
|
.run();
|
|
|
|
return await readSessionMeta(c, params.sessionId);
|
|
}
|
|
|
|
async function updateSessionMeta(c: any, sessionId: string, values: Record<string, unknown>): Promise<any> {
|
|
await ensureSessionMeta(c, { sessionId });
|
|
await c.db
|
|
.update(taskWorkspaceSessions)
|
|
.set({
|
|
...values,
|
|
updatedAt: Date.now(),
|
|
})
|
|
.where(eq(taskWorkspaceSessions.sessionId, sessionId))
|
|
.run();
|
|
return await readSessionMeta(c, sessionId);
|
|
}
|
|
|
|
async function readSessionMetaBySandboxSessionId(c: any, sandboxSessionId: string): Promise<any | null> {
|
|
const row = await c.db.select().from(taskWorkspaceSessions).where(eq(taskWorkspaceSessions.sandboxSessionId, sandboxSessionId)).get();
|
|
if (!row) {
|
|
return null;
|
|
}
|
|
return await readSessionMeta(c, row.sessionId);
|
|
}
|
|
|
|
async function requireReadySessionMeta(c: any, sessionId: string): Promise<any> {
|
|
const meta = await readSessionMeta(c, sessionId);
|
|
if (!meta) {
|
|
throw new Error(`Unknown workspace session: ${sessionId}`);
|
|
}
|
|
if (meta.status !== "ready" || !meta.sandboxSessionId) {
|
|
throw new Error(meta.errorMessage ?? "This workspace session is still preparing");
|
|
}
|
|
return meta;
|
|
}
|
|
|
|
export function requireSendableSessionMeta(meta: any, sessionId: string): any {
|
|
if (!meta) {
|
|
throw new Error(`Unknown workspace session: ${sessionId}`);
|
|
}
|
|
if (meta.status !== "ready" || !meta.sandboxSessionId) {
|
|
throw new Error(`Session is not ready (status: ${meta.status}). Wait for session provisioning to complete.`);
|
|
}
|
|
return meta;
|
|
}
|
|
|
|
function shellFragment(parts: string[]): string {
|
|
return parts.join(" && ");
|
|
}
|
|
|
|
function stableSandboxId(c: any): string {
|
|
return c.state.taskId;
|
|
}
|
|
|
|
async function getTaskSandboxRuntime(
|
|
c: any,
|
|
record: any,
|
|
): Promise<{
|
|
sandbox: any;
|
|
sandboxId: string;
|
|
sandboxProviderId: string;
|
|
switchTarget: string;
|
|
cwd: string;
|
|
}> {
|
|
const { config } = getActorRuntimeContext();
|
|
const sandboxId = stableSandboxId(c);
|
|
const sandboxProviderId = resolveSandboxProviderId(config, record.sandboxProviderId ?? null);
|
|
const sandbox = await getOrCreateTaskSandbox(c, c.state.organizationId, sandboxId, {});
|
|
const actorId = typeof sandbox.resolve === "function" ? await sandbox.resolve().catch(() => null) : null;
|
|
const switchTarget = sandboxProviderId === "local" ? `sandbox://local/${sandboxId}` : `sandbox://e2b/${sandboxId}`;
|
|
const now = Date.now();
|
|
|
|
await c.db
|
|
.insert(taskSandboxes)
|
|
.values({
|
|
sandboxId,
|
|
sandboxProviderId,
|
|
sandboxActorId: typeof actorId === "string" ? actorId : null,
|
|
switchTarget,
|
|
cwd: SANDBOX_REPO_CWD,
|
|
createdAt: now,
|
|
updatedAt: now,
|
|
})
|
|
.onConflictDoUpdate({
|
|
target: taskSandboxes.sandboxId,
|
|
set: {
|
|
sandboxProviderId,
|
|
sandboxActorId: typeof actorId === "string" ? actorId : null,
|
|
switchTarget,
|
|
cwd: SANDBOX_REPO_CWD,
|
|
updatedAt: now,
|
|
},
|
|
})
|
|
.run();
|
|
|
|
await c.db
|
|
.update(taskRuntime)
|
|
.set({
|
|
activeSandboxId: sandboxId,
|
|
activeSwitchTarget: switchTarget,
|
|
activeCwd: SANDBOX_REPO_CWD,
|
|
updatedAt: now,
|
|
})
|
|
.where(eq(taskRuntime.id, 1))
|
|
.run();
|
|
|
|
return {
|
|
sandbox,
|
|
sandboxId,
|
|
sandboxProviderId,
|
|
switchTarget,
|
|
cwd: SANDBOX_REPO_CWD,
|
|
};
|
|
}
|
|
|
|
/**
|
|
* Track whether the sandbox repo has been fully prepared (cloned + fetched + checked out)
|
|
* for the current actor lifecycle. Subsequent calls can skip the expensive `git fetch`
|
|
* when `skipFetch` is true (used by sendWorkspaceMessage to avoid blocking on every prompt).
|
|
*/
|
|
let sandboxRepoPrepared = false;
|
|
|
|
async function ensureSandboxRepo(c: any, sandbox: any, record: any, opts?: { skipFetchIfPrepared?: boolean; authSessionId?: string | null }): Promise<void> {
|
|
if (!record.branchName) {
|
|
throw new Error("cannot prepare a sandbox repo before the task branch exists");
|
|
}
|
|
|
|
// If the repo was already prepared and the caller allows skipping fetch, just return.
|
|
// The clone, fetch, and checkout already happened on a prior call.
|
|
if (opts?.skipFetchIfPrepared && sandboxRepoPrepared) {
|
|
return;
|
|
}
|
|
|
|
const auth = await resolveOrganizationGithubAuth(c, c.state.organizationId);
|
|
const metadata = await getRepositoryMetadata(c);
|
|
const baseRef = metadata.defaultBranch ?? "main";
|
|
const sandboxRepoRoot = dirname(SANDBOX_REPO_CWD);
|
|
const script = [
|
|
"set -euo pipefail",
|
|
`mkdir -p ${JSON.stringify(sandboxRepoRoot)}`,
|
|
"git config --global credential.helper '!f() { echo username=x-access-token; echo password=${GH_TOKEN:-$GITHUB_TOKEN}; }; f'",
|
|
`if [ ! -d ${JSON.stringify(`${SANDBOX_REPO_CWD}/.git`)} ]; then rm -rf ${JSON.stringify(SANDBOX_REPO_CWD)} && git clone ${JSON.stringify(
|
|
metadata.remoteUrl,
|
|
)} ${JSON.stringify(SANDBOX_REPO_CWD)}; fi`,
|
|
`cd ${JSON.stringify(SANDBOX_REPO_CWD)}`,
|
|
"git fetch origin --prune",
|
|
`if git show-ref --verify --quiet refs/remotes/origin/${JSON.stringify(record.branchName).slice(1, -1)}; then target_ref=${JSON.stringify(
|
|
`origin/${record.branchName}`,
|
|
)}; else target_ref=${JSON.stringify(baseRef)}; fi`,
|
|
`git checkout -B ${JSON.stringify(record.branchName)} \"$target_ref\"`,
|
|
];
|
|
const result = await sandbox.runProcess({
|
|
command: "bash",
|
|
args: ["-lc", script.join("; ")],
|
|
cwd: "/",
|
|
env: auth?.githubToken
|
|
? {
|
|
GH_TOKEN: auth.githubToken,
|
|
GITHUB_TOKEN: auth.githubToken,
|
|
}
|
|
: undefined,
|
|
timeoutMs: 5 * 60_000,
|
|
});
|
|
|
|
if ((result.exitCode ?? 0) !== 0) {
|
|
throw new Error(`sandbox repo preparation failed (${result.exitCode ?? 1}): ${[result.stdout, result.stderr].filter(Boolean).join("")}`);
|
|
}
|
|
|
|
// On first repo preparation, inject the task owner's git credentials into the sandbox
|
|
// so that push/commit operations are authenticated and attributed to the correct user.
|
|
if (!sandboxRepoPrepared && opts?.authSessionId) {
|
|
await maybeSwapTaskOwner(c, opts.authSessionId, sandbox);
|
|
}
|
|
|
|
sandboxRepoPrepared = true;
|
|
}
|
|
|
|
async function executeInSandbox(
|
|
c: any,
|
|
params: {
|
|
sandboxId: string;
|
|
cwd: string;
|
|
command: string;
|
|
label: string;
|
|
},
|
|
): Promise<{ exitCode: number; result: string }> {
|
|
const record = await ensureWorkspaceSeeded(c);
|
|
const runtime = await getTaskSandboxRuntime(c, record);
|
|
await ensureSandboxRepo(c, runtime.sandbox, record);
|
|
const response = await runtime.sandbox.runProcess({
|
|
command: "bash",
|
|
args: ["-lc", shellFragment([`cd ${JSON.stringify(params.cwd)}`, params.command])],
|
|
cwd: "/",
|
|
timeoutMs: 5 * 60_000,
|
|
});
|
|
|
|
return {
|
|
exitCode: response.exitCode ?? 0,
|
|
result: [response.stdout, response.stderr].filter(Boolean).join(""),
|
|
};
|
|
}
|
|
|
|
function parseGitStatus(output: string): Array<{ path: string; type: "M" | "A" | "D" }> {
|
|
return output
|
|
.split("\n")
|
|
.map((line) => line.trimEnd())
|
|
.filter(Boolean)
|
|
.map((line) => {
|
|
const status = line.slice(0, 2).trim();
|
|
const rawPath = line.slice(3).trim();
|
|
const path = rawPath.includes(" -> ") ? (rawPath.split(" -> ").pop() ?? rawPath) : rawPath;
|
|
const type = status.includes("D") ? "D" : status.includes("A") || status === "??" ? "A" : "M";
|
|
return { path, type };
|
|
});
|
|
}
|
|
|
|
function parseNumstat(output: string): Map<string, { added: number; removed: number }> {
|
|
const map = new Map<string, { added: number; removed: number }>();
|
|
for (const line of output.split("\n")) {
|
|
const trimmed = line.trim();
|
|
if (!trimmed) continue;
|
|
const [addedRaw, removedRaw, ...pathParts] = trimmed.split("\t");
|
|
const path = pathParts.join("\t").trim();
|
|
if (!path) continue;
|
|
map.set(path, {
|
|
added: Number.parseInt(addedRaw ?? "0", 10) || 0,
|
|
removed: Number.parseInt(removedRaw ?? "0", 10) || 0,
|
|
});
|
|
}
|
|
return map;
|
|
}
|
|
|
|
function buildFileTree(paths: string[]): Array<any> {
|
|
const root = {
|
|
children: new Map<string, any>(),
|
|
};
|
|
|
|
for (const path of paths) {
|
|
const parts = path.split("/").filter(Boolean);
|
|
let current = root;
|
|
let currentPath = "";
|
|
|
|
for (let index = 0; index < parts.length; index += 1) {
|
|
const part = parts[index]!;
|
|
currentPath = currentPath ? `${currentPath}/${part}` : part;
|
|
const isDir = index < parts.length - 1;
|
|
let node = current.children.get(part);
|
|
if (!node) {
|
|
node = {
|
|
name: part,
|
|
path: currentPath,
|
|
isDir,
|
|
children: isDir ? new Map<string, any>() : undefined,
|
|
};
|
|
current.children.set(part, node);
|
|
} else if (isDir && !(node.children instanceof Map)) {
|
|
node.children = new Map<string, any>();
|
|
}
|
|
current = node;
|
|
}
|
|
}
|
|
|
|
function sortNodes(nodes: Iterable<any>): Array<any> {
|
|
return [...nodes]
|
|
.map((node) =>
|
|
node.isDir
|
|
? {
|
|
name: node.name,
|
|
path: node.path,
|
|
isDir: true,
|
|
children: sortNodes(node.children?.values?.() ?? []),
|
|
}
|
|
: {
|
|
name: node.name,
|
|
path: node.path,
|
|
isDir: false,
|
|
},
|
|
)
|
|
.sort((left, right) => {
|
|
if (left.isDir !== right.isDir) {
|
|
return left.isDir ? -1 : 1;
|
|
}
|
|
return left.path.localeCompare(right.path);
|
|
});
|
|
}
|
|
|
|
return sortNodes(root.children.values());
|
|
}
|
|
|
|
async function collectWorkspaceGitState(c: any, record: any) {
|
|
const activeSandboxId = record.activeSandboxId;
|
|
const activeSandbox = activeSandboxId != null ? ((record.sandboxes ?? []).find((candidate: any) => candidate.sandboxId === activeSandboxId) ?? null) : null;
|
|
const cwd = activeSandbox?.cwd ?? record.sandboxes?.[0]?.cwd ?? null;
|
|
if (!activeSandboxId || !cwd) {
|
|
return {
|
|
fileChanges: [],
|
|
diffs: {},
|
|
fileTree: [],
|
|
};
|
|
}
|
|
|
|
const statusResult = await executeInSandbox(c, {
|
|
sandboxId: activeSandboxId,
|
|
cwd,
|
|
command: "git status --porcelain=v1 -uall",
|
|
label: "git status",
|
|
});
|
|
if (statusResult.exitCode !== 0) {
|
|
return {
|
|
fileChanges: [],
|
|
diffs: {},
|
|
fileTree: [],
|
|
};
|
|
}
|
|
|
|
const statusRows = parseGitStatus(statusResult.result);
|
|
const numstatResult = await executeInSandbox(c, {
|
|
sandboxId: activeSandboxId,
|
|
cwd,
|
|
command: "git diff --numstat",
|
|
label: "git diff numstat",
|
|
});
|
|
const numstat = parseNumstat(numstatResult.result);
|
|
|
|
const filesResult = await executeInSandbox(c, {
|
|
sandboxId: activeSandboxId,
|
|
cwd,
|
|
command: "git ls-files --cached --others --exclude-standard",
|
|
label: "git ls-files",
|
|
});
|
|
const allPaths = filesResult.result
|
|
.split("\n")
|
|
.map((line) => line.trim())
|
|
.filter(Boolean);
|
|
|
|
const diffs: Record<string, string> = {};
|
|
for (const row of statusRows) {
|
|
const diffResult = await executeInSandbox(c, {
|
|
sandboxId: activeSandboxId,
|
|
cwd,
|
|
command: `git diff -- ${JSON.stringify(row.path)}`,
|
|
label: `git diff ${row.path}`,
|
|
});
|
|
diffs[row.path] = diffResult.exitCode === 0 ? diffResult.result : "";
|
|
}
|
|
|
|
return {
|
|
fileChanges: statusRows.map((row) => {
|
|
const counts = numstat.get(row.path) ?? { added: 0, removed: 0 };
|
|
return {
|
|
path: row.path,
|
|
added: counts.added,
|
|
removed: counts.removed,
|
|
type: row.type,
|
|
};
|
|
}),
|
|
diffs,
|
|
fileTree: buildFileTree(allPaths),
|
|
};
|
|
}
|
|
|
|
async function readCachedGitState(c: any): Promise<{ fileChanges: Array<any>; diffs: Record<string, string>; fileTree: Array<any>; updatedAt: number | null }> {
|
|
const row = await c.db
|
|
.select({
|
|
gitStateJson: taskRuntime.gitStateJson,
|
|
gitStateUpdatedAt: taskRuntime.gitStateUpdatedAt,
|
|
})
|
|
.from(taskRuntime)
|
|
.where(eq(taskRuntime.id, 1))
|
|
.get();
|
|
const parsed = parseGitState(row?.gitStateJson);
|
|
return {
|
|
...parsed,
|
|
updatedAt: row?.gitStateUpdatedAt ?? null,
|
|
};
|
|
}
|
|
|
|
async function writeCachedGitState(c: any, gitState: { fileChanges: Array<any>; diffs: Record<string, string>; fileTree: Array<any> }): Promise<void> {
|
|
const now = Date.now();
|
|
await c.db
|
|
.update(taskRuntime)
|
|
.set({
|
|
gitStateJson: JSON.stringify(gitState),
|
|
gitStateUpdatedAt: now,
|
|
updatedAt: now,
|
|
})
|
|
.where(eq(taskRuntime.id, 1))
|
|
.run();
|
|
}
|
|
|
|
async function readSessionTranscript(c: any, record: any, sessionId: string) {
|
|
const sandboxId = record.activeSandboxId ?? stableSandboxId(c);
|
|
if (!sandboxId) {
|
|
return [];
|
|
}
|
|
|
|
const sandbox = getTaskSandbox(c, c.state.organizationId, sandboxId);
|
|
const page = await sandbox.getEvents({
|
|
sessionId,
|
|
limit: 100,
|
|
});
|
|
return page.items.map((event: any) => ({
|
|
id: event.id,
|
|
eventIndex: event.eventIndex,
|
|
sessionId: event.sessionId,
|
|
createdAt: event.createdAt,
|
|
connectionId: event.connectionId,
|
|
sender: event.sender,
|
|
payload: event.payload,
|
|
}));
|
|
}
|
|
|
|
async function writeSessionTranscript(c: any, sessionId: string, transcript: Array<any>): Promise<void> {
|
|
await updateSessionMeta(c, sessionId, {
|
|
transcriptJson: JSON.stringify(transcript),
|
|
transcriptUpdatedAt: Date.now(),
|
|
});
|
|
}
|
|
|
|
async function enqueueWorkspaceRefresh(
|
|
c: any,
|
|
command: "task.command.workspace.refresh_derived" | "task.command.workspace.refresh_session_transcript",
|
|
body: Record<string, unknown>,
|
|
): Promise<void> {
|
|
const self = selfTask(c);
|
|
await self.send(taskWorkflowQueueName(command as any), body, { wait: false });
|
|
}
|
|
|
|
async function enqueueWorkspaceEnsureSession(c: any, sessionId: string): Promise<void> {
|
|
const self = selfTask(c);
|
|
await self.send(taskWorkflowQueueName("task.command.workspace.ensure_session" as any), { sessionId }, { wait: false });
|
|
}
|
|
|
|
function pendingWorkspaceSessionStatus(record: any): "pending_provision" | "pending_session_create" {
|
|
return record.activeSandboxId ? "pending_session_create" : "pending_provision";
|
|
}
|
|
|
|
async function maybeScheduleWorkspaceRefreshes(c: any, record: any, sessions: Array<any>): Promise<void> {
|
|
const gitState = await readCachedGitState(c);
|
|
if (record.activeSandboxId && !gitState.updatedAt) {
|
|
await enqueueWorkspaceRefresh(c, "task.command.workspace.refresh_derived", {});
|
|
}
|
|
|
|
for (const session of sessions) {
|
|
if (session.closed || session.status !== "ready" || !session.sandboxSessionId || session.transcriptUpdatedAt) {
|
|
continue;
|
|
}
|
|
await enqueueWorkspaceRefresh(c, "task.command.workspace.refresh_session_transcript", {
|
|
sessionId: session.sandboxSessionId,
|
|
});
|
|
}
|
|
}
|
|
|
|
function computeWorkspaceTaskStatus(record: any, sessions: Array<any>) {
|
|
if (record.status && String(record.status).startsWith("init_")) {
|
|
return record.status;
|
|
}
|
|
if (record.status === "archived" || record.status === "killed") {
|
|
return record.status;
|
|
}
|
|
if (sessions.some((session) => session.closed !== true && session.thinkingSinceMs)) {
|
|
return "running";
|
|
}
|
|
if (sessions.some((session) => session.closed !== true && session.status === "error")) {
|
|
return "error";
|
|
}
|
|
return "idle";
|
|
}
|
|
|
|
export async function ensureWorkspaceSeeded(c: any): Promise<any> {
|
|
return await getCurrentRecord(c);
|
|
}
|
|
|
|
function buildSessionSummary(meta: any, userState?: any): any {
|
|
const derivedSandboxSessionId = meta.status === "ready" ? (meta.sandboxSessionId ?? null) : null;
|
|
const sessionStatus =
|
|
meta.status === "pending_provision" || meta.status === "pending_session_create"
|
|
? meta.status
|
|
: meta.thinkingSinceMs
|
|
? "running"
|
|
: meta.status === "error"
|
|
? "error"
|
|
: meta.status === "ready" && derivedSandboxSessionId
|
|
? "idle"
|
|
: "ready";
|
|
let thinkingSinceMs = meta.thinkingSinceMs ?? null;
|
|
let unread = Boolean(userState?.unread);
|
|
if (thinkingSinceMs && sessionStatus !== "running") {
|
|
thinkingSinceMs = null;
|
|
unread = true;
|
|
}
|
|
|
|
return {
|
|
id: meta.id,
|
|
sessionId: meta.sessionId,
|
|
sandboxSessionId: derivedSandboxSessionId,
|
|
sessionName: meta.sessionName,
|
|
agent: agentKindForModel(meta.model),
|
|
model: meta.model,
|
|
status: sessionStatus,
|
|
thinkingSinceMs: sessionStatus === "running" ? thinkingSinceMs : null,
|
|
unread,
|
|
created: Boolean(meta.created || derivedSandboxSessionId),
|
|
errorMessage: meta.errorMessage ?? null,
|
|
};
|
|
}
|
|
|
|
function buildSessionDetailFromMeta(meta: any, userState?: any): any {
|
|
const summary = buildSessionSummary(meta, userState);
|
|
return {
|
|
sessionId: meta.sessionId,
|
|
sandboxSessionId: summary.sandboxSessionId ?? null,
|
|
sessionName: summary.sessionName,
|
|
agent: summary.agent,
|
|
model: summary.model,
|
|
status: summary.status,
|
|
thinkingSinceMs: summary.thinkingSinceMs,
|
|
unread: summary.unread,
|
|
created: summary.created,
|
|
errorMessage: summary.errorMessage,
|
|
draft: {
|
|
text: userState?.draftText ?? "",
|
|
attachments: Array.isArray(userState?.draftAttachments) ? userState.draftAttachments : [],
|
|
updatedAtMs: userState?.draftUpdatedAtMs ?? null,
|
|
},
|
|
transcript: meta.transcript ?? [],
|
|
};
|
|
}
|
|
|
|
/**
|
|
* Builds a WorkspaceTaskSummary from local task actor state. Task actors push
|
|
* this to the parent organization actor so organization sidebar reads stay local.
|
|
*/
|
|
export async function buildTaskSummary(c: any, authSessionId?: string | null): Promise<any> {
|
|
const record = await ensureWorkspaceSeeded(c);
|
|
const repositoryMetadata = await getRepositoryMetadata(c);
|
|
const sessions = await listSessionMetaRows(c);
|
|
await maybeScheduleWorkspaceRefreshes(c, record, sessions);
|
|
const userTaskState = await getUserTaskState(c, authSessionId);
|
|
const taskStatus = computeWorkspaceTaskStatus(record, sessions);
|
|
const activeSessionId =
|
|
userTaskState.activeSessionId && sessions.some((meta) => meta.sessionId === userTaskState.activeSessionId) ? userTaskState.activeSessionId : null;
|
|
|
|
const owner = await readTaskOwner(c);
|
|
|
|
return {
|
|
id: c.state.taskId,
|
|
repoId: c.state.repoId,
|
|
title: record.title ?? "New Task",
|
|
status: taskStatus,
|
|
repoName: repoLabelFromRemote(repositoryMetadata.remoteUrl),
|
|
updatedAtMs: record.updatedAt,
|
|
branch: record.branchName,
|
|
pullRequest: record.pullRequest ?? null,
|
|
activeSessionId,
|
|
sessionsSummary: sessions.map((meta) => buildSessionSummary(meta, userTaskState.bySessionId.get(meta.sessionId))),
|
|
primaryUserLogin: owner?.primaryGithubLogin ?? null,
|
|
primaryUserAvatarUrl: owner?.primaryGithubAvatarUrl ?? null,
|
|
};
|
|
}
|
|
|
|
/**
|
|
* Builds a WorkspaceTaskDetail from local task actor state for direct task
|
|
* subscribers. This is a full replacement payload, not a patch.
|
|
*/
|
|
export async function buildTaskDetail(c: any, authSessionId?: string | null): Promise<any> {
|
|
const record = await ensureWorkspaceSeeded(c);
|
|
const gitState = await readCachedGitState(c);
|
|
const sessions = await listSessionMetaRows(c);
|
|
await maybeScheduleWorkspaceRefreshes(c, record, sessions);
|
|
const summary = await buildTaskSummary(c, authSessionId);
|
|
|
|
return {
|
|
...summary,
|
|
task: record.task,
|
|
fileChanges: gitState.fileChanges,
|
|
diffs: gitState.diffs,
|
|
fileTree: gitState.fileTree,
|
|
minutesUsed: 0,
|
|
sandboxes: (record.sandboxes ?? []).map((sandbox: any) => ({
|
|
sandboxProviderId: sandbox.sandboxProviderId,
|
|
sandboxId: sandbox.sandboxId,
|
|
cwd: sandbox.cwd ?? null,
|
|
})),
|
|
activeSandboxId: record.activeSandboxId ?? null,
|
|
};
|
|
}
|
|
|
|
/**
|
|
* Builds a WorkspaceSessionDetail for a specific session.
|
|
*/
|
|
export async function buildSessionDetail(c: any, sessionId: string, authSessionId?: string | null): Promise<any> {
|
|
const record = await ensureWorkspaceSeeded(c);
|
|
const meta = await readSessionMeta(c, sessionId);
|
|
if (!meta || meta.closed) {
|
|
throw new Error(`Unknown workspace session: ${sessionId}`);
|
|
}
|
|
const userTaskState = await getUserTaskState(c, authSessionId);
|
|
const userSessionState = userTaskState.bySessionId.get(sessionId);
|
|
|
|
// Skip live transcript fetch if the sandbox session doesn't exist yet or
|
|
// the session is still provisioning — the sandbox API will block/timeout.
|
|
const isPending = meta.status === "pending_provision" || meta.status === "pending_session_create";
|
|
if (!meta.sandboxSessionId || isPending) {
|
|
return buildSessionDetailFromMeta(meta, userSessionState);
|
|
}
|
|
|
|
try {
|
|
const transcript = await readSessionTranscript(c, record, meta.sandboxSessionId);
|
|
if (JSON.stringify(meta.transcript ?? []) !== JSON.stringify(transcript)) {
|
|
await writeSessionTranscript(c, meta.sessionId, transcript);
|
|
return buildSessionDetailFromMeta(
|
|
{
|
|
...meta,
|
|
transcript,
|
|
transcriptUpdatedAt: Date.now(),
|
|
},
|
|
userSessionState,
|
|
);
|
|
}
|
|
} catch (error) {
|
|
// Session detail reads degrade to cached transcript when sandbox is unavailable.
|
|
logActorWarning("task", "readSessionTranscript failed, using cached transcript", {
|
|
taskId: c.state.taskId,
|
|
sessionId,
|
|
error: resolveErrorMessage(error),
|
|
});
|
|
}
|
|
|
|
return buildSessionDetailFromMeta(meta, userSessionState);
|
|
}
|
|
|
|
export async function getTaskSummary(c: any): Promise<any> {
|
|
return await buildTaskSummary(c);
|
|
}
|
|
|
|
export async function getTaskDetail(c: any, authSessionId?: string): Promise<any> {
|
|
return await buildTaskDetail(c, authSessionId);
|
|
}
|
|
|
|
export async function getSessionDetail(c: any, sessionId: string, authSessionId?: string): Promise<any> {
|
|
return await buildSessionDetail(c, sessionId, authSessionId);
|
|
}
|
|
|
|
/**
|
|
* Replaces the old notifyWorkspaceUpdated pattern.
|
|
*
|
|
* The task actor emits two kinds of updates:
|
|
* - Push summary state up to the parent organization actor so the sidebar
|
|
* materialized projection stays current.
|
|
* - Broadcast full detail/session payloads down to direct task subscribers.
|
|
*/
|
|
export async function broadcastTaskUpdate(c: any, options?: { sessionId?: string }): Promise<void> {
|
|
const organization = await getOrCreateOrganization(c, c.state.organizationId);
|
|
await organization.send(
|
|
organizationWorkflowQueueName("organization.command.applyTaskSummaryUpdate"),
|
|
{ taskSummary: await buildTaskSummary(c) },
|
|
{ wait: false },
|
|
);
|
|
c.broadcast("taskUpdated", {
|
|
type: "taskUpdated",
|
|
detail: await buildTaskDetail(c),
|
|
});
|
|
|
|
if (options?.sessionId) {
|
|
c.broadcast("sessionUpdated", {
|
|
type: "sessionUpdated",
|
|
session: await buildSessionDetail(c, options.sessionId),
|
|
});
|
|
}
|
|
}
|
|
|
|
export async function refreshWorkspaceDerivedState(c: any): Promise<void> {
|
|
const record = await ensureWorkspaceSeeded(c);
|
|
const gitState = await collectWorkspaceGitState(c, record);
|
|
await writeCachedGitState(c, gitState);
|
|
await broadcastTaskUpdate(c);
|
|
}
|
|
|
|
export async function refreshWorkspaceSessionTranscript(c: any, sessionId: string): Promise<void> {
|
|
const record = await ensureWorkspaceSeeded(c);
|
|
const meta = (await readSessionMetaBySandboxSessionId(c, sessionId)) ?? (await readSessionMeta(c, sessionId));
|
|
if (!meta?.sandboxSessionId) {
|
|
return;
|
|
}
|
|
|
|
const transcript = await readSessionTranscript(c, record, meta.sandboxSessionId);
|
|
await writeSessionTranscript(c, meta.sessionId, transcript);
|
|
await broadcastTaskUpdate(c, { sessionId: meta.sessionId });
|
|
}
|
|
|
|
export async function renameWorkspaceTask(c: any, value: string): Promise<void> {
|
|
const nextTitle = value.trim();
|
|
if (!nextTitle) {
|
|
throw new Error("task title is required");
|
|
}
|
|
|
|
await c.db
|
|
.update(taskTable)
|
|
.set({
|
|
title: nextTitle,
|
|
updatedAt: Date.now(),
|
|
})
|
|
.where(eq(taskTable.id, 1))
|
|
.run();
|
|
await broadcastTaskUpdate(c);
|
|
}
|
|
|
|
export async function syncTaskPullRequest(c: any, pullRequest: any): Promise<void> {
|
|
const now = pullRequest?.updatedAtMs ?? Date.now();
|
|
await c.db
|
|
.update(taskTable)
|
|
.set({
|
|
pullRequestJson: pullRequest ? JSON.stringify(pullRequest) : null,
|
|
updatedAt: now,
|
|
})
|
|
.where(eq(taskTable.id, 1))
|
|
.run();
|
|
await broadcastTaskUpdate(c);
|
|
}
|
|
|
|
export async function createWorkspaceSession(c: any, model?: string, authSessionId?: string): Promise<{ sessionId: string }> {
|
|
const sessionId = `session-${randomUUID()}`;
|
|
const record = await ensureWorkspaceSeeded(c);
|
|
await ensureSessionMeta(c, {
|
|
sessionId,
|
|
model: model ?? (await resolveDefaultModel(c, authSessionId)),
|
|
authSessionId,
|
|
sandboxSessionId: null,
|
|
status: pendingWorkspaceSessionStatus(record),
|
|
created: false,
|
|
});
|
|
await upsertUserTaskState(c, authSessionId, sessionId, {
|
|
activeSessionId: sessionId,
|
|
unread: false,
|
|
});
|
|
await broadcastTaskUpdate(c, { sessionId: sessionId });
|
|
await enqueueWorkspaceEnsureSession(c, sessionId);
|
|
return { sessionId };
|
|
}
|
|
|
|
export async function ensureWorkspaceSession(c: any, sessionId: string, model?: string, authSessionId?: string): Promise<void> {
|
|
const meta = await readSessionMeta(c, sessionId);
|
|
if (!meta || meta.closed) {
|
|
return;
|
|
}
|
|
|
|
const record = await ensureWorkspaceSeeded(c);
|
|
if (meta.sandboxSessionId && meta.status === "ready") {
|
|
await enqueueWorkspaceRefresh(c, "task.command.workspace.refresh_session_transcript", {
|
|
sessionId: meta.sandboxSessionId,
|
|
});
|
|
await broadcastTaskUpdate(c, { sessionId: sessionId });
|
|
return;
|
|
}
|
|
|
|
await updateSessionMeta(c, sessionId, {
|
|
sandboxSessionId: meta.sandboxSessionId ?? sessionId,
|
|
status: "pending_session_create",
|
|
errorMessage: null,
|
|
});
|
|
|
|
try {
|
|
const runtime = await getTaskSandboxRuntime(c, record);
|
|
await ensureSandboxRepo(c, runtime.sandbox, record);
|
|
const resolvedModel = model ?? meta.model ?? (await resolveDefaultModel(c, authSessionId));
|
|
const resolvedAgent = await resolveSandboxAgentForModel(c, resolvedModel);
|
|
await runtime.sandbox.createSession({
|
|
id: meta.sandboxSessionId ?? sessionId,
|
|
agent: resolvedAgent,
|
|
model: resolvedModel,
|
|
sessionInit: {
|
|
cwd: runtime.cwd,
|
|
},
|
|
});
|
|
|
|
await updateSessionMeta(c, sessionId, {
|
|
sandboxSessionId: meta.sandboxSessionId ?? sessionId,
|
|
status: "ready",
|
|
errorMessage: null,
|
|
});
|
|
await enqueueWorkspaceRefresh(c, "task.command.workspace.refresh_session_transcript", {
|
|
sessionId: meta.sandboxSessionId ?? sessionId,
|
|
});
|
|
} catch (error) {
|
|
await updateSessionMeta(c, sessionId, {
|
|
status: "error",
|
|
errorMessage: error instanceof Error ? error.message : String(error),
|
|
});
|
|
}
|
|
|
|
await broadcastTaskUpdate(c, { sessionId: sessionId });
|
|
}
|
|
|
|
export async function enqueuePendingWorkspaceSessions(c: any): Promise<void> {
|
|
const pending = (await listSessionMetaRows(c, { includeClosed: true })).filter(
|
|
(row) => row.closed !== true && row.status !== "ready" && row.status !== "error",
|
|
);
|
|
|
|
const self = selfTask(c);
|
|
for (const row of pending) {
|
|
await self.send(taskWorkflowQueueName("task.command.workspace.ensure_session" as any), { sessionId: row.sessionId, model: row.model }, { wait: false });
|
|
}
|
|
}
|
|
|
|
export async function renameWorkspaceSession(c: any, sessionId: string, title: string): Promise<void> {
|
|
const trimmed = title.trim();
|
|
if (!trimmed) {
|
|
throw new Error("session title is required");
|
|
}
|
|
await updateSessionMeta(c, sessionId, {
|
|
sessionName: trimmed,
|
|
});
|
|
await broadcastTaskUpdate(c, { sessionId });
|
|
}
|
|
|
|
export async function selectWorkspaceSession(c: any, sessionId: string, authSessionId?: string): Promise<void> {
|
|
const meta = await readSessionMeta(c, sessionId);
|
|
if (!meta || meta.closed) {
|
|
return;
|
|
}
|
|
await upsertUserTaskState(c, authSessionId, sessionId, {
|
|
activeSessionId: sessionId,
|
|
});
|
|
await broadcastTaskUpdate(c, { sessionId });
|
|
}
|
|
|
|
export async function setWorkspaceSessionUnread(c: any, sessionId: string, unread: boolean, authSessionId?: string): Promise<void> {
|
|
await upsertUserTaskState(c, authSessionId, sessionId, {
|
|
unread,
|
|
});
|
|
await broadcastTaskUpdate(c, { sessionId });
|
|
}
|
|
|
|
export async function updateWorkspaceDraft(c: any, sessionId: string, text: string, attachments: Array<any>, authSessionId?: string): Promise<void> {
|
|
await upsertUserTaskState(c, authSessionId, sessionId, {
|
|
draftText: text,
|
|
draftAttachmentsJson: JSON.stringify(attachments),
|
|
draftUpdatedAt: Date.now(),
|
|
});
|
|
await broadcastTaskUpdate(c, { sessionId });
|
|
}
|
|
|
|
export async function changeWorkspaceModel(c: any, sessionId: string, model: string, _authSessionId?: string): Promise<void> {
|
|
const meta = await readSessionMeta(c, sessionId);
|
|
if (!meta || meta.closed) {
|
|
return;
|
|
}
|
|
|
|
if (meta.model === model) {
|
|
return;
|
|
}
|
|
|
|
const record = await ensureWorkspaceSeeded(c);
|
|
let nextMeta = await updateSessionMeta(c, sessionId, {
|
|
model,
|
|
});
|
|
let shouldEnsure = nextMeta.status === "pending_provision" || nextMeta.status === "pending_session_create" || nextMeta.status === "error";
|
|
|
|
if (shouldRecreateSessionForModelChange(nextMeta)) {
|
|
const sandbox = getTaskSandbox(c, c.state.organizationId, stableSandboxId(c));
|
|
await sandbox.destroySession(nextMeta.sandboxSessionId);
|
|
nextMeta = await updateSessionMeta(c, sessionId, {
|
|
sandboxSessionId: null,
|
|
status: pendingWorkspaceSessionStatus(record),
|
|
errorMessage: null,
|
|
transcriptJson: "[]",
|
|
transcriptUpdatedAt: null,
|
|
thinkingSinceMs: null,
|
|
});
|
|
shouldEnsure = true;
|
|
} else if (nextMeta.status === "ready" && nextMeta.sandboxSessionId) {
|
|
const sandbox = getTaskSandbox(c, c.state.organizationId, stableSandboxId(c));
|
|
if (typeof sandbox.rawSendSessionMethod === "function") {
|
|
try {
|
|
await sandbox.rawSendSessionMethod(nextMeta.sandboxSessionId, "session/set_config_option", {
|
|
configId: "model",
|
|
value: model,
|
|
});
|
|
} catch {
|
|
// Some agents do not allow live model updates. Preserve the new preference in metadata.
|
|
}
|
|
}
|
|
} else if (nextMeta.status !== "ready") {
|
|
nextMeta = await updateSessionMeta(c, sessionId, {
|
|
status: pendingWorkspaceSessionStatus(record),
|
|
errorMessage: null,
|
|
});
|
|
}
|
|
|
|
if (shouldEnsure) {
|
|
await enqueueWorkspaceEnsureSession(c, sessionId);
|
|
}
|
|
await broadcastTaskUpdate(c, { sessionId });
|
|
}
|
|
|
|
export async function sendWorkspaceMessage(c: any, sessionId: string, text: string, attachments: Array<any>, authSessionId?: string): Promise<void> {
|
|
const meta = requireSendableSessionMeta(await readSessionMeta(c, sessionId), sessionId);
|
|
const record = await ensureWorkspaceSeeded(c);
|
|
const runtime = await getTaskSandboxRuntime(c, record);
|
|
// Skip git fetch on subsequent messages — the repo was already prepared during session
|
|
// creation. This avoids a 5-30s network round-trip to GitHub on every prompt.
|
|
await ensureSandboxRepo(c, runtime.sandbox, record, { skipFetchIfPrepared: true, authSessionId });
|
|
|
|
// Check if the task owner needs to swap. If a different user is sending this message,
|
|
// update the owner record and inject their git credentials into the sandbox.
|
|
const ownerSwapped = await maybeSwapTaskOwner(c, authSessionId, runtime.sandbox);
|
|
if (ownerSwapped) {
|
|
await broadcastTaskUpdate(c);
|
|
}
|
|
const prompt = [text.trim(), ...attachments.map((attachment: any) => `@ ${attachment.filePath}:${attachment.lineNumber}\n${attachment.lineContent}`)].filter(
|
|
Boolean,
|
|
);
|
|
if (prompt.length === 0) {
|
|
throw new Error("message text is required");
|
|
}
|
|
|
|
await updateSessionMeta(c, sessionId, {
|
|
created: 1,
|
|
thinkingSinceMs: Date.now(),
|
|
});
|
|
await upsertUserTaskState(c, authSessionId, sessionId, {
|
|
unread: false,
|
|
draftText: "",
|
|
draftAttachmentsJson: "[]",
|
|
draftUpdatedAt: Date.now(),
|
|
activeSessionId: sessionId,
|
|
});
|
|
|
|
await syncWorkspaceSessionStatus(c, meta.sandboxSessionId, "running", Date.now());
|
|
|
|
try {
|
|
await runtime.sandbox.sendPrompt({
|
|
sessionId: meta.sandboxSessionId,
|
|
prompt: prompt.join("\n\n"),
|
|
});
|
|
await syncWorkspaceSessionStatus(c, meta.sandboxSessionId, "idle", Date.now());
|
|
} catch (error) {
|
|
await updateSessionMeta(c, sessionId, {
|
|
status: "error",
|
|
errorMessage: error instanceof Error ? error.message : String(error),
|
|
});
|
|
await syncWorkspaceSessionStatus(c, meta.sandboxSessionId, "error", Date.now());
|
|
throw error;
|
|
}
|
|
}
|
|
|
|
export async function stopWorkspaceSession(c: any, sessionId: string): Promise<void> {
|
|
const meta = await requireReadySessionMeta(c, sessionId);
|
|
const sandbox = getTaskSandbox(c, c.state.organizationId, stableSandboxId(c));
|
|
await sandbox.destroySession(meta.sandboxSessionId);
|
|
await updateSessionMeta(c, sessionId, {
|
|
thinkingSinceMs: null,
|
|
});
|
|
await broadcastTaskUpdate(c, { sessionId });
|
|
}
|
|
|
|
export async function syncWorkspaceSessionStatus(c: any, sessionId: string, status: "running" | "idle" | "error", at: number): Promise<void> {
|
|
const meta = (await readSessionMetaBySandboxSessionId(c, sessionId)) ?? (await ensureSessionMeta(c, { sessionId: sessionId, sandboxSessionId: sessionId }));
|
|
let changed = false;
|
|
|
|
if (status === "running") {
|
|
if (!meta.thinkingSinceMs) {
|
|
await updateSessionMeta(c, sessionId, {
|
|
thinkingSinceMs: at,
|
|
});
|
|
changed = true;
|
|
}
|
|
} else {
|
|
if (meta.thinkingSinceMs) {
|
|
await updateSessionMeta(c, sessionId, {
|
|
thinkingSinceMs: null,
|
|
});
|
|
changed = true;
|
|
}
|
|
}
|
|
|
|
if (changed) {
|
|
const sessions = await listSessionMetaRows(c, { includeClosed: true });
|
|
const nextStatus = computeWorkspaceTaskStatus(await ensureWorkspaceSeeded(c), sessions);
|
|
await c.db
|
|
.update(taskTable)
|
|
.set({
|
|
status: nextStatus,
|
|
updatedAt: at,
|
|
})
|
|
.where(eq(taskTable.id, 1))
|
|
.run();
|
|
await enqueueWorkspaceRefresh(c, "task.command.workspace.refresh_session_transcript", {
|
|
sessionId,
|
|
});
|
|
if (status !== "running") {
|
|
await enqueueWorkspaceRefresh(c, "task.command.workspace.refresh_derived", {});
|
|
}
|
|
await broadcastTaskUpdate(c, { sessionId: meta.sessionId });
|
|
}
|
|
}
|
|
|
|
export async function closeWorkspaceSession(c: any, sessionId: string, authSessionId?: string): Promise<void> {
|
|
const sessions = await listSessionMetaRows(c);
|
|
if (sessions.filter((candidate) => candidate.closed !== true).length <= 1) {
|
|
return;
|
|
}
|
|
|
|
const meta = await readSessionMeta(c, sessionId);
|
|
if (!meta) {
|
|
return;
|
|
}
|
|
if (meta.sandboxSessionId) {
|
|
const sandbox = getTaskSandbox(c, c.state.organizationId, stableSandboxId(c));
|
|
await sandbox.destroySession(meta.sandboxSessionId);
|
|
}
|
|
await updateSessionMeta(c, sessionId, {
|
|
closed: 1,
|
|
thinkingSinceMs: null,
|
|
});
|
|
const remainingSessions = sessions.filter((candidate) => candidate.sessionId !== sessionId && candidate.closed !== true);
|
|
const userTaskState = await getUserTaskState(c, authSessionId);
|
|
if (userTaskState.activeSessionId === sessionId && remainingSessions[0]) {
|
|
await upsertUserTaskState(c, authSessionId, remainingSessions[0].sessionId, {
|
|
activeSessionId: remainingSessions[0].sessionId,
|
|
});
|
|
}
|
|
await deleteUserTaskState(c, authSessionId, sessionId);
|
|
await broadcastTaskUpdate(c);
|
|
}
|
|
|
|
export async function markWorkspaceUnread(c: any, authSessionId?: string): Promise<void> {
|
|
const sessions = await listSessionMetaRows(c);
|
|
const latest = sessions[sessions.length - 1];
|
|
if (!latest) {
|
|
return;
|
|
}
|
|
await upsertUserTaskState(c, authSessionId, latest.sessionId, {
|
|
unread: true,
|
|
});
|
|
await broadcastTaskUpdate(c, { sessionId: latest.sessionId });
|
|
}
|
|
|
|
export async function publishWorkspacePr(c: any): Promise<void> {
|
|
const record = await ensureWorkspaceSeeded(c);
|
|
if (!record.branchName) {
|
|
throw new Error("cannot publish PR without a branch");
|
|
}
|
|
const metadata = await getRepositoryMetadata(c);
|
|
const repoFullName = metadata.fullName ?? githubRepoFullNameFromRemote(metadata.remoteUrl);
|
|
if (!repoFullName) {
|
|
throw new Error(`Unable to resolve GitHub repository for ${metadata.remoteUrl}`);
|
|
}
|
|
const { driver } = getActorRuntimeContext();
|
|
const auth = await resolveOrganizationGithubAuth(c, c.state.organizationId);
|
|
const created = await driver.github.createPr(repoFullName, record.branchName, record.title ?? record.task, undefined, {
|
|
githubToken: auth?.githubToken ?? null,
|
|
baseBranch: metadata.defaultBranch ?? undefined,
|
|
});
|
|
await syncTaskPullRequest(c, {
|
|
number: created.number,
|
|
status: "ready",
|
|
title: record.title ?? record.task,
|
|
body: null,
|
|
state: "open",
|
|
url: created.url,
|
|
headRefName: record.branchName,
|
|
baseRefName: metadata.defaultBranch ?? "main",
|
|
authorLogin: null,
|
|
isDraft: false,
|
|
merged: false,
|
|
updatedAtMs: Date.now(),
|
|
});
|
|
}
|
|
|
|
export async function revertWorkspaceFile(c: any, path: string): Promise<void> {
|
|
const record = await ensureWorkspaceSeeded(c);
|
|
if (!record.activeSandboxId) {
|
|
throw new Error("cannot revert file without an active sandbox");
|
|
}
|
|
const activeSandbox = (record.sandboxes ?? []).find((candidate: any) => candidate.sandboxId === record.activeSandboxId) ?? null;
|
|
if (!activeSandbox?.cwd) {
|
|
throw new Error("cannot revert file without a sandbox cwd");
|
|
}
|
|
|
|
const result = await executeInSandbox(c, {
|
|
sandboxId: record.activeSandboxId,
|
|
cwd: activeSandbox.cwd,
|
|
command: `if git ls-files --error-unmatch -- ${JSON.stringify(path)} >/dev/null 2>&1; then git restore --staged --worktree -- ${JSON.stringify(path)} || git checkout -- ${JSON.stringify(path)}; else rm -f ${JSON.stringify(path)}; fi`,
|
|
label: `git restore ${path}`,
|
|
});
|
|
if (result.exitCode !== 0) {
|
|
throw new Error(`file revert failed (${result.exitCode}): ${result.result}`);
|
|
}
|
|
await enqueueWorkspaceRefresh(c, "task.command.workspace.refresh_derived", {});
|
|
await broadcastTaskUpdate(c);
|
|
}
|