@rpcbase/db 0.2.0 → 0.3.0

package/dist/index.d.ts

@@ -1,3 +1,4 @@
 export * from './models';
 export * from './schema';
+export * from './loadModel';
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,UAAU,CAAA;AACxB,cAAc,UAAU,CAAA"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,UAAU,CAAA;AACxB,cAAc,UAAU,CAAA;AACxB,cAAc,aAAa,CAAA"}

package/dist/index.js

@@ -1,7 +1,8 @@
-import { Schema, isValidObjectId, Types, SchemaTypes } from "mongoose";
+import mongoose, { Schema, isValidObjectId, Types, SchemaTypes } from "mongoose";
 import { z, ZodMap, ZodRecord, ZodAny, ZodUnion, ZodNullable, ZodOptional, ZodDefault, ZodDate, ZodEnum, ZodBoolean, ZodArray, ZodObject, ZodNumber, ZodString } from "zod";
 export * from "zod";
 import { z as z2 } from "zod";
+import assert from "assert";
 const ZUser = z.object({
   email: z.string().email().optional(),
   password: z.string(),
@@ -20,6 +21,11 @@ const UserSchema = new Schema({
   email_verification_code: { type: String, required: false },
   email_verification_expires_at: { type: Date, required: false }
 });
+const __vite_glob_0_1 = /* @__PURE__ */ Object.freeze(/* @__PURE__ */ Object.defineProperty({
+  __proto__: null,
+  UserSchema,
+  ZUser
+}, Symbol.toStringTag, { value: "Module" }));
 const ZTenant = z.object({
   tenant_id: z.string(),
   parent_tenant_id: z.string().optional(),
@@ -30,6 +36,18 @@ const TenantSchema = new Schema({
   parent_tenant_id: { type: String },
   name: { type: String }
 });
+const __vite_glob_0_0 = /* @__PURE__ */ Object.freeze(/* @__PURE__ */ Object.defineProperty({
+  __proto__: null,
+  TenantSchema,
+  ZTenant
+}, Symbol.toStringTag, { value: "Module" }));
+const frameworkSchemas = /* @__PURE__ */ Object.freeze(/* @__PURE__ */ Object.defineProperty({
+  __proto__: null,
+  TenantSchema,
+  UserSchema,
+  ZTenant,
+  ZUser
+}, Symbol.toStringTag, { value: "Module" }));
 const zmAssert$4 = {
   string(f) {
     return f.constructor.name === "ZodString";
@@ -635,12 +653,75 @@ function parseMixed(required = true, def) {
     required
   };
 }
+const { APP_NAME, DB_PORT } = process.env;
+const connections = {};
+let cachedModels = null;
+const buildModels = () => {
+  const externalModelModules = globalThis.__rb_model_modules;
+  const modelModules = externalModelModules ?? /* @__PURE__ */ Object.assign({ "/src/models/Tenant.ts": __vite_glob_0_0, "/src/models/User.ts": __vite_glob_0_1, "/src/models/index.ts": frameworkSchemas });
+  const appModels = Object.entries(modelModules).flatMap(([key, exports$1]) => {
+    const modelName = key.split("/").pop()?.replace(/\.(t|j)sx?$/, "");
+    assert(modelName, "Model name not found");
+    const schemaEntry = Object.entries(exports$1).find(([key2]) => key2 === `${modelName}Schema`);
+    if (!schemaEntry) return [];
+    const schema = schemaEntry[1];
+    if (!(schema instanceof mongoose.Schema)) return [];
+    return [{
+      name: modelName,
+      model: mongoose.models[modelName] ?? mongoose.model(modelName, schema)
+    }];
+  });
+  const frameworkModels = Object.entries(frameworkSchemas).filter(([_, schema]) => schema instanceof mongoose.Schema).map(([key, schema]) => {
+    const name = key.replace(/Schema$/, "");
+    return {
+      name,
+      model: mongoose.models[name] ?? mongoose.model(name, schema)
+    };
+  });
+  return [...frameworkModels, ...appModels].reduce((acc, { name, model }) => {
+    acc[name] = model;
+    return acc;
+  }, {});
+};
+const loadModel = async (modelName, ctx) => {
+  const models = cachedModels ?? (cachedModels = buildModels());
+  const tenantId = ctx.req.session?.user?.current_tenant_id || "00000000";
+  assert(tenantId, "Tenant ID is missing from session");
+  const dbName = ["User", "Tenant"].includes(modelName) ? `${APP_NAME}-users-db` : `${APP_NAME}-${tenantId}-db`;
+  const connectionString = `mongodb://localhost:${DB_PORT}/${dbName}`;
+  if (!connections[dbName]) {
+    const connection = mongoose.createConnection(connectionString, {
+      sanitizeFilter: true
+    });
+    await new Promise((resolve, reject) => {
+      connection.once("open", resolve);
+      connection.on("error", reject);
+    });
+    connections[dbName] = connection;
+  } else {
+    const connection = connections[dbName];
+    if (connection.readyState !== 1) {
+      await new Promise((resolve, reject) => {
+        connection.once("open", resolve);
+        connection.on("error", reject);
+      });
+    }
+  }
+  const modelConnection = connections[dbName];
+  const model = models[modelName];
+  assert(model, `Model ${modelName} not registered. Available models: ${Object.keys(models).join(", ")}`);
+  if (!modelConnection.models[modelName]) {
+    modelConnection.model(modelName, model.schema);
+  }
+  return modelConnection.models[modelName];
+};
 export {
   TenantSchema,
   UserSchema,
   ZTenant,
   ZUser,
   extendZod,
+  loadModel,
   z2 as z,
   zId,
   zUUID,

package/dist/loadModel.d.ts

@@ -0,0 +1,13 @@
+import { default as mongoose } from '../../vite/node_modules/mongoose';
+type LoadModelCtx = {
+    req: {
+        session?: {
+            user?: {
+                current_tenant_id?: string;
+            };
+        } | null;
+    };
+};
+export declare const loadModel: (modelName: string, ctx: LoadModelCtx) => Promise<mongoose.Model<any, {}, {}, {}, any, any>>;
+export type { LoadModelCtx };
+//# sourceMappingURL=loadModel.d.ts.map

package/dist/loadModel.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"loadModel.d.ts","sourceRoot":"","sources":["../src/loadModel.ts"],"names":[],"mappings":"AAEA,OAAO,QAAQ,MAAM,UAAU,CAAA;AAK/B,KAAK,YAAY,GAAG;IAClB,GAAG,EAAE;QACH,OAAO,CAAC,EAAE;YACR,IAAI,CAAC,EAAE;gBACL,iBAAiB,CAAC,EAAE,MAAM,CAAC;aAC5B,CAAA;SACF,GAAG,IAAI,CAAC;KACV,CAAC;CACH,CAAC;AA+CF,eAAO,MAAM,SAAS,GAAU,WAAW,MAAM,EAAE,KAAK,YAAY,uDA2CnE,CAAA;AAED,YAAY,EAAE,YAAY,EAAE,CAAA"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@rpcbase/db",
-  "version": "0.2.0",
+  "version": "0.3.0",
   "type": "module",
   "files": [
     "dist"