@jingyi0605/codingns 0.3.6 → 0.4.0

package/dist/server/modules/assistant-capability/assistant-capability-service.js

@@ -1,5 +1,7 @@
 import { createId } from "../../shared/utils/id.js";
+import { AppError } from "../../shared/errors/app-error.js";
 import { nowIso } from "../../shared/utils/time.js";
+import { recordButlerProxyMessageOrigin } from "../sessions/session-message-origin-utils.js";
 const ASSISTANT_CAPABILITIES = [
     {
         name: "capabilities.list",
@@ -25,6 +27,18 @@ const ASSISTANT_CAPABILITIES = [
         enabled: true,
         summary: "列出指定项目下可操作的会话"
     },
+    {
+        name: "projects.sessions.start",
+        mode: "proxy_execute",
+        enabled: true,
+        summary: "按当前助手配置为项目新建真实会话"
+    },
+    {
+        name: "sessions.start",
+        mode: "proxy_execute",
+        enabled: true,
+        summary: "按 project/workspace/sandbox 目标新建真实会话；未指定目标时自动创建沙箱"
+    },
     {
         name: "sessions.get",
         mode: "read",
@@ -55,6 +69,90 @@ const ASSISTANT_CAPABILITIES = [
         enabled: true,
         summary: "从指定会话或消息点 fork 新会话"
     },
+    {
+        name: "automations.list",
+        mode: "read",
+        enabled: true,
+        summary: "列出当前助手自动化任务"
+    },
+    {
+        name: "automations.get",
+        mode: "read",
+        enabled: true,
+        summary: "读取单个助手自动化任务详情"
+    },
+    {
+        name: "automations.create",
+        mode: "proxy_execute",
+        enabled: true,
+        summary: "创建正式助手自动化任务"
+    },
+    {
+        name: "automations.cancel",
+        mode: "proxy_execute",
+        enabled: true,
+        summary: "取消助手自动化任务"
+    },
+    {
+        name: "automations.runs.list",
+        mode: "read",
+        enabled: true,
+        summary: "读取助手自动化执行记录"
+    },
+    {
+        name: "sandboxes.list",
+        mode: "read",
+        enabled: true,
+        summary: "列出当前助手沙箱"
+    },
+    {
+        name: "sandboxes.create",
+        mode: "proxy_execute",
+        enabled: true,
+        summary: "创建新的助手沙箱"
+    },
+    {
+        name: "sandboxes.promote",
+        mode: "proxy_execute",
+        enabled: true,
+        summary: "保留或晋升助手沙箱"
+    },
+    {
+        name: "sandboxes.expire",
+        mode: "proxy_execute",
+        enabled: true,
+        summary: "标记助手沙箱过期"
+    },
+    {
+        name: "sandboxes.remove",
+        mode: "proxy_execute",
+        enabled: true,
+        summary: "清理助手沙箱"
+    },
+    {
+        name: "timers.list",
+        mode: "read",
+        enabled: true,
+        summary: "列出当前助手会话相关的计时器"
+    },
+    {
+        name: "timers.get",
+        mode: "read",
+        enabled: true,
+        summary: "读取单个助手计时器详情"
+    },
+    {
+        name: "timers.create",
+        mode: "proxy_execute",
+        enabled: true,
+        summary: "创建助手控制会话的定时继续任务"
+    },
+    {
+        name: "timers.cancel",
+        mode: "proxy_execute",
+        enabled: true,
+        summary: "取消助手控制会话计时器"
+    },
     {
         name: "terminals.list",
         mode: "read",
@@ -221,6 +319,10 @@ const ASSISTANT_CAPABILITIES = [
 export class AssistantCapabilityService {
     butlerProjectService;
     butlerSessionService;
+    butlerControlSessionService;
+    assistantAutomationService;
+    assistantSandboxService;
+    butlerControlTimerService;
     sessionHistoryService;
     sessionLiveRuntimeService;
     terminalService;
@@ -230,9 +332,14 @@ export class AssistantCapabilityService {
     worktreeSyncService;
     worktreeMergeService;
     worktreeCleanupService;
-    constructor(butlerProjectService, butlerSessionService, sessionHistoryService, sessionLiveRuntimeService, terminalService, debugTargetService, workspaceService, worktreeManager, worktreeSyncService, worktreeMergeService, worktreeCleanupService) {
+    sessionMessageOriginRepository;
+    constructor(butlerProjectService, butlerSessionService, butlerControlSessionService, assistantAutomationService, assistantSandboxService, butlerControlTimerService, sessionHistoryService, sessionLiveRuntimeService, terminalService, debugTargetService, workspaceService, worktreeManager, worktreeSyncService, worktreeMergeService, worktreeCleanupService, sessionMessageOriginRepository = null) {
         this.butlerProjectService = butlerProjectService;
         this.butlerSessionService = butlerSessionService;
+        this.butlerControlSessionService = butlerControlSessionService;
+        this.assistantAutomationService = assistantAutomationService;
+        this.assistantSandboxService = assistantSandboxService;
+        this.butlerControlTimerService = butlerControlTimerService;
         this.sessionHistoryService = sessionHistoryService;
         this.sessionLiveRuntimeService = sessionLiveRuntimeService;
         this.terminalService = terminalService;
@@ -242,6 +349,7 @@ export class AssistantCapabilityService {
         this.worktreeSyncService = worktreeSyncService;
         this.worktreeMergeService = worktreeMergeService;
         this.worktreeCleanupService = worktreeCleanupService;
+        this.sessionMessageOriginRepository = sessionMessageOriginRepository;
     }
     listCapabilities() {
         return this.createReceipt("capabilities.list", {
@@ -289,6 +397,72 @@ export class AssistantCapabilityService {
             items
         });
     }
+    async startProjectSession(input) {
+        const config = this.resolveSessionLaunchConfig(input);
+        const session = await this.butlerSessionService.startSession(input.projectId, {
+            role: "adhoc",
+            ownershipMode: "managed",
+            content: input.content.trim(),
+            providerId: config.providerId,
+            model: config.model,
+            reasoningLevel: config.reasoningLevel,
+            permissionMode: config.permissionMode
+        }, input.userId);
+        return this.createReceipt("projects.sessions.start", {
+            kind: "project",
+            id: input.projectId
+        }, {
+            session
+        });
+    }
+    async startSession(input) {
+        const config = this.resolveSessionLaunchConfig(input);
+        const currentControlSession = this.butlerControlSessionService.getCurrentSession(input.userId);
+        const target = input.target
+            ? this.resolveAssistantSessionTarget(input.target, input.userId)
+            : await this.createAutomaticSandboxTarget(input, currentControlSession);
+        if (target.kind === "project") {
+            const session = await this.butlerSessionService.startSession(target.id, {
+                role: "adhoc",
+                ownershipMode: "managed",
+                content: input.content.trim(),
+                providerId: config.providerId,
+                model: config.model,
+                reasoningLevel: config.reasoningLevel,
+                permissionMode: config.permissionMode
+            }, input.userId);
+            return this.createReceipt("sessions.start", {
+                kind: "project",
+                id: target.id
+            }, {
+                session,
+                target
+            });
+        }
+        const session = await this.sessionLiveRuntimeService.startLiveSession({
+            workspaceId: target.workspaceId,
+            userId: input.userId,
+            provider: config.providerId,
+            content: input.content.trim(),
+            clientRequestId: null,
+            runtimeOptions: {
+                model: config.model,
+                reasoningLevel: config.reasoningLevel,
+                permissionMode: config.permissionMode,
+                attachments: []
+            }
+        });
+        if (target.kind === "sandbox") {
+            this.assistantSandboxService.markSandboxUsedByControlSession(target.id, input.userId, currentControlSession?.id ?? null);
+        }
+        return this.createReceipt("sessions.start", {
+            kind: target.kind,
+            id: target.id
+        }, {
+            session,
+            target
+        });
+    }
     getSession(sessionId, userId) {
         const session = this.sessionHistoryService.getSession(sessionId, userId);
         return this.createReceipt("sessions.get", {
@@ -317,11 +491,19 @@ export class AssistantCapabilityService {
         });
     }
     async sendSessionMessage(input) {
+        const requestedAt = nowIso();
+        const clientRequestId = recordButlerProxyMessageOrigin(this.sessionMessageOriginRepository, {
+            sessionId: input.sessionId,
+            clientRequestId: input.clientRequestId?.trim() || null,
+            content: input.content,
+            createdAt: requestedAt,
+            fallbackKey: `assistant-send:${input.sessionId}:${requestedAt}`
+        });
         const result = await this.sessionLiveRuntimeService.sendLiveMessage({
             sessionId: input.sessionId,
             userId: input.userId,
             content: input.content,
-            clientRequestId: input.clientRequestId?.trim() || null,
+            clientRequestId,
             runtimeOptions: {
                 model: input.model?.trim() || null,
                 reasoningLevel: input.reasoningLevel?.trim() || null,
@@ -329,6 +511,14 @@ export class AssistantCapabilityService {
                 attachments: []
             }
         });
+        recordButlerProxyMessageOrigin(this.sessionMessageOriginRepository, {
+            sessionId: input.sessionId,
+            clientRequestId,
+            messageId: result.message?.messageId ?? null,
+            content: input.content,
+            createdAt: result.acceptedAt,
+            fallbackKey: `assistant-send:${input.sessionId}:${result.acceptedAt}`
+        });
         return this.createReceipt("sessions.message.send", {
             kind: "session",
             id: input.sessionId
@@ -352,6 +542,229 @@ export class AssistantCapabilityService {
             session
         });
     }
+    listTimers(input) {
+        const items = this.butlerControlTimerService.listTimers({
+            userId: input.userId,
+            statuses: input.status ? [input.status] : undefined,
+            controlSessionId: input.controlSessionId ?? null,
+            limit: input.limit ?? undefined
+        });
+        return this.createReceipt("timers.list", {
+            kind: "none",
+            id: null
+        }, {
+            items
+        });
+    }
+    listAutomations(input) {
+        const items = this.assistantAutomationService.listTasks({
+            userId: input.userId,
+            statuses: input.status ? [input.status] : undefined,
+            controlSessionId: input.controlSessionId ?? null
+        });
+        return this.createReceipt("automations.list", {
+            kind: "none",
+            id: null
+        }, {
+            items
+        });
+    }
+    getAutomation(automationId, userId) {
+        const automation = this.assistantAutomationService.getTask(automationId, userId);
+        return this.createReceipt("automations.get", {
+            kind: "automation",
+            id: automationId
+        }, {
+            automation
+        });
+    }
+    createAutomation(input) {
+        const triggerType = input.triggerType ?? "once";
+        const automation = this.assistantAutomationService.createTask({
+            userId: input.userId,
+            controlSessionId: input.controlSessionId,
+            projectId: input.projectId,
+            title: input.title,
+            trigger: buildAssistantAutomationTriggerInput(triggerType, input),
+            action: {
+                type: "send_control_message",
+                content: input.content,
+                includeTriggerContext: input.includeTriggerContext ?? triggerType === "condition",
+                targetSessionId: input.targetSessionId ?? null
+            }
+        });
+        return this.createReceipt("automations.create", {
+            kind: "automation",
+            id: automation.id
+        }, {
+            automation
+        });
+    }
+    updateAutomation(input) {
+        const automation = this.assistantAutomationService.updateTask({
+            taskId: input.automationId,
+            userId: input.userId,
+            title: input.title,
+            content: input.content,
+            includeTriggerContext: input.includeTriggerContext,
+            dueAt: input.dueAt,
+            everySeconds: input.everySeconds,
+            everyMinutes: input.everyMinutes,
+            everyHours: input.everyHours,
+            stopAt: input.stopAt,
+            cronMinute: input.cronMinute,
+            cronHour: input.cronHour,
+            cronDaysOfWeek: input.cronDaysOfWeek,
+            pollIntervalSeconds: input.pollIntervalSeconds,
+            expiresAt: input.expiresAt,
+            maxChecks: input.maxChecks
+        });
+        return this.createReceipt("automations.update", {
+            kind: "automation",
+            id: input.automationId
+        }, {
+            automation
+        });
+    }
+    cancelAutomation(automationId, userId) {
+        const automation = this.assistantAutomationService.cancelTask(automationId, userId);
+        return this.createReceipt("automations.cancel", {
+            kind: "automation",
+            id: automationId
+        }, {
+            automation
+        });
+    }
+    skipAutomationWait(automationId, userId) {
+        const automation = this.assistantAutomationService.skipCurrentWait(automationId, userId);
+        return this.createReceipt("automations.wait.skip", {
+            kind: "automation",
+            id: automationId
+        }, {
+            automation
+        });
+    }
+    listAutomationRuns(automationId, userId) {
+        const items = this.assistantAutomationService.listRuns(automationId, userId);
+        return this.createReceipt("automations.runs.list", {
+            kind: "automation",
+            id: automationId
+        }, {
+            items
+        });
+    }
+    listRecentAutomationRuns(input) {
+        const items = this.assistantAutomationService.listRecentRuns({
+            userId: input.userId,
+            controlSessionId: input.controlSessionId ?? null,
+            limit: input.limit ?? undefined
+        });
+        return this.createReceipt("automations.runs.recent", {
+            kind: "none",
+            id: null
+        }, {
+            items
+        });
+    }
+    listSandboxes(input) {
+        const items = this.assistantSandboxService.listSandboxes({
+            userId: input.userId,
+            controlSessionId: input.controlSessionId ?? null,
+            statuses: input.status ? [input.status] : undefined
+        });
+        return this.createReceipt("sandboxes.list", {
+            kind: "none",
+            id: null
+        }, {
+            items
+        });
+    }
+    async createSandbox(input) {
+        const currentControlSession = this.butlerControlSessionService.getCurrentSession(input.userId);
+        const sandbox = await this.assistantSandboxService.createSandbox({
+            userId: input.userId,
+            controlSessionId: currentControlSession?.id ?? null,
+            title: input.title,
+            description: input.description,
+            purpose: input.purpose,
+            expiresAt: input.expiresAt,
+            source: input.sourceKind === "clone"
+                ? {
+                    kind: "clone",
+                    repositoryUrl: requireAssistantRepositoryUrl(input.repositoryUrl),
+                    directoryName: input.directoryName,
+                    auth: input.auth
+                }
+                : {
+                    kind: "blank",
+                    directoryName: input.directoryName
+                }
+        });
+        return this.createReceipt("sandboxes.create", {
+            kind: "sandbox",
+            id: sandbox.id
+        }, {
+            sandbox
+        });
+    }
+    promoteSandbox(input) {
+        const sandbox = this.assistantSandboxService.promoteSandbox(input.sandboxId, input.userId, {
+            mode: input.mode,
+            projectName: input.projectName,
+            defaultProvider: input.defaultProvider
+        });
+        return this.createReceipt("sandboxes.promote", {
+            kind: "sandbox",
+            id: input.sandboxId
+        }, {
+            sandbox
+        });
+    }
+    expireSandbox(sandboxId, userId) {
+        const sandbox = this.assistantSandboxService.expireSandbox(sandboxId, userId);
+        return this.createReceipt("sandboxes.expire", {
+            kind: "sandbox",
+            id: sandboxId
+        }, {
+            sandbox
+        });
+    }
+    removeSandbox(sandboxId, userId) {
+        const sandbox = this.assistantSandboxService.removeSandbox(sandboxId, userId);
+        return this.createReceipt("sandboxes.remove", {
+            kind: "sandbox",
+            id: sandboxId
+        }, {
+            sandbox
+        });
+    }
+    getTimer(timerId, userId) {
+        const timer = this.butlerControlTimerService.getTimer(timerId, userId);
+        return this.createReceipt("timers.get", {
+            kind: "timer",
+            id: timerId
+        }, {
+            timer
+        });
+    }
+    createTimer(input) {
+        const timer = this.butlerControlTimerService.createTimer(input);
+        return this.createReceipt("timers.create", {
+            kind: "timer",
+            id: timer.id
+        }, {
+            timer
+        });
+    }
+    cancelTimer(timerId, userId) {
+        const timer = this.butlerControlTimerService.cancelTimer(timerId, userId);
+        return this.createReceipt("timers.cancel", {
+            kind: "timer",
+            id: timerId
+        }, {
+            timer
+        });
+    }
     async listTerminals(input) {
         const workspaceId = input.projectId?.trim()
             ? this.butlerProjectService.getById(input.projectId).workspaceId
@@ -430,7 +843,7 @@ export class AssistantCapabilityService {
         });
     }
     async createDebugLaunchPlan(input) {
-        const plan = await this.debugTargetService.createLaunchPlan(input.targetId, input.portRequests);
+        const plan = await this.debugTargetService.createLaunchPlan(input.targetId, input.portRequests, input.userId);
         return this.createReceipt("debug-targets.launch-plan.create", {
             kind: "debug_target",
             id: input.targetId
@@ -630,5 +1043,154 @@ export class AssistantCapabilityService {
             payload
         };
     }
+    resolveSessionLaunchConfig(input) {
+        const controlSession = this.butlerControlSessionService.getCurrentSession(input.userId);
+        const providerId = input.providerId?.trim()
+            ?? controlSession?.providerId
+            ?? undefined;
+        if (!providerId) {
+            throw new AppError({
+                statusCode: 409,
+                errorCode: "ASSISTANT_CONTROL_SESSION_NOT_FOUND",
+                detail: "当前没有可用的助手控制会话，无法继承默认 provider"
+            });
+        }
+        return {
+            providerId,
+            model: normalizeAssistantText(input.model) ?? controlSession?.model ?? null,
+            reasoningLevel: normalizeAssistantText(input.reasoningLevel) ?? controlSession?.reasoningLevel ?? null,
+            permissionMode: normalizeAssistantText(input.permissionMode) ?? controlSession?.permissionMode ?? null
+        };
+    }
+    resolveAssistantSessionTarget(target, userId) {
+        if (target.kind === "project") {
+            const project = this.butlerProjectService.getById(target.projectId);
+            return {
+                kind: "project",
+                id: project.id,
+                workspaceId: project.workspaceId
+            };
+        }
+        if (target.kind === "workspace") {
+            return {
+                kind: "workspace",
+                id: target.workspaceId,
+                workspaceId: target.workspaceId
+            };
+        }
+        return {
+            kind: "sandbox",
+            id: target.sandboxId,
+            workspaceId: this.assistantSandboxService.resolveWorkspaceId(target.sandboxId, userId)
+        };
+    }
+    async createAutomaticSandboxTarget(input, controlSession) {
+        const sandbox = await this.assistantSandboxService.createSandbox({
+            userId: input.userId,
+            controlSessionId: controlSession?.id ?? null,
+            title: inferAutomaticSandboxTitle(controlSession, input.content),
+            purpose: "当前任务未指定工作区，系统自动创建",
+            source: {
+                kind: "blank"
+            }
+        });
+        return {
+            kind: "sandbox",
+            id: sandbox.id,
+            workspaceId: sandbox.workspaceId
+        };
+    }
+}
+function normalizeAssistantText(value) {
+    const normalized = value?.trim();
+    return normalized ? normalized : null;
+}
+function inferAutomaticSandboxTitle(controlSession, content) {
+    return normalizeAssistantText(controlSession?.title)
+        ?? normalizeAssistantText(controlSession?.lastSummary)
+        ?? summarizeAssistantText(content)
+        ?? "助手临时沙箱";
+}
+function summarizeAssistantText(content) {
+    const normalized = normalizeAssistantText(content);
+    if (!normalized) {
+        return null;
+    }
+    return normalized.length > 48 ? `${normalized.slice(0, 45)}...` : normalized;
+}
+function buildAssistantAutomationTriggerInput(triggerType, input) {
+    switch (triggerType) {
+        case "once":
+            return {
+                type: "once",
+                dueAt: input.dueAt ?? null,
+                afterSeconds: input.afterSeconds ?? null
+            };
+        case "interval":
+            return {
+                type: "interval",
+                seconds: input.everySeconds ?? null,
+                minutes: input.everyMinutes ?? null,
+                hours: input.everyHours ?? null,
+                stopAt: input.stopAt ?? null
+            };
+        case "cron":
+            return {
+                type: "cron",
+                minute: input.cronMinute ?? null,
+                hour: input.cronHour ?? null,
+                daysOfWeek: input.cronDaysOfWeek ?? null,
+                stopAt: input.stopAt ?? null
+            };
+        case "condition": {
+            const conditionKind = input.conditionKind;
+            if (conditionKind === "git.remote_tag_changed") {
+                return {
+                    type: "condition",
+                    conditionKind,
+                    repositoryUrl: input.repositoryUrl ?? null,
+                    pollIntervalSeconds: input.pollIntervalSeconds ?? null,
+                    expiresAt: input.expiresAt ?? null,
+                    maxChecks: input.maxChecks ?? null
+                };
+            }
+            return {
+                type: "condition",
+                conditionKind: requireAssistantConditionKind(conditionKind),
+                sessionId: input.conditionSessionId ?? null,
+                pollIntervalSeconds: input.pollIntervalSeconds ?? null,
+                expiresAt: input.expiresAt ?? null,
+                maxChecks: input.maxChecks ?? null
+            };
+        }
+        default:
+            return assertNeverAssistantAutomationTriggerType(triggerType);
+    }
+}
+function requireAssistantRepositoryUrl(value) {
+    const normalized = normalizeAssistantText(value);
+    if (!normalized) {
+        throw new AppError({
+            statusCode: 400,
+            errorCode: "INVALID_INPUT",
+            detail: "clone 沙箱必须提供 repositoryUrl",
+            field: "repositoryUrl"
+        });
+    }
+    return normalized;
+}
+function requireAssistantConditionKind(value) {
+    if (value === "git.remote_tag_changed" || value === "session.runtime_idle") {
+        return value;
+    }
+    throw new AppError({
+        statusCode: 400,
+        errorCode: "INVALID_INPUT",
+        detail: "condition 自动化必须提供 conditionKind",
+        field: "conditionKind"
+    });
+}
+function assertNeverAssistantAutomationTriggerType(value) {
+    throw new Error(`Unexpected assistant automation triggerType: ${String(value)}`);
 }
 //# sourceMappingURL=assistant-capability-service.js.map