cantian-mongodb 0.0.1 → 0.0.6

package/dist/bin/migrate.d.ts

@@ -0,0 +1,2 @@
+#!/usr/bin/env node
+export {};

package/dist/bin/migrate.js

@@ -0,0 +1,21 @@
+#!/usr/bin/env node
+import { glob } from 'node:fs';
+import { migrate } from '../migration.js';
+import { db } from '../mongoClient.js';
+let [, , modelDir] = process.argv;
+if (!modelDir) {
+    modelDir = 'dist/models';
+}
+const modelFiles = await new Promise((r) => glob(`${modelDir}/*.js`, (err, files) => {
+    if (err)
+        throw err;
+    r(files);
+}));
+for (const file of modelFiles) {
+    const model = (await import(`${process.cwd()}/${file}`)).default;
+    if (model) {
+        await migrate(db, model);
+    }
+}
+process.exit();
+//# sourceMappingURL=migrate.js.map

package/dist/bin/migrate.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"migrate.js","sourceRoot":"","sources":["../../src/bin/migrate.ts"],"names":[],"mappings":";AAEA,OAAO,EAAE,IAAI,EAAE,MAAM,SAAS,CAAC;AAC/B,OAAO,EAAE,OAAO,EAAE,MAAM,iBAAiB,CAAC;AAC1C,OAAO,EAAE,EAAE,EAAS,MAAM,mBAAmB,CAAC;AAE9C,IAAI,CAAC,EAAE,AAAD,EAAG,QAAQ,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;AAElC,IAAI,CAAC,QAAQ,EAAE,CAAC;IACd,QAAQ,GAAG,aAAa,CAAC;AAC3B,CAAC;AAED,MAAM,UAAU,GAAG,MAAM,IAAI,OAAO,CAAW,CAAC,CAAC,EAAE,EAAE,CACnD,IAAI,CAAC,GAAG,QAAQ,OAAO,EAAE,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE;IACtC,IAAI,GAAG;QAAE,MAAM,GAAG,CAAC;IACnB,CAAC,CAAC,KAAK,CAAC,CAAC;AACX,CAAC,CAAC,CACH,CAAC;AACF,KAAK,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;IAC9B,MAAM,KAAK,GAAG,CAAC,MAAM,MAAM,CAAC,GAAG,OAAO,CAAC,GAAG,EAAE,IAAI,IAAI,EAAE,CAAC,CAAC,CAAC,OAAgB,CAAC;IAC1E,IAAI,KAAK,EAAE,CAAC;QACV,MAAM,OAAO,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;IAC3B,CAAC;AACH,CAAC;AAED,OAAO,CAAC,IAAI,EAAE,CAAC"}

package/dist/migration.d.ts

@@ -0,0 +1,18 @@
+import { CreateIndexesOptions, Db, IndexSpecification } from 'mongodb';
+import { Model } from './mongoClient.js';
+export type Indexes = {
+    key: any;
+    options?: CreateIndexesOptions;
+    vectorOptions?: Record<string, any>;
+}[];
+export declare function generateIndexName(key: IndexSpecification): string;
+export declare const migrate: (db: Db, model: Model) => Promise<void>;
+export type MigrateRemoteEvent = {
+    dbName: string;
+    collections: {
+        collectionName: string;
+        schema: Model['schema'];
+        indexes: Model['indexes'];
+    }[];
+};
+export declare const migrateRemoteHandler: (event: MigrateRemoteEvent) => Promise<void>;

package/dist/migration.js

@@ -0,0 +1,66 @@
+import { mongoClient } from './mongoClient.js';
+export function generateIndexName(key) {
+    return Object.entries(key)
+        .map(([k, v]) => `${k}_${v}`)
+        .join('_');
+}
+async function migrateSchema(db, collectionName, schema) {
+    const existingCollections = await db
+        .listCollections({
+        name: collectionName,
+    })
+        .toArray();
+    if (existingCollections.length) {
+        await db.command({
+            collMod: collectionName,
+            validator: { $jsonSchema: schema },
+        });
+        console.info(`Update collection '${collectionName}' successfully with schema: ${JSON.stringify(schema)}`);
+    }
+    else {
+        await db.createCollection(collectionName, {
+            validator: { $jsonSchema: schema },
+        });
+        console.info(`Create collection '${collectionName}' successfully with schema: ${JSON.stringify(schema)}`);
+    }
+}
+export const migrate = async (db, model) => {
+    const collectionName = model.collection.collectionName;
+    await migrateSchema(db, collectionName, model.schema);
+    if (model.indexes?.length) {
+        for (const index of model.indexes) {
+            const indexOption = { ...index.options, name: generateIndexName(index.key) };
+            await model.collection.createIndex(index.key, indexOption);
+            console.info(`Create a general index for ${collectionName} successfully.`, indexOption);
+        }
+    }
+    const existingIndexes = await model.collection.listIndexes().toArray();
+    const existingIndexMap = {};
+    for (const existingIndex of existingIndexes) {
+        existingIndexMap[existingIndex.name] = existingIndex;
+    }
+    for (const existingIndex of existingIndexes) {
+        if (existingIndex.name === '_id_') {
+            continue;
+        }
+        const inUsed = model.indexes?.find((v) => {
+            return existingIndex.name === generateIndexName(v.key);
+        });
+        if (!inUsed) {
+            await model.collection.dropIndex(existingIndex.name);
+            console.info(`Removed an index from ${collectionName} successfully.`, existingIndex);
+        }
+    }
+};
+export const migrateRemoteHandler = async (event) => {
+    const db = mongoClient.db(event.dbName, { ignoreUndefined: true });
+    for (const collectionConfig of event.collections) {
+        const collection = db.collection(collectionConfig.collectionName);
+        await migrate(db, {
+            schema: collectionConfig.schema,
+            indexes: collectionConfig.indexes,
+            collection,
+        });
+    }
+};
+//# sourceMappingURL=migration.js.map

package/dist/migration.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"migration.js","sourceRoot":"","sources":["../src/migration.ts"],"names":[],"mappings":"AAEA,OAAO,EAAS,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAItD,MAAM,UAAU,iBAAiB,CAAC,GAAuB;IACvD,OAAO,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC;SACvB,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;SAC5B,IAAI,CAAC,GAAG,CAAC,CAAC;AACf,CAAC;AAED,KAAK,UAAU,aAAa,CAAC,EAAM,EAAE,cAAsB,EAAE,MAAkB;IAC7E,MAAM,mBAAmB,GAAG,MAAM,EAAE;SACjC,eAAe,CAAC;QACf,IAAI,EAAE,cAAc;KACrB,CAAC;SACD,OAAO,EAAE,CAAC;IACb,IAAI,mBAAmB,CAAC,MAAM,EAAE,CAAC;QAC/B,MAAM,EAAE,CAAC,OAAO,CAAC;YACf,OAAO,EAAE,cAAc;YACvB,SAAS,EAAE,EAAE,WAAW,EAAE,MAAM,EAAE;SACnC,CAAC,CAAC;QACH,OAAO,CAAC,IAAI,CAAC,sBAAsB,cAAc,+BAA+B,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;IAC5G,CAAC;SAAM,CAAC;QACN,MAAM,EAAE,CAAC,gBAAgB,CAAC,cAAc,EAAE;YACxC,SAAS,EAAE,EAAE,WAAW,EAAE,MAAM,EAAE;SACnC,CAAC,CAAC;QACH,OAAO,CAAC,IAAI,CAAC,sBAAsB,cAAc,+BAA+B,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;IAC5G,CAAC;AACH,CAAC;AAED,MAAM,CAAC,MAAM,OAAO,GAAG,KAAK,EAAE,EAAM,EAAE,KAAY,EAAE,EAAE;IACpD,MAAM,cAAc,GAAG,KAAK,CAAC,UAAU,CAAC,cAAc,CAAC;IACvD,MAAM,aAAa,CAAC,EAAE,EAAE,cAAc,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;IAEtD,IAAI,KAAK,CAAC,OAAO,EAAE,MAAM,EAAE,CAAC;QAC1B,KAAK,MAAM,KAAK,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;YAClC,MAAM,WAAW,GAAG,EAAE,GAAG,KAAK,CAAC,OAAO,EAAE,IAAI,EAAE,iBAAiB,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC;YAC7E,MAAM,KAAK,CAAC,UAAU,CAAC,WAAW,CAAC,KAAK,CAAC,GAAG,EAAE,WAAW,CAAC,CAAC;YAC3D,OAAO,CAAC,IAAI,CAAC,8BAA8B,cAAc,gBAAgB,EAAE,WAAW,CAAC,CAAC;QAC1F,CAAC;IACH,CAAC;IAED,MAAM,eAAe,GAAG,MAAM,KAAK,CAAC,UAAU,CAAC,WAAW,EAAE,CAAC,OAAO,EAAE,CAAC;IACvE,MAAM,gBAAgB,GAAwB,EAAE,CAAC;IACjD,KAAK,MAAM,aAAa,IAAI,eAAe,EAAE,CAAC;QAC5C,gBAAgB,CAAC,aAAa,CAAC,IAAI,CAAC,GAAG,aAAa,CAAC;IACvD,CAAC;IAED,KAAK,MAAM,aAAa,IAAI,eAAe,EAAE,CAAC;QAC5C,IAAI,aAAa,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAClC,SAAS;QACX,CAAC;QACD,MAAM,MAAM,GAAG,KAAK,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE;YACvC,OAAO,aAAa,CAAC,IAAI,KAAK,iBAAiB,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;QACzD,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,KAAK,CAAC,UAAU,CAAC,SAAS,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;YACrD,OAAO,CAAC,IAAI,CAAC,yBAAyB,cAAc,gBAAgB,EAAE,aAAa,CAAC,CAAC;QACvF,CAAC;IACH,CAAC;AACH,CAAC,CAAC;AAOF,MAAM,CAAC,MAAM,oBAAoB,GAAG,KAAK,EAAE,KAAyB,EAAE,EAAE;IACtE,MAAM,EAAE,GAAG,WAAW,CAAC,EAAE,CAAC,KAAK,CAAC,MAAM,EAAE,EAAE,eAAe,EAAE,IAAI,EAAE,CAAC,CAAC;IACnE,KAAK,MAAM,gBAAgB,IAAI,KAAK,CAAC,WAAW,EAAE,CAAC;QACjD,MAAM,UAAU,GAAG,EAAE,CAAC,UAAU,CAAC,gBAAgB,CAAC,cAAc,CAAC,CAAC;QAClE,MAAM,OAAO,CAAC,EAAE,EAAE;YAChB,MAAM,EAAE,gBAAgB,CAAC,MAAM;YAC/B,OAAO,EAAE,gBAAgB,CAAC,OAAO;YACjC,UAAU;SACX,CAAC,CAAC;IACL,CAAC;AACH,CAAC,CAAC"}

package/package.json

@@ -1,15 +1,19 @@
 {
   "name": "cantian-mongodb",
-  "version": "0.0.1",
+  "version": "0.0.6",
   "type": "module",
   "exports": {
-    ".": "./dist/index.js"
+    ".": "./dist/index.js",
+    "./migration": "./dist/migration.js"
   },
   "files": [
     "dist"
   ],
+  "bin": {
+    "cantian-mongodb-migrate": "./dist/bin/migrate.js"
+  },
   "scripts": {
-    "prepublishOnly": "rm -rf dist && tsc"
+    "prepublishOnly": "rm -rf dist && tsc && chmod +x dist/bin/*.js"
   },
   "devDependencies": {
     "json-schema-to-ts": "^3.1.1"