orchid-orm 1.4.21 → 1.4.22

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;");
   }
@@ -2197,17 +2201,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());