orchid-orm 1.17.32 → 1.17.34

package/codegen/index.mjs

@@ -395,13 +395,15 @@ const createTable = async (_a) => {
     logger,
     getTable,
     relations,
-    tables
+    tables,
+    delayed
   } = _b, params = __objRest$3(_b, [
     "ast",
     "logger",
     "getTable",
     "relations",
-    "tables"
+    "tables",
+    "delayed"
   ]);
   const key = toCamelCase(ast.name);
   const tablePath = params.tablePath(key);
@@ -431,36 +433,43 @@ const createTable = async (_a) => {
     columnsShapeToCode(ast.shape, ast, "t"),
     "}));"
   );
-  const relCode = await getRelations(
-    ast,
-    getTable,
-    tablePath,
-    imports,
-    relations,
-    ast.name
-  );
-  if (relCode) {
-    props.push("", ...relCode);
-  }
+  const importsCode = importsToCode(imports);
   const code = [
-    ...Object.entries(imports).map(
-      ([from, name]) => `import { ${name} } from '${from}';`
-    ),
-    "",
     `export class ${className} extends ${params.baseTable.exportAs} {`,
     props,
     "}\n"
   ];
   await fs.mkdir(path__default.dirname(tablePath), { recursive: true });
   try {
-    await fs.writeFile(tablePath, codeToString(code, "", "  "), { flag: "wx" });
-    logger == null ? void 0 : logger.log(`Created ${pathToLog(tablePath)}`);
+    const content = importsCode + "\n\n" + codeToString(code, "", "  ");
+    await fs.writeFile(tablePath, content, { flag: "wx" });
+    delayed.push(async () => {
+      const imports2 = {};
+      const relCode = await getRelations(
+        ast,
+        getTable,
+        tablePath,
+        imports2,
+        relations,
+        ast.name
+      );
+      if (relCode) {
+        const code2 = codeToString(relCode, "  ", "  ");
+        const updated = content.slice(0, importsCode.length) + `
+${importsToCode(imports2)}` + content.slice(importsCode.length, -2) + "  \n" + code2 + "\n" + content.slice(-2);
+        await fs.writeFile(tablePath, updated);
+      }
+      logger == null ? void 0 : logger.log(`Created ${pathToLog(tablePath)}`);
+    });
   } catch (err) {
     if (err.code !== "EEXIST") {
       throw err;
     }
   }
 };
+function importsToCode(imports) {
+  return Object.entries(imports).map(([from, name]) => `import { ${name} } from '${from}';`).join("\n");
+}
 const getRelations = async (ast, getTable, tablePath, imports, relations, tableName) => {
   const refs = [];
   for (const key in ast.shape) {
@@ -502,7 +511,10 @@ const getRelations = async (ast, getTable, tablePath, imports, relations, tableN
     imports[path2] = info.name;
     code.push(
       `${info.key}: this.belongsTo(() => ${info.name}, {`,
-      [`primaryKey: '${foreignColumns[0]}',`, `foreignKey: '${columns[0]}',`],
+      [
+        `columns: [${foreignColumns.map(singleQuote).join(", ")}],`,
+        `references: [${columns.map(singleQuote).join(", ")}],`
+      ],
       "}),"
     );
     await handleForeignKey({
@@ -1322,6 +1334,7 @@ class AppCodeUpdaterError extends Error {
 const makeGetTable = (path2, ormExportedAs, tables, imp) => {
   let orm;
   return async (tableName) => {
+    var _a;
     if (tables[tableName])
       return tables[tableName];
     if (!orm) {
@@ -1339,14 +1352,19 @@ const makeGetTable = (path2, ormExportedAs, tables, imp) => {
     }
     for (const key in orm) {
       const table = orm[key];
-      if (table && typeof table === "object" && table instanceof Db && table.table === tableName) {
-        const tableInfo = {
-          key,
-          name: table.name,
-          path: table.filePath
-        };
-        return tables[tableName] = tableInfo;
-      }
+      if (!table || typeof table !== "object" || !(table instanceof Db) || table.table !== tableName)
+        continue;
+      const name = table.name;
+      if (!name)
+        continue;
+      const path3 = (_a = table.getFilePath) == null ? void 0 : _a.call(table);
+      if (!path3)
+        continue;
+      return tables[tableName] = {
+        key,
+        name,
+        path: path3
+      };
     }
     return;
   };
@@ -1384,13 +1402,15 @@ const appCodeUpdater = ({
       options,
       logger
     );
+    const delayed = [];
     const promises = [
       updateTableFile(__spreadProps(__spreadValues({}, params), {
         ast,
         baseTable,
         getTable,
         relations: cache.relations,
-        tables: cache.tables
+        tables: cache.tables,
+        delayed
       }))
     ];
     if (!cache.createdBaseTable) {
@@ -1401,6 +1421,7 @@ const appCodeUpdater = ({
       );
     }
     await Promise.all(promises);
+    await Promise.all(delayed.map((fn) => fn()));
   },
   async afterAll({ cache, logger }) {
     const { relations } = cache;