aws-runtime-bridge 1.1.1 → 1.1.2

package/package/aws-client-agent-mcp/dist/memory-store.js

@@ -0,0 +1,268 @@
+const ALL_MEMORY_TYPES = ['task_list', 'recently_memory', 'memory'];
+const LISTABLE_MEMORY_TYPES = ['task_list', 'recently_memory'];
+const URI_PATTERN = /^([a-zA-Z_][a-zA-Z0-9_]*):\/\/(.*)$/;
+export class MemoryStore {
+    filePath;
+    constructor(filePath) {
+        this.filePath = filePath;
+    }
+    nowIso() {
+        return new Date().toISOString();
+    }
+    normalizeUri(input) {
+        const value = String(input || '').trim();
+        const matched = URI_PATTERN.exec(value);
+        if (!matched) {
+            throw new Error(`Invalid memory uri: ${input}`);
+        }
+        const domainPart = matched[1];
+        const pathPart = matched[2];
+        if (!domainPart || !pathPart) {
+            throw new Error(`Invalid memory uri: ${input}`);
+        }
+        const domain = domainPart.toLowerCase();
+        const path = pathPart.replace(/^\/+|\/+$/g, '');
+        if (!path) {
+            throw new Error(`Invalid memory uri path: ${input}`);
+        }
+        return {
+            uri: `${domain}://${path}`,
+            domain,
+            path
+        };
+    }
+    normalizeMemoryType(input, allowedTypes = ALL_MEMORY_TYPES) {
+        const value = String(input || '').trim();
+        if (!allowedTypes.includes(value)) {
+            throw new Error(`Invalid memory_type: ${input}`);
+        }
+        return value;
+    }
+    ensureTypeMatch(record, expectedType) {
+        if (expectedType && record.memory_type !== expectedType) {
+            throw new Error(`Memory type mismatch: expected ${expectedType}, actual ${record.memory_type}`);
+        }
+    }
+    async ensureParentDir() {
+        const { mkdir } = await import('node:fs/promises');
+        const path = await import('node:path');
+        const parent = path.dirname(this.filePath);
+        await mkdir(parent, { recursive: true });
+    }
+    async loadData() {
+        const { readFile } = await import('node:fs/promises');
+        try {
+            const raw = await readFile(this.filePath, 'utf-8');
+            const parsed = JSON.parse(raw);
+            const original = parsed.by_uri && typeof parsed.by_uri === 'object'
+                ? parsed.by_uri
+                : parsed.byUri && typeof parsed.byUri === 'object'
+                    ? parsed.byUri
+                    : {};
+            const normalizedByUri = {};
+            for (const [uri, record] of Object.entries(original)) {
+                const legacyType = record.memoryType;
+                const memory_type = this.normalizeMemoryType(record.memory_type || legacyType || 'memory');
+                normalizedByUri[uri] = {
+                    ...record,
+                    memory_type
+                };
+            }
+            return { by_uri: normalizedByUri };
+        }
+        catch {
+            return { by_uri: {} };
+        }
+    }
+    async saveData(data) {
+        const { writeFile } = await import('node:fs/promises');
+        await this.ensureParentDir();
+        await writeFile(this.filePath, JSON.stringify(data, null, 2), 'utf-8');
+    }
+    generateId(uri) {
+        const cleaned = uri.replace(/[^a-zA-Z0-9]+/g, '_').replace(/^_+|_+$/g, '');
+        return `${cleaned}_${Date.now().toString(36)}`;
+    }
+    async createMemory(input) {
+        const normalizedType = this.normalizeMemoryType(input.memory_type);
+        const normalized = this.normalizeUri(input.uri);
+        const data = await this.loadData();
+        if (data.by_uri[normalized.uri]) {
+            throw new Error(`Memory already exists: ${normalized.uri}`);
+        }
+        const now = this.nowIso();
+        const record = {
+            id: this.generateId(normalized.uri),
+            uri: normalized.uri,
+            domain: normalized.domain,
+            path: normalized.path,
+            content: String(input.content || ''),
+            memory_type: normalizedType,
+            createdAt: now,
+            updatedAt: now,
+            version: 1
+        };
+        data.by_uri[normalized.uri] = record;
+        await this.saveData(data);
+        return record;
+    }
+    async readMemory(input) {
+        const normalized = this.normalizeUri(input.uri);
+        const expectedType = typeof input.memory_type === 'string'
+            ? this.normalizeMemoryType(input.memory_type)
+            : undefined;
+        const data = await this.loadData();
+        const record = data.by_uri[normalized.uri] ?? null;
+        if (!record) {
+            return null;
+        }
+        this.ensureTypeMatch(record, expectedType);
+        return record;
+    }
+    async updateMemory(input) {
+        const normalized = this.normalizeUri(input.uri);
+        const expectedType = typeof input.memory_type === 'string'
+            ? this.normalizeMemoryType(input.memory_type)
+            : undefined;
+        const data = await this.loadData();
+        const existing = data.by_uri[normalized.uri];
+        if (!existing) {
+            throw new Error(`Memory not found: ${normalized.uri}`);
+        }
+        this.ensureTypeMatch(existing, expectedType);
+        let nextContent = existing.content;
+        if (typeof input.old_string === 'string' && typeof input.new_string === 'string') {
+            const oldString = input.old_string;
+            if (!oldString) {
+                throw new Error('old_string cannot be empty when replacing');
+            }
+            const occurrenceCount = nextContent.split(oldString).length - 1;
+            if (occurrenceCount === 0) {
+                throw new Error('old_string not found in memory content');
+            }
+            if (occurrenceCount > 1) {
+                throw new Error('old_string appears multiple times; replace is ambiguous');
+            }
+            nextContent = nextContent.replace(oldString, input.new_string);
+        }
+        if (typeof input.append === 'string' && input.append.length > 0) {
+            nextContent = `${nextContent}${input.append}`;
+        }
+        const updated = {
+            ...existing,
+            content: nextContent,
+            updatedAt: this.nowIso(),
+            version: existing.version + 1
+        };
+        data.by_uri[normalized.uri] = updated;
+        await this.saveData(data);
+        return updated;
+    }
+    async deleteMemory(input) {
+        const normalized = this.normalizeUri(input.uri);
+        const expectedType = typeof input.memory_type === 'string'
+            ? this.normalizeMemoryType(input.memory_type)
+            : undefined;
+        const data = await this.loadData();
+        const existing = data.by_uri[normalized.uri];
+        if (!existing) {
+            return false;
+        }
+        this.ensureTypeMatch(existing, expectedType);
+        delete data.by_uri[normalized.uri];
+        await this.saveData(data);
+        return true;
+    }
+    async searchMemory(input) {
+        const data = await this.loadData();
+        const query = String(input.query || '').trim().toLowerCase();
+        const memoryTypeFilter = input.memory_type
+            ? this.normalizeMemoryType(input.memory_type)
+            : undefined;
+        const limit = typeof input.limit === 'number' && input.limit > 0 ? input.limit : 10;
+        if (!query) {
+            return { query, results: [] };
+        }
+        const seenById = new Set();
+        const ranked = [];
+        for (const record of Object.values(data.by_uri)) {
+            if (memoryTypeFilter && record.memory_type !== memoryTypeFilter) {
+                continue;
+            }
+            if (seenById.has(record.id)) {
+                continue;
+            }
+            const haystack = `${record.uri}\n${record.content}`.toLowerCase();
+            if (!haystack.includes(query)) {
+                continue;
+            }
+            const inUri = record.uri.toLowerCase().includes(query) ? 2 : 0;
+            const inContent = record.content.toLowerCase().includes(query) ? 1 : 0;
+            const score = inUri + inContent;
+            seenById.add(record.id);
+            ranked.push({ score, record });
+        }
+        ranked.sort((a, b) => {
+            if (b.score !== a.score) {
+                return b.score - a.score;
+            }
+            return b.record.updatedAt.localeCompare(a.record.updatedAt);
+        });
+        return {
+            query,
+            results: ranked.slice(0, limit).map((item) => item.record)
+        };
+    }
+    async listByType(memory_type) {
+        const listType = this.normalizeMemoryType(memory_type, LISTABLE_MEMORY_TYPES);
+        const data = await this.loadData();
+        const seenById = new Set();
+        const list = [];
+        for (const record of Object.values(data.by_uri)) {
+            if (record.memory_type !== listType) {
+                continue;
+            }
+            if (seenById.has(record.id)) {
+                continue;
+            }
+            seenById.add(record.id);
+            list.push(record);
+        }
+        list.sort((a, b) => b.updatedAt.localeCompare(a.updatedAt));
+        return list;
+    }
+    async removeTaskListMemory(uri) {
+        const record = await this.readMemory({ uri });
+        if (!record) {
+            return false;
+        }
+        if (record.memory_type !== 'task_list') {
+            throw new Error('task_list_remove only supports task_list type');
+        }
+        return this.deleteMemory({ uri: record.uri, memory_type: 'task_list' });
+    }
+    async archiveRecentlyMemory(uri) {
+        const record = await this.readMemory({ uri });
+        if (!record) {
+            throw new Error(`Memory not found: ${uri}`);
+        }
+        if (record.memory_type !== 'recently_memory') {
+            throw new Error('recently_memory_archive only supports recently_memory type');
+        }
+        const data = await this.loadData();
+        const existing = data.by_uri[record.uri];
+        if (!existing) {
+            throw new Error(`Memory not found: ${uri}`);
+        }
+        const archived = {
+            ...existing,
+            memory_type: 'memory',
+            updatedAt: this.nowIso(),
+            version: existing.version + 1
+        };
+        data.by_uri[record.uri] = archived;
+        await this.saveData(data);
+        return archived;
+    }
+}
+//# sourceMappingURL=memory-store.js.map

package/package/aws-client-agent-mcp/dist/memory-store.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"memory-store.js","sourceRoot":"","sources":["../src/memory-store.ts"],"names":[],"mappings":"AAEA,MAAM,gBAAgB,GAAiB,CAAC,WAAW,EAAE,iBAAiB,EAAE,QAAQ,CAAC,CAAC;AAClF,MAAM,qBAAqB,GAAiB,CAAC,WAAW,EAAE,iBAAiB,CAAC,CAAC;AAgD7E,MAAM,WAAW,GAAG,qCAAqC,CAAC;AAE1D,MAAM,OAAO,WAAW;IACL,QAAQ,CAAS;IAElC,YAAY,QAAgB;QAC1B,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC3B,CAAC;IAEO,MAAM;QACZ,OAAO,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IAClC,CAAC;IAEO,YAAY,CAAC,KAAa;QAChC,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QACzC,MAAM,OAAO,GAAG,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACxC,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,IAAI,KAAK,CAAC,uBAAuB,KAAK,EAAE,CAAC,CAAC;QAClD,CAAC;QAED,MAAM,UAAU,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;QAC9B,MAAM,QAAQ,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;QAC5B,IAAI,CAAC,UAAU,IAAI,CAAC,QAAQ,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CAAC,uBAAuB,KAAK,EAAE,CAAC,CAAC;QAClD,CAAC;QAED,MAAM,MAAM,GAAG,UAAU,CAAC,WAAW,EAAE,CAAC;QACxC,MAAM,IAAI,GAAG,QAAQ,CAAC,OAAO,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC;QAChD,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,MAAM,IAAI,KAAK,CAAC,4BAA4B,KAAK,EAAE,CAAC,CAAC;QACvD,CAAC;QAED,OAAO;YACL,GAAG,EAAE,GAAG,MAAM,MAAM,IAAI,EAAE;YAC1B,MAAM;YACN,IAAI;SACL,CAAC;IACJ,CAAC;IAEO,mBAAmB,CAAC,KAAc,EAAE,YAAY,GAAG,gBAAgB;QACzE,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,IAAI,EAAgB,CAAC;QACvD,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;YAClC,MAAM,IAAI,KAAK,CAAC,wBAAwB,KAAK,EAAE,CAAC,CAAC;QACnD,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,eAAe,CAAC,MAAoB,EAAE,YAAyB;QACrE,IAAI,YAAY,IAAI,MAAM,CAAC,WAAW,KAAK,YAAY,EAAE,CAAC;YACxD,MAAM,IAAI,KAAK,CAAC,kCAAkC,YAAY,YAAY,MAAM,CAAC,WAAW,EAAE,CAAC,CAAC;QAClG,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,eAAe;QAC3B,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,kBAAkB,CAAC,CAAC;QACnD,MAAM,IAAI,GAAG,MAAM,MAAM,CAAC,WAAW,CAAC,CAAC;QACvC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC3C,MAAM,KAAK,CAAC,MAAM,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAC3C,CAAC;IAEO,KAAK,CAAC,QAAQ;QACpB,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,MAAM,CAAC,kBAAkB,CAAC,CAAC;QAEtD,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;YACnD,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAkH,CAAC;YAChJ,MAAM,QAAQ,GAAG,MAAM,CAAC,MAAM,IAAI,OAAO,MAAM,CAAC,MAAM,KAAK,QAAQ;gBACjE,CAAC,CAAC,MAAM,CAAC,MAAM;gBACf,CAAC,CAAC,MAAM,CAAC,KAAK,IAAI,OAAO,MAAM,CAAC,KAAK,KAAK,QAAQ;oBAChD,CAAC,CAAC,MAAM,CAAC,KAAK;oBACd,CAAC,CAAC,EAAE,CAAC;YAET,MAAM,eAAe,GAAiC,EAAE,CAAC;YACzD,KAAK,MAAM,CAAC,GAAG,EAAE,MAAM,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACrD,MAAM,UAAU,GAAI,MAA8D,CAAC,UAAU,CAAC;gBAC9F,MAAM,WAAW,GAAG,IAAI,CAAC,mBAAmB,CAAE,MAAgC,CAAC,WAAW,IAAI,UAAU,IAAI,QAAQ,CAAC,CAAC;gBACtH,eAAe,CAAC,GAAG,CAAC,GAAG;oBACrB,GAAG,MAAM;oBACT,WAAW;iBACZ,CAAC;YACJ,CAAC;YAED,OAAO,EAAE,MAAM,EAAE,eAAe,EAAE,CAAC;QACrC,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC;QACxB,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,QAAQ,CAAC,IAAyB;QAC9C,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,MAAM,CAAC,kBAAkB,CAAC,CAAC;QACvD,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;QAC7B,MAAM,SAAS,CAAC,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;IACzE,CAAC;IAEO,UAAU,CAAC,GAAW;QAC5B,MAAM,OAAO,GAAG,GAAG,CAAC,OAAO,CAAC,gBAAgB,EAAE,GAAG,CAAC,CAAC,OAAO,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;QAC3E,OAAO,GAAG,OAAO,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,EAAE,CAAC;IACjD,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,KAAwB;QACzC,MAAM,cAAc,GAAG,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;QACnE,MAAM,UAAU,GAAG,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAChD,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;QAEnC,IAAI,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;YAChC,MAAM,IAAI,KAAK,CAAC,0BAA0B,UAAU,CAAC,GAAG,EAAE,CAAC,CAAC;QAC9D,CAAC;QAED,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;QAC1B,MAAM,MAAM,GAAiB;YAC3B,EAAE,EAAE,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,GAAG,CAAC;YACnC,GAAG,EAAE,UAAU,CAAC,GAAG;YACnB,MAAM,EAAE,UAAU,CAAC,MAAM;YACzB,IAAI,EAAE,UAAU,CAAC,IAAI;YACrB,OAAO,EAAE,MAAM,CAAC,KAAK,CAAC,OAAO,IAAI,EAAE,CAAC;YACpC,WAAW,EAAE,cAAc;YAC3B,SAAS,EAAE,GAAG;YACd,SAAS,EAAE,GAAG;YACd,OAAO,EAAE,CAAC;SACX,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC;QACrC,MAAM,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QAC1B,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,KAAsB;QACrC,MAAM,UAAU,GAAG,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAChD,MAAM,YAAY,GAAG,OAAO,KAAK,CAAC,WAAW,KAAK,QAAQ;YACxD,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,WAAW,CAAC;YAC7C,CAAC,CAAC,SAAS,CAAC;QACd,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;QACnC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,IAAI,CAAC;QACnD,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO,IAAI,CAAC;QACd,CAAC;QACD,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;QAC3C,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,KAAwB;QACzC,MAAM,UAAU,GAAG,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAChD,MAAM,YAAY,GAAG,OAAO,KAAK,CAAC,WAAW,KAAK,QAAQ;YACxD,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,WAAW,CAAC;YAC7C,CAAC,CAAC,SAAS,CAAC;QACd,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;QACnC,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;QAC7C,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,MAAM,IAAI,KAAK,CAAC,qBAAqB,UAAU,CAAC,GAAG,EAAE,CAAC,CAAC;QACzD,CAAC;QACD,IAAI,CAAC,eAAe,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC;QAE7C,IAAI,WAAW,GAAG,QAAQ,CAAC,OAAO,CAAC;QAEnC,IAAI,OAAO,KAAK,CAAC,UAAU,KAAK,QAAQ,IAAI,OAAO,KAAK,CAAC,UAAU,KAAK,QAAQ,EAAE,CAAC;YACjF,MAAM,SAAS,GAAG,KAAK,CAAC,UAAU,CAAC;YACnC,IAAI,CAAC,SAAS,EAAE,CAAC;gBACf,MAAM,IAAI,KAAK,CAAC,2CAA2C,CAAC,CAAC;YAC/D,CAAC;YACD,MAAM,eAAe,GAAG,WAAW,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC;YAChE,IAAI,eAAe,KAAK,CAAC,EAAE,CAAC;gBAC1B,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;YAC5D,CAAC;YACD,IAAI,eAAe,GAAG,CAAC,EAAE,CAAC;gBACxB,MAAM,IAAI,KAAK,CAAC,yDAAyD,CAAC,CAAC;YAC7E,CAAC;YACD,WAAW,GAAG,WAAW,CAAC,OAAO,CAAC,SAAS,EAAE,KAAK,CAAC,UAAU,CAAC,CAAC;QACjE,CAAC;QAED,IAAI,OAAO,KAAK,CAAC,MAAM,KAAK,QAAQ,IAAI,KAAK,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAChE,WAAW,GAAG,GAAG,WAAW,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC;QAChD,CAAC;QAED,MAAM,OAAO,GAAiB;YAC5B,GAAG,QAAQ;YACX,OAAO,EAAE,WAAW;YACpB,SAAS,EAAE,IAAI,CAAC,MAAM,EAAE;YACxB,OAAO,EAAE,QAAQ,CAAC,OAAO,GAAG,CAAC;SAC9B,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,GAAG,CAAC,GAAG,OAAO,CAAC;QACtC,MAAM,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QAC1B,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,KAAwB;QACzC,MAAM,UAAU,GAAG,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAChD,MAAM,YAAY,GAAG,OAAO,KAAK,CAAC,WAAW,KAAK,QAAQ;YACxD,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,WAAW,CAAC;YAC7C,CAAC,CAAC,SAAS,CAAC;QACd,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;QACnC,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;QAC7C,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,OAAO,KAAK,CAAC;QACf,CAAC;QACD,IAAI,CAAC,eAAe,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC;QAE7C,OAAO,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;QACnC,MAAM,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QAC1B,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,KAAwB;QACzC,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;QACnC,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QAC7D,MAAM,gBAAgB,GAAG,KAAK,CAAC,WAAW;YACxC,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,WAAW,CAAC;YAC7C,CAAC,CAAC,SAAS,CAAC;QACd,MAAM,KAAK,GAAG,OAAO,KAAK,CAAC,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;QAEpF,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;QAChC,CAAC;QAED,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAU,CAAC;QACnC,MAAM,MAAM,GAAmD,EAAE,CAAC;QAElE,KAAK,MAAM,MAAM,IAAI,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;YAChD,IAAI,gBAAgB,IAAI,MAAM,CAAC,WAAW,KAAK,gBAAgB,EAAE,CAAC;gBAChE,SAAS;YACX,CAAC;YACD,IAAI,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC;gBAC5B,SAAS;YACX,CAAC;YAED,MAAM,QAAQ,GAAG,GAAG,MAAM,CAAC,GAAG,KAAK,MAAM,CAAC,OAAO,EAAE,CAAC,WAAW,EAAE,CAAC;YAClE,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC9B,SAAS;YACX,CAAC;YAED,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAC/D,MAAM,SAAS,GAAG,MAAM,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YACvE,MAAM,KAAK,GAAG,KAAK,GAAG,SAAS,CAAC;YAEhC,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YACxB,MAAM,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC;QACjC,CAAC;QAED,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;YACnB,IAAI,CAAC,CAAC,KAAK,KAAK,CAAC,CAAC,KAAK,EAAE,CAAC;gBACxB,OAAO,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC;YAC3B,CAAC;YACD,OAAO,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAC9D,CAAC,CAAC,CAAC;QAEH,OAAO;YACL,KAAK;YACL,OAAO,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC;SAC3D,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,WAAmB;QAClC,MAAM,QAAQ,GAAG,IAAI,CAAC,mBAAmB,CAAC,WAAW,EAAE,qBAAqB,CAAC,CAAC;QAC9E,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;QAEnC,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAU,CAAC;QACnC,MAAM,IAAI,GAAmB,EAAE,CAAC;QAEhC,KAAK,MAAM,MAAM,IAAI,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;YAChD,IAAI,MAAM,CAAC,WAAW,KAAK,QAAQ,EAAE,CAAC;gBACpC,SAAS;YACX,CAAC;YACD,IAAI,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC;gBAC5B,SAAS;YACX,CAAC;YACD,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YACxB,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACpB,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;QAC5D,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK,CAAC,oBAAoB,CAAC,GAAW;QACpC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;QAC9C,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO,KAAK,CAAC;QACf,CAAC;QACD,IAAI,MAAM,CAAC,WAAW,KAAK,WAAW,EAAE,CAAC;YACvC,MAAM,IAAI,KAAK,CAAC,+CAA+C,CAAC,CAAC;QACnE,CAAC;QAED,OAAO,IAAI,CAAC,YAAY,CAAC,EAAE,GAAG,EAAE,MAAM,CAAC,GAAG,EAAE,WAAW,EAAE,WAAW,EAAE,CAAC,CAAC;IAC1E,CAAC;IAED,KAAK,CAAC,qBAAqB,CAAC,GAAW;QACrC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;QAC9C,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,qBAAqB,GAAG,EAAE,CAAC,CAAC;QAC9C,CAAC;QACD,IAAI,MAAM,CAAC,WAAW,KAAK,iBAAiB,EAAE,CAAC;YAC7C,MAAM,IAAI,KAAK,CAAC,4DAA4D,CAAC,CAAC;QAChF,CAAC;QAED,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;QACnC,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QACzC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,MAAM,IAAI,KAAK,CAAC,qBAAqB,GAAG,EAAE,CAAC,CAAC;QAC9C,CAAC;QAED,MAAM,QAAQ,GAAiB;YAC7B,GAAG,QAAQ;YACX,WAAW,EAAE,QAAQ;YACrB,SAAS,EAAE,IAAI,CAAC,MAAM,EAAE;YACxB,OAAO,EAAE,QAAQ,CAAC,OAAO,GAAG,CAAC;SAC9B,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,QAAQ,CAAC;QAEnC,MAAM,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QAC1B,OAAO,QAAQ,CAAC;IAClB,CAAC;CACF"}

package/package/aws-client-agent-mcp/dist/memory-store.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=memory-store.test.d.ts.map

package/package/aws-client-agent-mcp/dist/memory-store.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"memory-store.test.d.ts","sourceRoot":"","sources":["../src/memory-store.test.ts"],"names":[],"mappings":""}

package/package/aws-client-agent-mcp/dist/memory-store.test.js

@@ -0,0 +1,164 @@
+import assert from 'node:assert/strict';
+import { mkdtemp, rm } from 'node:fs/promises';
+import os from 'node:os';
+import path from 'node:path';
+import { describe, it } from 'node:test';
+import { MemoryStore } from './memory-store.js';
+describe('MemoryStore', () => {
+    it('creates and reads memory by uri', async () => {
+        const dir = await mkdtemp(path.join(os.tmpdir(), 'aws-memory-store-'));
+        const storePath = path.join(dir, 'memory.json');
+        const store = new MemoryStore(storePath);
+        try {
+            const created = await store.createMemory({
+                memory_type: 'memory',
+                uri: 'core://agent/profile',
+                content: 'I am a backend expert.'
+            });
+            assert.equal(created.uri, 'core://agent/profile');
+            const loaded = await store.readMemory({ uri: 'core://agent/profile', memory_type: 'memory' });
+            assert.ok(loaded);
+            assert.equal(loaded?.content, 'I am a backend expert.');
+        }
+        finally {
+            await rm(dir, { recursive: true, force: true });
+        }
+    });
+    it('updates memory with exact replace and append', async () => {
+        const dir = await mkdtemp(path.join(os.tmpdir(), 'aws-memory-store-'));
+        const storePath = path.join(dir, 'memory.json');
+        const store = new MemoryStore(storePath);
+        try {
+            await store.createMemory({
+                memory_type: 'memory',
+                uri: 'notes://project/todo',
+                content: 'finish api docs'
+            });
+            const replaced = await store.updateMemory({
+                memory_type: 'memory',
+                uri: 'notes://project/todo',
+                old_string: 'api docs',
+                new_string: 'API docs v2'
+            });
+            assert.equal(replaced.content, 'finish API docs v2');
+            const appended = await store.updateMemory({
+                memory_type: 'memory',
+                uri: 'notes://project/todo',
+                append: '\n- add memory tests'
+            });
+            assert.match(appended.content, /add memory tests/);
+        }
+        finally {
+            await rm(dir, { recursive: true, force: true });
+        }
+    });
+    it('searches memories by query and supports limit', async () => {
+        const dir = await mkdtemp(path.join(os.tmpdir(), 'aws-memory-store-'));
+        const storePath = path.join(dir, 'memory.json');
+        const store = new MemoryStore(storePath);
+        try {
+            await store.createMemory({
+                memory_type: 'memory',
+                uri: 'core://agent/skills',
+                content: 'java spring boot mysql'
+            });
+            await store.createMemory({
+                memory_type: 'memory',
+                uri: 'notes://project/readme',
+                content: 'project setup and architecture'
+            });
+            const result = await store.searchMemory({ query: 'project', limit: 1, memory_type: 'memory' });
+            assert.equal(result.results.length, 1);
+            assert.equal(result.results[0]?.uri, 'notes://project/readme');
+        }
+        finally {
+            await rm(dir, { recursive: true, force: true });
+        }
+    });
+    it('deletes memory and no longer reads it', async () => {
+        const dir = await mkdtemp(path.join(os.tmpdir(), 'aws-memory-store-'));
+        const storePath = path.join(dir, 'memory.json');
+        const store = new MemoryStore(storePath);
+        try {
+            await store.createMemory({
+                memory_type: 'memory',
+                uri: 'core://agent/temp',
+                content: 'temporary memory'
+            });
+            const deleted = await store.deleteMemory({ uri: 'core://agent/temp', memory_type: 'memory' });
+            assert.equal(deleted, true);
+            const loaded = await store.readMemory({ uri: 'core://agent/temp', memory_type: 'memory' });
+            assert.equal(loaded, null);
+        }
+        finally {
+            await rm(dir, { recursive: true, force: true });
+        }
+    });
+    it('supports memory_type classification and list/filter', async () => {
+        const dir = await mkdtemp(path.join(os.tmpdir(), 'aws-memory-store-'));
+        const storePath = path.join(dir, 'memory.json');
+        const store = new MemoryStore(storePath);
+        try {
+            const types = ['task_list', 'recently_memory', 'memory'];
+            await store.createMemory({ memory_type: types[0], uri: 'task://todo/1', content: 'fix tests' });
+            await store.createMemory({ memory_type: types[1], uri: 'recent://ctx/1', content: 'just discussed API design' });
+            await store.createMemory({ memory_type: types[2], uri: 'core://archive/1', content: 'long-term architecture decision' });
+            const taskList = await store.listByType('task_list');
+            assert.equal(taskList.length, 1);
+            assert.equal(taskList[0]?.memory_type, 'task_list');
+            const recentList = await store.listByType('recently_memory');
+            assert.equal(recentList.length, 1);
+            assert.equal(recentList[0]?.memory_type, 'recently_memory');
+            await assert.rejects(() => store.listByType('memory'), /Invalid memory_type/);
+        }
+        finally {
+            await rm(dir, { recursive: true, force: true });
+        }
+    });
+    it('removes task_list memory only via task_list_remove semantic', async () => {
+        const dir = await mkdtemp(path.join(os.tmpdir(), 'aws-memory-store-'));
+        const storePath = path.join(dir, 'memory.json');
+        const store = new MemoryStore(storePath);
+        try {
+            await store.createMemory({ memory_type: 'task_list', uri: 'task://todo/remove-1', content: 'to be removed' });
+            await store.createMemory({ memory_type: 'memory', uri: 'core://kept/1', content: 'should stay' });
+            const removed = await store.removeTaskListMemory('task://todo/remove-1');
+            assert.equal(removed, true);
+            assert.equal(await store.readMemory({ uri: 'task://todo/remove-1' }), null);
+            await assert.rejects(() => store.removeTaskListMemory('core://kept/1'), /task_list_remove/);
+        }
+        finally {
+            await rm(dir, { recursive: true, force: true });
+        }
+    });
+    it('archives recently_memory into memory', async () => {
+        const dir = await mkdtemp(path.join(os.tmpdir(), 'aws-memory-store-'));
+        const storePath = path.join(dir, 'memory.json');
+        const store = new MemoryStore(storePath);
+        try {
+            await store.createMemory({ memory_type: 'recently_memory', uri: 'recent://note/1', content: 'hot memory' });
+            const archived = await store.archiveRecentlyMemory('recent://note/1');
+            assert.equal(archived.memory_type, 'memory');
+            const loaded = await store.readMemory({ uri: 'recent://note/1', memory_type: 'memory' });
+            assert.ok(loaded);
+            assert.equal(loaded?.memory_type, 'memory');
+            await store.createMemory({ memory_type: 'memory', uri: 'core://cold/2', content: 'cold memory' });
+            await assert.rejects(() => store.archiveRecentlyMemory('core://cold/2'), /recently_memory_archive/);
+        }
+        finally {
+            await rm(dir, { recursive: true, force: true });
+        }
+    });
+    it('rejects invalid memory_type on create', async () => {
+        const dir = await mkdtemp(path.join(os.tmpdir(), 'aws-memory-store-'));
+        const storePath = path.join(dir, 'memory.json');
+        const store = new MemoryStore(storePath);
+        try {
+            await assert.rejects(() => store.createMemory({ memory_type: 'invalid_type', uri: 'x://1', content: 'x' }), /Invalid memory_type/);
+        }
+        finally {
+            await rm(dir, { recursive: true, force: true });
+        }
+    });
+});
+//# sourceMappingURL=memory-store.test.js.map

package/package/aws-client-agent-mcp/dist/memory-store.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"memory-store.test.js","sourceRoot":"","sources":["../src/memory-store.test.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,oBAAoB,CAAC;AACxC,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,MAAM,kBAAkB,CAAC;AAC/C,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,WAAW,CAAC;AAEzC,OAAO,EAAE,WAAW,EAAmB,MAAM,mBAAmB,CAAC;AAEjE,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;IAC3B,EAAE,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;QAC/C,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,mBAAmB,CAAC,CAAC,CAAC;QACvE,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,aAAa,CAAC,CAAC;QAChD,MAAM,KAAK,GAAG,IAAI,WAAW,CAAC,SAAS,CAAC,CAAC;QAEzC,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,KAAK,CAAC,YAAY,CAAC;gBACvC,WAAW,EAAE,QAAQ;gBACrB,GAAG,EAAE,sBAAsB;gBAC3B,OAAO,EAAE,wBAAwB;aAClC,CAAC,CAAC;YAEH,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,EAAE,sBAAsB,CAAC,CAAC;YAClD,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,UAAU,CAAC,EAAE,GAAG,EAAE,sBAAsB,EAAE,WAAW,EAAE,QAAQ,EAAE,CAAC,CAAC;YAC9F,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC;YAClB,MAAM,CAAC,KAAK,CAAC,MAAM,EAAE,OAAO,EAAE,wBAAwB,CAAC,CAAC;QAC1D,CAAC;gBAAS,CAAC;YACT,MAAM,EAAE,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QAClD,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;QAC5D,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,mBAAmB,CAAC,CAAC,CAAC;QACvE,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,aAAa,CAAC,CAAC;QAChD,MAAM,KAAK,GAAG,IAAI,WAAW,CAAC,SAAS,CAAC,CAAC;QAEzC,IAAI,CAAC;YACH,MAAM,KAAK,CAAC,YAAY,CAAC;gBACvB,WAAW,EAAE,QAAQ;gBACrB,GAAG,EAAE,sBAAsB;gBAC3B,OAAO,EAAE,iBAAiB;aAC3B,CAAC,CAAC;YAEH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,YAAY,CAAC;gBACxC,WAAW,EAAE,QAAQ;gBACrB,GAAG,EAAE,sBAAsB;gBAC3B,UAAU,EAAE,UAAU;gBACtB,UAAU,EAAE,aAAa;aAC1B,CAAC,CAAC;YACH,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,EAAE,oBAAoB,CAAC,CAAC;YAErD,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,YAAY,CAAC;gBACxC,WAAW,EAAE,QAAQ;gBACrB,GAAG,EAAE,sBAAsB;gBAC3B,MAAM,EAAE,sBAAsB;aAC/B,CAAC,CAAC;YACH,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,EAAE,kBAAkB,CAAC,CAAC;QACrD,CAAC;gBAAS,CAAC;YACT,MAAM,EAAE,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QAClD,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;QAC7D,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,mBAAmB,CAAC,CAAC,CAAC;QACvE,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,aAAa,CAAC,CAAC;QAChD,MAAM,KAAK,GAAG,IAAI,WAAW,CAAC,SAAS,CAAC,CAAC;QAEzC,IAAI,CAAC;YACH,MAAM,KAAK,CAAC,YAAY,CAAC;gBACvB,WAAW,EAAE,QAAQ;gBACrB,GAAG,EAAE,qBAAqB;gBAC1B,OAAO,EAAE,wBAAwB;aAClC,CAAC,CAAC;YACH,MAAM,KAAK,CAAC,YAAY,CAAC;gBACvB,WAAW,EAAE,QAAQ;gBACrB,GAAG,EAAE,wBAAwB;gBAC7B,OAAO,EAAE,gCAAgC;aAC1C,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,YAAY,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,EAAE,WAAW,EAAE,QAAQ,EAAE,CAAC,CAAC;YAC/F,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;YACvC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,wBAAwB,CAAC,CAAC;QACjE,CAAC;gBAAS,CAAC;YACT,MAAM,EAAE,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QAClD,CAAC;IACH,CAAC,CAAC,CAAC;IAED,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;QACvD,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,mBAAmB,CAAC,CAAC,CAAC;QACvE,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,aAAa,CAAC,CAAC;QAChD,MAAM,KAAK,GAAG,IAAI,WAAW,CAAC,SAAS,CAAC,CAAC;QAEzC,IAAI,CAAC;YACH,MAAM,KAAK,CAAC,YAAY,CAAC;gBACvB,WAAW,EAAE,QAAQ;gBACrB,GAAG,EAAE,mBAAmB;gBACxB,OAAO,EAAE,kBAAkB;aAC5B,CAAC,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,KAAK,CAAC,YAAY,CAAC,EAAE,GAAG,EAAE,mBAAmB,EAAE,WAAW,EAAE,QAAQ,EAAE,CAAC,CAAC;YAC9F,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;YAE5B,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,UAAU,CAAC,EAAE,GAAG,EAAE,mBAAmB,EAAE,WAAW,EAAE,QAAQ,EAAE,CAAC,CAAC;YAC3F,MAAM,CAAC,KAAK,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QAC7B,CAAC;gBAAS,CAAC;YACT,MAAM,EAAE,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QAClD,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QACnE,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,mBAAmB,CAAC,CAAC,CAAC;QACvE,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,aAAa,CAAC,CAAC;QAChD,MAAM,KAAK,GAAG,IAAI,WAAW,CAAC,SAAS,CAAC,CAAC;QAEzC,IAAI,CAAC;YACH,MAAM,KAAK,GAAiB,CAAC,WAAW,EAAE,iBAAiB,EAAE,QAAQ,CAAC,CAAC;YACvE,MAAM,KAAK,CAAC,YAAY,CAAC,EAAE,WAAW,EAAE,KAAK,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,eAAe,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC,CAAC;YAChG,MAAM,KAAK,CAAC,YAAY,CAAC,EAAE,WAAW,EAAE,KAAK,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,gBAAgB,EAAE,OAAO,EAAE,2BAA2B,EAAE,CAAC,CAAC;YACjH,MAAM,KAAK,CAAC,YAAY,CAAC,EAAE,WAAW,EAAE,KAAK,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,kBAAkB,EAAE,OAAO,EAAE,iCAAiC,EAAE,CAAC,CAAC;YAEzH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;YACrD,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;YACjC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,WAAW,CAAC,CAAC;YAEpD,MAAM,UAAU,GAAG,MAAM,KAAK,CAAC,UAAU,CAAC,iBAAiB,CAAC,CAAC;YAC7D,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;YACnC,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,iBAAiB,CAAC,CAAC;YAE5D,MAAM,MAAM,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,qBAAqB,CAAC,CAAC;QAChF,CAAC;gBAAS,CAAC;YACT,MAAM,EAAE,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QAClD,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;QAC3E,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,mBAAmB,CAAC,CAAC,CAAC;QACvE,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,aAAa,CAAC,CAAC;QAChD,MAAM,KAAK,GAAG,IAAI,WAAW,CAAC,SAAS,CAAC,CAAC;QAEzC,IAAI,CAAC;YACH,MAAM,KAAK,CAAC,YAAY,CAAC,EAAE,WAAW,EAAE,WAAW,EAAE,GAAG,EAAE,sBAAsB,EAAE,OAAO,EAAE,eAAe,EAAE,CAAC,CAAC;YAC9G,MAAM,KAAK,CAAC,YAAY,CAAC,EAAE,WAAW,EAAE,QAAQ,EAAE,GAAG,EAAE,eAAe,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC,CAAC;YAElG,MAAM,OAAO,GAAG,MAAM,KAAK,CAAC,oBAAoB,CAAC,sBAAsB,CAAC,CAAC;YACzE,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;YAC5B,MAAM,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC,UAAU,CAAC,EAAE,GAAG,EAAE,sBAAsB,EAAE,CAAC,EAAE,IAAI,CAAC,CAAC;YAE5E,MAAM,MAAM,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,oBAAoB,CAAC,eAAe,CAAC,EAAE,kBAAkB,CAAC,CAAC;QAC9F,CAAC;gBAAS,CAAC;YACT,MAAM,EAAE,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QAClD,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;QACpD,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,mBAAmB,CAAC,CAAC,CAAC;QACvE,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,aAAa,CAAC,CAAC;QAChD,MAAM,KAAK,GAAG,IAAI,WAAW,CAAC,SAAS,CAAC,CAAC;QAEzC,IAAI,CAAC;YACH,MAAM,KAAK,CAAC,YAAY,CAAC,EAAE,WAAW,EAAE,iBAAiB,EAAE,GAAG,EAAE,iBAAiB,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC,CAAC;YAC5G,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,qBAAqB,CAAC,iBAAiB,CAAC,CAAC;YACtE,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;YAE7C,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,UAAU,CAAC,EAAE,GAAG,EAAE,iBAAiB,EAAE,WAAW,EAAE,QAAQ,EAAE,CAAC,CAAC;YACzF,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC;YAClB,MAAM,CAAC,KAAK,CAAC,MAAM,EAAE,WAAW,EAAE,QAAQ,CAAC,CAAC;YAE5C,MAAM,KAAK,CAAC,YAAY,CAAC,EAAE,WAAW,EAAE,QAAQ,EAAE,GAAG,EAAE,eAAe,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC,CAAC;YAClG,MAAM,MAAM,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,qBAAqB,CAAC,eAAe,CAAC,EAAE,yBAAyB,CAAC,CAAC;QACtG,CAAC;gBAAS,CAAC;YACT,MAAM,EAAE,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QAClD,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;QACrD,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,mBAAmB,CAAC,CAAC,CAAC;QACvE,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,aAAa,CAAC,CAAC;QAChD,MAAM,KAAK,GAAG,IAAI,WAAW,CAAC,SAAS,CAAC,CAAC;QAEzC,IAAI,CAAC;YACH,MAAM,MAAM,CAAC,OAAO,CAClB,GAAG,EAAE,CAAC,KAAK,CAAC,YAAY,CAAC,EAAE,WAAW,EAAE,cAA4B,EAAE,GAAG,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC,EACnG,qBAAqB,CACtB,CAAC;QACJ,CAAC;gBAAS,CAAC;YACT,MAAM,EAAE,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QAClD,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/package/aws-client-agent-mcp/dist/message-buffer.d.ts

@@ -0,0 +1,87 @@
+/**
+ * 消息缓冲与派发模块
+ * 负责消息的缓冲、分类和派发
+ */
+import type { DirectMessage, GroupMessage, ProfileInfo } from './types.js';
+/** 分类后的消息 */
+export interface ClassifiedMessages {
+    directMessages: DirectMessage[];
+    groupMessages: GroupMessage[];
+}
+/** 消息到达回调类型 */
+export type OnMessageArrivedCallback = () => void;
+/**
+ * 消息缓冲区
+ * 管理消息的存储、分类和派发
+ */
+export declare class MessageBuffer {
+    private buffer;
+    private resolvers;
+    private onMessageArrived;
+    private isPolling;
+    /**
+     * 设置消息到达回调
+     * 当有新消息到达且正在 poll 等待时，会立即调用此回调
+     */
+    setOnMessageArrived(callback: OnMessageArrivedCallback | null): void;
+    /**
+     * 设置是否正在 poll 等待
+     */
+    setIsPolling(polling: boolean): void;
+    /**
+     * 添加消息到缓冲区
+     */
+    push(message: DirectMessage | GroupMessage): void;
+    /**
+     * 接收 Profile 推送。
+     * 主流程：记录日志 -> 不进入未读缓冲区 -> 引导调用方通过 get_profile 获取身份详情。
+     */
+    setProfileMessage(profile: ProfileInfo): void;
+    /**
+     * 移除 Profile 系统消息。
+     * 当前未再向未读消息注入 profile 信息，保留该方法仅为兼容既有调用链。
+     */
+    removeProfileMessage(): void;
+    /**
+     * 检查是否有 Profile 消息。
+     * 当前身份信息不会进入未读消息，因此固定返回 false。
+     */
+    hasProfileMessage(): boolean;
+    /**
+     * 获取 Profile 消息。
+     * 当前身份信息不会进入未读消息，因此固定返回 null。
+     */
+    getProfileMessage(): null;
+    /**
+       * 阻塞等待新消息
+       * 如果缓冲区有消息则立即返回，否则阻塞等待
+       * @param timeoutMs 可选超时时间（毫秒），超时后返回空消息而不继续阻塞
+       */
+    poll(timeoutMs?: number): Promise<ClassifiedMessages>;
+    /**
+     * 非阻塞获取消息
+     */
+    get(): ClassifiedMessages;
+    /**
+     * 获取缓冲区大小
+     */
+    size(): number;
+    /**
+     * 清空缓冲区
+     */
+    clear(): void;
+    /**
+       * 尝试派发缓冲区消息给等待的 resolver
+       */
+    private tryDispatch;
+    /**
+     * 消费缓冲区所有消息并分类
+     */
+    private consumeAll;
+    /**
+     * 判断消息类型
+     * DirectMessage 的 msgId 是 string，GroupMessage 的 msgId 是 number
+     */
+    private isDirectMessage;
+}
+//# sourceMappingURL=message-buffer.d.ts.map

package/package/aws-client-agent-mcp/dist/message-buffer.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"message-buffer.d.ts","sourceRoot":"","sources":["../src/message-buffer.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAGH,OAAO,KAAK,EAAE,aAAa,EAAE,YAAY,EAAE,WAAW,EAAE,MAAM,YAAY,CAAC;AAE3E,aAAa;AACb,MAAM,WAAW,kBAAkB;IACjC,cAAc,EAAE,aAAa,EAAE,CAAC;IAChC,aAAa,EAAE,YAAY,EAAE,CAAC;CAC/B;AAKD,eAAe;AACf,MAAM,MAAM,wBAAwB,GAAG,MAAM,IAAI,CAAC;AAElD;;;GAGG;AACH,qBAAa,aAAa;IACxB,OAAO,CAAC,MAAM,CAAwC;IACtD,OAAO,CAAC,SAAS,CAAyB;IAC1C,OAAO,CAAC,gBAAgB,CAAyC;IACjE,OAAO,CAAC,SAAS,CAAkB;IAEnC;;;OAGG;IACH,mBAAmB,CAAC,QAAQ,EAAE,wBAAwB,GAAG,IAAI,GAAG,IAAI;IAIpE;;OAEG;IACH,YAAY,CAAC,OAAO,EAAE,OAAO,GAAG,IAAI;IAIpC;;OAEG;IACH,IAAI,CAAC,OAAO,EAAE,aAAa,GAAG,YAAY,GAAG,IAAI;IAajD;;;OAGG;IACH,iBAAiB,CAAC,OAAO,EAAE,WAAW,GAAG,IAAI;IAI7C;;;OAGG;IACH,oBAAoB,IAAI,IAAI;IAI5B;;;OAGG;IACH,iBAAiB,IAAI,OAAO;IAI5B;;;OAGG;IACH,iBAAiB,IAAI,IAAI;IAI3B;;;;SAIK;IACH,IAAI,CAAC,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,kBAAkB,CAAC;IAsDrD;;OAEG;IACH,GAAG,IAAI,kBAAkB;IASzB;;OAEG;IACH,IAAI,IAAI,MAAM;IAId;;OAEG;IACH,KAAK,IAAI,IAAI;IAKf;;SAEK;IACH,OAAO,CAAC,WAAW;IAWnB;;OAEG;IACH,OAAO,CAAC,UAAU;IAkBlB;;;OAGG;IACH,OAAO,CAAC,eAAe;CAGxB"}