orchid-orm 1.9.2 → 1.9.3

package/dist/index.js

@@ -6,6 +6,7 @@ var node_async_hooks = require('node:async_hooks');
 var path = require('path');
 var fs = require('fs/promises');
 var typescript = require('typescript');
+var pluralize = require('pluralize');
 
 function _interopNamespaceDefault(e) {
   var n = Object.create(null);
@@ -49,26 +50,35 @@ const create = (columnTypes, filePath, snakeCase) => {
   const base = (_a = class {
     constructor() {
       this.snakeCase = snakeCase;
-      this.setColumns = (fn) => {
-        columnTypes[orchidCore.snakeCaseKey] = this.snakeCase;
-        const shape = pqb.getColumnTypes(columnTypes, fn);
-        if (this.snakeCase) {
-          for (const key in shape) {
-            const column = shape[key];
-            if (column.data.name)
-              continue;
-            const snakeName = orchidCore.toSnakeCase(key);
-            if (snakeName !== key) {
-              column.data.name = snakeName;
-            }
+      this.columnTypes = columnTypes;
+    }
+    setColumns(fn) {
+      if (!this.filePath) {
+        const filePath2 = orchidCore.getCallerFilePath();
+        if (!filePath2) {
+          throw new Error(
+            `Failed to determine file path for table ${this.constructor.name}. Please set \`filePath\` property manually`
+          );
+        }
+        this.filePath = filePath2;
+      }
+      columnTypes[orchidCore.snakeCaseKey] = this.snakeCase;
+      const shape = pqb.getColumnTypes(columnTypes, fn);
+      if (this.snakeCase) {
+        for (const key in shape) {
+          const column = shape[key];
+          if (column.data.name)
+            continue;
+          const snakeName = orchidCore.toSnakeCase(key);
+          if (snakeName !== key) {
+            column.data.name = snakeName;
           }
         }
-        return {
-          shape,
-          type: void 0
-        };
+      }
+      return {
+        shape,
+        type: void 0
       };
-      this.columnTypes = columnTypes;
     }
     belongsTo(fn, options) {
       return {
@@ -1405,6 +1415,8 @@ const orchidORM = (_a, tables) => {
     );
     dbTable.definedAs = key;
     dbTable.db = result;
+    dbTable.filePath = table.filePath;
+    dbTable.name = table.constructor.name;
     result[key] = dbTable;
   }
   applyRelations(qb, tableInstances, result);
@@ -1778,6 +1790,48 @@ const getTablesListObject = (importName, statements) => {
   return;
 };
 
+const handleForeignKey = async ({
+  getTable,
+  relations,
+  tableName,
+  columns,
+  foreignTableName,
+  foreignColumns,
+  skipBelongsTo
+}) => {
+  var _a, _b;
+  const table = await getTable(tableName);
+  if (!table)
+    return;
+  const foreignTable = await getTable(foreignTableName);
+  if (!foreignTable)
+    return;
+  if (!skipBelongsTo) {
+    (_a = relations[tableName]) != null ? _a : relations[tableName] = {
+      path: table.path,
+      relations: []
+    };
+    relations[tableName].relations.push({
+      kind: "belongsTo",
+      columns,
+      className: foreignTable.name,
+      path: foreignTable.path,
+      foreignColumns
+    });
+  }
+  (_b = relations[foreignTableName]) != null ? _b : relations[foreignTableName] = {
+    path: foreignTable.path,
+    relations: []
+  };
+  relations[foreignTableName].relations.push({
+    kind: "hasMany",
+    columns: foreignColumns,
+    className: table.name,
+    path: table.path,
+    foreignColumns: columns
+  });
+};
+
 var __getOwnPropSymbols$4 = Object.getOwnPropertySymbols;
 var __hasOwnProp$4 = Object.prototype.hasOwnProperty;
 var __propIsEnum$4 = Object.prototype.propertyIsEnumerable;
@@ -1796,13 +1850,29 @@ var __objRest$2 = (source, exclude) => {
 const createTable = async (_a) => {
   var _b = _a, {
     ast,
-    logger
+    logger,
+    getTable,
+    relations,
+    tables
   } = _b, params = __objRest$2(_b, [
     "ast",
-    "logger"
+    "logger",
+    "getTable",
+    "relations",
+    "tables"
   ]);
-  const tablePath = params.tablePath(orchidCore.toCamelCase(ast.name));
+  const key = orchidCore.toCamelCase(ast.name);
+  const tablePath = params.tablePath(key);
   const baseTablePath = orchidCore.getImportPath(tablePath, params.baseTable.filePath);
+  const className = `${orchidCore.toPascalCase(ast.name)}Table`;
+  tables[ast.name] = {
+    key,
+    name: className,
+    path: tablePath
+  };
+  const imports = {
+    [baseTablePath]: params.baseTable.name
+  };
   const props = [];
   if (ast.schema) {
     props.push(`schema = ${orchidCore.singleQuote(ast.schema)};`);
@@ -1816,10 +1886,23 @@ const createTable = async (_a) => {
     pqb.columnsShapeToCode(ast.shape, ast, "t"),
     "}));"
   );
+  const relCode = await getRelations(
+    ast,
+    getTable,
+    tablePath,
+    imports,
+    relations,
+    ast.name
+  );
+  if (relCode) {
+    props.push("", ...relCode);
+  }
   const code = [
-    `import { ${params.baseTable.name} } from '${baseTablePath}';
-`,
-    `export class ${orchidCore.toPascalCase(ast.name)}Table extends ${params.baseTable.name} {`,
+    ...Object.entries(imports).map(
+      ([from, name]) => `import { ${name} } from '${from}';`
+    ),
+    "",
+    `export class ${className} extends ${params.baseTable.name} {`,
     props,
     "}\n"
   ];
@@ -1833,6 +1916,62 @@ const createTable = async (_a) => {
     }
   }
 };
+const getRelations = async (ast, getTable, tablePath, imports, relations, tableName) => {
+  const refs = [];
+  for (const key in ast.shape) {
+    const item = ast.shape[key];
+    if (!(item instanceof pqb.ColumnType) || !item.data.foreignKeys)
+      continue;
+    for (const fkey of item.data.foreignKeys) {
+      if ("table" in fkey) {
+        refs.push({
+          table: fkey.table,
+          columns: [key],
+          foreignColumns: fkey.columns
+        });
+      }
+    }
+  }
+  if (ast.constraints) {
+    for (const { references: ref } of ast.constraints) {
+      if (ref && typeof ref.fnOrTable === "string") {
+        refs.push({
+          table: ref.fnOrTable,
+          columns: ref.columns,
+          foreignColumns: ref.foreignColumns
+        });
+      }
+    }
+  }
+  if (!refs.length)
+    return;
+  const code = [];
+  for (const ref of refs) {
+    const { columns, foreignColumns } = ref;
+    if (columns.length > 1 || foreignColumns.length > 1)
+      continue;
+    const info = await getTable(ref.table);
+    if (!info)
+      continue;
+    const path2 = orchidCore.getImportPath(tablePath, info.path);
+    imports[path2] = info.name;
+    code.push(
+      `${info.key}: this.belongsTo(() => ${info.name}, {`,
+      [`primaryKey: '${foreignColumns[0]}',`, `foreignKey: '${columns[0]}',`],
+      "}),"
+    );
+    await handleForeignKey({
+      getTable,
+      relations,
+      tableName,
+      columns: ref.columns,
+      foreignTableName: ref.table,
+      foreignColumns: ref.foreignColumns,
+      skipBelongsTo: true
+    });
+  }
+  return code.length ? ["relations = {", code, "};"] : void 0;
+};
 
 var __defProp$2 = Object.defineProperty;
 var __getOwnPropSymbols$3 = Object.getOwnPropertySymbols;
@@ -2427,6 +2566,18 @@ const updateTableFile = async (params) => {
     await changeTable(__spreadProps$1(__spreadValues$1({}, params), { ast }));
   } else if (ast.type === "renameTable") {
     await renameTable(__spreadProps$1(__spreadValues$1({}, params), { ast }));
+  } else if (ast.type === "constraint" && ast.references) {
+    const ref = ast.references;
+    if (typeof ref.fnOrTable === "string") {
+      await handleForeignKey({
+        getTable: params.getTable,
+        relations: params.relations,
+        tableName: ast.tableName,
+        columns: ref.columns,
+        foreignTableName: ref.fnOrTable,
+        foreignColumns: ref.foreignColumns
+      });
+    }
   }
 };
 
@@ -2453,6 +2604,131 @@ export const ${baseTable.name} = createBaseTable();
   });
 };
 
+const updateRelations = async ({
+  relations,
+  logger
+}) => {
+  await Promise.all(
+    Object.entries(relations).map(
+      ([tableName, item]) => updateRelationItem(tableName, item, logger)
+    )
+  );
+};
+const updateRelationItem = async (tableName, item, logger) => {
+  var _a;
+  const content = await fs.readFile(item.path, "utf-8").catch(() => void 0);
+  if (!content)
+    return;
+  const changes = new FileChanges(content);
+  const statements = ts.getStatements(content);
+  const dirName = path.dirname(item.path);
+  const imports = {};
+  for (const relation of item.relations) {
+    if (!imports[relation.path]) {
+      imports[relation.path] = relation.className;
+    }
+  }
+  let importsEnd = 0;
+  for (const node of ts.import.iterate(statements)) {
+    const source = ts.import.getSource(node);
+    const full = path.join(dirName, source + ".ts");
+    if (imports[full]) {
+      delete imports[full];
+    }
+    importsEnd = node.end;
+  }
+  const addImports = Object.entries(imports).map(
+    ([path2, name]) => `import { ${name} } from '${orchidCore.getImportPath(item.path, path2)}';`
+  ).join("\n");
+  if (addImports) {
+    changes.add(importsEnd, `
+${addImports}`);
+  }
+  const targetClass = findClassByTableName(statements, tableName);
+  if (targetClass) {
+    const relationsMember = findRelationsMember(targetClass.members);
+    if (relationsMember) {
+      const { initializer } = relationsMember;
+      const takenKeys = {};
+      if (ts.is.objectLiteral(initializer)) {
+        const props = initializer.properties;
+        for (const prop of props) {
+          const name = (_a = prop.name) == null ? void 0 : _a.getText();
+          if (name)
+            takenKeys[name] = true;
+        }
+        const addRelations = [];
+        for (const rel of item.relations) {
+          if (!checkRelation(rel))
+            continue;
+          const name = makeRelationName(rel);
+          if (takenKeys[name])
+            continue;
+          addRelations.push(relationToCode(rel));
+        }
+        if (addRelations.length) {
+          const pos = props.end;
+          changes.add(pos, addRelations.join(""));
+        }
+      }
+    } else {
+      changes.add(
+        targetClass.end - 1,
+        `
+  relations = {${item.relations.filter(checkRelation).map((rel) => relationToCode(rel)).join("")}
+  };
+`
+      );
+    }
+  }
+  await fs.writeFile(item.path, changes.apply());
+  logger == null ? void 0 : logger.log(`Updated ${orchidCore.pathToLog(item.path)}`);
+};
+const findClassByTableName = (statements, tableName) => {
+  for (const node of ts.class.iterate(statements)) {
+    for (const member of node.members) {
+      const name = ts.prop.getName(member);
+      if (name !== "table")
+        continue;
+      const { initializer: value } = member;
+      if (!value || !ts.is.stringLiteral(value))
+        continue;
+      if (value.text === tableName) {
+        return node;
+      }
+    }
+  }
+  return;
+};
+const findRelationsMember = (members) => {
+  for (const member of members) {
+    const name = ts.prop.getName(member);
+    if (name === "relations")
+      return member;
+  }
+  return;
+};
+const checkRelation = (rel) => {
+  return rel.columns.length === 1 && rel.foreignColumns.length === 1;
+};
+const makeRelationName = (rel) => {
+  return pluralize.plural(
+    rel.className[0].toLowerCase() + rel.className.slice(1).replace(/Table$/, "")
+  );
+};
+const relationToCode = (rel, name = makeRelationName(rel)) => {
+  const code = [`
+    ${name}: this.${rel.kind}(() => ${rel.className}, {`];
+  const pk = rel[rel.kind === "hasMany" ? "columns" : "foreignColumns"][0];
+  const fk = rel[rel.kind === "hasMany" ? "foreignColumns" : "columns"][0];
+  code.push(
+    `      primaryKey: '${pk}',`,
+    `      foreignKey: '${fk}',`,
+    "    }),"
+  );
+  return code.join("\n");
+};
+
 var __defProp = Object.defineProperty;
 var __defProps = Object.defineProperties;
 var __getOwnPropDescs = Object.getOwnPropertyDescriptors;
@@ -2474,37 +2750,68 @@ var __spreadValues = (a, b) => {
 var __spreadProps = (a, b) => __defProps(a, __getOwnPropDescs(b));
 class AppCodeUpdaterError extends Error {
 }
+const makeGetTable = (path2, ormExportedAs, tables) => {
+  let orm;
+  return async (tableName) => {
+    if (tables[tableName])
+      return tables[tableName];
+    if (!orm) {
+      orm = (await import(path2))[ormExportedAs];
+      if (!orm) {
+        throw new Error(`ORM is not exported as ${ormExportedAs} from ${path2}`);
+      }
+    }
+    for (const key in orm) {
+      const table = orm[key];
+      if (table && typeof table === "object" && table instanceof pqb.Db && table.table === tableName) {
+        const tableInfo = {
+          key,
+          name: table.name,
+          path: table.filePath
+        };
+        return tables[tableName] = tableInfo;
+      }
+    }
+    return;
+  };
+};
 const appCodeUpdater = ({
   tablePath,
-  mainFilePath
-}) => {
-  return async ({
+  ormPath,
+  ormExportedAs = "db"
+}) => ({
+  async process({
     ast,
     options,
     basePath,
     cache: cacheObject,
     logger,
     baseTable
-  }) => {
+  }) {
+    var _a, _b;
     const params = {
       tablePath(name) {
         const file = tablePath(name);
         return resolvePath(basePath, file);
       },
-      mainFilePath: resolvePath(basePath, mainFilePath),
+      ormPath: resolvePath(basePath, ormPath),
+      ormExportedAs,
       logger
     };
+    const cache = cacheObject;
+    (_a = cache.relations) != null ? _a : cache.relations = {};
+    (_b = cache.tables) != null ? _b : cache.tables = {};
+    const getTable = makeGetTable(params.ormPath, ormExportedAs, cache.tables);
     const promises = [
-      updateMainFile(
-        params.mainFilePath,
-        params.tablePath,
+      updateMainFile(params.ormPath, params.tablePath, ast, options, logger),
+      updateTableFile(__spreadProps(__spreadValues({}, params), {
         ast,
-        options,
-        logger
-      ),
-      updateTableFile(__spreadProps(__spreadValues({}, params), { ast, baseTable }))
+        baseTable,
+        getTable,
+        relations: cache.relations,
+        tables: cache.tables
+      }))
     ];
-    const cache = cacheObject;
     if (!cache.createdBaseTable) {
       promises.push(
         createBaseTableFile({ logger: params.logger, baseTable }).then(() => {
@@ -2513,8 +2820,17 @@ const appCodeUpdater = ({
       );
     }
     await Promise.all(promises);
-  };
-};
+  },
+  async afterAll({ cache, logger }) {
+    const { relations } = cache;
+    if (!relations)
+      return;
+    await updateRelations({
+      relations,
+      logger
+    });
+  }
+});
 const resolvePath = (basePath, filePath) => path__namespace.isAbsolute(filePath) ? filePath : path__namespace.resolve(basePath, filePath);
 
 Object.defineProperty(exports, 'OrchidOrmError', {