@multiplayer-app/ai-agent-mongo 0.1.0-beta.1 → 0.1.0-beta.3

package/dist/cjs/__tests__/schemas/AgentMessageSchema.test.js

@@ -0,0 +1,295 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const vitest_1 = require("vitest");
+const AgentMessageSchema_js_1 = require("../../schemas/AgentMessageSchema.js");
+const ai_agent_types_1 = require("@multiplayer-app/ai-agent-types");
+const index_js_1 = require("../../mongo/index.js");
+(0, vitest_1.describe)('AgentMessageSchema', () => {
+    const testChatId = new index_js_1.ObjectId().toString();
+    (0, vitest_1.describe)('validation', () => {
+        (0, vitest_1.it)('should create a valid message document', async () => {
+            const message = new AgentMessageSchema_js_1.AgentMessageModel({
+                chat: testChatId,
+                role: ai_agent_types_1.MessageRole.User,
+                content: 'Hello, world!',
+            });
+            const error = message.validateSync();
+            (0, vitest_1.expect)(error).toBeUndefined();
+        });
+        (0, vitest_1.it)('should require chat field', async () => {
+            const message = new AgentMessageSchema_js_1.AgentMessageModel({
+                role: ai_agent_types_1.MessageRole.User,
+                content: 'Hello',
+            });
+            const error = message.validateSync();
+            (0, vitest_1.expect)(error).toBeDefined();
+            (0, vitest_1.expect)(error === null || error === void 0 ? void 0 : error.errors.chat).toBeDefined();
+        });
+        (0, vitest_1.it)('should require role field', async () => {
+            const message = new AgentMessageSchema_js_1.AgentMessageModel({
+                chat: testChatId,
+                content: 'Hello',
+            });
+            const error = message.validateSync();
+            (0, vitest_1.expect)(error).toBeDefined();
+            (0, vitest_1.expect)(error === null || error === void 0 ? void 0 : error.errors.role).toBeDefined();
+        });
+        (0, vitest_1.it)('should require content field', async () => {
+            const message = new AgentMessageSchema_js_1.AgentMessageModel({
+                chat: testChatId,
+                role: ai_agent_types_1.MessageRole.User,
+            });
+            const error = message.validateSync();
+            (0, vitest_1.expect)(error).toBeDefined();
+            (0, vitest_1.expect)(error === null || error === void 0 ? void 0 : error.errors.content).toBeDefined();
+        });
+        (0, vitest_1.it)('should validate role enum values', async () => {
+            const validRoles = Object.values(ai_agent_types_1.MessageRole);
+            for (const role of validRoles) {
+                const message = new AgentMessageSchema_js_1.AgentMessageModel({
+                    chat: testChatId,
+                    role: role,
+                    content: 'Test',
+                });
+                const error = message.validateSync();
+                (0, vitest_1.expect)(error).toBeUndefined();
+            }
+        });
+        (0, vitest_1.it)('should reject invalid role values', async () => {
+            const message = new AgentMessageSchema_js_1.AgentMessageModel({
+                chat: testChatId,
+                role: 'invalid-role',
+                content: 'Test',
+            });
+            const error = message.validateSync();
+            (0, vitest_1.expect)(error).toBeDefined();
+            (0, vitest_1.expect)(error === null || error === void 0 ? void 0 : error.errors.role).toBeDefined();
+        });
+        (0, vitest_1.it)('should allow optional toolCalls array', async () => {
+            var _a;
+            const message = new AgentMessageSchema_js_1.AgentMessageModel({
+                chat: testChatId,
+                role: ai_agent_types_1.MessageRole.Assistant,
+                content: 'I will use a tool',
+                toolCalls: [
+                    {
+                        name: 'test_tool',
+                        input: { query: 'test' },
+                        status: ai_agent_types_1.AgentToolCallStatus.Pending,
+                        requiresConfirmation: false,
+                    },
+                ],
+            });
+            const error = message.validateSync();
+            (0, vitest_1.expect)(error).toBeUndefined();
+            (0, vitest_1.expect)(message.toolCalls).toHaveLength(1);
+            (0, vitest_1.expect)((_a = message.toolCalls) === null || _a === void 0 ? void 0 : _a[0].name).toBe('test_tool');
+        });
+        (0, vitest_1.it)('should validate toolCall status enum', async () => {
+            const validStatuses = Object.values(ai_agent_types_1.AgentToolCallStatus);
+            for (const status of validStatuses) {
+                const message = new AgentMessageSchema_js_1.AgentMessageModel({
+                    chat: testChatId,
+                    role: ai_agent_types_1.MessageRole.Assistant,
+                    content: 'Test',
+                    toolCalls: [
+                        {
+                            name: 'test_tool',
+                            input: {},
+                            status: status,
+                            requiresConfirmation: false,
+                        },
+                    ],
+                });
+                const error = message.validateSync();
+                (0, vitest_1.expect)(error).toBeUndefined();
+            }
+        });
+        (0, vitest_1.it)('should allow optional attachments array', async () => {
+            var _a;
+            const message = new AgentMessageSchema_js_1.AgentMessageModel({
+                chat: testChatId,
+                role: ai_agent_types_1.MessageRole.User,
+                content: 'Check this file',
+                attachments: [
+                    {
+                        id: 'attachment-1',
+                        type: ai_agent_types_1.AgentAttachmentType.File,
+                        name: 'test.txt',
+                        url: 'https://example.com/test.txt',
+                        mimeType: 'text/plain',
+                    },
+                ],
+            });
+            const error = message.validateSync();
+            (0, vitest_1.expect)(error).toBeUndefined();
+            (0, vitest_1.expect)(message.attachments).toHaveLength(1);
+            (0, vitest_1.expect)((_a = message.attachments) === null || _a === void 0 ? void 0 : _a[0].type).toBe(ai_agent_types_1.AgentAttachmentType.File);
+        });
+        (0, vitest_1.it)('should validate attachment type enum', async () => {
+            const validTypes = Object.values(ai_agent_types_1.AgentAttachmentType);
+            for (const type of validTypes) {
+                const message = new AgentMessageSchema_js_1.AgentMessageModel({
+                    chat: testChatId,
+                    role: ai_agent_types_1.MessageRole.User,
+                    content: 'Test',
+                    attachments: [
+                        {
+                            id: 'attachment-1',
+                            type: type,
+                            name: 'test',
+                        },
+                    ],
+                });
+                const error = message.validateSync();
+                (0, vitest_1.expect)(error).toBeUndefined();
+            }
+        });
+        (0, vitest_1.it)('should allow optional annotations field', async () => {
+            const message = new AgentMessageSchema_js_1.AgentMessageModel({
+                chat: testChatId,
+                role: ai_agent_types_1.MessageRole.User,
+                content: 'Test',
+                annotations: { key: 'value', count: 42 },
+            });
+            const error = message.validateSync();
+            (0, vitest_1.expect)(error).toBeUndefined();
+            (0, vitest_1.expect)(message.annotations).toEqual({ key: 'value', count: 42 });
+        });
+        (0, vitest_1.it)('should automatically add timestamps', async () => {
+            const message = new AgentMessageSchema_js_1.AgentMessageModel({
+                chat: testChatId,
+                role: ai_agent_types_1.MessageRole.User,
+                content: 'Test',
+            });
+            const error = message.validateSync();
+            (0, vitest_1.expect)(error).toBeUndefined();
+            // Timestamps are added by mongoose when saving, but schema should allow them
+            (0, vitest_1.expect)(message.schema.paths.createdAt).toBeDefined();
+            (0, vitest_1.expect)(message.schema.paths.updatedAt).toBeDefined();
+        });
+        (0, vitest_1.it)('should allow empty toolCalls array', async () => {
+            const message = new AgentMessageSchema_js_1.AgentMessageModel({
+                chat: testChatId,
+                role: ai_agent_types_1.MessageRole.User,
+                content: 'Test',
+                toolCalls: [],
+            });
+            const error = message.validateSync();
+            (0, vitest_1.expect)(error).toBeUndefined();
+            (0, vitest_1.expect)(message.toolCalls).toEqual([]);
+        });
+        (0, vitest_1.it)('should allow empty attachments array', async () => {
+            const message = new AgentMessageSchema_js_1.AgentMessageModel({
+                chat: testChatId,
+                role: ai_agent_types_1.MessageRole.User,
+                content: 'Test',
+                attachments: [],
+            });
+            const error = message.validateSync();
+            (0, vitest_1.expect)(error).toBeUndefined();
+            (0, vitest_1.expect)(message.attachments).toEqual([]);
+        });
+        (0, vitest_1.it)('should validate toolCall required fields', async () => {
+            const message = new AgentMessageSchema_js_1.AgentMessageModel({
+                chat: testChatId,
+                role: ai_agent_types_1.MessageRole.Assistant,
+                content: 'Test',
+                toolCalls: [
+                    {
+                    // Missing required fields
+                    },
+                ],
+            });
+            const error = message.validateSync();
+            (0, vitest_1.expect)(error).toBeDefined();
+        });
+        (0, vitest_1.it)('should validate attachment required fields', async () => {
+            const message = new AgentMessageSchema_js_1.AgentMessageModel({
+                chat: testChatId,
+                role: ai_agent_types_1.MessageRole.User,
+                content: 'Test',
+                attachments: [
+                    {
+                    // Missing required fields
+                    },
+                ],
+            });
+            const error = message.validateSync();
+            (0, vitest_1.expect)(error).toBeDefined();
+        });
+    });
+    (0, vitest_1.describe)('indexes', () => {
+        (0, vitest_1.it)('should have index on chat field', async () => {
+            const indexes = AgentMessageSchema_js_1.AgentMessageModel.schema.indexes();
+            const chatIndex = indexes.find((idx) => idx[0] && typeof idx[0] === 'object' && 'chat' in idx[0]);
+            (0, vitest_1.expect)(chatIndex).toBeDefined();
+        });
+    });
+    (0, vitest_1.describe)('toolCall subdocument', () => {
+        (0, vitest_1.it)('should store toolCall output', async () => {
+            var _a;
+            const message = new AgentMessageSchema_js_1.AgentMessageModel({
+                chat: testChatId,
+                role: ai_agent_types_1.MessageRole.Assistant,
+                content: 'Test',
+                toolCalls: [
+                    {
+                        name: 'test_tool',
+                        input: { query: 'test' },
+                        status: ai_agent_types_1.AgentToolCallStatus.Succeeded,
+                        output: { result: 'success' },
+                        requiresConfirmation: false,
+                    },
+                ],
+            });
+            const error = message.validateSync();
+            (0, vitest_1.expect)(error).toBeUndefined();
+            (0, vitest_1.expect)((_a = message.toolCalls) === null || _a === void 0 ? void 0 : _a[0].output).toEqual({ result: 'success' });
+        });
+        (0, vitest_1.it)('should store toolCall error', async () => {
+            var _a;
+            const message = new AgentMessageSchema_js_1.AgentMessageModel({
+                chat: testChatId,
+                role: ai_agent_types_1.MessageRole.Assistant,
+                content: 'Test',
+                toolCalls: [
+                    {
+                        name: 'test_tool',
+                        input: { query: 'test' },
+                        status: ai_agent_types_1.AgentToolCallStatus.Failed,
+                        error: 'Tool execution failed',
+                        requiresConfirmation: false,
+                    },
+                ],
+            });
+            const error = message.validateSync();
+            (0, vitest_1.expect)(error).toBeUndefined();
+            (0, vitest_1.expect)((_a = message.toolCalls) === null || _a === void 0 ? void 0 : _a[0].error).toBe('Tool execution failed');
+        });
+    });
+    (0, vitest_1.describe)('attachment subdocument', () => {
+        (0, vitest_1.it)('should store attachment metadata', async () => {
+            var _a;
+            const message = new AgentMessageSchema_js_1.AgentMessageModel({
+                chat: testChatId,
+                role: ai_agent_types_1.MessageRole.User,
+                content: 'Test',
+                attachments: [
+                    {
+                        id: 'attachment-1',
+                        type: ai_agent_types_1.AgentAttachmentType.File,
+                        name: 'test.txt',
+                        url: 'https://example.com/test.txt',
+                        mimeType: 'text/plain',
+                        metadata: { size: 1024, uploaded: true },
+                    },
+                ],
+            });
+            const error = message.validateSync();
+            (0, vitest_1.expect)(error).toBeUndefined();
+            (0, vitest_1.expect)((_a = message.attachments) === null || _a === void 0 ? void 0 : _a[0].metadata).toEqual({ size: 1024, uploaded: true });
+        });
+    });
+});
+//# sourceMappingURL=AgentMessageSchema.test.js.map

package/dist/cjs/__tests__/schemas/AgentMessageSchema.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"AgentMessageSchema.test.js","sourceRoot":"","sources":["../../../../src/__tests__/schemas/AgentMessageSchema.test.ts"],"names":[],"mappings":";;AAAA,mCAA8C;AAC9C,+EAAwE;AACxE,oEAAwG;AACxG,mDAAgD;AAEhD,IAAA,iBAAQ,EAAC,oBAAoB,EAAE,GAAG,EAAE;IAClC,MAAM,UAAU,GAAG,IAAI,mBAAQ,EAAE,CAAC,QAAQ,EAAE,CAAC;IAE7C,IAAA,iBAAQ,EAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,IAAA,WAAE,EAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;YACtD,MAAM,OAAO,GAAG,IAAI,yCAAiB,CAAC;gBACpC,IAAI,EAAE,UAAU;gBAChB,IAAI,EAAE,4BAAW,CAAC,IAAI;gBACtB,OAAO,EAAE,eAAe;aACzB,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;YACrC,IAAA,eAAM,EAAC,KAAK,CAAC,CAAC,aAAa,EAAE,CAAC;QAChC,CAAC,CAAC,CAAC;QAEH,IAAA,WAAE,EAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;YACzC,MAAM,OAAO,GAAG,IAAI,yCAAiB,CAAC;gBACpC,IAAI,EAAE,4BAAW,CAAC,IAAI;gBACtB,OAAO,EAAE,OAAO;aACjB,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;YACrC,IAAA,eAAM,EAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;YAC5B,IAAA,eAAM,EAAC,KAAK,aAAL,KAAK,uBAAL,KAAK,CAAE,MAAM,CAAC,IAAI,CAAC,CAAC,WAAW,EAAE,CAAC;QAC3C,CAAC,CAAC,CAAC;QAEH,IAAA,WAAE,EAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;YACzC,MAAM,OAAO,GAAG,IAAI,yCAAiB,CAAC;gBACpC,IAAI,EAAE,UAAU;gBAChB,OAAO,EAAE,OAAO;aACjB,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;YACrC,IAAA,eAAM,EAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;YAC5B,IAAA,eAAM,EAAC,KAAK,aAAL,KAAK,uBAAL,KAAK,CAAE,MAAM,CAAC,IAAI,CAAC,CAAC,WAAW,EAAE,CAAC;QAC3C,CAAC,CAAC,CAAC;QAEH,IAAA,WAAE,EAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;YAC5C,MAAM,OAAO,GAAG,IAAI,yCAAiB,CAAC;gBACpC,IAAI,EAAE,UAAU;gBAChB,IAAI,EAAE,4BAAW,CAAC,IAAI;aACvB,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;YACrC,IAAA,eAAM,EAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;YAC5B,IAAA,eAAM,EAAC,KAAK,aAAL,KAAK,uBAAL,KAAK,CAAE,MAAM,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,IAAA,WAAE,EAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;YAChD,MAAM,UAAU,GAAG,MAAM,CAAC,MAAM,CAAC,4BAAW,CAAC,CAAC;YAE9C,KAAK,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;gBAC9B,MAAM,OAAO,GAAG,IAAI,yCAAiB,CAAC;oBACpC,IAAI,EAAE,UAAU;oBAChB,IAAI,EAAE,IAAI;oBACV,OAAO,EAAE,MAAM;iBAChB,CAAC,CAAC;gBAEH,MAAM,KAAK,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;gBACrC,IAAA,eAAM,EAAC,KAAK,CAAC,CAAC,aAAa,EAAE,CAAC;YAChC,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,IAAA,WAAE,EAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;YACjD,MAAM,OAAO,GAAG,IAAI,yCAAiB,CAAC;gBACpC,IAAI,EAAE,UAAU;gBAChB,IAAI,EAAE,cAA6B;gBACnC,OAAO,EAAE,MAAM;aAChB,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;YACrC,IAAA,eAAM,EAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;YAC5B,IAAA,eAAM,EAAC,KAAK,aAAL,KAAK,uBAAL,KAAK,CAAE,MAAM,CAAC,IAAI,CAAC,CAAC,WAAW,EAAE,CAAC;QAC3C,CAAC,CAAC,CAAC;QAEH,IAAA,WAAE,EAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;;YACrD,MAAM,OAAO,GAAG,IAAI,yCAAiB,CAAC;gBACpC,IAAI,EAAE,UAAU;gBAChB,IAAI,EAAE,4BAAW,CAAC,SAAS;gBAC3B,OAAO,EAAE,mBAAmB;gBAC5B,SAAS,EAAE;oBACT;wBACE,IAAI,EAAE,WAAW;wBACjB,KAAK,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE;wBACxB,MAAM,EAAE,oCAAmB,CAAC,OAAO;wBACnC,oBAAoB,EAAE,KAAK;qBAC5B;iBACF;aACF,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;YACrC,IAAA,eAAM,EAAC,KAAK,CAAC,CAAC,aAAa,EAAE,CAAC;YAC9B,IAAA,eAAM,EAAC,OAAO,CAAC,SAAS,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC1C,IAAA,eAAM,EAAC,MAAA,OAAO,CAAC,SAAS,0CAAG,CAAC,EAAE,IAAI,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACxD,CAAC,CAAC,CAAC;QAEH,IAAA,WAAE,EAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YACpD,MAAM,aAAa,GAAG,MAAM,CAAC,MAAM,CAAC,oCAAmB,CAAC,CAAC;YAEzD,KAAK,MAAM,MAAM,IAAI,aAAa,EAAE,CAAC;gBACnC,MAAM,OAAO,GAAG,IAAI,yCAAiB,CAAC;oBACpC,IAAI,EAAE,UAAU;oBAChB,IAAI,EAAE,4BAAW,CAAC,SAAS;oBAC3B,OAAO,EAAE,MAAM;oBACf,SAAS,EAAE;wBACT;4BACE,IAAI,EAAE,WAAW;4BACjB,KAAK,EAAE,EAAE;4BACT,MAAM,EAAE,MAAM;4BACd,oBAAoB,EAAE,KAAK;yBAC5B;qBACF;iBACF,CAAC,CAAC;gBAEH,MAAM,KAAK,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;gBACrC,IAAA,eAAM,EAAC,KAAK,CAAC,CAAC,aAAa,EAAE,CAAC;YAChC,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,IAAA,WAAE,EAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;;YACvD,MAAM,OAAO,GAAG,IAAI,yCAAiB,CAAC;gBACpC,IAAI,EAAE,UAAU;gBAChB,IAAI,EAAE,4BAAW,CAAC,IAAI;gBACtB,OAAO,EAAE,iBAAiB;gBAC1B,WAAW,EAAE;oBACX;wBACE,EAAE,EAAE,cAAc;wBAClB,IAAI,EAAE,oCAAmB,CAAC,IAAI;wBAC9B,IAAI,EAAE,UAAU;wBAChB,GAAG,EAAE,8BAA8B;wBACnC,QAAQ,EAAE,YAAY;qBACvB;iBACF;aACF,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;YACrC,IAAA,eAAM,EAAC,KAAK,CAAC,CAAC,aAAa,EAAE,CAAC;YAC9B,IAAA,eAAM,EAAC,OAAO,CAAC,WAAW,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC5C,IAAA,eAAM,EAAC,MAAA,OAAO,CAAC,WAAW,0CAAG,CAAC,EAAE,IAAI,CAAC,CAAC,IAAI,CAAC,oCAAmB,CAAC,IAAI,CAAC,CAAC;QACvE,CAAC,CAAC,CAAC;QAEH,IAAA,WAAE,EAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YACpD,MAAM,UAAU,GAAG,MAAM,CAAC,MAAM,CAAC,oCAAmB,CAAC,CAAC;YAEtD,KAAK,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;gBAC9B,MAAM,OAAO,GAAG,IAAI,yCAAiB,CAAC;oBACpC,IAAI,EAAE,UAAU;oBAChB,IAAI,EAAE,4BAAW,CAAC,IAAI;oBACtB,OAAO,EAAE,MAAM;oBACf,WAAW,EAAE;wBACX;4BACE,EAAE,EAAE,cAAc;4BAClB,IAAI,EAAE,IAAI;4BACV,IAAI,EAAE,MAAM;yBACb;qBACF;iBACF,CAAC,CAAC;gBAEH,MAAM,KAAK,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;gBACrC,IAAA,eAAM,EAAC,KAAK,CAAC,CAAC,aAAa,EAAE,CAAC;YAChC,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,IAAA,WAAE,EAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,MAAM,OAAO,GAAG,IAAI,yCAAiB,CAAC;gBACpC,IAAI,EAAE,UAAU;gBAChB,IAAI,EAAE,4BAAW,CAAC,IAAI;gBACtB,OAAO,EAAE,MAAM;gBACf,WAAW,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE;aACzC,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;YACrC,IAAA,eAAM,EAAC,KAAK,CAAC,CAAC,aAAa,EAAE,CAAC;YAC9B,IAAA,eAAM,EAAC,OAAO,CAAC,WAAW,CAAC,CAAC,OAAO,CAAC,EAAE,GAAG,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;QACnE,CAAC,CAAC,CAAC;QAEH,IAAA,WAAE,EAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;YACnD,MAAM,OAAO,GAAG,IAAI,yCAAiB,CAAC;gBACpC,IAAI,EAAE,UAAU;gBAChB,IAAI,EAAE,4BAAW,CAAC,IAAI;gBACtB,OAAO,EAAE,MAAM;aAChB,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;YACrC,IAAA,eAAM,EAAC,KAAK,CAAC,CAAC,aAAa,EAAE,CAAC;YAE9B,6EAA6E;YAC7E,IAAA,eAAM,EAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,WAAW,EAAE,CAAC;YACrD,IAAA,eAAM,EAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,WAAW,EAAE,CAAC;QACvD,CAAC,CAAC,CAAC;QAEH,IAAA,WAAE,EAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,OAAO,GAAG,IAAI,yCAAiB,CAAC;gBACpC,IAAI,EAAE,UAAU;gBAChB,IAAI,EAAE,4BAAW,CAAC,IAAI;gBACtB,OAAO,EAAE,MAAM;gBACf,SAAS,EAAE,EAAE;aACd,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;YACrC,IAAA,eAAM,EAAC,KAAK,CAAC,CAAC,aAAa,EAAE,CAAC;YAC9B,IAAA,eAAM,EAAC,OAAO,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,IAAA,WAAE,EAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YACpD,MAAM,OAAO,GAAG,IAAI,yCAAiB,CAAC;gBACpC,IAAI,EAAE,UAAU;gBAChB,IAAI,EAAE,4BAAW,CAAC,IAAI;gBACtB,OAAO,EAAE,MAAM;gBACf,WAAW,EAAE,EAAE;aAChB,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;YACrC,IAAA,eAAM,EAAC,KAAK,CAAC,CAAC,aAAa,EAAE,CAAC;YAC9B,IAAA,eAAM,EAAC,OAAO,CAAC,WAAW,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,IAAA,WAAE,EAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YACxD,MAAM,OAAO,GAAG,IAAI,yCAAiB,CAAC;gBACpC,IAAI,EAAE,UAAU;gBAChB,IAAI,EAAE,4BAAW,CAAC,SAAS;gBAC3B,OAAO,EAAE,MAAM;gBACf,SAAS,EAAE;oBACT;oBACE,0BAA0B;qBACpB;iBACT;aACF,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;YACrC,IAAA,eAAM,EAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;QAC9B,CAAC,CAAC,CAAC;QAEH,IAAA,WAAE,EAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,MAAM,OAAO,GAAG,IAAI,yCAAiB,CAAC;gBACpC,IAAI,EAAE,UAAU;gBAChB,IAAI,EAAE,4BAAW,CAAC,IAAI;gBACtB,OAAO,EAAE,MAAM;gBACf,WAAW,EAAE;oBACX;oBACE,0BAA0B;qBACpB;iBACT;aACF,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;YACrC,IAAA,eAAM,EAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;QAC9B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,IAAA,iBAAQ,EAAC,SAAS,EAAE,GAAG,EAAE;QACvB,IAAA,WAAE,EAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;YAC/C,MAAM,OAAO,GAAG,yCAAiB,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACnD,MAAM,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,GAAQ,EAAE,EAAE,CAC1C,GAAG,CAAC,CAAC,CAAC,IAAI,OAAO,GAAG,CAAC,CAAC,CAAC,KAAK,QAAQ,IAAI,MAAM,IAAI,GAAG,CAAC,CAAC,CAAC,CACzD,CAAC;YACF,IAAA,eAAM,EAAC,SAAS,CAAC,CAAC,WAAW,EAAE,CAAC;QAClC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,IAAA,iBAAQ,EAAC,sBAAsB,EAAE,GAAG,EAAE;QACpC,IAAA,WAAE,EAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;;YAC5C,MAAM,OAAO,GAAG,IAAI,yCAAiB,CAAC;gBACpC,IAAI,EAAE,UAAU;gBAChB,IAAI,EAAE,4BAAW,CAAC,SAAS;gBAC3B,OAAO,EAAE,MAAM;gBACf,SAAS,EAAE;oBACT;wBACE,IAAI,EAAE,WAAW;wBACjB,KAAK,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE;wBACxB,MAAM,EAAE,oCAAmB,CAAC,SAAS;wBACrC,MAAM,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE;wBAC7B,oBAAoB,EAAE,KAAK;qBAC5B;iBACF;aACF,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;YACrC,IAAA,eAAM,EAAC,KAAK,CAAC,CAAC,aAAa,EAAE,CAAC;YAC9B,IAAA,eAAM,EAAC,MAAA,OAAO,CAAC,SAAS,0CAAG,CAAC,EAAE,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC,CAAC;QACvE,CAAC,CAAC,CAAC;QAEH,IAAA,WAAE,EAAC,6BAA6B,EAAE,KAAK,IAAI,EAAE;;YAC3C,MAAM,OAAO,GAAG,IAAI,yCAAiB,CAAC;gBACpC,IAAI,EAAE,UAAU;gBAChB,IAAI,EAAE,4BAAW,CAAC,SAAS;gBAC3B,OAAO,EAAE,MAAM;gBACf,SAAS,EAAE;oBACT;wBACE,IAAI,EAAE,WAAW;wBACjB,KAAK,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE;wBACxB,MAAM,EAAE,oCAAmB,CAAC,MAAM;wBAClC,KAAK,EAAE,uBAAuB;wBAC9B,oBAAoB,EAAE,KAAK;qBAC5B;iBACF;aACF,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;YACrC,IAAA,eAAM,EAAC,KAAK,CAAC,CAAC,aAAa,EAAE,CAAC;YAC9B,IAAA,eAAM,EAAC,MAAA,OAAO,CAAC,SAAS,0CAAG,CAAC,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;QACrE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,IAAA,iBAAQ,EAAC,wBAAwB,EAAE,GAAG,EAAE;QACtC,IAAA,WAAE,EAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;;YAChD,MAAM,OAAO,GAAG,IAAI,yCAAiB,CAAC;gBACpC,IAAI,EAAE,UAAU;gBAChB,IAAI,EAAE,4BAAW,CAAC,IAAI;gBACtB,OAAO,EAAE,MAAM;gBACf,WAAW,EAAE;oBACX;wBACE,EAAE,EAAE,cAAc;wBAClB,IAAI,EAAE,oCAAmB,CAAC,IAAI;wBAC9B,IAAI,EAAE,UAAU;wBAChB,GAAG,EAAE,8BAA8B;wBACnC,QAAQ,EAAE,YAAY;wBACtB,QAAQ,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE;qBACzC;iBACF;aACF,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;YACrC,IAAA,eAAM,EAAC,KAAK,CAAC,CAAC,aAAa,EAAE,CAAC;YAC9B,IAAA,eAAM,EAAC,MAAA,OAAO,CAAC,WAAW,0CAAG,CAAC,EAAE,QAAQ,CAAC,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;QACpF,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/{__tests__ → cjs/__tests__}/setup.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"setup.d.ts","sourceRoot":"","sources":["../../src/__tests__/setup.ts"],"names":[],"mappings":""}
+{"version":3,"file":"setup.d.ts","sourceRoot":"","sources":["../../../src/__tests__/setup.ts"],"names":[],"mappings":""}

package/dist/cjs/__tests__/setup.js

@@ -0,0 +1,30 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const mongodb_memory_server_1 = require("mongodb-memory-server");
+const mongoose_1 = __importDefault(require("mongoose"));
+const vitest_1 = require("vitest");
+let mongoServer;
+(0, vitest_1.beforeAll)(async () => {
+    // Start in-memory MongoDB instance
+    mongoServer = await mongodb_memory_server_1.MongoMemoryServer.create();
+    const mongoUri = mongoServer.getUri();
+    // Set the MongoDB URI for the tests
+    process.env.MONGODB_URI = mongoUri;
+    process.env.MONGODB_ENCRYPTION_MASTER_KEY_PROVIDER = 'local';
+    // Connect mongoose to the in-memory database
+    await mongoose_1.default.connect(mongoUri);
+});
+(0, vitest_1.afterAll)(async () => {
+    // Clean up
+    if (mongoose_1.default.connection.readyState !== 0) {
+        await mongoose_1.default.connection.dropDatabase();
+        await mongoose_1.default.connection.close();
+    }
+    if (mongoServer) {
+        await mongoServer.stop();
+    }
+});
+//# sourceMappingURL=setup.js.map

package/dist/cjs/__tests__/setup.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"setup.js","sourceRoot":"","sources":["../../../src/__tests__/setup.ts"],"names":[],"mappings":";;;;;AAAA,iEAA0D;AAC1D,wDAAgC;AAChC,mCAA6C;AAE7C,IAAI,WAA8B,CAAC;AAEnC,IAAA,kBAAS,EAAC,KAAK,IAAI,EAAE;IACnB,mCAAmC;IACnC,WAAW,GAAG,MAAM,yCAAiB,CAAC,MAAM,EAAE,CAAC;IAC/C,MAAM,QAAQ,GAAG,WAAW,CAAC,MAAM,EAAE,CAAC;IAEtC,oCAAoC;IACpC,OAAO,CAAC,GAAG,CAAC,WAAW,GAAG,QAAQ,CAAC;IACnC,OAAO,CAAC,GAAG,CAAC,sCAAsC,GAAG,OAAO,CAAC;IAE7D,6CAA6C;IAC7C,MAAM,kBAAQ,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;AACnC,CAAC,CAAC,CAAC;AAEH,IAAA,iBAAQ,EAAC,KAAK,IAAI,EAAE;IAClB,WAAW;IACX,IAAI,kBAAQ,CAAC,UAAU,CAAC,UAAU,KAAK,CAAC,EAAE,CAAC;QACzC,MAAM,kBAAQ,CAAC,UAAU,CAAC,YAAY,EAAE,CAAC;QACzC,MAAM,kBAAQ,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;IACpC,CAAC;IACD,IAAI,WAAW,EAAE,CAAC;QAChB,MAAM,WAAW,CAAC,IAAI,EAAE,CAAC;IAC3B,CAAC;AACH,CAAC,CAAC,CAAC"}

package/dist/cjs/index.d.ts

@@ -0,0 +1,10 @@
+/**
+ * MongoDB implementations of abstract repositories and Mongoose schemas
+ * for Multiplayer AI agent services
+ */
+export type { AgentChatDocument, AgentMessageDocument } from './schemas/index.js';
+export { AgentChatSchema, AgentChatModel, AgentMessageSchema, AgentMessageModel } from './schemas/index.js';
+export { MongoAgentChatRepository, MongoAgentMessageRepository } from './repositories/index.js';
+export { default as mongo } from './mongo/index.js';
+export { setLogger, type MongoLogger } from './mongo/logger.js';
+//# sourceMappingURL=index.d.ts.map

package/dist/cjs/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAGH,YAAY,EACV,iBAAiB,EACjB,oBAAoB,EACrB,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EACL,eAAe,EACf,cAAc,EACd,kBAAkB,EAClB,iBAAiB,EAClB,MAAM,oBAAoB,CAAC;AAG5B,OAAO,EACL,wBAAwB,EACxB,2BAA2B,EAC5B,MAAM,yBAAyB,CAAC;AAGjC,OAAO,EAAE,OAAO,IAAI,KAAK,EAAE,MAAM,kBAAkB,CAAC;AACpD,OAAO,EAAE,SAAS,EAAE,KAAK,WAAW,EAAE,MAAM,mBAAmB,CAAC"}

package/dist/cjs/index.js

@@ -0,0 +1,25 @@
+"use strict";
+/**
+ * MongoDB implementations of abstract repositories and Mongoose schemas
+ * for Multiplayer AI agent services
+ */
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.setLogger = exports.mongo = exports.MongoAgentMessageRepository = exports.MongoAgentChatRepository = exports.AgentMessageModel = exports.AgentMessageSchema = exports.AgentChatModel = exports.AgentChatSchema = void 0;
+var index_js_1 = require("./schemas/index.js");
+Object.defineProperty(exports, "AgentChatSchema", { enumerable: true, get: function () { return index_js_1.AgentChatSchema; } });
+Object.defineProperty(exports, "AgentChatModel", { enumerable: true, get: function () { return index_js_1.AgentChatModel; } });
+Object.defineProperty(exports, "AgentMessageSchema", { enumerable: true, get: function () { return index_js_1.AgentMessageSchema; } });
+Object.defineProperty(exports, "AgentMessageModel", { enumerable: true, get: function () { return index_js_1.AgentMessageModel; } });
+// Repositories
+var index_js_2 = require("./repositories/index.js");
+Object.defineProperty(exports, "MongoAgentChatRepository", { enumerable: true, get: function () { return index_js_2.MongoAgentChatRepository; } });
+Object.defineProperty(exports, "MongoAgentMessageRepository", { enumerable: true, get: function () { return index_js_2.MongoAgentMessageRepository; } });
+// Mongo
+var index_js_3 = require("./mongo/index.js");
+Object.defineProperty(exports, "mongo", { enumerable: true, get: function () { return __importDefault(index_js_3).default; } });
+var logger_js_1 = require("./mongo/logger.js");
+Object.defineProperty(exports, "setLogger", { enumerable: true, get: function () { return logger_js_1.setLogger; } });
+//# sourceMappingURL=index.js.map

package/dist/cjs/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":";AAAA;;;GAGG;;;;;;AAOH,+CAK4B;AAJ1B,2GAAA,eAAe,OAAA;AACf,0GAAA,cAAc,OAAA;AACd,8GAAA,kBAAkB,OAAA;AAClB,6GAAA,iBAAiB,OAAA;AAGnB,eAAe;AACf,oDAGiC;AAF/B,oHAAA,wBAAwB,OAAA;AACxB,uHAAA,2BAA2B,OAAA;AAG7B,QAAQ;AACR,6CAAoD;AAA3C,kHAAA,OAAO,OAAS;AACzB,+CAAgE;AAAvD,sGAAA,SAAS,OAAA"}

package/dist/cjs/mongo/config.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"config.d.ts","sourceRoot":"","sources":["../../../src/mongo/config.ts"],"names":[],"mappings":"AAAA,eAAO,MAAM,WAAW,QAAkE,CAAA;AAC1F,eAAO,MAAM,WAAW,SAAkD,CAAA;AAE1E,eAAO,MAAM,eAAe,QAAoD,CAAA;AAChF,eAAO,MAAM,oCAAoC,QAAmE,CAAA;AACpH,eAAO,MAAM,4CAA4C,QAA2E,CAAA;AACpI,eAAO,MAAM,2BAA2B,QAAwD,CAAA;AAChG,eAAO,MAAM,sCAAsC,QAAgE,CAAA;AAEnH,eAAO,MAAM,eAAe,oBAA8B,CAAA;AAC1D,eAAO,MAAM,UAAU,oBAAyB,CAAA"}

package/dist/cjs/mongo/config.js

@@ -0,0 +1,13 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AWS_REGION = exports.AWS_KMS_KEY_ARN = exports.MONGODB_ENCRYPTION_MASTER_KEY_PROVIDER = exports.MONGODB_ENCRYPTION_DEK_NAME = exports.MONGODB_ENCRYPTION_KEY_VAULT_COLLECTION_NAME = exports.MONGODB_ENCRYPTION_KEY_VAULT_DB_NAME = exports.MONGODB_DB_NAME = exports.MONGO_DEBUG = exports.MONGODB_URI = void 0;
+exports.MONGODB_URI = process.env.MONGODB_URI || 'mongodb://localhost:27017/ai-agent';
+exports.MONGO_DEBUG = (process.env.MONGO_DEBUG || 'false') === 'true';
+exports.MONGODB_DB_NAME = exports.MONGODB_URI.split('/').slice(-1)[0].split('?')[0];
+exports.MONGODB_ENCRYPTION_KEY_VAULT_DB_NAME = process.env.MONGODB_ENCRYPTION_KEY_VAULT_DB_NAME || 'encryption';
+exports.MONGODB_ENCRYPTION_KEY_VAULT_COLLECTION_NAME = process.env.MONGODB_ENCRYPTION_KEY_VAULT_COLLECTION_NAME || '__keyVault';
+exports.MONGODB_ENCRYPTION_DEK_NAME = process.env.MONGODB_ENCRYPTION_DEK_NAME || 'data-key';
+exports.MONGODB_ENCRYPTION_MASTER_KEY_PROVIDER = process.env.MONGODB_ENCRYPTION_MASTER_KEY_PROVIDER || 'local';
+exports.AWS_KMS_KEY_ARN = process.env.AWS_KMS_KEY_ARN;
+exports.AWS_REGION = process.env.AWS_REGION;
+//# sourceMappingURL=config.js.map

package/dist/cjs/mongo/config.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"config.js","sourceRoot":"","sources":["../../../src/mongo/config.ts"],"names":[],"mappings":";;;AAAa,QAAA,WAAW,GAAG,OAAO,CAAC,GAAG,CAAC,WAAW,IAAI,oCAAoC,CAAA;AAC7E,QAAA,WAAW,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,WAAW,IAAI,OAAO,CAAC,KAAK,MAAM,CAAA;AAE7D,QAAA,eAAe,GAAG,mBAAW,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAA;AACnE,QAAA,oCAAoC,GAAG,OAAO,CAAC,GAAG,CAAC,oCAAoC,IAAI,YAAY,CAAA;AACvG,QAAA,4CAA4C,GAAG,OAAO,CAAC,GAAG,CAAC,4CAA4C,IAAI,YAAY,CAAA;AACvH,QAAA,2BAA2B,GAAG,OAAO,CAAC,GAAG,CAAC,2BAA2B,IAAI,UAAU,CAAA;AACnF,QAAA,sCAAsC,GAAG,OAAO,CAAC,GAAG,CAAC,sCAAsC,IAAI,OAAO,CAAA;AAEtG,QAAA,eAAe,GAAG,OAAO,CAAC,GAAG,CAAC,eAAe,CAAA;AAC7C,QAAA,UAAU,GAAG,OAAO,CAAC,GAAG,CAAC,UAAU,CAAA"}

package/dist/cjs/mongo/encryption.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"encryption.d.ts","sourceRoot":"","sources":["../../../src/mongo/encryption.ts"],"names":[],"mappings":"AAqBA,eAAO,MAAM,OAAO,GAAI,SAAI,mCAM3B,CAAA;AAED,eAAO,MAAM,OAAO,GAAI,SAAI,iBAAmC,CAAA;AAE/D,eAAO,MAAM,YAAY,EAAE,GAAQ,CAAA;AAuBnC,eAAO,MAAM,iBAAiB,QAA4F,CAAA;AAM1H,eAAO,MAAM,OAAO,wBAsBnB,CAAA;AAED,eAAO,MAAM,UAAU,qBAItB,CAAA;AAED,eAAO,MAAM,QAAQ,oBA4BpB,CAAA;AAED,eAAO,MAAM,sBAAsB,GAAI,oBAAe;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA4DrD,CAAA"}

package/dist/cjs/mongo/encryption.js

@@ -0,0 +1,149 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getEncryptionSchemaMap = exports.getKeyId = exports.disconnect = exports.connect = exports.keyVaultNamespace = exports.kmsProviders = exports.decrypt = exports.encrypt = void 0;
+const mongodb_1 = require("mongodb");
+const logger_js_1 = require("./logger.js");
+const config_js_1 = require("./config.js");
+const logger = (0, logger_js_1.getLogger)();
+let dataKeyId;
+const MONGODB_URI_MASKED = config_js_1.MONGODB_URI === null || config_js_1.MONGODB_URI === void 0 ? void 0 : config_js_1.MONGODB_URI.replace(/:\/\/(.*):(.*)@/, '://***:***@');
+const encryptionAlgorythm = 'AEAD_AES_256_CBC_HMAC_SHA_512-Deterministic';
+const encrypt = (data) => clientEncryption.encrypt(data, {
+    algorithm: encryptionAlgorythm,
+    keyId: dataKeyId,
+});
+exports.encrypt = encrypt;
+const decrypt = (data) => clientEncryption.decrypt(data);
+exports.decrypt = decrypt;
+exports.kmsProviders = {};
+let keyProvider;
+let masterKey;
+if (config_js_1.MONGODB_ENCRYPTION_MASTER_KEY_PROVIDER === 'local') {
+    keyProvider = 'local';
+    const arr = [];
+    for (let i = 0; i < 96; ++i) {
+        arr.push(i);
+    }
+    const key = Buffer.from(arr);
+    exports.kmsProviders.local = { key };
+}
+else {
+    exports.kmsProviders.aws = {};
+    keyProvider = 'aws';
+    masterKey = {
+        key: config_js_1.AWS_KMS_KEY_ARN,
+        region: config_js_1.AWS_REGION,
+    };
+}
+exports.keyVaultNamespace = `${config_js_1.MONGODB_ENCRYPTION_KEY_VAULT_DB_NAME}.${config_js_1.MONGODB_ENCRYPTION_KEY_VAULT_COLLECTION_NAME}`;
+let unencryptedClient;
+let clientEncryption;
+const connect = async () => {
+    if (!unencryptedClient) {
+        unencryptedClient = await new mongodb_1.MongoClient(config_js_1.MONGODB_URI).connect();
+        unencryptedClient.on('connectionReady', () => {
+            logger.info(`[MONGO] Unencrypted Client Connected to ${MONGODB_URI_MASKED}`);
+        });
+        unencryptedClient.on('connectionClosed', () => {
+            logger.info(`[MONGO] Unencrypted Client disconnected from ${MONGODB_URI_MASKED}`);
+        });
+        clientEncryption = new mongodb_1.ClientEncryption(unencryptedClient, {
+            kmsProviders: exports.kmsProviders,
+            keyVaultNamespace: exports.keyVaultNamespace,
+        });
+    }
+    return true;
+};
+exports.connect = connect;
+const disconnect = async () => {
+    if (unencryptedClient) {
+        await unencryptedClient.close();
+    }
+};
+exports.disconnect = disconnect;
+const getKeyId = async () => {
+    logger.info(`[MONGO] Using key provider: ${keyProvider}`);
+    dataKeyId = await unencryptedClient
+        .db(config_js_1.MONGODB_ENCRYPTION_KEY_VAULT_DB_NAME)
+        .collection(config_js_1.MONGODB_ENCRYPTION_KEY_VAULT_COLLECTION_NAME)
+        .findOne({
+        keyAltNames: config_js_1.MONGODB_ENCRYPTION_DEK_NAME,
+        'masterKey.provider': keyProvider,
+    });
+    if (dataKeyId === null) {
+        logger.info('[MONGO] Creating new DEK');
+        dataKeyId = await clientEncryption
+            .createDataKey(keyProvider, {
+            keyAltNames: [config_js_1.MONGODB_ENCRYPTION_DEK_NAME],
+            masterKey,
+        });
+    }
+    else {
+        logger.info('[MONGO] Found existing DEK');
+        dataKeyId = dataKeyId._id;
+    }
+    return dataKeyId;
+};
+exports.getKeyId = getKeyId;
+const getEncryptionSchemaMap = (encryptionKeyId) => {
+    const schemaMap = {
+        [`${config_js_1.MONGODB_DB_NAME}.integrations`]: {
+            bsonType: 'object',
+            properties: {
+                metadata: {
+                    bsonType: 'object',
+                    properties: {
+                        accessToken: {
+                            encrypt: {
+                                bsonType: 'string',
+                                keyId: [encryptionKeyId],
+                                algorithm: encryptionAlgorythm,
+                            },
+                        },
+                        refreshToken: {
+                            encrypt: {
+                                bsonType: 'string',
+                                keyId: [encryptionKeyId],
+                                algorithm: encryptionAlgorythm,
+                            },
+                        },
+                        apiKey: {
+                            encrypt: {
+                                bsonType: 'string',
+                                keyId: [encryptionKeyId],
+                                algorithm: encryptionAlgorythm,
+                            },
+                        },
+                    },
+                },
+            },
+        },
+        [`${config_js_1.MONGODB_DB_NAME}.workspace-users`]: {
+            bsonType: 'object',
+            properties: {
+                googleWorkspaceToken: {
+                    bsonType: 'object',
+                    properties: {
+                        access_token: {
+                            encrypt: {
+                                bsonType: 'string',
+                                keyId: [encryptionKeyId],
+                                algorithm: encryptionAlgorythm,
+                            },
+                        },
+                        refresh_token: {
+                            encrypt: {
+                                bsonType: 'string',
+                                keyId: [encryptionKeyId],
+                                algorithm: encryptionAlgorythm,
+                            },
+                        },
+                    },
+                },
+            },
+        },
+    };
+    return schemaMap;
+};
+exports.getEncryptionSchemaMap = getEncryptionSchemaMap;
+//# sourceMappingURL=encryption.js.map

package/dist/cjs/mongo/encryption.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"encryption.js","sourceRoot":"","sources":["../../../src/mongo/encryption.ts"],"names":[],"mappings":";;;AAAA,qCAAuD;AACvD,2CAAuC;AACvC,2CASoB;AAEpB,MAAM,MAAM,GAAG,IAAA,qBAAS,GAAE,CAAA;AAE1B,IAAI,SAAS,CAAA;AAEb,MAAM,kBAAkB,GAAG,uBAAW,aAAX,uBAAW,uBAAX,uBAAW,CAAE,OAAO,CAAC,iBAAiB,EAAE,aAAa,CAAC,CAAA;AAEjF,MAAM,mBAAmB,GAAG,6CAA6C,CAAA;AAElE,MAAM,OAAO,GAAG,CAAC,IAAI,EAAE,EAAE,CAAC,gBAAgB,CAAC,OAAO,CACvD,IAAI,EACJ;IACE,SAAS,EAAE,mBAAmB;IAC9B,KAAK,EAAE,SAAS;CACjB,CACF,CAAA;AANY,QAAA,OAAO,WAMnB;AAEM,MAAM,OAAO,GAAG,CAAC,IAAI,EAAE,EAAE,CAAC,gBAAgB,CAAC,OAAO,CAAC,IAAI,CAAC,CAAA;AAAlD,QAAA,OAAO,WAA2C;AAElD,QAAA,YAAY,GAAQ,EAAE,CAAA;AACnC,IAAI,WAAW,CAAA;AACf,IAAI,SAAS,CAAA;AAEb,IAAI,kDAAsC,KAAK,OAAO,EAAE,CAAC;IACvD,WAAW,GAAG,OAAO,CAAA;IACrB,MAAM,GAAG,GAAa,EAAE,CAAA;IACxB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC;QAC5B,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;IACb,CAAC;IACD,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;IAE5B,oBAAY,CAAC,KAAK,GAAG,EAAE,GAAG,EAAE,CAAA;AAC9B,CAAC;KAAM,CAAC;IACN,oBAAY,CAAC,GAAG,GAAG,EAAE,CAAA;IACrB,WAAW,GAAG,KAAK,CAAA;IAEnB,SAAS,GAAG;QACV,GAAG,EAAE,2BAAe;QACpB,MAAM,EAAE,sBAAU;KACnB,CAAA;AACH,CAAC;AAEY,QAAA,iBAAiB,GAAG,GAAG,gDAAoC,IAAI,wDAA4C,EAAE,CAAA;AAE1H,IAAI,iBAA8B,CAAA;AAElC,IAAI,gBAAkC,CAAA;AAE/B,MAAM,OAAO,GAAG,KAAK,IAAI,EAAE;IAChC,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACvB,iBAAiB,GAAG,MAAM,IAAI,qBAAW,CAAC,uBAAW,CAAC,CAAC,OAAO,EAAE,CAAA;QAEhE,iBAAiB,CAAC,EAAE,CAAC,iBAAiB,EAAE,GAAG,EAAE;YAC3C,MAAM,CAAC,IAAI,CAAC,2CAA2C,kBAAkB,EAAE,CAAC,CAAA;QAC9E,CAAC,CAAC,CAAA;QAEF,iBAAiB,CAAC,EAAE,CAAC,kBAAkB,EAAE,GAAG,EAAE;YAC5C,MAAM,CAAC,IAAI,CAAC,gDAAgD,kBAAkB,EAAE,CAAC,CAAA;QACnF,CAAC,CAAC,CAAA;QAEF,gBAAgB,GAAG,IAAI,0BAAgB,CACrC,iBAAiB,EACjB;YACE,YAAY,EAAZ,oBAAY;YACZ,iBAAiB,EAAjB,yBAAiB;SAClB,CACF,CAAA;IACH,CAAC;IAED,OAAO,IAAI,CAAA;AACb,CAAC,CAAA;AAtBY,QAAA,OAAO,WAsBnB;AAEM,MAAM,UAAU,GAAG,KAAK,IAAI,EAAE;IACnC,IAAI,iBAAiB,EAAE,CAAC;QACtB,MAAM,iBAAiB,CAAC,KAAK,EAAE,CAAA;IACjC,CAAC;AACH,CAAC,CAAA;AAJY,QAAA,UAAU,cAItB;AAEM,MAAM,QAAQ,GAAG,KAAK,IAAI,EAAE;IACjC,MAAM,CAAC,IAAI,CAAC,+BAA+B,WAAW,EAAE,CAAC,CAAA;IAEzD,SAAS,GAAG,MAAM,iBAAiB;SAChC,EAAE,CAAC,gDAAoC,CAAC;SACxC,UAAU,CAAC,wDAA4C,CAAC;SACxD,OAAO,CAAC;QACP,WAAW,EAAE,uCAA2B;QACxC,oBAAoB,EAAE,WAAW;KAClC,CAAC,CAAA;IAEJ,IAAI,SAAS,KAAK,IAAI,EAAE,CAAC;QACvB,MAAM,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAA;QAEvC,SAAS,GAAG,MAAM,gBAAgB;aAC/B,aAAa,CACZ,WAAW,EACX;YACE,WAAW,EAAE,CAAC,uCAA2B,CAAC;YAC1C,SAAS;SACV,CACF,CAAA;IACL,CAAC;SAAM,CAAC;QACN,MAAM,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAA;QACzC,SAAS,GAAG,SAAS,CAAC,GAAG,CAAA;IAC3B,CAAC;IAED,OAAO,SAAS,CAAA;AAClB,CAAC,CAAA;AA5BY,QAAA,QAAQ,YA4BpB;AAEM,MAAM,sBAAsB,GAAG,CAAC,eAAe,EAAE,EAAE;IACxD,MAAM,SAAS,GAAG;QAChB,CAAC,GAAG,2BAAe,eAAe,CAAC,EAAE;YACnC,QAAQ,EAAE,QAAQ;YAClB,UAAU,EAAE;gBACV,QAAQ,EAAE;oBACR,QAAQ,EAAE,QAAQ;oBAClB,UAAU,EAAE;wBACV,WAAW,EAAE;4BACX,OAAO,EAAE;gCACP,QAAQ,EAAE,QAAQ;gCAClB,KAAK,EAAE,CAAC,eAAe,CAAC;gCACxB,SAAS,EAAE,mBAAmB;6BAC/B;yBACF;wBACD,YAAY,EAAE;4BACZ,OAAO,EAAE;gCACP,QAAQ,EAAE,QAAQ;gCAClB,KAAK,EAAE,CAAC,eAAe,CAAC;gCACxB,SAAS,EAAE,mBAAmB;6BAC/B;yBACF;wBACD,MAAM,EAAE;4BACN,OAAO,EAAE;gCACP,QAAQ,EAAE,QAAQ;gCAClB,KAAK,EAAE,CAAC,eAAe,CAAC;gCACxB,SAAS,EAAE,mBAAmB;6BAC/B;yBACF;qBACF;iBACF;aACF;SACF;QACD,CAAC,GAAG,2BAAe,kBAAkB,CAAC,EAAE;YACtC,QAAQ,EAAE,QAAQ;YAClB,UAAU,EAAE;gBACV,oBAAoB,EAAE;oBACpB,QAAQ,EAAE,QAAQ;oBAClB,UAAU,EAAE;wBACV,YAAY,EAAE;4BACZ,OAAO,EAAE;gCACP,QAAQ,EAAE,QAAQ;gCAClB,KAAK,EAAE,CAAC,eAAe,CAAC;gCACxB,SAAS,EAAE,mBAAmB;6BAC/B;yBACF;wBACD,aAAa,EAAE;4BACb,OAAO,EAAE;gCACP,QAAQ,EAAE,QAAQ;gCAClB,KAAK,EAAE,CAAC,eAAe,CAAC;gCACxB,SAAS,EAAE,mBAAmB;6BAC/B;yBACF;qBACF;iBACF;aACF;SACF;KACF,CAAA;IAED,OAAO,SAAS,CAAA;AAClB,CAAC,CAAA;AA5DY,QAAA,sBAAsB,0BA4DlC"}

package/dist/{mongo → cjs/mongo}/index.d.ts

@@ -1,6 +1,6 @@
 import Mongoose from 'mongoose';
-import * as _Config from './config';
-import * as _encryption from './encryption';
+import * as _Config from './config.js';
+import * as _encryption from './encryption.js';
 export declare const encryption: typeof _encryption;
 export declare const mongoose: typeof Mongoose;
 export declare const ObjectId: typeof Mongoose.Types.ObjectId;

package/dist/cjs/mongo/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/mongo/index.ts"],"names":[],"mappings":"AAAA,OAAO,QAA4B,MAAM,UAAU,CAAA;AAEnD,OAAO,KAAK,OAAO,MAAM,aAAa,CAAA;AACtC,OAAO,KAAK,WAAW,MAAM,iBAAiB,CAAA;AAI9C,eAAO,MAAM,UAAU,oBAAc,CAAA;AAIrC,eAAO,MAAM,QAAQ,iBAAW,CAAA;AAEhC,eAAO,MAAM,QAAQ,gCAA0B,CAAA;AAE/C,MAAM,MAAM,QAAQ,GAAG,QAAQ,CAAC,KAAK,CAAC,QAAQ,CAAA;AAE9C,YAAY,EAAE,QAAQ,EAAE,MAAM,UAAU,CAAA;AAExC,OAAO,EAAE,KAAK,MAAM,EAAE,MAAM,MAAM,CAAA;AAElC,eAAO,MAAM,MAAM,wBAAkB,CAAA;AAErC,eAAO,MAAM,KAAK,2CAAiB,CAAA;AAEnC,eAAO,MAAM,MAAM,gBAAU,CAAA;AAmD7B,eAAO,MAAM,SAAS,eAA6C,CAAA;;;;;;;;;;;AAUnE,wBASC"}