aws-runtime-bridge 1.0.0

package/dist/services/instance-state.test.js

@@ -0,0 +1,213 @@
+import { describe, it, expect, beforeEach, afterEach } from 'vitest';
+import os from 'node:os';
+import path from 'node:path';
+import { promises as fs } from 'node:fs';
+import { getInstanceStateFile, loadInstanceState, saveInstanceState, getAgentInitRoot, } from './instance-state.js';
+describe('instance-state service', () => {
+    const testAgentId = 'test-agent-123';
+    const stateFilePath = getInstanceStateFile(testAgentId);
+    beforeEach(async () => {
+        // 清理测试文件
+        try {
+            await fs.rm(path.dirname(stateFilePath), { recursive: true, force: true });
+        }
+        catch {
+            // 目录不存在，忽略
+        }
+    });
+    afterEach(async () => {
+        // 清理测试文件
+        try {
+            await fs.rm(path.dirname(stateFilePath), { recursive: true, force: true });
+        }
+        catch {
+            // 目录不存在，忽略
+        }
+    });
+    describe('getInstanceStateFile', () => {
+        it('should return correct path in temp directory', () => {
+            const result = getInstanceStateFile(testAgentId);
+            const expected = path.join(os.tmpdir(), 'agentswork-runtime-init', testAgentId, 'state.json');
+            expect(result).toBe(expected);
+        });
+        it('should handle agentId with special characters', () => {
+            const specialAgentId = 'agent-with-dashes_and_underscores';
+            const result = getInstanceStateFile(specialAgentId);
+            expect(result).toContain(specialAgentId);
+            expect(result).toContain('state.json');
+        });
+        it('should convert numeric agentId to string', () => {
+            const numericId = 12345;
+            const result = getInstanceStateFile(numericId);
+            expect(result).toContain('12345');
+        });
+    });
+    describe('getAgentInitRoot', () => {
+        it('should return correct init root directory', () => {
+            const result = getAgentInitRoot(testAgentId);
+            const expected = path.join(os.tmpdir(), 'agentswork-runtime-init', testAgentId);
+            expect(result).toBe(expected);
+        });
+    });
+    describe('loadInstanceState', () => {
+        it('should return default state when file does not exist', async () => {
+            const state = await loadInstanceState('non-existent-agent-id');
+            expect(state.initialized).toBe(false);
+            expect(state.skills).toEqual([]);
+            expect(state.mcpServers).toEqual([]);
+            expect(state.enabledTools).toEqual(['claude', 'opencode', 'codex']);
+            expect(state.toolStatus).toEqual({});
+            expect(state.skillEnabled).toBe(true);
+            expect(state.mcpEnabled).toBe(true);
+            expect(state.updatedAt).toBeNull();
+        });
+        it('should load existing state from file', async () => {
+            // 先保存一个状态
+            const savedState = {
+                initialized: true,
+                skills: [
+                    {
+                        id: 'skill-1',
+                        name: 'test-skill',
+                        fileName: 'test.zip',
+                        filePath: '/path/to/test.zip',
+                        bytes: 1024,
+                        installedAt: '2024-01-01T00:00:00Z',
+                    },
+                ],
+                enabledTools: ['claude', 'gemini'],
+                skillEnabled: false,
+                mcpEnabled: true,
+            };
+            await saveInstanceState(testAgentId, savedState);
+            // 然后加载
+            const loadedState = await loadInstanceState(testAgentId);
+            expect(loadedState.initialized).toBe(true);
+            expect(loadedState.skills).toHaveLength(1);
+            expect(loadedState.skills[0].name).toBe('test-skill');
+            expect(loadedState.enabledTools).toEqual(['claude', 'gemini']);
+            expect(loadedState.skillEnabled).toBe(false);
+            expect(loadedState.mcpEnabled).toBe(true);
+            expect(loadedState.updatedAt).not.toBeNull();
+        });
+        it('should handle corrupted JSON file gracefully', async () => {
+            // 创建损坏的 JSON 文件
+            await fs.mkdir(path.dirname(stateFilePath), { recursive: true });
+            await fs.writeFile(stateFilePath, '{ invalid json }', 'utf-8');
+            const state = await loadInstanceState(testAgentId);
+            // 应该返回默认状态
+            expect(state.initialized).toBe(false);
+            expect(state.skills).toEqual([]);
+        });
+    });
+    describe('saveInstanceState', () => {
+        it('should save complete state with all fields', async () => {
+            const partialState = {
+                initialized: true,
+                skills: [],
+                mcpServers: [],
+                enabledTools: ['opencode'],
+                toolStatus: {
+                    claude: {
+                        tool: 'claude',
+                        installed: true,
+                        executable: '/usr/local/bin/claude',
+                        version: '1.0.0',
+                        installing: false,
+                        error: null,
+                    },
+                },
+                skillEnabled: false,
+                mcpEnabled: true,
+            };
+            const result = await saveInstanceState(testAgentId, partialState);
+            expect(result.initialized).toBe(true);
+            expect(result.enabledTools).toEqual(['opencode']);
+            expect(result.toolStatus.claude.installed).toBe(true);
+            expect(result.skillEnabled).toBe(false);
+            expect(result.mcpEnabled).toBe(true);
+            expect(result.updatedAt).not.toBeNull();
+            // 验证文件已创建
+            const fileContent = await fs.readFile(stateFilePath, 'utf-8');
+            const parsed = JSON.parse(fileContent);
+            expect(parsed.initialized).toBe(true);
+        });
+        it('should use default values for missing arrays', async () => {
+            const partialState = {
+                initialized: true,
+            };
+            const result = await saveInstanceState(testAgentId, partialState);
+            expect(result.skills).toEqual([]);
+            expect(result.mcpServers).toEqual([]);
+            expect(result.enabledTools).toEqual(['claude', 'opencode', 'codex']);
+        });
+        it('should handle empty state object', async () => {
+            const result = await saveInstanceState(testAgentId, {});
+            expect(result.initialized).toBe(false);
+            expect(result.skills).toEqual([]);
+            expect(result.mcpServers).toEqual([]);
+            expect(result.enabledTools).toEqual(['claude', 'opencode', 'codex']);
+            expect(result.toolStatus).toEqual({});
+            expect(result.skillEnabled).toBe(true);
+            expect(result.mcpEnabled).toBe(true);
+        });
+        it('should handle null/undefined values', async () => {
+            const partialState = {
+                initialized: undefined,
+                skills: null,
+                mcpServers: undefined,
+                enabledTools: null,
+                toolStatus: null,
+                skillEnabled: undefined,
+                mcpEnabled: null,
+            };
+            const result = await saveInstanceState(testAgentId, partialState);
+            expect(result.initialized).toBe(false);
+            expect(result.skills).toEqual([]);
+            expect(result.mcpServers).toEqual([]);
+            expect(result.enabledTools).toEqual(['claude', 'opencode', 'codex']);
+            expect(result.toolStatus).toEqual({});
+            expect(result.skillEnabled).toBe(true);
+            expect(result.mcpEnabled).toBe(true);
+        });
+        it('should handle invalid toolStatus type', async () => {
+            const partialState = {
+                toolStatus: 'invalid',
+            };
+            const result = await saveInstanceState(testAgentId, partialState);
+            expect(result.toolStatus).toEqual({});
+        });
+        it('should handle array as toolStatus', async () => {
+            const partialState = {
+                toolStatus: [],
+            };
+            const result = await saveInstanceState(testAgentId, partialState);
+            expect(result.toolStatus).toEqual({});
+        });
+        it('should deduplicate and normalize enabledTools', async () => {
+            const partialState = {
+                enabledTools: ['CLAUDE', 'claude', 'OpenCode', '  gemini  ', ''],
+            };
+            const result = await saveInstanceState(testAgentId, partialState);
+            expect(result.enabledTools).toEqual(['claude', 'opencode', 'gemini']);
+        });
+        it('should overwrite existing state', async () => {
+            // 第一次保存
+            await saveInstanceState(testAgentId, { initialized: true });
+            // 第二次保存
+            const result = await saveInstanceState(testAgentId, { initialized: false });
+            expect(result.initialized).toBe(false);
+            // 验证文件内容
+            const fileContent = await fs.readFile(stateFilePath, 'utf-8');
+            const parsed = JSON.parse(fileContent);
+            expect(parsed.initialized).toBe(false);
+        });
+        it('should update updatedAt timestamp on each save', async () => {
+            const first = await saveInstanceState(testAgentId, { initialized: true });
+            // 等待一小段时间确保时间戳不同
+            await new Promise((resolve) => setTimeout(resolve, 10));
+            const second = await saveInstanceState(testAgentId, { initialized: false });
+            expect(second.updatedAt).not.toBe(first.updatedAt);
+        });
+    });
+});

package/dist/services/memory-service.d.ts

@@ -0,0 +1,195 @@
+/**
+ * 记忆服务
+ *
+ * 支持三文件存储结构：
+ * - todo-memory.json: task_list 类型记忆
+ * - hot-memory.json: recently_memory 类型记忆
+ * - cold-memory.json: memory 类型记忆（长期记忆）
+ *
+ * 文件存储在项目根目录下的 vibe-coding/memory/ 目录
+ *
+ * 权限控制：
+ * - 私有记忆（isPublic=false）：仅创建者可读写
+ * - 公开记忆（isPublic=true）：小组内所有成员可读取，仅创建者可编辑删除
+ */
+export type MemoryType = "task_list" | "recently_memory" | "memory";
+export interface MemoryRecord {
+    id: string;
+    uri: string;
+    domain: string;
+    path: string;
+    content: string;
+    memory_type: MemoryType;
+    createdAt: string;
+    updatedAt: string;
+    version: number;
+    creatorId: string;
+    creatorName: string;
+    isPublic: boolean;
+    lastEditorId?: string;
+    lastEditorName?: string;
+}
+export interface CreateMemoryInput {
+    memory_type: MemoryType;
+    uri: string;
+    content: string;
+    creatorId: string;
+    creatorName: string;
+    isPublic?: boolean;
+}
+export interface UpdateMemoryInput {
+    memory_type?: MemoryType;
+    uri: string;
+    old_string?: string;
+    new_string?: string;
+    append?: string;
+    editorId: string;
+    editorName: string;
+}
+export interface ReadMemoryInput {
+    memory_type?: MemoryType;
+    uri: string;
+    requestorId: string;
+}
+export interface DeleteMemoryInput {
+    memory_type?: MemoryType;
+    uri: string;
+    requestorId: string;
+}
+export interface SearchMemoryInput {
+    query: string;
+    limit?: number;
+    memory_type?: MemoryType;
+    requestorId: string;
+}
+export interface ListMemoryInput {
+    memory_type: MemoryType;
+    requestorId: string;
+}
+/**
+ * 获取记忆文件路径（异步版本）
+ *
+ * 记忆文件存储在 vibe-coding/memory/ 目录下（与 my.yml 同级）
+ */
+export declare function getMemoryFilePath(workspacePath: string, memoryType: MemoryType): Promise<string>;
+/**
+ * 获取记忆目录路径（异步版本）
+ */
+export declare function getMemoryDirPath(workspacePath: string): Promise<string>;
+export declare class MemoryService {
+    /**
+     * 确保记忆目录存在（已使用异步 API）
+     */
+    private ensureMemoryDir;
+    /**
+     * 生成当前时间的 ISO 字符串
+     */
+    private nowIso;
+    /**
+     * 规范化 URI
+     */
+    private normalizeUri;
+    /**
+     * 规范化记忆类型
+     */
+    private normalizeMemoryType;
+    /**
+     * 确保类型匹配
+     */
+    private ensureTypeMatch;
+    /**
+     * 生成记忆 ID
+     */
+    private generateId;
+    /**
+     * 加载指定类型的记忆文件（已使用异步 API）
+     */
+    private loadMemoryFile;
+    /**
+     * 保存记忆文件（已使用异步 API）
+     */
+    private saveMemoryFile;
+    /**
+     * 创建记忆
+     */
+    createMemory(workspacePath: string, input: CreateMemoryInput): Promise<MemoryRecord>;
+    /**
+     * 读取记忆（带权限检查）
+     *
+     * 权限规则：
+     * - 自己的记忆：可读取
+     * - 公开的记忆：可读取，但会标注来源信息
+     * - 他人私有记忆：无法读取
+     */
+    readMemory(workspacePath: string, input: ReadMemoryInput): Promise<(MemoryRecord & {
+        isOwn: boolean;
+        source?: string;
+    }) | null>;
+    /**
+     * 更新记忆（带权限检查）
+     *
+     * 权限规则：只有创建者可以更新自己的记忆
+     * 更新时会记录最后编辑者信息
+     */
+    updateMemory(workspacePath: string, input: UpdateMemoryInput): Promise<MemoryRecord>;
+    /**
+     * 删除记忆（带权限检查）
+     *
+     * 权限规则：只有创建者可以删除自己的记忆
+     */
+    deleteMemory(workspacePath: string, input: DeleteMemoryInput): Promise<{
+        deleted: boolean;
+        reason?: string;
+    }>;
+    /**
+     * 搜索记忆（带权限过滤）
+     *
+     * 权限规则：只返回自己的记忆 + 公开的记忆
+     * 公开的记忆会标注来源信息
+     */
+    searchMemory(workspacePath: string, input: SearchMemoryInput): Promise<{
+        query: string;
+        results: Array<MemoryRecord & {
+            isOwn: boolean;
+            source?: string;
+        }>;
+        totalFound: number;
+    }>;
+    /**
+     * 按类型列出记忆（带权限过滤）
+     *
+     * 权限规则：只列出自己的记忆 + 公开的记忆
+     */
+    listByType(workspacePath: string, input: ListMemoryInput): Promise<Array<MemoryRecord & {
+        isOwn: boolean;
+        source?: string;
+    }>>;
+    /**
+     * 删除任务列表记忆（仅限 task_list 类型，带权限检查）
+     */
+    removeTaskListMemory(workspacePath: string, uri: string, requestorId: string): Promise<{
+        deleted: boolean;
+        reason?: string;
+    }>;
+    /**
+     * 归档短期记忆（将 recently_memory 转为 memory，带权限检查）
+     *
+     * 权限规则：只有创建者可以归档自己的记忆
+     */
+    archiveRecentlyMemory(workspacePath: string, uri: string, requestorId: string): Promise<MemoryRecord>;
+    /**
+     * 获取所有记忆统计信息（已使用异步 API）
+     */
+    getStats(workspacePath: string, requestorId?: string): Promise<{
+        task_list: number;
+        recently_memory: number;
+        memory: number;
+        total: number;
+        accessible: number;
+        own: number;
+        public: number;
+        memoryDir: string;
+    }>;
+}
+export declare const memoryService: MemoryService;
+//# sourceMappingURL=memory-service.d.ts.map

package/dist/services/memory-service.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"memory-service.d.ts","sourceRoot":"","sources":["../../src/services/memory-service.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;GAaG;AAQH,MAAM,MAAM,UAAU,GAAG,WAAW,GAAG,iBAAiB,GAAG,QAAQ,CAAC;AAepE,MAAM,WAAW,YAAY;IAC3B,EAAE,EAAE,MAAM,CAAC;IACX,GAAG,EAAE,MAAM,CAAC;IACZ,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;IAChB,WAAW,EAAE,UAAU,CAAC;IACxB,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC;IAEhB,SAAS,EAAE,MAAM,CAAC;IAClB,WAAW,EAAE,MAAM,CAAC;IAEpB,QAAQ,EAAE,OAAO,CAAC;IAElB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,cAAc,CAAC,EAAE,MAAM,CAAC;CACzB;AAED,MAAM,WAAW,iBAAiB;IAChC,WAAW,EAAE,UAAU,CAAC;IACxB,GAAG,EAAE,MAAM,CAAC;IACZ,OAAO,EAAE,MAAM,CAAC;IAEhB,SAAS,EAAE,MAAM,CAAC;IAClB,WAAW,EAAE,MAAM,CAAC;IAEpB,QAAQ,CAAC,EAAE,OAAO,CAAC;CACpB;AAED,MAAM,WAAW,iBAAiB;IAChC,WAAW,CAAC,EAAE,UAAU,CAAC;IACzB,GAAG,EAAE,MAAM,CAAC;IACZ,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,MAAM,CAAC,EAAE,MAAM,CAAC;IAEhB,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,eAAe;IAC9B,WAAW,CAAC,EAAE,UAAU,CAAC;IACzB,GAAG,EAAE,MAAM,CAAC;IAEZ,WAAW,EAAE,MAAM,CAAC;CACrB;AAED,MAAM,WAAW,iBAAiB;IAChC,WAAW,CAAC,EAAE,UAAU,CAAC;IACzB,GAAG,EAAE,MAAM,CAAC;IAEZ,WAAW,EAAE,MAAM,CAAC;CACrB;AAED,MAAM,WAAW,iBAAiB;IAChC,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,WAAW,CAAC,EAAE,UAAU,CAAC;IAEzB,WAAW,EAAE,MAAM,CAAC;CACrB;AAED,MAAM,WAAW,eAAe;IAC9B,WAAW,EAAE,UAAU,CAAC;IAExB,WAAW,EAAE,MAAM,CAAC;CACrB;AASD;;;;GAIG;AACH,wBAAsB,iBAAiB,CACrC,aAAa,EAAE,MAAM,EACrB,UAAU,EAAE,UAAU,GACrB,OAAO,CAAC,MAAM,CAAC,CAKjB;AAED;;GAEG;AACH,wBAAsB,gBAAgB,CAAC,aAAa,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAG7E;AA+GD,qBAAa,aAAa;IACxB;;OAEG;YACW,eAAe;IAM7B;;OAEG;IACH,OAAO,CAAC,MAAM;IAId;;OAEG;IACH,OAAO,CAAC,YAAY;IA8BpB;;OAEG;IACH,OAAO,CAAC,mBAAmB;IAW3B;;OAEG;IACH,OAAO,CAAC,eAAe;IAWvB;;OAEG;IACH,OAAO,CAAC,UAAU;IAKlB;;OAEG;YACW,cAAc;IA4C5B;;OAEG;YACW,cAAc;IAc5B;;OAEG;IACG,YAAY,CAChB,aAAa,EAAE,MAAM,EACrB,KAAK,EAAE,iBAAiB,GACvB,OAAO,CAAC,YAAY,CAAC;IA6CxB;;;;;;;OAOG;IACG,UAAU,CACd,aAAa,EAAE,MAAM,EACrB,KAAK,EAAE,eAAe,GACrB,OAAO,CAAC,CAAC,YAAY,GAAG;QAAE,KAAK,EAAE,OAAO,CAAC;QAAC,MAAM,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC,GAAG,IAAI,CAAC;IA4CvE;;;;;OAKG;IACG,YAAY,CAChB,aAAa,EAAE,MAAM,EACrB,KAAK,EAAE,iBAAiB,GACvB,OAAO,CAAC,YAAY,CAAC;IA2FxB;;;;OAIG;IACG,YAAY,CAChB,aAAa,EAAE,MAAM,EACrB,KAAK,EAAE,iBAAiB,GACvB,OAAO,CAAC;QAAE,OAAO,EAAE,OAAO,CAAC;QAAC,MAAM,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC;IA2CjD;;;;;OAKG;IACG,YAAY,CAChB,aAAa,EAAE,MAAM,EACrB,KAAK,EAAE,iBAAiB,GACvB,OAAO,CAAC;QACT,KAAK,EAAE,MAAM,CAAC;QACd,OAAO,EAAE,KAAK,CAAC,YAAY,GAAG;YAAE,KAAK,EAAE,OAAO,CAAC;YAAC,MAAM,CAAC,EAAE,MAAM,CAAA;SAAE,CAAC,CAAC;QACnE,UAAU,EAAE,MAAM,CAAC;KACpB,CAAC;IAoFF;;;;OAIG;IACG,UAAU,CACd,aAAa,EAAE,MAAM,EACrB,KAAK,EAAE,eAAe,GACrB,OAAO,CAAC,KAAK,CAAC,YAAY,GAAG;QAAE,KAAK,EAAE,OAAO,CAAC;QAAC,MAAM,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAqCrE;;OAEG;IACG,oBAAoB,CACxB,aAAa,EAAE,MAAM,EACrB,GAAG,EAAE,MAAM,EACX,WAAW,EAAE,MAAM,GAClB,OAAO,CAAC;QAAE,OAAO,EAAE,OAAO,CAAC;QAAC,MAAM,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC;IAmBjD;;;;OAIG;IACG,qBAAqB,CACzB,aAAa,EAAE,MAAM,EACrB,GAAG,EAAE,MAAM,EACX,WAAW,EAAE,MAAM,GAClB,OAAO,CAAC,YAAY,CAAC;IA6CxB;;OAEG;IACG,QAAQ,CACZ,aAAa,EAAE,MAAM,EACrB,WAAW,CAAC,EAAE,MAAM,GACnB,OAAO,CAAC;QACT,SAAS,EAAE,MAAM,CAAC;QAClB,eAAe,EAAE,MAAM,CAAC;QACxB,MAAM,EAAE,MAAM,CAAC;QACf,KAAK,EAAE,MAAM,CAAC;QACd,UAAU,EAAE,MAAM,CAAC;QACnB,GAAG,EAAE,MAAM,CAAC;QACZ,MAAM,EAAE,MAAM,CAAC;QACf,SAAS,EAAE,MAAM,CAAC;KACnB,CAAC;CAuCH;AAGD,eAAO,MAAM,aAAa,eAAsB,CAAC"}