drizzle-multitenant 1.0.0

package/dist/cross-schema/index.js

@@ -0,0 +1,219 @@
+import { sql, getTableName } from 'drizzle-orm';
+
+// src/cross-schema/cross-schema.ts
+var CrossSchemaQueryBuilder = class {
+  constructor(context) {
+    this.context = context;
+  }
+  fromTable = null;
+  joins = [];
+  selectFields = {};
+  whereCondition = null;
+  orderByFields = [];
+  limitValue = null;
+  offsetValue = null;
+  /**
+   * Set the main table to query from
+   */
+  from(source, table) {
+    const schemaName = this.getSchemaName(source);
+    this.fromTable = { table, source, schemaName };
+    return this;
+  }
+  /**
+   * Add an inner join
+   */
+  innerJoin(source, table, condition) {
+    return this.addJoin(source, table, condition, "inner");
+  }
+  /**
+   * Add a left join
+   */
+  leftJoin(source, table, condition) {
+    return this.addJoin(source, table, condition, "left");
+  }
+  /**
+   * Add a right join
+   */
+  rightJoin(source, table, condition) {
+    return this.addJoin(source, table, condition, "right");
+  }
+  /**
+   * Add a full outer join
+   */
+  fullJoin(source, table, condition) {
+    return this.addJoin(source, table, condition, "full");
+  }
+  /**
+   * Select specific fields
+   */
+  select(fields) {
+    this.selectFields = fields;
+    return this;
+  }
+  /**
+   * Add a where condition
+   */
+  where(condition) {
+    this.whereCondition = condition;
+    return this;
+  }
+  /**
+   * Add order by
+   */
+  orderBy(...fields) {
+    this.orderByFields = fields;
+    return this;
+  }
+  /**
+   * Set limit
+   */
+  limit(value) {
+    this.limitValue = value;
+    return this;
+  }
+  /**
+   * Set offset
+   */
+  offset(value) {
+    this.offsetValue = value;
+    return this;
+  }
+  /**
+   * Execute the query and return typed results
+   */
+  async execute() {
+    if (!this.fromTable) {
+      throw new Error("[drizzle-multitenant] No table specified. Use .from() first.");
+    }
+    const sqlQuery = this.buildSql();
+    const result = await this.context.tenantDb.execute(sqlQuery);
+    return result.rows;
+  }
+  /**
+   * Build the SQL query
+   */
+  buildSql() {
+    if (!this.fromTable) {
+      throw new Error("[drizzle-multitenant] No table specified");
+    }
+    const parts = [];
+    const selectParts = Object.entries(this.selectFields).map(([alias, column]) => {
+      const columnName = column.name;
+      return sql`${sql.raw(`"${columnName}"`)} as ${sql.raw(`"${alias}"`)}`;
+    });
+    if (selectParts.length === 0) {
+      parts.push(sql`SELECT *`);
+    } else {
+      parts.push(sql`SELECT ${sql.join(selectParts, sql`, `)}`);
+    }
+    const fromTableRef = this.getFullTableName(this.fromTable.schemaName, this.fromTable.table);
+    parts.push(sql` FROM ${sql.raw(fromTableRef)}`);
+    for (const join of this.joins) {
+      const joinTableRef = this.getFullTableName(join.schemaName, join.table);
+      const joinType = this.getJoinKeyword(join.type);
+      parts.push(sql` ${sql.raw(joinType)} ${sql.raw(joinTableRef)} ON ${join.condition}`);
+    }
+    if (this.whereCondition) {
+      parts.push(sql` WHERE ${this.whereCondition}`);
+    }
+    if (this.orderByFields.length > 0) {
+      parts.push(sql` ORDER BY ${sql.join(this.orderByFields, sql`, `)}`);
+    }
+    if (this.limitValue !== null) {
+      parts.push(sql` LIMIT ${sql.raw(this.limitValue.toString())}`);
+    }
+    if (this.offsetValue !== null) {
+      parts.push(sql` OFFSET ${sql.raw(this.offsetValue.toString())}`);
+    }
+    return sql.join(parts, sql``);
+  }
+  /**
+   * Add a join to the query
+   */
+  addJoin(source, table, condition, type) {
+    const schemaName = this.getSchemaName(source);
+    this.joins.push({ table, source, schemaName, condition, type });
+    return this;
+  }
+  /**
+   * Get schema name for a source
+   */
+  getSchemaName(source) {
+    if (source === "tenant") {
+      return this.context.tenantSchema ?? "tenant";
+    }
+    return this.context.sharedSchema ?? "public";
+  }
+  /**
+   * Get fully qualified table name
+   */
+  getFullTableName(schemaName, table) {
+    const tableName = getTableName(table);
+    return `"${schemaName}"."${tableName}"`;
+  }
+  /**
+   * Get SQL keyword for join type
+   */
+  getJoinKeyword(type) {
+    switch (type) {
+      case "inner":
+        return "INNER JOIN";
+      case "left":
+        return "LEFT JOIN";
+      case "right":
+        return "RIGHT JOIN";
+      case "full":
+        return "FULL OUTER JOIN";
+    }
+  }
+};
+function createCrossSchemaQuery(context) {
+  return new CrossSchemaQueryBuilder(context);
+}
+async function withSharedLookup(config) {
+  const {
+    tenantDb,
+    tenantTable,
+    sharedTable,
+    foreignKey,
+    sharedKey = "id",
+    sharedFields,
+    where: whereCondition
+  } = config;
+  const tenantTableName = getTableName(tenantTable);
+  const sharedTableName = getTableName(sharedTable);
+  const sharedFieldList = sharedFields.map((field) => `s."${String(field)}"`).join(", ");
+  const queryParts = [
+    `SELECT t.*, ${sharedFieldList}`,
+    `FROM "${tenantTableName}" t`,
+    `LEFT JOIN "public"."${sharedTableName}" s ON t."${String(foreignKey)}" = s."${String(sharedKey)}"`
+  ];
+  if (whereCondition) {
+    queryParts.push("WHERE");
+  }
+  const sqlQuery = sql.raw(queryParts.join(" "));
+  const result = await tenantDb.execute(sqlQuery);
+  return result.rows;
+}
+async function crossSchemaRaw(db, options) {
+  const { tenantSchema, sharedSchema, sql: rawSql } = options;
+  const processedSql = rawSql.replace(/\$tenant\./g, `"${tenantSchema}".`).replace(/\$shared\./g, `"${sharedSchema}".`);
+  const query = sql.raw(processedSql);
+  const result = await db.execute(query);
+  return result.rows;
+}
+function buildCrossSchemaSelect(fields, tenantSchema, _sharedSchema) {
+  const columns = Object.entries(fields).map(([alias, column]) => {
+    const columnName = column.name;
+    return `"${columnName}" as "${alias}"`;
+  });
+  const getSchema = () => {
+    return tenantSchema;
+  };
+  return { columns, getSchema };
+}
+
+export { CrossSchemaQueryBuilder, buildCrossSchemaSelect, createCrossSchemaQuery, crossSchemaRaw, withSharedLookup };
+//# sourceMappingURL=index.js.map
+//# sourceMappingURL=index.js.map

package/dist/cross-schema/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/cross-schema/cross-schema.ts"],"names":[],"mappings":";;;AAgCO,IAAM,0BAAN,MAGL;AAAA,EAeA,YAA6B,OAAA,EAA2D;AAA3D,IAAA,IAAA,CAAA,OAAA,GAAA,OAAA;AAAA,EAA4D;AAAA,EAdjF,SAAA,GAA+E,IAAA;AAAA,EAC/E,QAMH,EAAC;AAAA,EACE,eAAuC,EAAC;AAAA,EACxC,cAAA,GAAsC,IAAA;AAAA,EACtC,gBAAgC,EAAC;AAAA,EACjC,UAAA,GAA4B,IAAA;AAAA,EAC5B,WAAA,GAA6B,IAAA;AAAA;AAAA;AAAA;AAAA,EAOrC,IAAA,CAAsB,QAAsB,KAAA,EAAgB;AAC1D,IAAA,MAAM,UAAA,GAAa,IAAA,CAAK,aAAA,CAAc,MAAM,CAAA;AAC5C,IAAA,IAAA,CAAK,SAAA,GAAY,EAAE,KAAA,EAAO,MAAA,EAAQ,UAAA,EAAW;AAC7C,IAAA,OAAO,IAAA;AAAA,EACT;AAAA;AAAA;AAAA;AAAA,EAKA,SAAA,CAA2B,MAAA,EAAsB,KAAA,EAAU,SAAA,EAAgC;AACzF,IAAA,OAAO,IAAA,CAAK,OAAA,CAAQ,MAAA,EAAQ,KAAA,EAAO,WAAW,OAAO,CAAA;AAAA,EACvD;AAAA;AAAA;AAAA;AAAA,EAKA,QAAA,CAA0B,MAAA,EAAsB,KAAA,EAAU,SAAA,EAAgC;AACxF,IAAA,OAAO,IAAA,CAAK,OAAA,CAAQ,MAAA,EAAQ,KAAA,EAAO,WAAW,MAAM,CAAA;AAAA,EACtD;AAAA;AAAA;AAAA;AAAA,EAKA,SAAA,CAA2B,MAAA,EAAsB,KAAA,EAAU,SAAA,EAAgC;AACzF,IAAA,OAAO,IAAA,CAAK,OAAA,CAAQ,MAAA,EAAQ,KAAA,EAAO,WAAW,OAAO,CAAA;AAAA,EACvD;AAAA;AAAA;AAAA;AAAA,EAKA,QAAA,CAA0B,MAAA,EAAsB,KAAA,EAAU,SAAA,EAAgC;AACxF,IAAA,OAAO,IAAA,CAAK,OAAA,CAAQ,MAAA,EAAQ,KAAA,EAAO,WAAW,MAAM,CAAA;AAAA,EACtD;AAAA;AAAA;AAAA;AAAA,EAKA,OAAyC,MAAA,EAAiB;AACxD,IAAA,IAAA,CAAK,YAAA,GAAe,MAAA;AACpB,IAAA,OAAO,IAAA;AAAA,EACT;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,SAAA,EAA+B;AACnC,IAAA,IAAA,CAAK,cAAA,GAAiB,SAAA;AACtB,IAAA,OAAO,IAAA;AAAA,EACT;AAAA;AAAA;AAAA;AAAA,EAKA,WAAW,MAAA,EAA8B;AACvC,IAAA,IAAA,CAAK,aAAA,GAAgB,MAAA;AACrB,IAAA,OAAO,IAAA;AAAA,EACT;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,KAAA,EAAqB;AACzB,IAAA,IAAA,CAAK,UAAA,GAAa,KAAA;AAClB,IAAA,OAAO,IAAA;AAAA,EACT;AAAA;AAAA;AAAA;AAAA,EAKA,OAAO,KAAA,EAAqB;AAC1B,IAAA,IAAA,CAAK,WAAA,GAAc,KAAA;AACnB,IAAA,OAAO,IAAA;AAAA,EACT;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,OAAA,GAAiE;AACrE,IAAA,IAAI,CAAC,KAAK,SAAA,EAAW;AACnB,MAAA,MAAM,IAAI,MAAM,8DAA8D,CAAA;AAAA,IAChF;AAEA,IAAA,MAAM,QAAA,GAAW,KAAK,QAAA,EAAS;AAG/B,IAAA,MAAM,SAAS,MAAM,IAAA,CAAK,OAAA,CAAQ,QAAA,CAAS,QAAQ,QAAQ,CAAA;AAE3D,IAAA,OAAO,MAAA,CAAO,IAAA;AAAA,EAChB;AAAA;AAAA;AAAA;AAAA,EAKQ,QAAA,GAAyB;AAC/B,IAAA,IAAI,CAAC,KAAK,SAAA,EAAW;AACnB,MAAA,MAAM,IAAI,MAAM,0CAA0C,CAAA;AAAA,IAC5D;AAEA,IAAA,MAAM,QAAwB,EAAC;AAG/B,IAAA,MAAM,WAAA,GAAc,MAAA,CAAO,OAAA,CAAQ,IAAA,CAAK,YAAY,CAAA,CAAE,GAAA,CAAI,CAAC,CAAC,KAAA,EAAO,MAAM,CAAA,KAAM;AAC7E,MAAA,MAAM,aAAa,MAAA,CAAO,IAAA;AAC1B,MAAA,OAAO,GAAA,CAAA,EAAM,GAAA,CAAI,GAAA,CAAI,CAAA,CAAA,EAAI,UAAU,CAAA,CAAA,CAAG,CAAC,CAAA,IAAA,EAAO,GAAA,CAAI,GAAA,CAAI,CAAA,CAAA,EAAI,KAAK,GAAG,CAAC,CAAA,CAAA;AAAA,IACrE,CAAC,CAAA;AAED,IAAA,IAAI,WAAA,CAAY,WAAW,CAAA,EAAG;AAC5B,MAAA,KAAA,CAAM,KAAK,GAAA,CAAA,QAAA,CAAa,CAAA;AAAA,IAC1B,CAAA,MAAO;AACL,MAAA,KAAA,CAAM,KAAK,GAAA,CAAA,OAAA,EAAa,GAAA,CAAI,KAAK,WAAA,EAAa,GAAA,CAAA,EAAA,CAAO,CAAC,CAAA,CAAE,CAAA;AAAA,IAC1D;AAGA,IAAA,MAAM,YAAA,GAAe,KAAK,gBAAA,CAAiB,IAAA,CAAK,UAAU,UAAA,EAAY,IAAA,CAAK,UAAU,KAAK,CAAA;AAC1F,IAAA,KAAA,CAAM,KAAK,GAAA,CAAA,MAAA,EAAY,GAAA,CAAI,GAAA,CAAI,YAAY,CAAC,CAAA,CAAE,CAAA;AAG9C,IAAA,KAAA,MAAW,IAAA,IAAQ,KAAK,KAAA,EAAO;AAC7B,MAAA,MAAM,eAAe,IAAA,CAAK,gBAAA,CAAiB,IAAA,CAAK,UAAA,EAAY,KAAK,KAAK,CAAA;AACtE,MAAA,MAAM,QAAA,GAAW,IAAA,CAAK,cAAA,CAAe,IAAA,CAAK,IAAI,CAAA;AAC9C,MAAA,KAAA,CAAM,IAAA,CAAK,GAAA,CAAA,CAAA,EAAO,GAAA,CAAI,GAAA,CAAI,QAAQ,CAAC,CAAA,CAAA,EAAI,GAAA,CAAI,GAAA,CAAI,YAAY,CAAC,CAAA,IAAA,EAAO,IAAA,CAAK,SAAS,CAAA,CAAE,CAAA;AAAA,IACrF;AAGA,IAAA,IAAI,KAAK,cAAA,EAAgB;AACvB,MAAA,KAAA,CAAM,IAAA,CAAK,GAAA,CAAA,OAAA,EAAa,IAAA,CAAK,cAAc,CAAA,CAAE,CAAA;AAAA,IAC/C;AAGA,IAAA,IAAI,IAAA,CAAK,aAAA,CAAc,MAAA,GAAS,CAAA,EAAG;AACjC,MAAA,KAAA,CAAM,IAAA,CAAK,gBAAgB,GAAA,CAAI,IAAA,CAAK,KAAK,aAAA,EAAe,GAAA,CAAA,EAAA,CAAO,CAAC,CAAA,CAAE,CAAA;AAAA,IACpE;AAGA,IAAA,IAAI,IAAA,CAAK,eAAe,IAAA,EAAM;AAC5B,MAAA,KAAA,CAAM,IAAA,CAAK,aAAa,GAAA,CAAI,GAAA,CAAI,KAAK,UAAA,CAAW,QAAA,EAAU,CAAC,CAAA,CAAE,CAAA;AAAA,IAC/D;AAGA,IAAA,IAAI,IAAA,CAAK,gBAAgB,IAAA,EAAM;AAC7B,MAAA,KAAA,CAAM,IAAA,CAAK,cAAc,GAAA,CAAI,GAAA,CAAI,KAAK,WAAA,CAAY,QAAA,EAAU,CAAC,CAAA,CAAE,CAAA;AAAA,IACjE;AAEA,IAAA,OAAO,GAAA,CAAI,IAAA,CAAK,KAAA,EAAO,GAAA,CAAA,CAAK,CAAA;AAAA,EAC9B;AAAA;AAAA;AAAA;AAAA,EAKQ,OAAA,CACN,MAAA,EACA,KAAA,EACA,SAAA,EACA,IAAA,EACM;AACN,IAAA,MAAM,UAAA,GAAa,IAAA,CAAK,aAAA,CAAc,MAAM,CAAA;AAC5C,IAAA,IAAA,CAAK,KAAA,CAAM,KAAK,EAAE,KAAA,EAAO,QAAQ,UAAA,EAAY,SAAA,EAAW,MAAM,CAAA;AAC9D,IAAA,OAAO,IAAA;AAAA,EACT;AAAA;AAAA;AAAA;AAAA,EAKQ,cAAc,MAAA,EAA8B;AAClD,IAAA,IAAI,WAAW,QAAA,EAAU;AACvB,MAAA,OAAO,IAAA,CAAK,QAAQ,YAAA,IAAgB,QAAA;AAAA,IACtC;AACA,IAAA,OAAO,IAAA,CAAK,QAAQ,YAAA,IAAgB,QAAA;AAAA,EACtC;AAAA;AAAA;AAAA;AAAA,EAKQ,gBAAA,CAAiB,YAAoB,KAAA,EAAsB;AACjE,IAAA,MAAM,SAAA,GAAY,aAAa,KAAK,CAAA;AACpC,IAAA,OAAO,CAAA,CAAA,EAAI,UAAU,CAAA,GAAA,EAAM,SAAS,CAAA,CAAA,CAAA;AAAA,EACtC;AAAA;AAAA;AAAA;AAAA,EAKQ,eAAe,IAAA,EAAwB;AAC7C,IAAA,QAAQ,IAAA;AAAM,MACZ,KAAK,OAAA;AACH,QAAA,OAAO,YAAA;AAAA,MACT,KAAK,MAAA;AACH,QAAA,OAAO,WAAA;AAAA,MACT,KAAK,OAAA;AACH,QAAA,OAAO,YAAA;AAAA,MACT,KAAK,MAAA;AACH,QAAA,OAAO,iBAAA;AAAA;AACX,EACF;AACF;AAKO,SAAS,uBAId,OAAA,EACuD;AACvD,EAAA,OAAO,IAAI,wBAAwB,OAAO,CAAA;AAC5C;AAiBA,eAAsB,iBAIpB,MAAA,EAAgH;AAChH,EAAA,MAAM;AAAA,IACJ,QAAA;AAAA,IACA,WAAA;AAAA,IACA,WAAA;AAAA,IACA,UAAA;AAAA,IACA,SAAA,GAAY,IAAA;AAAA,IACZ,YAAA;AAAA,IACA,KAAA,EAAO;AAAA,GACT,GAAI,MAAA;AAGJ,EAAA,MAAM,eAAA,GAAkB,aAAa,WAAW,CAAA;AAChD,EAAA,MAAM,eAAA,GAAkB,aAAa,WAAW,CAAA;AAGhD,EAAA,MAAM,eAAA,GAAkB,YAAA,CACrB,GAAA,CAAI,CAAC,KAAA,KAAU,CAAA,GAAA,EAAM,MAAA,CAAO,KAAK,CAAC,CAAA,CAAA,CAAG,CAAA,CACrC,IAAA,CAAK,IAAI,CAAA;AAGZ,EAAA,MAAM,UAAA,GAAa;AAAA,IACjB,eAAe,eAAe,CAAA,CAAA;AAAA,IAC9B,SAAS,eAAe,CAAA,GAAA,CAAA;AAAA,IACxB,CAAA,oBAAA,EAAuB,eAAe,CAAA,UAAA,EAAa,MAAA,CAAO,UAAU,CAAC,CAAA,OAAA,EAAU,MAAA,CAAO,SAAS,CAAC,CAAA,CAAA;AAAA,GAClG;AAEA,EAAA,IAAI,cAAA,EAAgB;AAClB,IAAA,UAAA,CAAW,KAAK,OAAO,CAAA;AAAA,EAEzB;AAEA,EAAA,MAAM,WAAW,GAAA,CAAI,GAAA,CAAI,UAAA,CAAW,IAAA,CAAK,GAAG,CAAC,CAAA;AAE7C,EAAA,MAAM,MAAA,GAAS,MAAM,QAAA,CAAS,OAAA,CAAQ,QAAQ,CAAA;AAE9C,EAAA,OAAO,MAAA,CAAO,IAAA;AAChB;AAkBA,eAAsB,cAAA,CACpB,IACA,OAAA,EACoB;AACpB,EAAA,MAAM,EAAE,YAAA,EAAc,YAAA,EAAc,GAAA,EAAK,QAAO,GAAI,OAAA;AAGpD,EAAA,MAAM,YAAA,GAAe,MAAA,CAClB,OAAA,CAAQ,aAAA,EAAe,CAAA,CAAA,EAAI,YAAY,CAAA,EAAA,CAAI,CAAA,CAC3C,OAAA,CAAQ,aAAA,EAAe,CAAA,CAAA,EAAI,YAAY,CAAA,EAAA,CAAI,CAAA;AAE9C,EAAA,MAAM,KAAA,GAAQ,GAAA,CAAI,GAAA,CAAI,YAAY,CAAA;AAElC,EAAA,MAAM,MAAA,GAAS,MAAM,EAAA,CAAG,OAAA,CAAQ,KAAK,CAAA;AAErC,EAAA,OAAO,MAAA,CAAO,IAAA;AAChB;AAsBO,SAAS,sBAAA,CACd,MAAA,EACA,YAAA,EACA,aAAA,EACgD;AAChD,EAAA,MAAM,OAAA,GAAU,MAAA,CAAO,OAAA,CAAQ,MAAM,CAAA,CAAE,IAAI,CAAC,CAAC,KAAA,EAAO,MAAM,CAAA,KAAM;AAC9D,IAAA,MAAM,aAAa,MAAA,CAAO,IAAA;AAC1B,IAAA,OAAO,CAAA,CAAA,EAAI,UAAU,CAAA,MAAA,EAAS,KAAK,CAAA,CAAA,CAAA;AAAA,EACrC,CAAC,CAAA;AAED,EAAA,MAAM,YAAY,MAAc;AAG9B,IAAA,OAAO,YAAA;AAAA,EACT,CAAA;AAEA,EAAA,OAAO,EAAE,SAAS,SAAA,EAAU;AAC9B","file":"index.js","sourcesContent":["import { sql, getTableName } from 'drizzle-orm';\nimport type { NodePgDatabase } from 'drizzle-orm/node-postgres';\nimport type { SQL, Table, Column } from 'drizzle-orm';\nimport type {\n  CrossSchemaContext,\n  SchemaSource,\n  JoinType,\n  JoinCondition,\n  SharedLookupConfig,\n  CrossSchemaRawOptions,\n} from './types.js';\n\n/**\n * Cross-schema query builder for joining tenant and shared data\n *\n * @example\n * ```typescript\n * const query = createCrossSchemaQuery({\n *   tenantDb: tenants.getDb('tenant-uuid'),\n *   sharedDb: tenants.getSharedDb(),\n * });\n *\n * const results = await query\n *   .from('tenant', orders)\n *   .leftJoin('shared', subscriptionPlans, eq(orders.planId, subscriptionPlans.id))\n *   .select({\n *     orderId: orders.id,\n *     planName: subscriptionPlans.name,\n *   })\n *   .execute();\n * ```\n */\nexport class CrossSchemaQueryBuilder<\n  TTenantSchema extends Record<string, unknown> = Record<string, unknown>,\n  TSharedSchema extends Record<string, unknown> = Record<string, unknown>,\n> {\n  private fromTable: { table: Table; source: SchemaSource; schemaName: string } | null = null;\n  private joins: Array<{\n    table: Table;\n    source: SchemaSource;\n    schemaName: string;\n    condition: JoinCondition;\n    type: JoinType;\n  }> = [];\n  private selectFields: Record<string, Column> = {};\n  private whereCondition: SQL<unknown> | null = null;\n  private orderByFields: SQL<unknown>[] = [];\n  private limitValue: number | null = null;\n  private offsetValue: number | null = null;\n\n  constructor(private readonly context: CrossSchemaContext<TTenantSchema, TSharedSchema>) {}\n\n  /**\n   * Set the main table to query from\n   */\n  from<T extends Table>(source: SchemaSource, table: T): this {\n    const schemaName = this.getSchemaName(source);\n    this.fromTable = { table, source, schemaName };\n    return this;\n  }\n\n  /**\n   * Add an inner join\n   */\n  innerJoin<T extends Table>(source: SchemaSource, table: T, condition: JoinCondition): this {\n    return this.addJoin(source, table, condition, 'inner');\n  }\n\n  /**\n   * Add a left join\n   */\n  leftJoin<T extends Table>(source: SchemaSource, table: T, condition: JoinCondition): this {\n    return this.addJoin(source, table, condition, 'left');\n  }\n\n  /**\n   * Add a right join\n   */\n  rightJoin<T extends Table>(source: SchemaSource, table: T, condition: JoinCondition): this {\n    return this.addJoin(source, table, condition, 'right');\n  }\n\n  /**\n   * Add a full outer join\n   */\n  fullJoin<T extends Table>(source: SchemaSource, table: T, condition: JoinCondition): this {\n    return this.addJoin(source, table, condition, 'full');\n  }\n\n  /**\n   * Select specific fields\n   */\n  select<T extends Record<string, Column>>(fields: T): this {\n    this.selectFields = fields;\n    return this;\n  }\n\n  /**\n   * Add a where condition\n   */\n  where(condition: SQL<unknown>): this {\n    this.whereCondition = condition;\n    return this;\n  }\n\n  /**\n   * Add order by\n   */\n  orderBy(...fields: SQL<unknown>[]): this {\n    this.orderByFields = fields;\n    return this;\n  }\n\n  /**\n   * Set limit\n   */\n  limit(value: number): this {\n    this.limitValue = value;\n    return this;\n  }\n\n  /**\n   * Set offset\n   */\n  offset(value: number): this {\n    this.offsetValue = value;\n    return this;\n  }\n\n  /**\n   * Execute the query and return typed results\n   */\n  async execute<TResult = Record<string, unknown>>(): Promise<TResult[]> {\n    if (!this.fromTable) {\n      throw new Error('[drizzle-multitenant] No table specified. Use .from() first.');\n    }\n\n    const sqlQuery = this.buildSql();\n\n    // Use the tenant db to execute (it has access to both schemas via search_path)\n    const result = await this.context.tenantDb.execute(sqlQuery);\n\n    return result.rows as TResult[];\n  }\n\n  /**\n   * Build the SQL query\n   */\n  private buildSql(): SQL<unknown> {\n    if (!this.fromTable) {\n      throw new Error('[drizzle-multitenant] No table specified');\n    }\n\n    const parts: SQL<unknown>[] = [];\n\n    // SELECT clause\n    const selectParts = Object.entries(this.selectFields).map(([alias, column]) => {\n      const columnName = column.name;\n      return sql`${sql.raw(`\"${columnName}\"`)} as ${sql.raw(`\"${alias}\"`)}`;\n    });\n\n    if (selectParts.length === 0) {\n      parts.push(sql`SELECT *`);\n    } else {\n      parts.push(sql`SELECT ${sql.join(selectParts, sql`, `)}`);\n    }\n\n    // FROM clause\n    const fromTableRef = this.getFullTableName(this.fromTable.schemaName, this.fromTable.table);\n    parts.push(sql` FROM ${sql.raw(fromTableRef)}`);\n\n    // JOIN clauses\n    for (const join of this.joins) {\n      const joinTableRef = this.getFullTableName(join.schemaName, join.table);\n      const joinType = this.getJoinKeyword(join.type);\n      parts.push(sql` ${sql.raw(joinType)} ${sql.raw(joinTableRef)} ON ${join.condition}`);\n    }\n\n    // WHERE clause\n    if (this.whereCondition) {\n      parts.push(sql` WHERE ${this.whereCondition}`);\n    }\n\n    // ORDER BY clause\n    if (this.orderByFields.length > 0) {\n      parts.push(sql` ORDER BY ${sql.join(this.orderByFields, sql`, `)}`);\n    }\n\n    // LIMIT clause\n    if (this.limitValue !== null) {\n      parts.push(sql` LIMIT ${sql.raw(this.limitValue.toString())}`);\n    }\n\n    // OFFSET clause\n    if (this.offsetValue !== null) {\n      parts.push(sql` OFFSET ${sql.raw(this.offsetValue.toString())}`);\n    }\n\n    return sql.join(parts, sql``);\n  }\n\n  /**\n   * Add a join to the query\n   */\n  private addJoin<T extends Table>(\n    source: SchemaSource,\n    table: T,\n    condition: JoinCondition,\n    type: JoinType\n  ): this {\n    const schemaName = this.getSchemaName(source);\n    this.joins.push({ table, source, schemaName, condition, type });\n    return this;\n  }\n\n  /**\n   * Get schema name for a source\n   */\n  private getSchemaName(source: SchemaSource): string {\n    if (source === 'tenant') {\n      return this.context.tenantSchema ?? 'tenant';\n    }\n    return this.context.sharedSchema ?? 'public';\n  }\n\n  /**\n   * Get fully qualified table name\n   */\n  private getFullTableName(schemaName: string, table: Table): string {\n    const tableName = getTableName(table);\n    return `\"${schemaName}\".\"${tableName}\"`;\n  }\n\n  /**\n   * Get SQL keyword for join type\n   */\n  private getJoinKeyword(type: JoinType): string {\n    switch (type) {\n      case 'inner':\n        return 'INNER JOIN';\n      case 'left':\n        return 'LEFT JOIN';\n      case 'right':\n        return 'RIGHT JOIN';\n      case 'full':\n        return 'FULL OUTER JOIN';\n    }\n  }\n}\n\n/**\n * Create a cross-schema query builder\n */\nexport function createCrossSchemaQuery<\n  TTenantSchema extends Record<string, unknown> = Record<string, unknown>,\n  TSharedSchema extends Record<string, unknown> = Record<string, unknown>,\n>(\n  context: CrossSchemaContext<TTenantSchema, TSharedSchema>\n): CrossSchemaQueryBuilder<TTenantSchema, TSharedSchema> {\n  return new CrossSchemaQueryBuilder(context);\n}\n\n/**\n * Helper for common pattern: tenant table with shared lookup\n *\n * @example\n * ```typescript\n * const ordersWithPlans = await withSharedLookup({\n *   tenantDb,\n *   sharedDb,\n *   tenantTable: orders,\n *   sharedTable: subscriptionPlans,\n *   foreignKey: 'planId',\n *   sharedFields: ['name', 'features', 'price'],\n * });\n * ```\n */\nexport async function withSharedLookup<\n  TTenantTable extends Table,\n  TSharedTable extends Table,\n  TSharedFields extends keyof TSharedTable['_']['columns'],\n>(config: SharedLookupConfig<TTenantTable, TSharedTable, TSharedFields>): Promise<Array<Record<string, unknown>>> {\n  const {\n    tenantDb,\n    tenantTable,\n    sharedTable,\n    foreignKey,\n    sharedKey = 'id' as keyof TSharedTable['_']['columns'],\n    sharedFields,\n    where: whereCondition,\n  } = config;\n\n  // Get table names using Drizzle's utility\n  const tenantTableName = getTableName(tenantTable);\n  const sharedTableName = getTableName(sharedTable);\n\n  // Build field list for shared table\n  const sharedFieldList = sharedFields\n    .map((field) => `s.\"${String(field)}\"`)\n    .join(', ');\n\n  // Build the query\n  const queryParts = [\n    `SELECT t.*, ${sharedFieldList}`,\n    `FROM \"${tenantTableName}\" t`,\n    `LEFT JOIN \"public\".\"${sharedTableName}\" s ON t.\"${String(foreignKey)}\" = s.\"${String(sharedKey)}\"`,\n  ];\n\n  if (whereCondition) {\n    queryParts.push('WHERE');\n    // We'll need to use raw SQL for the where condition\n  }\n\n  const sqlQuery = sql.raw(queryParts.join(' '));\n\n  const result = await tenantDb.execute(sqlQuery);\n\n  return result.rows as Array<Record<string, unknown>>;\n}\n\n/**\n * Execute raw cross-schema SQL with type safety\n *\n * @example\n * ```typescript\n * const result = await crossSchemaRaw<{ userName: string; planName: string }>({\n *   tenantSchema: 'tenant_abc123',\n *   sharedSchema: 'public',\n *   sql: `\n *     SELECT u.name as \"userName\", p.name as \"planName\"\n *     FROM $tenant.users u\n *     JOIN $shared.plans p ON u.plan_id = p.id\n *   `,\n * });\n * ```\n */\nexport async function crossSchemaRaw<TResult = Record<string, unknown>>(\n  db: NodePgDatabase<Record<string, unknown>>,\n  options: CrossSchemaRawOptions\n): Promise<TResult[]> {\n  const { tenantSchema, sharedSchema, sql: rawSql } = options;\n\n  // Replace $tenant and $shared placeholders\n  const processedSql = rawSql\n    .replace(/\\$tenant\\./g, `\"${tenantSchema}\".`)\n    .replace(/\\$shared\\./g, `\"${sharedSchema}\".`);\n\n  const query = sql.raw(processedSql);\n\n  const result = await db.execute(query);\n\n  return result.rows as TResult[];\n}\n\n/**\n * Create a typed cross-schema query using SQL template\n *\n * @example\n * ```typescript\n * const users = await crossSchemaSelect(tenantDb, {\n *   tenantSchema: 'tenant_abc',\n *   sharedSchema: 'public',\n *   select: {\n *     id: users.id,\n *     name: users.name,\n *     planName: plans.name,\n *   },\n *   from: { table: users, schema: 'tenant' },\n *   joins: [\n *     { table: plans, schema: 'shared', on: eq(users.planId, plans.id), type: 'left' },\n *   ],\n * });\n * ```\n */\nexport function buildCrossSchemaSelect<T extends Record<string, Column>>(\n  fields: T,\n  tenantSchema: string,\n  _sharedSchema: string\n): { columns: string[]; getSchema: () => string } {\n  const columns = Object.entries(fields).map(([alias, column]) => {\n    const columnName = column.name;\n    return `\"${columnName}\" as \"${alias}\"`;\n  });\n\n  const getSchema = (): string => {\n    // This would need more context to determine which schema a column belongs to\n    // For now, return tenant schema as default\n    return tenantSchema;\n  };\n\n  return { columns, getSchema };\n}\n"]}

package/dist/index.d.ts

@@ -0,0 +1,71 @@
+import { C as Config, T as TenantManager } from './types-DKVaTaIb.js';
+export { b as ConnectionConfig, D as DEFAULT_CONFIG, H as Hooks, I as IsolationConfig, c as IsolationStrategy, M as MetricsConfig, P as PoolEntry, d as SchemasConfig, S as SharedDb, a as TenantDb } from './types-DKVaTaIb.js';
+export { B as BaseTenantContext, a as TenantContext, T as TenantContextData, c as createTenantContext } from './context-DBerWr50.js';
+export { AppliedMigration, CreateTenantOptions, DropTenantOptions, MigrateOptions, MigrationErrorHandler, MigrationFile, MigrationHooks, MigrationProgressCallback, MigrationResults, Migrator, MigratorConfig, TenantMigrationResult, TenantMigrationStatus, createMigrator } from './migrator/index.js';
+export { ColumnSelection, CrossSchemaContext, CrossSchemaQueryBuilder, CrossSchemaRawOptions, InferSelectedColumns, JoinCondition, JoinDefinition, JoinType, LookupResult, SchemaSource, SharedLookupConfig, TableReference, buildCrossSchemaSelect, createCrossSchemaQuery, crossSchemaRaw, withSharedLookup } from './cross-schema/index.js';
+import 'pg';
+import 'drizzle-orm/node-postgres';
+import 'drizzle-orm';
+
+/**
+ * Define configuration for drizzle-multitenant
+ *
+ * @example
+ * ```typescript
+ * import { defineConfig } from 'drizzle-multitenant';
+ * import * as tenantSchema from './schemas/tenant';
+ * import * as sharedSchema from './schemas/shared';
+ *
+ * export default defineConfig({
+ *   connection: {
+ *     url: process.env.DATABASE_URL!,
+ *     poolConfig: {
+ *       max: 10,
+ *       idleTimeoutMillis: 30000,
+ *     },
+ *   },
+ *   isolation: {
+ *     strategy: 'schema',
+ *     schemaNameTemplate: (tenantId) => `tenant_${tenantId.replace(/-/g, '_')}`,
+ *     maxPools: 50,
+ *     poolTtlMs: 60 * 60 * 1000,
+ *   },
+ *   schemas: {
+ *     tenant: tenantSchema,
+ *     shared: sharedSchema,
+ *   },
+ * });
+ * ```
+ */
+declare function defineConfig<TTenantSchema extends Record<string, unknown>, TSharedSchema extends Record<string, unknown> = Record<string, unknown>>(config: Config<TTenantSchema, TSharedSchema>): Config<TTenantSchema, TSharedSchema>;
+
+/**
+ * Create a tenant manager instance
+ *
+ * @example
+ * ```typescript
+ * import { createTenantManager, defineConfig } from 'drizzle-multitenant';
+ *
+ * const config = defineConfig({
+ *   connection: { url: process.env.DATABASE_URL! },
+ *   isolation: {
+ *     strategy: 'schema',
+ *     schemaNameTemplate: (id) => `tenant_${id}`,
+ *   },
+ *   schemas: { tenant: tenantSchema },
+ * });
+ *
+ * const tenants = createTenantManager(config);
+ *
+ * // Get database for a specific tenant
+ * const db = tenants.getDb('tenant-uuid');
+ * const users = await db.select().from(schema.users);
+ *
+ * // Get shared database
+ * const sharedDb = tenants.getSharedDb();
+ * const plans = await sharedDb.select().from(sharedSchema.plans);
+ * ```
+ */
+declare function createTenantManager<TTenantSchema extends Record<string, unknown>, TSharedSchema extends Record<string, unknown> = Record<string, unknown>>(config: Config<TTenantSchema, TSharedSchema>): TenantManager<TTenantSchema, TSharedSchema>;
+
+export { Config, TenantManager, createTenantManager, defineConfig };