orchid-orm 1.4.21 → 1.5.0

package/dist/index.js

@@ -1754,7 +1754,11 @@ const createTable = async (_a) => {
   ]);
   const tablePath = params.tablePath(ast.name);
   const baseTablePath = getImportPath(tablePath, params.baseTablePath);
-  const props = [`table = ${pqb.singleQuote(ast.name)};`];
+  const props = [];
+  if (ast.schema) {
+    props.push(`schema = ${pqb.singleQuote(ast.schema)};`);
+  }
+  props.push(`table = ${pqb.singleQuote(ast.name)};`);
   if (ast.noPrimaryKey === "ignore") {
     props.push("noPrimaryKey = true;");
   }
@@ -1967,20 +1971,32 @@ const changeColumn = ({ changes, t, spaces }, changeItem, prop) => {
     }
   }
   const changedProps = {};
+  const replaced = {};
   for (const item of items.reverse()) {
     if (!ts.is.propertyAccess(item.expression))
       continue;
     const { name } = item.expression;
-    const key = name.escapedText.toString();
+    let key = name.escapedText.toString();
+    if (key === "index")
+      key = "indexes";
+    else if (key === "foreignKey")
+      key = "foreignKeys";
     if (!propsToChange[key])
       continue;
-    const value = getColumnMethodArgs(to, key);
-    if (value) {
-      const code = [`${key}(`];
-      pqb.addCode(code, value);
-      pqb.addCode(code, ")");
-      changes.replace(name.pos, item.end, pqb.codeToString(code, "", "  ").trim());
-    } else {
+    let remove = true;
+    if (!replaced[key]) {
+      const code = getColumnMethodArgs(to, key);
+      if (code) {
+        changes.replace(
+          item.expression.expression.end,
+          item.end,
+          pqb.codeToString(code, spaces + "  ", "  ").trim()
+        );
+        replaced[key] = true;
+        remove = false;
+      }
+    }
+    if (remove) {
       changes.remove(item.expression.expression.end, item.end);
     }
     changedProps[key] = true;
@@ -1989,12 +2005,9 @@ const changeColumn = ({ changes, t, spaces }, changeItem, prop) => {
   for (const key in propsToChange) {
     if (changedProps[key])
       continue;
-    const value = getColumnMethodArgs(to, key);
-    if (value !== void 0) {
-      const code = [`.${key}(`];
-      pqb.addCode(code, value);
-      pqb.addCode(code, ")");
-      append += pqb.codeToString(code, "", "  ").trim();
+    const code = getColumnMethodArgs(to, key);
+    if (code) {
+      append += pqb.codeToString(code, spaces + "  ", "  ").trim();
     }
   }
   if (append) {
@@ -2049,15 +2062,22 @@ const getColumnMethodArgs = (to, key) => {
   const value = to[key];
   if (!value)
     return;
+  if (key === "indexes") {
+    return pqb.columnIndexesToCode(value);
+  }
+  if (key === "foreignKeys") {
+    return pqb.columnForeignKeysToCode(value);
+  }
+  const code = [`.${key}(`];
   if (key === "collate" || key === "compression") {
-    return pqb.singleQuote(value);
+    pqb.addCode(code, pqb.singleQuote(value));
   } else if (key === "default") {
-    return pqb.columnDefaultArgumentToCode(value);
-  } else if (key === "nullable" || key === "primaryKey") {
-    return "";
-  } else {
+    pqb.addCode(code, pqb.columnDefaultArgumentToCode(value));
+  } else if (key !== "nullable" && key !== "primaryKey") {
     return;
   }
+  pqb.addCode(code, ")");
+  return code;
 };
 const dropMatchingIndexes = (context, prop, i, call, items) => {
   if (!items.length)
@@ -2197,17 +2217,38 @@ const renameTable = async (_a) => {
   const changes = new FileChanges(content);
   const statements = ts.getStatements(content);
   const className = toPascalCase(ast.from);
+  const changeSchema = ast.fromSchema !== ast.toSchema;
   for (const node of ts.class.iterate(statements)) {
     if (((_a2 = node.name) == null ? void 0 : _a2.escapedText) !== className)
       continue;
+    const addSchema = changeSchema && ast.toSchema && !node.members.some((member) => ts.prop.getName(member) === "schema");
+    if (addSchema && ast.toSchema) {
+      changes.add(
+        node.members.pos,
+        `
+  schema = ${pqb.singleQuote(ast.toSchema)};`
+      );
+    }
     for (const member of node.members) {
       const name = ts.prop.getName(member);
-      if (name !== "table")
+      if (name !== "table" && !(changeSchema && name === "schema"))
         continue;
       const { initializer: value } = member;
       if (!value)
         continue;
-      changes.replace(value.pos, value.end, ` ${pqb.singleQuote(ast.to)}`);
+      if (name === "schema") {
+        if (ast.toSchema) {
+          changes.replace(
+            value.pos,
+            value.end,
+            ` ${pqb.singleQuote(ast.toSchema)}`
+          );
+        } else {
+          changes.remove(member.pos, member.end);
+        }
+      } else {
+        changes.replace(value.pos, value.end, ` ${pqb.singleQuote(ast.to)}`);
+      }
     }
   }
   await fs__default["default"].writeFile(params.tablePath(ast.to), changes.apply());