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

package/dist/mongo/encryption.js

@@ -0,0 +1,140 @@
+import { MongoClient, ClientEncryption } from 'mongodb';
+import { getLogger } from './logger';
+import { MONGODB_DB_NAME, MONGODB_URI, MONGODB_ENCRYPTION_KEY_VAULT_DB_NAME, MONGODB_ENCRYPTION_KEY_VAULT_COLLECTION_NAME, MONGODB_ENCRYPTION_DEK_NAME, MONGODB_ENCRYPTION_MASTER_KEY_PROVIDER, AWS_KMS_KEY_ARN, AWS_REGION, } from './config';
+const logger = getLogger();
+let dataKeyId;
+const MONGODB_URI_MASKED = MONGODB_URI === null || MONGODB_URI === void 0 ? void 0 : MONGODB_URI.replace(/:\/\/(.*):(.*)@/, '://***:***@');
+const encryptionAlgorythm = 'AEAD_AES_256_CBC_HMAC_SHA_512-Deterministic';
+export const encrypt = (data) => clientEncryption.encrypt(data, {
+    algorithm: encryptionAlgorythm,
+    keyId: dataKeyId,
+});
+export const decrypt = (data) => clientEncryption.decrypt(data);
+export const kmsProviders = {};
+let keyProvider;
+let masterKey;
+if (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);
+    kmsProviders.local = { key };
+}
+else {
+    kmsProviders.aws = {};
+    keyProvider = 'aws';
+    masterKey = {
+        key: AWS_KMS_KEY_ARN,
+        region: AWS_REGION,
+    };
+}
+export const keyVaultNamespace = `${MONGODB_ENCRYPTION_KEY_VAULT_DB_NAME}.${MONGODB_ENCRYPTION_KEY_VAULT_COLLECTION_NAME}`;
+let unencryptedClient;
+let clientEncryption;
+export const connect = async () => {
+    if (!unencryptedClient) {
+        unencryptedClient = await new MongoClient(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 ClientEncryption(unencryptedClient, {
+            kmsProviders,
+            keyVaultNamespace,
+        });
+    }
+    return true;
+};
+export const disconnect = async () => {
+    if (unencryptedClient) {
+        await unencryptedClient.close();
+    }
+};
+export const getKeyId = async () => {
+    logger.info(`[MONGO] Using key provider: ${keyProvider}`);
+    dataKeyId = await unencryptedClient
+        .db(MONGODB_ENCRYPTION_KEY_VAULT_DB_NAME)
+        .collection(MONGODB_ENCRYPTION_KEY_VAULT_COLLECTION_NAME)
+        .findOne({
+        keyAltNames: MONGODB_ENCRYPTION_DEK_NAME,
+        'masterKey.provider': keyProvider,
+    });
+    if (dataKeyId === null) {
+        logger.info('[MONGO] Creating new DEK');
+        dataKeyId = await clientEncryption
+            .createDataKey(keyProvider, {
+            keyAltNames: [MONGODB_ENCRYPTION_DEK_NAME],
+            masterKey,
+        });
+    }
+    else {
+        logger.info('[MONGO] Found existing DEK');
+        dataKeyId = dataKeyId._id;
+    }
+    return dataKeyId;
+};
+export const getEncryptionSchemaMap = (encryptionKeyId) => {
+    const schemaMap = {
+        [`${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,
+                            },
+                        },
+                    },
+                },
+            },
+        },
+        [`${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;
+};
+//# sourceMappingURL=encryption.js.map

package/dist/mongo/encryption.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"encryption.js","sourceRoot":"","sources":["../../src/mongo/encryption.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,gBAAgB,EAAE,MAAM,SAAS,CAAA;AACvD,OAAO,EAAE,SAAS,EAAE,MAAM,UAAU,CAAA;AACpC,OAAO,EACL,eAAe,EACf,WAAW,EACX,oCAAoC,EACpC,4CAA4C,EAC5C,2BAA2B,EAC3B,sCAAsC,EACtC,eAAe,EACf,UAAU,GACX,MAAM,UAAU,CAAA;AAEjB,MAAM,MAAM,GAAG,SAAS,EAAE,CAAA;AAE1B,IAAI,SAAS,CAAA;AAEb,MAAM,kBAAkB,GAAG,WAAW,aAAX,WAAW,uBAAX,WAAW,CAAE,OAAO,CAAC,iBAAiB,EAAE,aAAa,CAAC,CAAA;AAEjF,MAAM,mBAAmB,GAAG,6CAA6C,CAAA;AAEzE,MAAM,CAAC,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;AAED,MAAM,CAAC,MAAM,OAAO,GAAG,CAAC,IAAI,EAAE,EAAE,CAAC,gBAAgB,CAAC,OAAO,CAAC,IAAI,CAAC,CAAA;AAE/D,MAAM,CAAC,MAAM,YAAY,GAAQ,EAAE,CAAA;AACnC,IAAI,WAAW,CAAA;AACf,IAAI,SAAS,CAAA;AAEb,IAAI,sCAAsC,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,YAAY,CAAC,KAAK,GAAG,EAAE,GAAG,EAAE,CAAA;AAC9B,CAAC;KAAM,CAAC;IACN,YAAY,CAAC,GAAG,GAAG,EAAE,CAAA;IACrB,WAAW,GAAG,KAAK,CAAA;IAEnB,SAAS,GAAG;QACV,GAAG,EAAE,eAAe;QACpB,MAAM,EAAE,UAAU;KACnB,CAAA;AACH,CAAC;AAED,MAAM,CAAC,MAAM,iBAAiB,GAAG,GAAG,oCAAoC,IAAI,4CAA4C,EAAE,CAAA;AAE1H,IAAI,iBAA8B,CAAA;AAElC,IAAI,gBAAkC,CAAA;AAEtC,MAAM,CAAC,MAAM,OAAO,GAAG,KAAK,IAAI,EAAE;IAChC,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACvB,iBAAiB,GAAG,MAAM,IAAI,WAAW,CAAC,WAAW,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,gBAAgB,CACrC,iBAAiB,EACjB;YACE,YAAY;YACZ,iBAAiB;SAClB,CACF,CAAA;IACH,CAAC;IAED,OAAO,IAAI,CAAA;AACb,CAAC,CAAA;AAED,MAAM,CAAC,MAAM,UAAU,GAAG,KAAK,IAAI,EAAE;IACnC,IAAI,iBAAiB,EAAE,CAAC;QACtB,MAAM,iBAAiB,CAAC,KAAK,EAAE,CAAA;IACjC,CAAC;AACH,CAAC,CAAA;AAED,MAAM,CAAC,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,oCAAoC,CAAC;SACxC,UAAU,CAAC,4CAA4C,CAAC;SACxD,OAAO,CAAC;QACP,WAAW,EAAE,2BAA2B;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,2BAA2B,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;AAED,MAAM,CAAC,MAAM,sBAAsB,GAAG,CAAC,eAAe,EAAE,EAAE;IACxD,MAAM,SAAS,GAAG;QAChB,CAAC,GAAG,eAAe,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,eAAe,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"}

package/dist/mongo/index.d.ts

@@ -0,0 +1,25 @@
+import Mongoose from 'mongoose';
+import * as _Config from './config';
+import * as _encryption from './encryption';
+export declare const encryption: typeof _encryption;
+export declare const mongoose: typeof Mongoose;
+export declare const ObjectId: typeof Mongoose.Types.ObjectId;
+export type ObjectId = Mongoose.Types.ObjectId;
+export type { Document } from 'mongoose';
+export { type Binary } from 'bson';
+export declare const Schema: typeof Mongoose.Schema;
+export declare const Model: Mongoose.Model<any, {}, {}, {}, any, any>;
+export declare const Config: typeof _Config;
+export declare const connected: () => boolean;
+declare const _default: {
+    encryption: typeof _encryption;
+    connect: () => Promise<void>;
+    disconnect: () => Promise<void>;
+    connected: () => boolean;
+    mongoose: typeof Mongoose;
+    ObjectId: typeof Mongoose.Types.ObjectId;
+    Schema: typeof Mongoose.Schema;
+    Model: Mongoose.Model<any, {}, {}, {}, any, any>;
+};
+export default _default;
+//# sourceMappingURL=index.d.ts.map

package/dist/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,UAAU,CAAA;AACnC,OAAO,KAAK,WAAW,MAAM,cAAc,CAAA;AAI3C,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"}

package/dist/mongo/index.js

@@ -0,0 +1,72 @@
+var _a;
+import Mongoose from 'mongoose';
+import { getLogger } from './logger';
+import * as _Config from './config';
+import * as _encryption from './encryption';
+const logger = getLogger();
+export const encryption = _encryption;
+const MONGODB_URI_MASKED = (_a = _Config.MONGODB_URI) === null || _a === void 0 ? void 0 : _a.replace(/:\/\/(.*):(.*)@/, '://***:***@');
+export const mongoose = Mongoose;
+export const ObjectId = Mongoose.Types.ObjectId;
+export const Schema = Mongoose.Schema;
+export const Model = Mongoose.Model;
+export const Config = _Config;
+Mongoose.set('debug', _Config.MONGO_DEBUG);
+Mongoose.connection.on('connected', () => {
+    logger.info(`[MONGO] Connected to ${MONGODB_URI_MASKED}`);
+});
+Mongoose.connection.on('error', error => {
+    logger.error(error, `[MONGO] Connection to MongoDB failed: ${error.message}`);
+});
+Mongoose.connection.on('reconnected', () => {
+    logger.info(`[MONGO] Reconnected to ${MONGODB_URI_MASKED}`);
+});
+Mongoose.connection.on('disconnected', () => {
+    logger.info(`[MONGO] Disconnected from ${MONGODB_URI_MASKED}`);
+});
+Mongoose.connection.on('close', () => {
+    logger.info(`[MONGO] Connection closed ${MONGODB_URI_MASKED}`);
+});
+Mongoose.connection.on('reconnectFailed', () => {
+    logger.error(`[MONGO] Reconnect Failed ${MONGODB_URI_MASKED}`);
+});
+const connect = async () => {
+    await encryption.connect();
+    const encryptionKeyId = await encryption.getKeyId();
+    const options = {
+        autoIndex: true,
+        connectTimeoutMS: 10000,
+        socketTimeoutMS: 25000,
+        minPoolSize: 3,
+        autoEncryption: {
+            bypassAutoEncryption: true,
+            keyVaultNamespace: encryption.keyVaultNamespace,
+            kmsProviders: encryption.kmsProviders,
+            schemaMap: encryption.getEncryptionSchemaMap(encryptionKeyId),
+            extraOptions: {
+                mongocryptdURI: Config.MONGODB_URI,
+            },
+        },
+    };
+    await Mongoose.connect(_Config.MONGODB_URI, options);
+};
+export const connected = () => Mongoose.connection.readyState === 1;
+const disconnect = async () => {
+    if (!connected())
+        return;
+    await Promise.allSettled([
+        Mongoose.connection.close(),
+        encryption.disconnect(),
+    ]);
+};
+export default {
+    encryption,
+    connect,
+    disconnect,
+    connected,
+    mongoose,
+    ObjectId: Mongoose.Types.ObjectId,
+    Schema: Mongoose.Schema,
+    Model: Mongoose.Model,
+};
+//# sourceMappingURL=index.js.map

package/dist/mongo/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/mongo/index.ts"],"names":[],"mappings":";AAAA,OAAO,QAA4B,MAAM,UAAU,CAAA;AACnD,OAAO,EAAE,SAAS,EAAE,MAAM,UAAU,CAAA;AACpC,OAAO,KAAK,OAAO,MAAM,UAAU,CAAA;AACnC,OAAO,KAAK,WAAW,MAAM,cAAc,CAAA;AAE3C,MAAM,MAAM,GAAG,SAAS,EAAE,CAAA;AAE1B,MAAM,CAAC,MAAM,UAAU,GAAG,WAAW,CAAA;AAErC,MAAM,kBAAkB,GAAG,MAAA,OAAO,CAAC,WAAW,0CAAE,OAAO,CAAC,iBAAiB,EAAE,aAAa,CAAC,CAAA;AAEzF,MAAM,CAAC,MAAM,QAAQ,GAAG,QAAQ,CAAA;AAEhC,MAAM,CAAC,MAAM,QAAQ,GAAG,QAAQ,CAAC,KAAK,CAAC,QAAQ,CAAA;AAQ/C,MAAM,CAAC,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAA;AAErC,MAAM,CAAC,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAA;AAEnC,MAAM,CAAC,MAAM,MAAM,GAAG,OAAO,CAAA;AAE7B,QAAQ,CAAC,GAAG,CAAC,OAAO,EAAE,OAAO,CAAC,WAAW,CAAC,CAAA;AAE1C,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC,WAAW,EAAE,GAAG,EAAE;IACvC,MAAM,CAAC,IAAI,CAAC,wBAAwB,kBAAkB,EAAE,CAAC,CAAA;AAC3D,CAAC,CAAC,CAAA;AAEF,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC,OAAO,EAAE,KAAK,CAAC,EAAE;IACtC,MAAM,CAAC,KAAK,CAAC,KAAK,EAAE,yCAAyC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAA;AAC/E,CAAC,CAAC,CAAA;AAEF,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC,aAAa,EAAE,GAAG,EAAE;IACzC,MAAM,CAAC,IAAI,CAAC,0BAA0B,kBAAkB,EAAE,CAAC,CAAA;AAC7D,CAAC,CAAC,CAAA;AAEF,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC,cAAc,EAAE,GAAG,EAAE;IAC1C,MAAM,CAAC,IAAI,CAAC,6BAA6B,kBAAkB,EAAE,CAAC,CAAA;AAChE,CAAC,CAAC,CAAA;AAEF,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;IACnC,MAAM,CAAC,IAAI,CAAC,6BAA6B,kBAAkB,EAAE,CAAC,CAAA;AAChE,CAAC,CAAC,CAAA;AAEF,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC7C,MAAM,CAAC,KAAK,CAAC,4BAA4B,kBAAkB,EAAE,CAAC,CAAA;AAChE,CAAC,CAAC,CAAA;AAEF,MAAM,OAAO,GAAG,KAAK,IAAI,EAAE;IACzB,MAAM,UAAU,CAAC,OAAO,EAAE,CAAA;IAE1B,MAAM,eAAe,GAAG,MAAM,UAAU,CAAC,QAAQ,EAAE,CAAA;IAEnD,MAAM,OAAO,GAAmB;QAC9B,SAAS,EAAE,IAAI;QACf,gBAAgB,EAAE,KAAK;QACvB,eAAe,EAAE,KAAK;QACtB,WAAW,EAAE,CAAC;QACd,cAAc,EAAE;YACd,oBAAoB,EAAE,IAAI;YAC1B,iBAAiB,EAAE,UAAU,CAAC,iBAAiB;YAC/C,YAAY,EAAE,UAAU,CAAC,YAAY;YACrC,SAAS,EAAE,UAAU,CAAC,sBAAsB,CAAC,eAAe,CAAC;YAC7D,YAAY,EAAE;gBACZ,cAAc,EAAE,MAAM,CAAC,WAAW;aACnC;SACF;KACF,CAAA;IACD,MAAM,QAAQ,CAAC,OAAO,CAAC,OAAO,CAAC,WAAW,EAAE,OAAO,CAAC,CAAA;AACtD,CAAC,CAAA;AAED,MAAM,CAAC,MAAM,SAAS,GAAG,GAAG,EAAE,CAAC,QAAQ,CAAC,UAAU,CAAC,UAAU,KAAK,CAAC,CAAA;AAEnE,MAAM,UAAU,GAAG,KAAK,IAAI,EAAE;IAC5B,IAAI,CAAC,SAAS,EAAE;QAAE,OAAM;IACxB,MAAM,OAAO,CAAC,UAAU,CAAC;QACvB,QAAQ,CAAC,UAAU,CAAC,KAAK,EAAE;QAC3B,UAAU,CAAC,UAAU,EAAE;KACxB,CAAC,CAAA;AACJ,CAAC,CAAA;AAED,eAAe;IACb,UAAU;IACV,OAAO;IACP,UAAU;IACV,SAAS;IACT,QAAQ;IACR,QAAQ,EAAE,QAAQ,CAAC,KAAK,CAAC,QAAQ;IACjC,MAAM,EAAE,QAAQ,CAAC,MAAM;IACvB,KAAK,EAAE,QAAQ,CAAC,KAAK;CACtB,CAAA"}

package/dist/mongo/logger.d.ts

@@ -0,0 +1,31 @@
+/**
+ * Logger interface for MongoDB operations.
+ * Compatible with most common Node.js loggers (pino, winston, bunyan, etc.)
+ */
+export interface MongoLogger {
+    info(message: string): void;
+    error(errorOrMessage: Error | unknown | string, message?: string): void;
+    warn?(message: string): void;
+    debug?(message: string): void;
+}
+/**
+ * Sets a custom logger for MongoDB operations.
+ * If not called, all logging is silently discarded.
+ *
+ * @param customLogger - Logger instance implementing MongoLogger interface
+ *
+ * @example
+ * ```ts
+ * import { setLogger } from '@multiplayer-app/ai-agent-mongo';
+ * import pino from 'pino';
+ *
+ * setLogger(pino());
+ * ```
+ */
+export declare function setLogger(customLogger: MongoLogger): void;
+/**
+ * Gets the current logger instance.
+ * @internal
+ */
+export declare function getLogger(): MongoLogger;
+//# sourceMappingURL=logger.d.ts.map

package/dist/mongo/logger.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"logger.d.ts","sourceRoot":"","sources":["../../src/mongo/logger.ts"],"names":[],"mappings":"AAAA;;;GAGG;AACH,MAAM,WAAW,WAAW;IAC1B,IAAI,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;IAC5B,KAAK,CAAC,cAAc,EAAE,KAAK,GAAG,OAAO,GAAG,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACxE,IAAI,CAAC,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;IAC7B,KAAK,CAAC,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;CAC/B;AAeD;;;;;;;;;;;;;GAaG;AACH,wBAAgB,SAAS,CAAC,YAAY,EAAE,WAAW,GAAG,IAAI,CAEzD;AAED;;;GAGG;AACH,wBAAgB,SAAS,IAAI,WAAW,CAEvC"}

package/dist/mongo/logger.js

@@ -0,0 +1,36 @@
+/**
+ * No-op logger that discards all log messages.
+ * Used as default when no logger is provided.
+ */
+const noOpLogger = {
+    info: () => { },
+    error: () => { },
+    warn: () => { },
+    debug: () => { },
+};
+let logger = noOpLogger;
+/**
+ * Sets a custom logger for MongoDB operations.
+ * If not called, all logging is silently discarded.
+ *
+ * @param customLogger - Logger instance implementing MongoLogger interface
+ *
+ * @example
+ * ```ts
+ * import { setLogger } from '@multiplayer-app/ai-agent-mongo';
+ * import pino from 'pino';
+ *
+ * setLogger(pino());
+ * ```
+ */
+export function setLogger(customLogger) {
+    logger = customLogger;
+}
+/**
+ * Gets the current logger instance.
+ * @internal
+ */
+export function getLogger() {
+    return logger;
+}
+//# sourceMappingURL=logger.js.map

package/dist/mongo/logger.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"logger.js","sourceRoot":"","sources":["../../src/mongo/logger.ts"],"names":[],"mappings":"AAWA;;;GAGG;AACH,MAAM,UAAU,GAAgB;IAC9B,IAAI,EAAE,GAAG,EAAE,GAAE,CAAC;IACd,KAAK,EAAE,GAAG,EAAE,GAAE,CAAC;IACf,IAAI,EAAE,GAAG,EAAE,GAAE,CAAC;IACd,KAAK,EAAE,GAAG,EAAE,GAAE,CAAC;CAChB,CAAC;AAEF,IAAI,MAAM,GAAgB,UAAU,CAAC;AAErC;;;;;;;;;;;;;GAaG;AACH,MAAM,UAAU,SAAS,CAAC,YAAyB;IACjD,MAAM,GAAG,YAAY,CAAC;AACxB,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,SAAS;IACvB,OAAO,MAAM,CAAC;AAChB,CAAC"}

package/dist/repositories/MongoAgentChatRepository.d.ts

@@ -0,0 +1,37 @@
+import type { AgentChat, AgentChatResponse } from '@multiplayer-app/ai-agent-types';
+import { SortOrder } from '@multiplayer-app/ai-agent-types';
+import type { AgentChatRepository } from '@multiplayer-app/ai-agent-db';
+/**
+ * MongoDB implementation of AgentChatRepository
+ */
+export declare class MongoAgentChatRepository implements AgentChatRepository {
+    findById(id: string): Promise<AgentChat | null>;
+    find(filter?: Partial<AgentChat>, options?: {
+        sort?: {
+            field: string;
+            order: SortOrder;
+        };
+        limit?: number;
+    }): Promise<AgentChat[]>;
+    findOne(filter: Partial<AgentChat>): Promise<AgentChat | null>;
+    create(entity: Omit<AgentChat, 'id' | 'createdAt' | 'updatedAt'>): Promise<AgentChat>;
+    update(id: string, updates: Partial<Omit<AgentChat, 'id' | 'createdAt'>>): Promise<AgentChat | null>;
+    delete(id: string): Promise<boolean>;
+    exists(id: string): Promise<boolean>;
+    count(filter?: Partial<AgentChat>): Promise<number>;
+    findByUserId(userId: string): Promise<AgentChat[]>;
+    findByContextKey(contextKey: string): Promise<AgentChat[]>;
+    findByUserIdAndContextKey(userId: string, contextKey: string): Promise<AgentChat[]>;
+    findGuestChatsByContextKey(contextKey: string): Promise<AgentChat[]>;
+    updateTitle(id: string, title: string): Promise<AgentChat | null>;
+    deleteByUserId(userId: string): Promise<number>;
+    deleteByContextKey(contextKey: string): Promise<number>;
+    findWithMessages(filter?: Partial<AgentChat>, options?: {
+        sort?: {
+            field: string;
+            order: SortOrder;
+        };
+        limit?: number;
+    }): Promise<AgentChatResponse[]>;
+}
+//# sourceMappingURL=MongoAgentChatRepository.d.ts.map

package/dist/repositories/MongoAgentChatRepository.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"MongoAgentChatRepository.d.ts","sourceRoot":"","sources":["../../src/repositories/MongoAgentChatRepository.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,iBAAiB,EAAE,MAAM,iCAAiC,CAAC;AACpF,OAAO,EAAY,SAAS,EAAE,MAAM,iCAAiC,CAAC;AACtE,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,8BAA8B,CAAC;AA0BxE;;GAEG;AACH,qBAAa,wBAAyB,YAAW,mBAAmB;IAC5D,QAAQ,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC;IAK/C,IAAI,CACR,MAAM,CAAC,EAAE,OAAO,CAAC,SAAS,CAAC,EAC3B,OAAO,CAAC,EAAE;QAAE,IAAI,CAAC,EAAE;YAAE,KAAK,EAAE,MAAM,CAAC;YAAC,KAAK,EAAE,SAAS,CAAA;SAAE,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE,GACvE,OAAO,CAAC,SAAS,EAAE,CAAC;IA4BjB,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,SAAS,CAAC,GAAG,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC;IAa9D,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,SAAS,EAAE,IAAI,GAAG,WAAW,GAAG,WAAW,CAAC,GAAG,OAAO,CAAC,SAAS,CAAC;IAgBrF,MAAM,CAAC,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,GAAG,WAAW,CAAC,CAAC,GAAG,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC;IAUpG,MAAM,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAUpC,MAAM,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAKpC,KAAK,CAAC,MAAM,CAAC,EAAE,OAAO,CAAC,SAAS,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;IAcnD,YAAY,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;IAKlD,gBAAgB,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;IAK1D,yBAAyB,CAAC,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;IAKnF,0BAA0B,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;IAQpE,WAAW,CAAC,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC;IAUjE,cAAc,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAK/C,kBAAkB,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAKvD,gBAAgB,CACpB,MAAM,CAAC,EAAE,OAAO,CAAC,SAAS,CAAC,EAC3B,OAAO,CAAC,EAAE;QAAE,IAAI,CAAC,EAAE;YAAE,KAAK,EAAE,MAAM,CAAC;YAAC,KAAK,EAAE,SAAS,CAAA;SAAE,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE,GACvE,OAAO,CAAC,iBAAiB,EAAE,CAAC;CA2GhC"}

package/dist/repositories/MongoAgentChatRepository.js

@@ -0,0 +1,235 @@
+import { ChatType, SortOrder } from '@multiplayer-app/ai-agent-types';
+import { AgentChatModel } from '../schemas/AgentChatSchema';
+import { AgentMessageModel } from '../schemas/AgentMessageSchema';
+import { ObjectId } from '../mongo';
+/**
+ * Helper function to convert Mongoose document to AgentChat type
+ */
+function toAgentChat(doc) {
+    if (!doc)
+        return null;
+    const chat = doc.toObject ? doc.toObject() : doc;
+    return {
+        id: doc._id.toString(),
+        title: chat.title,
+        type: chat.type || ChatType.Chat,
+        status: chat.status,
+        contextKey: chat.contextKey,
+        model: chat.model,
+        createdAt: chat.createdAt,
+        updatedAt: chat.updatedAt,
+        metadata: chat.metadata,
+        userId: chat.userId
+    };
+}
+/**
+ * MongoDB implementation of AgentChatRepository
+ */
+export class MongoAgentChatRepository {
+    async findById(id) {
+        const doc = await AgentChatModel.findById(id).exec();
+        return toAgentChat(doc);
+    }
+    async find(filter, options) {
+        const query = {};
+        if (filter) {
+            if (filter.id) {
+                query._id = filter.id;
+                delete filter.id;
+            }
+            Object.assign(query, filter);
+        }
+        let queryBuilder = AgentChatModel.find(query);
+        // Apply sorting
+        if (options === null || options === void 0 ? void 0 : options.sort) {
+            const sortOrder = options.sort.order === SortOrder.Asc ? 1 : -1;
+            queryBuilder = queryBuilder.sort({ [options.sort.field]: sortOrder });
+        }
+        // Apply limit
+        if (options === null || options === void 0 ? void 0 : options.limit) {
+            queryBuilder = queryBuilder.limit(options.limit);
+        }
+        const docs = await queryBuilder.exec();
+        return docs.map(doc => toAgentChat(doc)).filter(Boolean);
+    }
+    async findOne(filter) {
+        const query = {};
+        if (filter.id) {
+            query._id = filter.id;
+            delete filter.id;
+        }
+        Object.assign(query, filter);
+        const doc = await AgentChatModel.findOne(query).exec();
+        return toAgentChat(doc);
+    }
+    async create(entity) {
+        const now = new Date().toISOString();
+        const doc = new AgentChatModel({
+            ...entity,
+            createdAt: now,
+            updatedAt: now,
+        });
+        const saved = await doc.save();
+        const result = toAgentChat(saved);
+        if (!result) {
+            throw new Error('Failed to create chat');
+        }
+        return result;
+    }
+    async update(id, updates) {
+        const doc = await AgentChatModel.findByIdAndUpdate(id, { $set: { ...updates, updatedAt: new Date().toISOString() } }, { new: true }).exec();
+        return toAgentChat(doc);
+    }
+    async delete(id) {
+        // Delete all related messages first
+        await AgentMessageModel.deleteMany({ chat: id }).exec();
+        // Then delete the chat
+        const result = await AgentChatModel.findByIdAndDelete(id).exec();
+        return !!result;
+    }
+    async exists(id) {
+        const count = await AgentChatModel.countDocuments({ _id: id }).exec();
+        return count > 0;
+    }
+    async count(filter) {
+        const query = {};
+        if (filter) {
+            if (filter.id) {
+                query._id = filter.id;
+                delete filter.id;
+            }
+            Object.assign(query, filter);
+        }
+        return AgentChatModel.countDocuments(query).exec();
+    }
+    async findByUserId(userId) {
+        const docs = await AgentChatModel.find({ userId }).exec();
+        return docs.map(doc => toAgentChat(doc)).filter(Boolean);
+    }
+    async findByContextKey(contextKey) {
+        const docs = await AgentChatModel.find({ contextKey }).exec();
+        return docs.map(doc => toAgentChat(doc)).filter(Boolean);
+    }
+    async findByUserIdAndContextKey(userId, contextKey) {
+        const docs = await AgentChatModel.find({ userId, contextKey }).exec();
+        return docs.map(doc => toAgentChat(doc)).filter(Boolean);
+    }
+    async findGuestChatsByContextKey(contextKey) {
+        const docs = await AgentChatModel.find({
+            contextKey,
+            userId: { $exists: false }
+        }).exec();
+        return docs.map(doc => toAgentChat(doc)).filter(Boolean);
+    }
+    async updateTitle(id, title) {
+        const doc = await AgentChatModel.findByIdAndUpdate(id, { $set: { title, updatedAt: new Date().toISOString() } }, { new: true }).exec();
+        return toAgentChat(doc);
+    }
+    async deleteByUserId(userId) {
+        const result = await AgentChatModel.deleteMany({ userId }).exec();
+        return result.deletedCount || 0;
+    }
+    async deleteByContextKey(contextKey) {
+        const result = await AgentChatModel.deleteMany({ contextKey }).exec();
+        return result.deletedCount || 0;
+    }
+    async findWithMessages(filter, options) {
+        // Build match stage for filtering
+        const matchStage = {};
+        if (filter) {
+            if (filter.id) {
+                matchStage._id = new ObjectId(filter.id);
+            }
+            if (filter.userId !== undefined) {
+                matchStage.userId = filter.userId;
+            }
+            if (filter.contextKey !== undefined) {
+                matchStage.contextKey = filter.contextKey;
+            }
+            if (filter.type !== undefined) {
+                matchStage.type = filter.type;
+            }
+            if (filter.status !== undefined) {
+                matchStage.status = filter.status;
+            }
+            if (filter.model !== undefined) {
+                matchStage.model = filter.model;
+            }
+        }
+        // Build aggregation pipeline
+        const pipeline = [
+            { $match: matchStage }
+        ];
+        // Add sort stage before lookup (for better performance)
+        if (options === null || options === void 0 ? void 0 : options.sort) {
+            const sortField = options.sort.field === 'id' ? '_id' : options.sort.field;
+            const sortOrder = options.sort.order === SortOrder.Asc ? 1 : -1;
+            pipeline.push({ $sort: { [sortField]: sortOrder } });
+        }
+        // Add limit before lookup if specified (for better performance)
+        if (options === null || options === void 0 ? void 0 : options.limit) {
+            pipeline.push({ $limit: options.limit });
+        }
+        // Lookup messages
+        pipeline.push({
+            $lookup: {
+                from: 'agentmessages',
+                localField: '_id',
+                foreignField: 'chat',
+                as: 'messages'
+            }
+        });
+        // Transform messages
+        pipeline.push({
+            $addFields: {
+                messages: {
+                    $map: {
+                        input: { $sortArray: { input: '$messages', sortBy: { createdAt: 1 } } },
+                        as: 'msg',
+                        in: {
+                            id: { $toString: '$$msg._id' },
+                            chat: { $toString: '$$msg.chat' },
+                            role: '$$msg.role',
+                            content: '$$msg.content',
+                            createdAt: '$$msg.createdAt',
+                            toolCalls: { $ifNull: ['$$msg.toolCalls', []] },
+                            attachments: { $ifNull: ['$$msg.attachments', []] },
+                            annotations: '$$msg.annotations'
+                        }
+                    }
+                }
+            }
+        });
+        // Project final shape
+        pipeline.push({
+            $project: {
+                id: { $toString: '$_id' },
+                title: 1,
+                type: { $ifNull: ['$type', ChatType.Chat] },
+                status: 1,
+                contextKey: 1,
+                model: 1,
+                createdAt: 1,
+                updatedAt: 1,
+                metadata: 1,
+                userId: 1,
+                messages: 1
+            }
+        });
+        const results = await AgentChatModel.aggregate(pipeline).exec();
+        return results.map((result) => ({
+            id: result.id,
+            title: result.title,
+            type: result.type || ChatType.Chat,
+            status: result.status,
+            contextKey: result.contextKey,
+            model: result.model,
+            createdAt: result.createdAt,
+            updatedAt: result.updatedAt,
+            metadata: result.metadata,
+            userId: result.userId,
+            messages: result.messages || []
+        }));
+    }
+}
+//# sourceMappingURL=MongoAgentChatRepository.js.map

package/dist/repositories/MongoAgentChatRepository.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"MongoAgentChatRepository.js","sourceRoot":"","sources":["../../src/repositories/MongoAgentChatRepository.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,iCAAiC,CAAC;AAEtE,OAAO,EAAE,cAAc,EAA0B,MAAM,4BAA4B,CAAC;AACpF,OAAO,EAAE,iBAAiB,EAAE,MAAM,+BAA+B,CAAC;AAClE,OAAO,EAAE,QAAQ,EAAE,MAAM,UAAU,CAAC;AAEpC;;GAEG;AACH,SAAS,WAAW,CAAC,GAA6B;IAChD,IAAI,CAAC,GAAG;QAAE,OAAO,IAAI,CAAC;IAEtB,MAAM,IAAI,GAAI,GAAW,CAAC,QAAQ,CAAC,CAAC,CAAE,GAAW,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC;IACnE,OAAO;QACL,EAAE,EAAE,GAAG,CAAC,GAAG,CAAC,QAAQ,EAAE;QACtB,KAAK,EAAE,IAAI,CAAC,KAAK;QACjB,IAAI,EAAE,IAAI,CAAC,IAAI,IAAI,QAAQ,CAAC,IAAI;QAChC,MAAM,EAAE,IAAI,CAAC,MAAM;QACnB,UAAU,EAAE,IAAI,CAAC,UAAU;QAC3B,KAAK,EAAE,IAAI,CAAC,KAAK;QACjB,SAAS,EAAE,IAAI,CAAC,SAAS;QACzB,SAAS,EAAE,IAAI,CAAC,SAAS;QACzB,QAAQ,EAAE,IAAI,CAAC,QAAQ;QACvB,MAAM,EAAE,IAAI,CAAC,MAAM;KACpB,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,OAAO,wBAAwB;IACnC,KAAK,CAAC,QAAQ,CAAC,EAAU;QACvB,MAAM,GAAG,GAAG,MAAM,cAAc,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QACrD,OAAO,WAAW,CAAC,GAAG,CAAC,CAAC;IAC1B,CAAC;IAED,KAAK,CAAC,IAAI,CACR,MAA2B,EAC3B,OAAwE;QAExE,MAAM,KAAK,GAAQ,EAAE,CAAC;QAEtB,IAAI,MAAM,EAAE,CAAC;YACX,IAAI,MAAM,CAAC,EAAE,EAAE,CAAC;gBACd,KAAK,CAAC,GAAG,GAAG,MAAM,CAAC,EAAE,CAAC;gBACtB,OAAO,MAAM,CAAC,EAAE,CAAC;YACnB,CAAC;YACD,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QAC/B,CAAC;QAED,IAAI,YAAY,GAAG,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAE9C,gBAAgB;QAChB,IAAI,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,IAAI,EAAE,CAAC;YAClB,MAAM,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC,KAAK,KAAK,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAChE,YAAY,GAAG,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC;QACxE,CAAC;QAED,cAAc;QACd,IAAI,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,KAAK,EAAE,CAAC;YACnB,YAAY,GAAG,YAAY,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACnD,CAAC;QAED,MAAM,IAAI,GAAG,MAAM,YAAY,CAAC,IAAI,EAAE,CAAC;QACvC,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,WAAW,CAAC,GAAG,CAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAC5D,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAA0B;QACtC,MAAM,KAAK,GAAQ,EAAE,CAAC;QAEtB,IAAI,MAAM,CAAC,EAAE,EAAE,CAAC;YACd,KAAK,CAAC,GAAG,GAAG,MAAM,CAAC,EAAE,CAAC;YACtB,OAAO,MAAM,CAAC,EAAE,CAAC;QACnB,CAAC;QACD,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QAE7B,MAAM,GAAG,GAAG,MAAM,cAAc,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,CAAC;QACvD,OAAO,WAAW,CAAC,GAAG,CAAC,CAAC;IAC1B,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,MAAyD;QACpE,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QACrC,MAAM,GAAG,GAAG,IAAI,cAAc,CAAC;YAC7B,GAAG,MAAM;YACT,SAAS,EAAE,GAAG;YACd,SAAS,EAAE,GAAG;SACf,CAAC,CAAC;QAEH,MAAM,KAAK,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC;QAC/B,MAAM,MAAM,GAAG,WAAW,CAAC,KAAK,CAAC,CAAC;QAClC,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;QAC3C,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,EAAU,EAAE,OAAqD;QAC5E,MAAM,GAAG,GAAG,MAAM,cAAc,CAAC,iBAAiB,CAChD,EAAE,EACF,EAAE,IAAI,EAAE,EAAE,GAAG,OAAO,EAAE,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,EAAE,EAAE,EAC7D,EAAE,GAAG,EAAE,IAAI,EAAE,CACd,CAAC,IAAI,EAAE,CAAC;QAET,OAAO,WAAW,CAAC,GAAG,CAAC,CAAC;IAC1B,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,EAAU;QACrB,oCAAoC;QACpC,MAAM,iBAAiB,CAAC,UAAU,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QAExD,uBAAuB;QACvB,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QAEjE,OAAO,CAAC,CAAC,MAAM,CAAC;IAClB,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,EAAU;QACrB,MAAM,KAAK,GAAG,MAAM,cAAc,CAAC,cAAc,CAAC,EAAE,GAAG,EAAE,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QACtE,OAAO,KAAK,GAAG,CAAC,CAAC;IACnB,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,MAA2B;QACrC,MAAM,KAAK,GAAQ,EAAE,CAAC;QAEtB,IAAI,MAAM,EAAE,CAAC;YACX,IAAI,MAAM,CAAC,EAAE,EAAE,CAAC;gBACd,KAAK,CAAC,GAAG,GAAG,MAAM,CAAC,EAAE,CAAC;gBACtB,OAAO,MAAM,CAAC,EAAE,CAAC;YACnB,CAAC;YACD,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QAC/B,CAAC;QAED,OAAO,cAAc,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,CAAC;IACrD,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,MAAc;QAC/B,MAAM,IAAI,GAAG,MAAM,cAAc,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QAC1D,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,WAAW,CAAC,GAAG,CAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAC5D,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,UAAkB;QACvC,MAAM,IAAI,GAAG,MAAM,cAAc,CAAC,IAAI,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QAC9D,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,WAAW,CAAC,GAAG,CAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAC5D,CAAC;IAED,KAAK,CAAC,yBAAyB,CAAC,MAAc,EAAE,UAAkB;QAChE,MAAM,IAAI,GAAG,MAAM,cAAc,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QACtE,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,WAAW,CAAC,GAAG,CAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAC5D,CAAC;IAED,KAAK,CAAC,0BAA0B,CAAC,UAAkB;QACjD,MAAM,IAAI,GAAG,MAAM,cAAc,CAAC,IAAI,CAAC;YACrC,UAAU;YACV,MAAM,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE;SAC3B,CAAC,CAAC,IAAI,EAAE,CAAC;QACV,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,WAAW,CAAC,GAAG,CAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAC5D,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,EAAU,EAAE,KAAa;QACzC,MAAM,GAAG,GAAG,MAAM,cAAc,CAAC,iBAAiB,CAChD,EAAE,EACF,EAAE,IAAI,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,EAAE,EAAE,EACxD,EAAE,GAAG,EAAE,IAAI,EAAE,CACd,CAAC,IAAI,EAAE,CAAC;QAET,OAAO,WAAW,CAAC,GAAG,CAAC,CAAC;IAC1B,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,MAAc;QACjC,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,UAAU,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QAClE,OAAO,MAAM,CAAC,YAAY,IAAI,CAAC,CAAC;IAClC,CAAC;IAED,KAAK,CAAC,kBAAkB,CAAC,UAAkB;QACzC,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,UAAU,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QACtE,OAAO,MAAM,CAAC,YAAY,IAAI,CAAC,CAAC;IAClC,CAAC;IAED,KAAK,CAAC,gBAAgB,CACpB,MAA2B,EAC3B,OAAwE;QAExE,kCAAkC;QAClC,MAAM,UAAU,GAAQ,EAAE,CAAC;QAE3B,IAAI,MAAM,EAAE,CAAC;YACX,IAAI,MAAM,CAAC,EAAE,EAAE,CAAC;gBACd,UAAU,CAAC,GAAG,GAAG,IAAI,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YAC3C,CAAC;YACD,IAAI,MAAM,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;gBAChC,UAAU,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;YACpC,CAAC;YACD,IAAI,MAAM,CAAC,UAAU,KAAK,SAAS,EAAE,CAAC;gBACpC,UAAU,CAAC,UAAU,GAAG,MAAM,CAAC,UAAU,CAAC;YAC5C,CAAC;YACD,IAAI,MAAM,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;gBAC9B,UAAU,CAAC,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC;YAChC,CAAC;YACD,IAAI,MAAM,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;gBAChC,UAAU,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;YACpC,CAAC;YACD,IAAI,MAAM,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;gBAC/B,UAAU,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;YAClC,CAAC;QACH,CAAC;QAED,6BAA6B;QAC7B,MAAM,QAAQ,GAAU;YACtB,EAAE,MAAM,EAAE,UAAU,EAAE;SACvB,CAAC;QAEF,wDAAwD;QACxD,IAAI,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,IAAI,EAAE,CAAC;YAClB,MAAM,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC,KAAK,KAAK,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC;YAC3E,MAAM,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC,KAAK,KAAK,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAChE,QAAQ,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,EAAE,CAAC,SAAS,CAAC,EAAE,SAAS,EAAE,EAAE,CAAC,CAAC;QACvD,CAAC;QAED,gEAAgE;QAChE,IAAI,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,KAAK,EAAE,CAAC;YACnB,QAAQ,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC;QAC3C,CAAC;QAED,kBAAkB;QAClB,QAAQ,CAAC,IAAI,CAAC;YACZ,OAAO,EAAE;gBACP,IAAI,EAAE,eAAe;gBACrB,UAAU,EAAE,KAAK;gBACjB,YAAY,EAAE,MAAM;gBACpB,EAAE,EAAE,UAAU;aACf;SACF,CAAC,CAAC;QAEH,qBAAqB;QACrB,QAAQ,CAAC,IAAI,CAAC;YACZ,UAAU,EAAE;gBACV,QAAQ,EAAE;oBACR,IAAI,EAAE;wBACJ,KAAK,EAAE,EAAE,UAAU,EAAE,EAAE,KAAK,EAAE,WAAW,EAAE,MAAM,EAAE,EAAE,SAAS,EAAE,CAAC,EAAE,EAAE,EAAE;wBACvE,EAAE,EAAE,KAAK;wBACT,EAAE,EAAE;4BACF,EAAE,EAAE,EAAE,SAAS,EAAE,WAAW,EAAE;4BAC9B,IAAI,EAAE,EAAE,SAAS,EAAE,YAAY,EAAE;4BACjC,IAAI,EAAE,YAAY;4BAClB,OAAO,EAAE,eAAe;4BACxB,SAAS,EAAE,iBAAiB;4BAC5B,SAAS,EAAE,EAAE,OAAO,EAAE,CAAC,iBAAiB,EAAE,EAAE,CAAC,EAAE;4BAC/C,WAAW,EAAE,EAAE,OAAO,EAAE,CAAC,mBAAmB,EAAE,EAAE,CAAC,EAAE;4BACnD,WAAW,EAAE,mBAAmB;yBACjC;qBACF;iBACF;aACF;SACF,CAAC,CAAC;QAEH,sBAAsB;QACtB,QAAQ,CAAC,IAAI,CAAC;YACZ,QAAQ,EAAE;gBACR,EAAE,EAAE,EAAE,SAAS,EAAE,MAAM,EAAE;gBACzB,KAAK,EAAE,CAAC;gBACR,IAAI,EAAE,EAAE,OAAO,EAAE,CAAC,OAAO,EAAE,QAAQ,CAAC,IAAI,CAAC,EAAE;gBAC3C,MAAM,EAAE,CAAC;gBACT,UAAU,EAAE,CAAC;gBACb,KAAK,EAAE,CAAC;gBACR,SAAS,EAAE,CAAC;gBACZ,SAAS,EAAE,CAAC;gBACZ,QAAQ,EAAE,CAAC;gBACX,MAAM,EAAE,CAAC;gBACT,QAAQ,EAAE,CAAC;aACZ;SACF,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,MAAM,cAAc,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,IAAI,EAAE,CAAC;QAEhE,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;YAC9B,EAAE,EAAE,MAAM,CAAC,EAAE;YACb,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,IAAI,EAAE,MAAM,CAAC,IAAI,IAAI,QAAQ,CAAC,IAAI;YAClC,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,UAAU,EAAE,MAAM,CAAC,UAAU;YAC7B,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,QAAQ,EAAE,MAAM,CAAC,QAAQ,IAAI,EAAE;SAChC,CAAC,CAAC,CAAC;IACN,CAAC;CACF"}

package/dist/repositories/MongoAgentMessageRepository.d.ts

@@ -0,0 +1,24 @@
+import type { AgentMessage, MessageRole } from '@multiplayer-app/ai-agent-types';
+import type { AgentMessageRepository } from '@multiplayer-app/ai-agent-db';
+/**
+ * MongoDB implementation of AgentMessageRepository
+ */
+export declare class MongoAgentMessageRepository implements AgentMessageRepository {
+    findById(id: string): Promise<AgentMessage | null>;
+    find(filter?: Partial<AgentMessage>): Promise<AgentMessage[]>;
+    findOne(filter: Partial<AgentMessage>): Promise<AgentMessage | null>;
+    create(entity: Omit<AgentMessage, 'id' | 'createdAt'>, id?: string): Promise<AgentMessage>;
+    update(id: string, updates: Partial<Omit<AgentMessage, 'id' | 'createdAt'>>): Promise<AgentMessage | null>;
+    delete(id: string): Promise<boolean>;
+    exists(id: string): Promise<boolean>;
+    count(filter?: Partial<AgentMessage>): Promise<number>;
+    findByChatId(chatId: string): Promise<AgentMessage[]>;
+    findByRole(role: MessageRole): Promise<AgentMessage[]>;
+    findByChatIdAndRole(chatId: string, role?: MessageRole): Promise<AgentMessage[]>;
+    findWithToolCalls(chatId?: string): Promise<AgentMessage[]>;
+    findWithAttachments(chatId?: string): Promise<AgentMessage[]>;
+    deleteByChatId(chatId: string): Promise<number>;
+    deleteByRole(role: MessageRole): Promise<number>;
+    deleteByChatIdAndRole(chatId: string, role: MessageRole): Promise<number>;
+}
+//# sourceMappingURL=MongoAgentMessageRepository.d.ts.map

package/dist/repositories/MongoAgentMessageRepository.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"MongoAgentMessageRepository.d.ts","sourceRoot":"","sources":["../../src/repositories/MongoAgentMessageRepository.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,WAAW,EAAE,MAAM,iCAAiC,CAAC;AACjF,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,8BAA8B,CAAC;AA0B3E;;GAEG;AACH,qBAAa,2BAA4B,YAAW,sBAAsB;IAClE,QAAQ,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,GAAG,IAAI,CAAC;IAKlD,IAAI,CAAC,MAAM,CAAC,EAAE,OAAO,CAAC,YAAY,CAAC,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;IAe7D,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,YAAY,CAAC,GAAG,OAAO,CAAC,YAAY,GAAG,IAAI,CAAC;IAapE,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,YAAY,EAAE,IAAI,GAAG,WAAW,CAAC,EAAE,EAAE,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC;IAgB1F,MAAM,CAAC,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,IAAI,CAAC,YAAY,EAAE,IAAI,GAAG,WAAW,CAAC,CAAC,GAAG,OAAO,CAAC,YAAY,GAAG,IAAI,CAAC;IAU1G,MAAM,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAKpC,MAAM,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAKpC,KAAK,CAAC,MAAM,CAAC,EAAE,OAAO,CAAC,YAAY,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;IActD,YAAY,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;IAKrD,UAAU,CAAC,IAAI,EAAE,WAAW,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;IAKtD,mBAAmB,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;IAShF,iBAAiB,CAAC,MAAM,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;IAa3D,mBAAmB,CAAC,MAAM,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;IAa7D,cAAc,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAK/C,YAAY,CAAC,IAAI,EAAE,WAAW,GAAG,OAAO,CAAC,MAAM,CAAC;IAKhD,qBAAqB,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,WAAW,GAAG,OAAO,CAAC,MAAM,CAAC;CAIhF"}