orchid-orm 1.15.2 → 1.16.0

package/dist/index.mjs

@@ -4,39 +4,36 @@ import { getStackTrace, applyMixins, getCallerFilePath, snakeCaseKey, toSnakeCas
 import { AsyncLocalStorage } from 'node:async_hooks';
 
 const createBaseTable = ({
-  columnTypes: columnTypes$1,
+  columnTypes: columnTypesArg,
   snakeCase,
-  filePath,
+  filePath: filePathArg,
   nowSQL,
-  exportAs,
-  language
-} = { columnTypes: columnTypes }) => {
-  const ct = typeof columnTypes$1 === "function" ? columnTypes$1(columnTypes) : columnTypes$1 || columnTypes;
-  return create(
-    ct,
-    // stack is needed only if filePath wasn't given
-    filePath || getStackTrace(),
-    snakeCase,
-    nowSQL,
-    exportAs,
-    language
-  );
-};
-const create = (columnTypes, filePathOrStack, snakeCase, nowSQL, exportAs = "BaseTable", language) => {
+  exportAs = "BaseTable",
+  language,
+  schemaProvider: schemaProviderArg
+} = {}) => {
   var _a;
+  const columnTypes$1 = typeof columnTypesArg === "function" ? columnTypesArg(columnTypes) : columnTypesArg || columnTypes;
+  const filePathOrStack = filePathArg || getStackTrace();
   let filePath;
+  function schemaProvider() {
+    const schema = schemaProviderArg.call(this);
+    this.schema = () => schema;
+    return schema;
+  }
   const base = (_a = class {
     constructor() {
       this.snakeCase = snakeCase;
-      this.columnTypes = columnTypes;
+      this.columnTypes = columnTypes$1;
       this.q = {};
       this.language = language;
     }
     static getFilePath() {
       if (filePath)
         return filePath;
-      if (typeof filePathOrStack === "string")
+      if (typeof filePathOrStack === "string") {
         return filePath = filePathOrStack;
+      }
       filePath = getCallerFilePath(filePathOrStack);
       if (filePath)
         return filePath;
@@ -60,8 +57,8 @@ const create = (columnTypes, filePathOrStack, snakeCase, nowSQL, exportAs = "Bas
       );
     }
     setColumns(fn) {
-      columnTypes[snakeCaseKey] = this.snakeCase;
-      const shape = getColumnTypes(columnTypes, fn, nowSQL, this.language);
+      columnTypes$1[snakeCaseKey] = this.snakeCase;
+      const shape = getColumnTypes(columnTypes$1, fn, nowSQL, this.language);
       if (this.snakeCase) {
         for (const key in shape) {
           const column = shape[key];
@@ -73,10 +70,7 @@ const create = (columnTypes, filePathOrStack, snakeCase, nowSQL, exportAs = "Bas
           }
         }
       }
-      return {
-        shape,
-        type: void 0
-      };
+      return this.constructor.prototype.columns = shape;
     }
     belongsTo(fn, options) {
       return {
@@ -106,9 +100,9 @@ const create = (columnTypes, filePathOrStack, snakeCase, nowSQL, exportAs = "Bas
         options
       };
     }
-  }, _a.nowSQL = nowSQL, _a.exportAs = exportAs, _a);
+  }, _a.nowSQL = nowSQL, _a.exportAs = exportAs, _a.schema = schemaProvider, _a);
   applyMixins(base, [QueryHooks]);
-  base.prototype.columnTypes = columnTypes;
+  base.prototype.columnTypes = columnTypes$1;
   return base;
 };
 
@@ -1712,7 +1706,7 @@ const orchidORM = (_a, tables) => {
       // eslint-disable-next-line @typescript-eslint/no-explicit-any
       qb,
       table.table,
-      table.columns.shape,
+      table.columns,
       table.columnTypes,
       transactionStorage,
       options2