grafio-mongo 1.0.0

package/dist/MongoStorageProvider.js

@@ -0,0 +1,417 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MongoStorageProvider = void 0;
+const mongodb_1 = require("mongodb");
+const grafio_1 = require("grafio");
+const errors_1 = require("grafio/errors");
+class MongoStorageProvider {
+    _nodes;
+    _edges;
+    _graphId;
+    _batchSize;
+    _client = null;
+    constructor(db, opts = {}) {
+        const nodesColl = opts.nodesCollection ?? 'sgdb_nodes';
+        const edgesColl = opts.edgesCollection ?? 'sgdb_edges';
+        this._nodes = db.collection(nodesColl);
+        this._edges = db.collection(edgesColl);
+        this._graphId = opts.graphId ?? 'default';
+        if (opts.batchSize !== undefined && opts.batchSize <= 0) {
+            throw new Error(`batchSize must be a positive integer, got: ${opts.batchSize}`);
+        }
+        this._batchSize = opts.batchSize ?? 1000;
+        if ('client' in db && db.client instanceof mongodb_1.MongoClient) {
+            this._client = db.client;
+        }
+    }
+    async ensureIndexes() {
+        await this._nodes.createIndex({ graphId: 1, id: 1 }, { unique: true, name: 'node_graph_id_unique' });
+        await this._nodes.createIndex({ graphId: 1, type: 1 }, { name: 'node_graph_type' });
+        await this._edges.createIndex({ graphId: 1, id: 1 }, { unique: true, name: 'edge_graph_id_unique' });
+        await this._edges.createIndex({ graphId: 1, type: 1 }, { name: 'edge_graph_type' });
+        await this._edges.createIndex({ graphId: 1, sourceId: 1, type: 1 }, { name: 'edge_graph_source_type' });
+        await this._edges.createIndex({ graphId: 1, targetId: 1, type: 1 }, { name: 'edge_graph_target_type' });
+    }
+    async clear() {
+        await Promise.all([
+            this._nodes.deleteMany({ graphId: this._graphId }),
+            this._edges.deleteMany({ graphId: this._graphId }),
+        ]);
+    }
+    async insertNode(node, transaction) {
+        const session = transaction?.context;
+        try {
+            await this._nodes.insertOne({
+                id: node.id,
+                graphId: this._graphId,
+                type: node.type,
+                properties: node.properties,
+            }, { session });
+        }
+        catch (e) {
+            if (this._isDuplicateKeyError(e))
+                throw new errors_1.NodeAlreadyExistsError(node.id);
+            throw e;
+        }
+    }
+    async deleteNode(id, transaction) {
+        const session = transaction?.context;
+        await this._nodes.deleteOne({ graphId: this._graphId, id }, { session });
+    }
+    async hasNode(id, transaction) {
+        const session = transaction?.context;
+        const doc = await this._nodes.findOne({ graphId: this._graphId, id }, { projection: { _id: 1 }, session });
+        return doc !== null;
+    }
+    async getNode(id, transaction) {
+        const session = transaction?.context;
+        const doc = await this._nodes.findOne({ graphId: this._graphId, id }, { session });
+        return doc ? this._docToNode(doc) : undefined;
+    }
+    async getAllNodes(limit, transaction) {
+        const session = transaction?.context;
+        const nodes = [];
+        const cursor = this._nodes.find({ graphId: this._graphId }, { session }).batchSize(this._batchSize);
+        if (limit)
+            cursor.limit(limit);
+        for await (const doc of cursor) {
+            nodes.push(this._docToNode(doc));
+        }
+        return nodes;
+    }
+    async getNodesByType(type, transaction) {
+        const session = transaction?.context;
+        const docs = await this._nodes.find({ graphId: this._graphId, type }, { session }).toArray();
+        return docs.map(d => this._docToNode(d));
+    }
+    async getNodesByProperty(key, value, nodeType, transaction) {
+        const session = transaction?.context;
+        const filter = {
+            graphId: this._graphId,
+            [`properties.${key}`]: value,
+        };
+        if (nodeType !== undefined) {
+            filter.type = nodeType;
+        }
+        const docs = await this._nodes.find(filter, { session }).toArray();
+        return docs.map(d => this._docToNode(d));
+    }
+    async getEdgesByProperty(key, value, edgeType, transaction) {
+        const session = transaction?.context;
+        const filter = {
+            graphId: this._graphId,
+            [`properties.${key}`]: value,
+        };
+        if (edgeType !== undefined) {
+            filter.type = edgeType;
+        }
+        const docs = await this._edges.find(filter, { session }).toArray();
+        return docs.map(d => this._docToEdge(d));
+    }
+    async insertEdge(edge, transaction) {
+        const session = transaction?.context;
+        try {
+            await this._edges.insertOne({
+                id: edge.id,
+                graphId: this._graphId,
+                sourceId: edge.sourceId,
+                targetId: edge.targetId,
+                type: edge.type,
+                properties: edge.properties,
+            }, { session });
+        }
+        catch (e) {
+            if (this._isDuplicateKeyError(e))
+                throw new errors_1.EdgeAlreadyExistsError(edge.id);
+            throw e;
+        }
+    }
+    async deleteEdge(id, transaction) {
+        const session = transaction?.context;
+        await this._edges.deleteOne({ graphId: this._graphId, id }, { session });
+    }
+    async hasEdge(id, transaction) {
+        const session = transaction?.context;
+        const doc = await this._edges.findOne({ graphId: this._graphId, id }, { projection: { _id: 1 }, session });
+        return doc !== null;
+    }
+    async getEdge(id, transaction) {
+        const session = transaction?.context;
+        const doc = await this._edges.findOne({ graphId: this._graphId, id }, { session });
+        return doc ? this._docToEdge(doc) : undefined;
+    }
+    async getAllEdges(transaction) {
+        const session = transaction?.context;
+        const edges = [];
+        const cursor = this._edges.find({ graphId: this._graphId }, { session }).batchSize(this._batchSize);
+        for await (const doc of cursor) {
+            edges.push(this._docToEdge(doc));
+        }
+        return edges;
+    }
+    async getEdgesByType(type, transaction) {
+        const session = transaction?.context;
+        const docs = await this._edges.find({ graphId: this._graphId, type }, { session }).toArray();
+        return docs.map(d => this._docToEdge(d));
+    }
+    async getEdgesBySource(nodeId, type, transaction) {
+        const session = transaction?.context;
+        const filter = { graphId: this._graphId, sourceId: nodeId };
+        if (type)
+            filter.type = type;
+        const docs = await this._edges.find(filter, { session }).toArray();
+        return docs.map(d => this._docToEdge(d));
+    }
+    async getEdgesByTarget(nodeId, type, transaction) {
+        const session = transaction?.context;
+        const filter = { graphId: this._graphId, targetId: nodeId };
+        if (type)
+            filter.type = type;
+        const docs = await this._edges.find(filter, { session }).toArray();
+        return docs.map(d => this._docToEdge(d));
+    }
+    async exportJSON() {
+        const nodes = [];
+        const edges = [];
+        const nodesCursor = this._nodes.find({ graphId: this._graphId }).batchSize(this._batchSize);
+        for await (const doc of nodesCursor) {
+            nodes.push(this._docToNode(doc));
+        }
+        const edgesCursor = this._edges.find({ graphId: this._graphId }).batchSize(this._batchSize);
+        for await (const doc of edgesCursor) {
+            edges.push(this._docToEdge(doc));
+        }
+        return {
+            graphId: this._graphId,
+            nodes,
+            edges,
+        };
+    }
+    async importJSON(data) {
+        const nodeIdSet = new Set();
+        for (const n of data.nodes) {
+            if (nodeIdSet.has(n.id))
+                throw new errors_1.NodeAlreadyExistsError(n.id);
+            nodeIdSet.add(n.id);
+        }
+        const edgeIdSet = new Set();
+        for (const e of data.edges) {
+            if (edgeIdSet.has(e.id))
+                throw new errors_1.EdgeAlreadyExistsError(e.id);
+            edgeIdSet.add(e.id);
+        }
+        const nodeIds = data.nodes.map(n => n.id);
+        const edgeIds = data.edges.map(e => e.id);
+        const [conflictNode, conflictEdge] = await Promise.all([
+            data.nodes.length > 0
+                ? this._nodes.findOne({ graphId: this._graphId, id: { $in: nodeIds } })
+                : Promise.resolve(null),
+            data.edges.length > 0
+                ? this._edges.findOne({ graphId: this._graphId, id: { $in: edgeIds } })
+                : Promise.resolve(null),
+        ]);
+        if (conflictNode)
+            throw new errors_1.NodeAlreadyExistsError(conflictNode.id);
+        if (conflictEdge)
+            throw new errors_1.EdgeAlreadyExistsError(conflictEdge.id);
+        const referencedIds = [...nodeIdSet];
+        for (const e of data.edges) {
+            referencedIds.push(e.sourceId, e.targetId);
+        }
+        const uniqueReferencedIds = [...new Set(referencedIds)];
+        const existingIdSet = new Set(await this._nodes
+            .find({ graphId: this._graphId, id: { $in: uniqueReferencedIds } }, { projection: { id: 1 } })
+            .toArray()
+            .then(docs => docs.map(d => d.id)));
+        for (const id of nodeIdSet)
+            existingIdSet.add(id);
+        for (const e of data.edges) {
+            if (!existingIdSet.has(e.sourceId))
+                throw new errors_1.NodeNotFoundError(e.sourceId);
+            if (!existingIdSet.has(e.targetId))
+                throw new errors_1.NodeNotFoundError(e.targetId);
+        }
+        if (data.nodes.length > 0) {
+            const nodeDocs = data.nodes.map(n => ({
+                id: n.id,
+                graphId: this._graphId,
+                type: n.type,
+                properties: n.properties,
+            }));
+            for (let i = 0; i < nodeDocs.length; i += this._batchSize) {
+                await this._nodes.insertMany(nodeDocs.slice(i, i + this._batchSize));
+            }
+        }
+        if (data.edges.length > 0) {
+            const edgeDocs = data.edges.map(e => ({
+                id: e.id,
+                graphId: this._graphId,
+                sourceId: e.sourceId,
+                targetId: e.targetId,
+                type: e.type,
+                properties: e.properties,
+            }));
+            for (let i = 0; i < edgeDocs.length; i += this._batchSize) {
+                await this._edges.insertMany(edgeDocs.slice(i, i + this._batchSize));
+            }
+        }
+    }
+    async createIndex(target, propertyKey, type) {
+        if (target === 'node') {
+            const indexFields = { graphId: 1, [`properties.${propertyKey}`]: 1 };
+            if (type && type !== '*') {
+                indexFields['type'] = 1;
+                await this._nodes.createIndex(indexFields, {
+                    name: `node_graphId_type_${propertyKey}`,
+                    background: true
+                });
+            }
+            else {
+                await this._nodes.createIndex(indexFields, {
+                    name: `node_graphId_${propertyKey}`,
+                    background: true
+                });
+            }
+        }
+        else {
+            const indexFields = { graphId: 1, [`properties.${propertyKey}`]: 1 };
+            if (type && type !== '*') {
+                indexFields['type'] = 1;
+                await this._edges.createIndex(indexFields, {
+                    name: `edge_graphId_type_${propertyKey}`,
+                    background: true
+                });
+            }
+            else {
+                await this._edges.createIndex(indexFields, {
+                    name: `edge_graphId_${propertyKey}`,
+                    background: true
+                });
+            }
+        }
+    }
+    async addProperty(target, id, key, value, transaction) {
+        if (!(0, grafio_1.isPrimitive)(value)) {
+            throw new errors_1.InvalidPropertyError(key, value);
+        }
+        const session = transaction?.context;
+        const collection = target === 'node' ? this._nodes : this._edges;
+        const result = await collection.updateOne({ graphId: this._graphId, id, [`properties.${key}`]: { $exists: false } }, { $set: { [`properties.${key}`]: value } }, { session });
+        if (result.matchedCount === 0) {
+            const record = target === 'node' ? await this.getNode(id, transaction) : await this.getEdge(id, transaction);
+            if (!record) {
+                throw target === 'node' ? new errors_1.NodeNotFoundError(id) : new errors_1.EdgeNotFoundError(id);
+            }
+            throw new errors_1.PropertyAlreadyExistsError(target, id, key);
+        }
+    }
+    async updateProperty(target, id, key, value, transaction) {
+        if (!(0, grafio_1.isPrimitive)(value)) {
+            throw new errors_1.InvalidPropertyError(key, value);
+        }
+        const session = transaction?.context;
+        const collection = target === 'node' ? this._nodes : this._edges;
+        const result = await collection.updateOne({ graphId: this._graphId, id, [`properties.${key}`]: { $exists: true } }, { $set: { [`properties.${key}`]: value } }, { session });
+        if (result.matchedCount === 0) {
+            const record = target === 'node' ? await this.getNode(id, transaction) : await this.getEdge(id, transaction);
+            if (!record) {
+                throw target === 'node' ? new errors_1.NodeNotFoundError(id) : new errors_1.EdgeNotFoundError(id);
+            }
+            throw new errors_1.PropertyNotFoundError(target, id, key);
+        }
+    }
+    async deleteProperty(target, id, key, transaction) {
+        const session = transaction?.context;
+        const collection = target === 'node' ? this._nodes : this._edges;
+        const record = target === 'node' ? await this.getNode(id, transaction) : await this.getEdge(id, transaction);
+        if (!record) {
+            if (target === 'node') {
+                throw new errors_1.NodeNotFoundError(id);
+            }
+            else {
+                throw new errors_1.EdgeNotFoundError(id);
+            }
+        }
+        await collection.updateOne({ graphId: this._graphId, id }, { $unset: { [`properties.${key}`]: '' } }, { session });
+    }
+    async clearProperties(target, id, transaction) {
+        const session = transaction?.context;
+        const collection = target === 'node' ? this._nodes : this._edges;
+        const record = target === 'node' ? await this.getNode(id, transaction) : await this.getEdge(id, transaction);
+        if (!record) {
+            if (target === 'node') {
+                throw new errors_1.NodeNotFoundError(id);
+            }
+            else {
+                throw new errors_1.EdgeNotFoundError(id);
+            }
+        }
+        const updateObj = {};
+        for (const key of Object.keys(record.properties)) {
+            updateObj[`properties.${key}`] = '';
+        }
+        if (Object.keys(updateObj).length > 0) {
+            await collection.updateOne({ graphId: this._graphId, id }, { $unset: updateObj }, { session });
+        }
+    }
+    supportsTransactions() {
+        return this._client !== null;
+    }
+    async beginTransaction() {
+        if (!this._client) {
+            throw new Error('MongoDB client not available for transactions. Ensure the Db instance has access to a MongoClient.');
+        }
+        const session = this._client.startSession();
+        session.startTransaction();
+        return {
+            id: `txn-${Date.now()}-${Math.random().toString(36).slice(2, 9)}`,
+            context: session,
+        };
+    }
+    async commitTransaction(handle) {
+        const session = handle.context;
+        if (!session.inTransaction()) {
+            throw new Error('No active transaction to commit');
+        }
+        try {
+            await session.commitTransaction();
+        }
+        finally {
+            session.endSession();
+        }
+    }
+    async rollbackTransaction(handle) {
+        const session = handle.context;
+        if (session.inTransaction()) {
+            try {
+                await session.abortTransaction();
+            }
+            finally {
+                session.endSession();
+            }
+        }
+        else {
+            session.endSession();
+        }
+    }
+    _docToNode(doc) {
+        return {
+            id: doc.id,
+            type: doc.type,
+            properties: doc.properties,
+        };
+    }
+    _docToEdge(doc) {
+        return {
+            id: doc.id,
+            sourceId: doc.sourceId,
+            targetId: doc.targetId,
+            type: doc.type,
+            properties: doc.properties,
+        };
+    }
+    _isDuplicateKeyError(e) {
+        return e instanceof Error && 'code' in e && e.code === 11000;
+    }
+}
+exports.MongoStorageProvider = MongoStorageProvider;

package/dist/MongoStorageProvider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"MongoStorageProvider.js","sourceRoot":"","sources":["../src/MongoStorageProvider.ts"],"names":[],"mappings":";;;AAAA,qCAAsC;AAUtC,mCAOgB;AAEhB,0CAQuB;AA0DvB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAmCG;AACH,MAAa,oBAAoB;IACd,MAAM,CAAsB;IAC5B,MAAM,CAAsB;IAC5B,QAAQ,CAAS;IACjB,UAAU,CAAS;IACnB,OAAO,GAAuB,IAAI,CAAC;IAEpD;;;OAGG;IACH,YAAY,EAAM,EAAE,OAAoC,EAAE;QACxD,MAAM,SAAS,GAAG,IAAI,CAAC,eAAe,IAAI,YAAY,CAAC;QACvD,MAAM,SAAS,GAAG,IAAI,CAAC,eAAe,IAAI,YAAY,CAAC;QAEvD,IAAI,CAAC,MAAM,GAAG,EAAE,CAAC,UAAU,CAAU,SAAS,CAAC,CAAC;QAChD,IAAI,CAAC,MAAM,GAAG,EAAE,CAAC,UAAU,CAAU,SAAS,CAAC,CAAC;QAChD,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,OAAO,IAAI,SAAS,CAAC;QAC1C,IAAI,IAAI,CAAC,SAAS,KAAK,SAAS,IAAI,IAAI,CAAC,SAAS,IAAI,CAAC,EAAE,CAAC;YACxD,MAAM,IAAI,KAAK,CAAC,8CAA8C,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC;QAClF,CAAC;QACD,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC;QAEzC,oCAAoC;QACpC,4DAA4D;QAC5D,IAAI,QAAQ,IAAI,EAAE,IAAI,EAAE,CAAC,MAAM,YAAY,qBAAW,EAAE,CAAC;YACvD,IAAI,CAAC,OAAO,GAAG,EAAE,CAAC,MAAM,CAAC;QAC3B,CAAC;IACH,CAAC;IAED,8EAA8E;IAC9E,mBAAmB;IACnB,8EAA8E;IAE9E;;;;;;OAMG;IACH,KAAK,CAAC,aAAa;QACjB,gGAAgG;QAChG,MAAM,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,sBAAsB,EAAE,CAAC,CAAC;QACrG,MAAM,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,iBAAiB,EAAE,CAAC,CAAC;QAEpF,eAAe;QACf,MAAM,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,sBAAsB,EAAE,CAAC,CAAC;QACrG,MAAM,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,iBAAiB,EAAE,CAAC,CAAC;QACpF,MAAM,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,wBAAwB,EAAE,CAAC,CAAC;QACxG,MAAM,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,wBAAwB,EAAE,CAAC,CAAC;IAC1G,CAAC;IAED,8EAA8E;IAC9E,YAAY;IACZ,8EAA8E;IAE9E,KAAK,CAAC,KAAK;QACT,MAAM,OAAO,CAAC,GAAG,CAAC;YAChB,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,EAAE,OAAO,EAAE,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClD,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,EAAE,OAAO,EAAE,IAAI,CAAC,QAAQ,EAAE,CAAC;SACnD,CAAC,CAAC;IACL,CAAC;IAED,8EAA8E;IAC9E,iBAAiB;IACjB,8EAA8E;IAE9E,KAAK,CAAC,UAAU,CAAC,IAAc,EAAE,WAAgC;QAC/D,MAAM,OAAO,GAAG,WAAW,EAAE,OAAoC,CAAC;QAClE,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC;gBAC1B,EAAE,EAAE,IAAI,CAAC,EAAE;gBACX,OAAO,EAAE,IAAI,CAAC,QAAQ;gBACtB,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,UAAU,EAAE,IAAI,CAAC,UAAU;aACjB,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;QAC7B,CAAC;QAAC,OAAO,CAAU,EAAE,CAAC;YACpB,IAAI,IAAI,CAAC,oBAAoB,CAAC,CAAC,CAAC;gBAAE,MAAM,IAAI,+BAAsB,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAC5E,MAAM,CAAC,CAAC;QACV,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,EAAU,EAAE,WAAgC;QAC3D,MAAM,OAAO,GAAG,WAAW,EAAE,OAAoC,CAAC;QAClE,MAAM,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,OAAO,EAAE,IAAI,CAAC,QAAQ,EAAE,EAAE,EAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;IAC3E,CAAC;IAED,8EAA8E;IAC9E,eAAe;IACf,8EAA8E;IAE9E,KAAK,CAAC,OAAO,CAAC,EAAU,EAAE,WAAgC;QACxD,MAAM,OAAO,GAAG,WAAW,EAAE,OAAoC,CAAC;QAClE,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CACnC,EAAE,OAAO,EAAE,IAAI,CAAC,QAAQ,EAAE,EAAE,EAAE,EAC9B,EAAE,UAAU,EAAE,EAAE,GAAG,EAAE,CAAC,EAAE,EAAE,OAAO,EAAE,CACpC,CAAC;QACF,OAAO,GAAG,KAAK,IAAI,CAAC;IACtB,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,EAAU,EAAE,WAAgC;QACxD,MAAM,OAAO,GAAG,WAAW,EAAE,OAAoC,CAAC;QAClE,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,IAAI,CAAC,QAAQ,EAAE,EAAE,EAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;QACnF,OAAO,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAChD,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,KAAc,EAAE,WAAgC;QAChE,MAAM,OAAO,GAAG,WAAW,EAAE,OAAoC,CAAC;QAClE,MAAM,KAAK,GAAe,EAAE,CAAC;QAC7B,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,IAAI,CAAC,QAAQ,EAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACpG,IAAI,KAAK;YAAE,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QAE/B,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;YAC/B,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC;QACnC,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,IAAY,EAAE,WAAgC;QACjE,MAAM,OAAO,GAAG,WAAW,EAAE,OAAoC,CAAC;QAClE,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,IAAI,CAAC,QAAQ,EAAE,IAAI,EAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC;QAC7F,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC;IAC3C,CAAC;IAED,KAAK,CAAC,kBAAkB,CAAC,GAAW,EAAE,KAAc,EAAE,QAAiB,EAAE,WAAgC;QACvG,MAAM,OAAO,GAAG,WAAW,EAAE,OAAoC,CAAC;QAClE,MAAM,MAAM,GAAoB;YAC9B,OAAO,EAAE,IAAI,CAAC,QAAQ;YACtB,CAAC,cAAc,GAAG,EAAE,CAAC,EAAE,KAA0D;SAClF,CAAC;QACF,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;YAC3B,MAAM,CAAC,IAAI,GAAG,QAAQ,CAAC;QACzB,CAAC;QACD,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC;QACnE,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC;IAC3C,CAAC;IAED,KAAK,CAAC,kBAAkB,CAAC,GAAW,EAAE,KAAc,EAAE,QAAiB,EAAE,WAAgC;QACvG,MAAM,OAAO,GAAG,WAAW,EAAE,OAAoC,CAAC;QAClE,MAAM,MAAM,GAAoB;YAC9B,OAAO,EAAE,IAAI,CAAC,QAAQ;YACtB,CAAC,cAAc,GAAG,EAAE,CAAC,EAAE,KAA0D;SAClF,CAAC;QACF,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;YAC3B,MAAM,CAAC,IAAI,GAAG,QAAQ,CAAC;QACzB,CAAC;QACD,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC;QACnE,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC;IAC3C,CAAC;IAED,8EAA8E;IAC9E,iBAAiB;IACjB,8EAA8E;IAE9E,KAAK,CAAC,UAAU,CAAC,IAAc,EAAE,WAAgC;QAC/D,MAAM,OAAO,GAAG,WAAW,EAAE,OAAoC,CAAC;QAClE,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC;gBAC1B,EAAE,EAAE,IAAI,CAAC,EAAE;gBACX,OAAO,EAAE,IAAI,CAAC,QAAQ;gBACtB,QAAQ,EAAE,IAAI,CAAC,QAAQ;gBACvB,QAAQ,EAAE,IAAI,CAAC,QAAQ;gBACvB,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,UAAU,EAAE,IAAI,CAAC,UAAU;aACjB,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;QAC7B,CAAC;QAAC,OAAO,CAAU,EAAE,CAAC;YACpB,IAAI,IAAI,CAAC,oBAAoB,CAAC,CAAC,CAAC;gBAAE,MAAM,IAAI,+BAAsB,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAC5E,MAAM,CAAC,CAAC;QACV,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,EAAU,EAAE,WAAgC;QAC3D,MAAM,OAAO,GAAG,WAAW,EAAE,OAAoC,CAAC;QAClE,MAAM,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,OAAO,EAAE,IAAI,CAAC,QAAQ,EAAE,EAAE,EAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;IAC3E,CAAC;IAED,8EAA8E;IAC9E,eAAe;IACf,8EAA8E;IAE9E,KAAK,CAAC,OAAO,CAAC,EAAU,EAAE,WAAgC;QACxD,MAAM,OAAO,GAAG,WAAW,EAAE,OAAoC,CAAC;QAClE,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CACnC,EAAE,OAAO,EAAE,IAAI,CAAC,QAAQ,EAAE,EAAE,EAAE,EAC9B,EAAE,UAAU,EAAE,EAAE,GAAG,EAAE,CAAC,EAAE,EAAE,OAAO,EAAE,CACpC,CAAC;QACF,OAAO,GAAG,KAAK,IAAI,CAAC;IACtB,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,EAAU,EAAE,WAAgC;QACxD,MAAM,OAAO,GAAG,WAAW,EAAE,OAAoC,CAAC;QAClE,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,IAAI,CAAC,QAAQ,EAAE,EAAE,EAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;QACnF,OAAO,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAChD,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,WAAgC;QAChD,MAAM,OAAO,GAAG,WAAW,EAAE,OAAoC,CAAC;QAClE,MAAM,KAAK,GAAe,EAAE,CAAC;QAC7B,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,IAAI,CAAC,QAAQ,EAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAEpG,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;YAC/B,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC;QACnC,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,IAAY,EAAE,WAAgC;QACjE,MAAM,OAAO,GAAG,WAAW,EAAE,OAAoC,CAAC;QAClE,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,IAAI,CAAC,QAAQ,EAAE,IAAI,EAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC;QAC7F,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC;IAC3C,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,MAAc,EAAE,IAAa,EAAE,WAAgC;QACpF,MAAM,OAAO,GAAG,WAAW,EAAE,OAAoC,CAAC;QAClE,MAAM,MAAM,GAAoB,EAAE,OAAO,EAAE,IAAI,CAAC,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC;QAC7E,IAAI,IAAI;YAAE,MAAM,CAAC,IAAI,GAAG,IAAI,CAAC;QAC7B,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC;QACnE,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC;IAC3C,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,MAAc,EAAE,IAAa,EAAE,WAAgC;QACpF,MAAM,OAAO,GAAG,WAAW,EAAE,OAAoC,CAAC;QAClE,MAAM,MAAM,GAAoB,EAAE,OAAO,EAAE,IAAI,CAAC,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC;QAC7E,IAAI,IAAI;YAAE,MAAM,CAAC,IAAI,GAAG,IAAI,CAAC;QAC7B,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC;QACnE,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC;IAC3C,CAAC;IAED,8EAA8E;IAC9E,mBAAmB;IACnB,8EAA8E;IAE9E,KAAK,CAAC,UAAU;QACd,MAAM,KAAK,GAAe,EAAE,CAAC;QAC7B,MAAM,KAAK,GAAe,EAAE,CAAC;QAE7B,MAAM,WAAW,GAAG,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC5F,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,WAAW,EAAE,CAAC;YACpC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC;QACnC,CAAC;QAED,MAAM,WAAW,GAAG,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC5F,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,WAAW,EAAE,CAAC;YACpC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC;QACnC,CAAC;QAED,OAAO;YACL,OAAO,EAAE,IAAI,CAAC,QAAQ;YACtB,KAAK;YACL,KAAK;SACN,CAAC;IACJ,CAAC;IAED;;;;;;;OAOG;IACH,KAAK,CAAC,UAAU,CAAC,IAAe;QAC9B,yDAAyD;QACzD,MAAM,SAAS,GAAG,IAAI,GAAG,EAAU,CAAC;QACpC,KAAK,MAAM,CAAC,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YAC3B,IAAI,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;gBAAE,MAAM,IAAI,+BAAsB,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YAChE,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QACtB,CAAC;QACD,MAAM,SAAS,GAAG,IAAI,GAAG,EAAU,CAAC;QACpC,KAAK,MAAM,CAAC,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YAC3B,IAAI,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;gBAAE,MAAM,IAAI,+BAAsB,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YAChE,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QACtB,CAAC;QAED,iFAAiF;QACjF,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QAC1C,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QAE1C,MAAM,CAAC,YAAY,EAAE,YAAY,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;YACrD,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC;gBACnB,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,IAAI,CAAC,QAAQ,EAAE,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,EAAE,CAAC;gBACvE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC;YACzB,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC;gBACnB,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,IAAI,CAAC,QAAQ,EAAE,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,EAAE,CAAC;gBACvE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC;SAC1B,CAAC,CAAC;QACH,IAAI,YAAY;YAAE,MAAM,IAAI,+BAAsB,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;QACpE,IAAI,YAAY;YAAE,MAAM,IAAI,+BAAsB,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;QAEpE,mDAAmD;QACnD,0FAA0F;QAC1F,MAAM,aAAa,GAAG,CAAC,GAAG,SAAS,CAAC,CAAC,CAAC,8CAA8C;QACpF,KAAK,MAAM,CAAC,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YAC3B,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC;QAC7C,CAAC;QACD,MAAM,mBAAmB,GAAG,CAAC,GAAG,IAAI,GAAG,CAAC,aAAa,CAAC,CAAC,CAAC;QACxD,MAAM,aAAa,GAAG,IAAI,GAAG,CAC3B,MAAM,IAAI,CAAC,MAAM;aACd,IAAI,CAAC,EAAE,OAAO,EAAE,IAAI,CAAC,QAAQ,EAAE,EAAE,EAAE,EAAE,GAAG,EAAE,mBAAmB,EAAE,EAAE,EAAE,EAAE,UAAU,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;aAC7F,OAAO,EAAE;aACT,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CACrC,CAAC;QAEF,KAAK,MAAM,EAAE,IAAI,SAAS;YAAE,aAAa,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAElD,KAAK,MAAM,CAAC,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YAC3B,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,QAAQ,CAAC;gBAAE,MAAM,IAAI,0BAAiB,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;YAC5E,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,QAAQ,CAAC;gBAAE,MAAM,IAAI,0BAAiB,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;QAC9E,CAAC;QAED,kCAAkC;QAClC,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC1B,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;gBACpC,EAAE,EAAE,CAAC,CAAC,EAAE;gBACR,OAAO,EAAE,IAAI,CAAC,QAAQ;gBACtB,IAAI,EAAE,CAAC,CAAC,IAAI;gBACZ,UAAU,EAAE,CAAC,CAAC,UAAU;aACb,CAAA,CAAC,CAAC;YACf,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;gBAC1D,MAAM,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC;YACvE,CAAC;QACH,CAAC;QACD,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC1B,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;gBACpC,EAAE,EAAE,CAAC,CAAC,EAAE;gBACR,OAAO,EAAE,IAAI,CAAC,QAAQ;gBACtB,QAAQ,EAAE,CAAC,CAAC,QAAQ;gBACpB,QAAQ,EAAE,CAAC,CAAC,QAAQ;gBACpB,IAAI,EAAE,CAAC,CAAC,IAAI;gBACZ,UAAU,EAAE,CAAC,CAAC,UAAU;aACb,CAAA,CAAC,CAAC;YACf,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;gBAC1D,MAAM,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC;YACvE,CAAC;QACH,CAAC;IACH,CAAC;IAED;;;;;;OAMG;IACH,KAAK,CAAC,WAAW,CAAC,MAAuB,EAAE,WAAmB,EAAE,IAAa;QAC3E,IAAI,MAAM,KAAK,MAAM,EAAE,CAAC;YACtB,sEAAsE;YACtE,MAAM,WAAW,GAAsB,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC,cAAc,WAAW,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC;YAExF,8EAA8E;YAC9E,IAAI,IAAI,IAAI,IAAI,KAAK,GAAG,EAAE,CAAC;gBACzB,WAAW,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;gBACxB,MAAM,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,WAAW,EAAE;oBACzC,IAAI,EAAE,qBAAqB,WAAW,EAAE;oBACxC,UAAU,EAAE,IAAI;iBACjB,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,MAAM,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,WAAW,EAAE;oBACzC,IAAI,EAAE,gBAAgB,WAAW,EAAE;oBACnC,UAAU,EAAE,IAAI;iBACjB,CAAC,CAAC;YACL,CAAC;QACH,CAAC;aAAM,CAAC;YACN,sEAAsE;YACtE,MAAM,WAAW,GAAsB,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC,cAAc,WAAW,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC;YAExF,8EAA8E;YAC9E,IAAI,IAAI,IAAI,IAAI,KAAK,GAAG,EAAE,CAAC;gBACzB,WAAW,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;gBACxB,MAAM,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,WAAW,EAAE;oBACzC,IAAI,EAAE,qBAAqB,WAAW,EAAE;oBACxC,UAAU,EAAE,IAAI;iBACjB,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,MAAM,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,WAAW,EAAE;oBACzC,IAAI,EAAE,gBAAgB,WAAW,EAAE;oBACnC,UAAU,EAAE,IAAI;iBACjB,CAAC,CAAC;YACL,CAAC;QACH,CAAC;IACH,CAAC;IAED,8EAA8E;IAC9E,qBAAqB;IACrB,8EAA8E;IAE9E;;;;;OAKG;IACH,KAAK,CAAC,WAAW,CAAC,MAAuB,EAAE,EAAU,EAAE,GAAW,EAAE,KAAc,EAAE,WAAgC;QAClH,IAAI,CAAC,IAAA,oBAAW,EAAC,KAAK,CAAC,EAAE,CAAC;YACxB,MAAM,IAAI,6BAAoB,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;QAC7C,CAAC;QAED,MAAM,OAAO,GAAG,WAAW,EAAE,OAAoC,CAAC;QAClE,MAAM,UAAU,GAAG,MAAM,KAAK,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC;QAEjE,+DAA+D;QAC/D,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,SAAS,CACvC,EAAE,OAAO,EAAE,IAAI,CAAC,QAAQ,EAAE,EAAE,EAAE,CAAC,cAAc,GAAG,EAAE,CAAC,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EACzE,EAAE,IAAI,EAAE,EAAE,CAAC,cAAc,GAAG,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAC1C,EAAE,OAAO,EAAE,CACZ,CAAC;QAEF,IAAI,MAAM,CAAC,YAAY,KAAK,CAAC,EAAE,CAAC;YAC9B,wFAAwF;YACxF,MAAM,MAAM,GAAG,MAAM,KAAK,MAAM,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC;YAC7G,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,MAAM,MAAM,KAAK,MAAM,CAAC,CAAC,CAAC,IAAI,0BAAiB,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,0BAAiB,CAAC,EAAE,CAAC,CAAC;YAClF,CAAC;YACD,MAAM,IAAI,mCAA0B,CAAC,MAAM,EAAE,EAAE,EAAE,GAAG,CAAC,CAAC;QACxD,CAAC;IACH,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,cAAc,CAAC,MAAuB,EAAE,EAAU,EAAE,GAAW,EAAE,KAAc,EAAE,WAAgC;QACrH,IAAI,CAAC,IAAA,oBAAW,EAAC,KAAK,CAAC,EAAE,CAAC;YACxB,MAAM,IAAI,6BAAoB,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;QAC7C,CAAC;QAED,MAAM,OAAO,GAAG,WAAW,EAAE,OAAoC,CAAC;QAClE,MAAM,UAAU,GAAG,MAAM,KAAK,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC;QAEjE,8DAA8D;QAC9D,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,SAAS,CACvC,EAAE,OAAO,EAAE,IAAI,CAAC,QAAQ,EAAE,EAAE,EAAE,CAAC,cAAc,GAAG,EAAE,CAAC,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,EACxE,EAAE,IAAI,EAAE,EAAE,CAAC,cAAc,GAAG,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAC1C,EAAE,OAAO,EAAE,CACZ,CAAC;QAEF,IAAI,MAAM,CAAC,YAAY,KAAK,CAAC,EAAE,CAAC;YAC9B,wEAAwE;YACxE,MAAM,MAAM,GAAG,MAAM,KAAK,MAAM,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC;YAC7G,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,MAAM,MAAM,KAAK,MAAM,CAAC,CAAC,CAAC,IAAI,0BAAiB,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,0BAAiB,CAAC,EAAE,CAAC,CAAC;YAClF,CAAC;YACD,MAAM,IAAI,8BAAqB,CAAC,MAAM,EAAE,EAAE,EAAE,GAAG,CAAC,CAAC;QACnD,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,cAAc,CAAC,MAAuB,EAAE,EAAU,EAAE,GAAW,EAAE,WAAgC;QACrG,MAAM,OAAO,GAAG,WAAW,EAAE,OAAoC,CAAC;QAClE,MAAM,UAAU,GAAG,MAAM,KAAK,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC;QACjE,MAAM,MAAM,GAAG,MAAM,KAAK,MAAM,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC;QAE7G,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,IAAI,MAAM,KAAK,MAAM,EAAE,CAAC;gBACtB,MAAM,IAAI,0BAAiB,CAAC,EAAE,CAAC,CAAC;YAClC,CAAC;iBAAM,CAAC;gBACN,MAAM,IAAI,0BAAiB,CAAC,EAAE,CAAC,CAAC;YAClC,CAAC;QACH,CAAC;QAED,MAAM,UAAU,CAAC,SAAS,CACxB,EAAE,OAAO,EAAE,IAAI,CAAC,QAAQ,EAAE,EAAE,EAAE,EAC9B,EAAE,MAAM,EAAE,EAAE,CAAC,cAAc,GAAG,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EACzC,EAAE,OAAO,EAAE,CACZ,CAAC;IACJ,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,eAAe,CAAC,MAAuB,EAAE,EAAU,EAAE,WAAgC;QACzF,MAAM,OAAO,GAAG,WAAW,EAAE,OAAoC,CAAC;QAClE,MAAM,UAAU,GAAG,MAAM,KAAK,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC;QACjE,MAAM,MAAM,GAAG,MAAM,KAAK,MAAM,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC;QAE7G,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,IAAI,MAAM,KAAK,MAAM,EAAE,CAAC;gBACtB,MAAM,IAAI,0BAAiB,CAAC,EAAE,CAAC,CAAC;YAClC,CAAC;iBAAM,CAAC;gBACN,MAAM,IAAI,0BAAiB,CAAC,EAAE,CAAC,CAAC;YAClC,CAAC;QACH,CAAC;QAED,MAAM,SAAS,GAA4B,EAAE,CAAC;QAC9C,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,EAAE,CAAC;YACjD,SAAS,CAAC,cAAc,GAAG,EAAE,CAAC,GAAG,EAAE,CAAC;QACtC,CAAC;QAED,IAAI,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACtC,MAAM,UAAU,CAAC,SAAS,CACxB,EAAE,OAAO,EAAE,IAAI,CAAC,QAAQ,EAAE,EAAE,EAAE,EAC9B,EAAE,MAAM,EAAE,SAAS,EAAE,EACrB,EAAE,OAAO,EAAE,CACZ,CAAC;QACJ,CAAC;IACH,CAAC;IAED,8EAA8E;IAC9E,kBAAkB;IAClB,8EAA8E;IAE9E,8EAA8E;IAC9E,sBAAsB;IACtB,8EAA8E;IAE9E;;;OAGG;IACH,oBAAoB;QAClB,OAAO,IAAI,CAAC,OAAO,KAAK,IAAI,CAAC;IAC/B,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,gBAAgB;QACpB,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAClB,MAAM,IAAI,KAAK,CAAC,oGAAoG,CAAC,CAAC;QACxH,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE,CAAC;QAC5C,OAAO,CAAC,gBAAgB,EAAE,CAAC,CAAC,uCAAuC;QAEnE,OAAO;YACL,EAAE,EAAE,OAAO,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE;YACjE,OAAO,EAAE,OAAO;SACjB,CAAC;IACJ,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,iBAAiB,CAAC,MAA0B;QAChD,MAAM,OAAO,GAAG,MAAM,CAAC,OAAwB,CAAC;QAChD,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAC;QACrD,CAAC;QACD,IAAI,CAAC;YACH,MAAM,OAAO,CAAC,iBAAiB,EAAE,CAAC;QACpC,CAAC;gBAAS,CAAC;YACT,OAAO,CAAC,UAAU,EAAE,CAAC;QACvB,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,mBAAmB,CAAC,MAA0B;QAClD,MAAM,OAAO,GAAG,MAAM,CAAC,OAAwB,CAAC;QAChD,IAAI,OAAO,CAAC,aAAa,EAAE,EAAE,CAAC;YAC5B,IAAI,CAAC;gBACH,MAAM,OAAO,CAAC,gBAAgB,EAAE,CAAC;YACnC,CAAC;oBAAS,CAAC;gBACT,OAAO,CAAC,UAAU,EAAE,CAAC;YACvB,CAAC;QACH,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,UAAU,EAAE,CAAC;QACvB,CAAC;IACH,CAAC;IAED,8EAA8E;IAC9E,kBAAkB;IAClB,8EAA8E;IAEtE,UAAU,CAAC,GAAoB;QACrC,OAAO;YACL,EAAE,EAAE,GAAG,CAAC,EAAE;YACV,IAAI,EAAE,GAAG,CAAC,IAAI;YACd,UAAU,EAAE,GAAG,CAAC,UAAU;SAC3B,CAAC;IACJ,CAAC;IAEO,UAAU,CAAC,GAAoB;QACrC,OAAO;YACL,EAAE,EAAE,GAAG,CAAC,EAAE;YACV,QAAQ,EAAE,GAAG,CAAC,QAAQ;YACtB,QAAQ,EAAE,GAAG,CAAC,QAAQ;YACtB,IAAI,EAAE,GAAG,CAAC,IAAI;YACd,UAAU,EAAE,GAAG,CAAC,UAAU;SAC3B,CAAC;IACJ,CAAC;IAEO,oBAAoB,CAAC,CAAU;QACrC,OAAO,CAAC,YAAY,KAAK,IAAI,MAAM,IAAI,CAAC,IAAK,CAAsB,CAAC,IAAI,KAAK,KAAK,CAAC;IACrF,CAAC;CACF;AArlBD,oDAqlBC"}

package/dist/index.d.ts

@@ -0,0 +1,3 @@
+export { MongoStorageProvider } from './MongoStorageProvider';
+export type { MongoStorageProviderOptions } from './MongoStorageProvider';
+export { MongoGraphFactory } from './MongoGraphFactory';

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,oBAAoB,EAAE,MAAM,wBAAwB,CAAC;AAC9D,YAAY,EAAE,2BAA2B,EAAE,MAAM,wBAAwB,CAAC;AAC1E,OAAO,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC"}

package/dist/index.js

@@ -0,0 +1,7 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MongoGraphFactory = exports.MongoStorageProvider = void 0;
+var MongoStorageProvider_1 = require("./MongoStorageProvider");
+Object.defineProperty(exports, "MongoStorageProvider", { enumerable: true, get: function () { return MongoStorageProvider_1.MongoStorageProvider; } });
+var MongoGraphFactory_1 = require("./MongoGraphFactory");
+Object.defineProperty(exports, "MongoGraphFactory", { enumerable: true, get: function () { return MongoGraphFactory_1.MongoGraphFactory; } });

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;AAAA,sDAAsD;AACtD,+DAA8D;AAArD,4HAAA,oBAAoB,OAAA;AAE7B,yDAAwD;AAA/C,sHAAA,iBAAiB,OAAA"}

package/jest.config.js

@@ -0,0 +1,27 @@
+/** @type {import('ts-jest').JestConfigWithTsJest} */
+module.exports = {
+  preset: 'ts-jest',
+  testEnvironment: 'node',
+  roots: ['<rootDir>/tests'],
+  testMatch: ['**/*.test.ts'],
+  moduleFileExtensions: ['ts', 'js', 'json'],
+  collectCoverageFrom: ['src/**/*.ts'],
+  coveragePathIgnorePatterns: ['/node_modules/', '/dist/'],
+  coverageThreshold: {
+    global: {
+      branches: 80,
+      functions: 80,
+      lines: 80,
+      statements: 80
+    }
+  },
+  testTimeout: 60000,
+  maxWorkers: 1,
+  coverageReporters: [
+    "lcov",
+    "json-summary"
+  ],
+  transform: {
+    '^.+\\.tsx?$': ['ts-jest', { tsconfig: { resolveJsonModule: true } }],
+  },
+};

package/package.json

@@ -0,0 +1,41 @@
+{
+  "name": "grafio-mongo",
+  "version": "1.0.0",
+  "description": "MongoDB storage backend for grafio",
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "scripts": {
+    "build": "tsc",
+    "build:prod": "tsc -p tsconfig.prod.json",
+    "test": "jest",
+    "test:coverage": "jest --coverage --coverageReporters=text-summary --coverageReporters=html"
+  },
+  "peerDependencies": {
+    "grafio": ">=5.0.0",
+    "mongodb": ">=5.0.0"
+  },
+  "devDependencies": {
+    "@types/jest": "^29.5.0",
+    "@types/node": "^25.6.0",
+    "grafio": "^5.4.2",
+    "jest": "^29.5.0",
+    "mongodb": "^7.2.0",
+    "mongodb-memory-server": "^11.1.0",
+    "ts-jest": "^29.1.0",
+    "ts-node": "^10.9.0",
+    "typescript": "^5.0.0"
+  },
+  "exports": {
+    ".": {
+      "import": "./dist/index.js",
+      "require": "./dist/index.js",
+      "types": "./dist/index.d.ts"
+    }
+  },
+  "author": "Satya Jugran",
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/satya-jugran/grafio-mongo"
+  }
+}

package/src/MongoGraphFactory.ts

@@ -0,0 +1,93 @@
+import type { Db } from 'mongodb';
+import {
+  MongoStorageProvider,
+  type MongoStorageProviderOptions,
+} from './MongoStorageProvider';
+import { Graph, GraphData, IGraphFactory } from 'grafio';
+
+/**
+ * Factory options for MongoGraphFactory — omits graphId since that is set per-call.
+ */
+type MongoGraphFactoryOptions = Omit<MongoStorageProviderOptions, 'graphId'>;
+
+/**
+ * MongoDB-backed factory for `Graph` instances.
+ *
+ * Holds the long-lived `Db` connection reference. `forGraph(graphId)` returns a
+ * lightweight `Graph` scoped to the given partition — no new I/O resources are
+ * created on each call (Collection handles are stateless).
+ *
+ * Call `ensureIndexes()` once at application startup before any graph operations.
+ *
+ * @example
+ * const factory = new MongoGraphFactory(db);
+ * await factory.ensureIndexes();
+ *
+ * const graph = factory.forGraph('user-123');
+ * await graph.addNode('Person', { name: 'Alice' });
+ */
+export class MongoGraphFactory implements IGraphFactory {
+  private readonly _db: Db;
+  private readonly _opts: MongoGraphFactoryOptions;
+
+  /**
+   * @param db   - An already-connected Mongo `Db` instance. The factory does not
+   *               manage the connection lifecycle — that is the consumer's responsibility.
+   * @param opts - Optional collection name overrides (no graphId here — that is
+   *               set per-call in `forGraph`).
+   */
+  constructor(db: Db, opts: MongoGraphFactoryOptions = {}) {
+    this._db = db;
+    this._opts = opts;
+  }
+
+  /**
+   * Creates all required MongoDB indexes.
+   *
+   * Call ONCE at application startup before any graph operations.
+   * Safe to call multiple times (MongoDB's `createIndex` is idempotent).
+   */
+  async ensureIndexes(): Promise<void> {
+    const probe = new MongoStorageProvider(this._db, { ...this._opts, graphId: 'default' });
+    await probe.ensureIndexes();
+  }
+
+  /**
+   * Returns a `Graph` instance scoped to the given `graphId`.
+   *
+   * Cheap to call — creates only stateless `Collection` references, no I/O.
+   * Each returned `Graph` operates on its own isolated partition of the same
+   * MongoDB collections.
+   *
+   * @param graphId - Defaults to `"default"` when omitted.
+   */
+  forGraph(graphId: string = 'default'): Graph {
+    const provider = new MongoStorageProvider(this._db, { ...this._opts, graphId });
+    return new Graph(provider);
+  }
+
+  /**
+   * Creates a Graph instance and imports the given graph data into it.
+   * Convenience method that combines forGraph() + importJSON().
+   *
+   * @param data - The GraphData to import
+   * @param graphId - Graph partition ID (defaults to 'default')
+   * @returns A Graph instance with the imported data
+   */
+  async fromGraphData(data: GraphData, graphId: string = 'default'): Promise<Graph> {
+    const provider = new MongoStorageProvider(this._db, { ...this._opts, graphId });
+    const graph = new Graph(provider);
+
+    // Filter data to only import nodes/edges that belong to this graphId
+    // If data.graphId is set and doesn't match, skip (data is from different partition)
+    // If data.graphId is not set, import all (backward compatible for legacy data)
+    const filteredData: GraphData = {
+      graphId,
+      nodes: data.graphId === undefined || data.graphId === graphId ? data.nodes : [],
+      edges: data.graphId === undefined || data.graphId === graphId ? data.edges : [],
+    };
+
+    await Graph.importJSON(filteredData, provider);
+    return graph;
+  }
+}