orchid-orm 1.4.20 → 1.4.22

package/src/codegen/updateTableFile/renameTable.test.ts

@@ -0,0 +1,124 @@
+import { updateTableFile } from './updateTableFile';
+import { asMock, ast, makeTestWritten, tablePath } from '../testUtils';
+import path from 'path';
+import fs from 'fs/promises';
+
+jest.mock('fs/promises', () => ({
+  mkdir: jest.fn(),
+  readFile: jest.fn(),
+  writeFile: jest.fn(),
+}));
+
+const baseTablePath = path.resolve('baseTable.ts');
+const baseTableName = 'BaseTable';
+const params = { baseTablePath, baseTableName, tablePath };
+
+const testWritten = makeTestWritten(tablePath('renamedTable'));
+
+describe('renameTable', () => {
+  beforeEach(() => {
+    jest.resetAllMocks();
+  });
+
+  it('should change `table` property', async () => {
+    asMock(fs.readFile)
+      .mockResolvedValue(`import { BaseTable } from '../baseTable';
+
+export class Table extends BaseTable {
+  table = 'table';
+  columns = this.setColumns((t) => ({}));
+}`);
+
+    await updateTableFile({
+      ...params,
+      ast: ast.renameTable,
+    });
+
+    testWritten(`import { BaseTable } from '../baseTable';
+
+export class Table extends BaseTable {
+  table = 'renamedTable';
+  columns = this.setColumns((t) => ({}));
+}`);
+  });
+
+  it('should change `schema` property', async () => {
+    asMock(fs.readFile)
+      .mockResolvedValue(`import { BaseTable } from '../baseTable';
+
+export class Table extends BaseTable {
+  schema = 'one';
+  table = 'table';
+  columns = this.setColumns((t) => ({}));
+}`);
+
+    await updateTableFile({
+      ...params,
+      ast: {
+        ...ast.renameTable,
+        fromSchema: 'one',
+        toSchema: 'two',
+      },
+    });
+
+    testWritten(`import { BaseTable } from '../baseTable';
+
+export class Table extends BaseTable {
+  schema = 'two';
+  table = 'renamedTable';
+  columns = this.setColumns((t) => ({}));
+}`);
+  });
+
+  it('should remove `schema` property', async () => {
+    asMock(fs.readFile)
+      .mockResolvedValue(`import { BaseTable } from '../baseTable';
+
+export class Table extends BaseTable {
+  schema = 'one';
+  table = 'table';
+  columns = this.setColumns((t) => ({}));
+}`);
+
+    await updateTableFile({
+      ...params,
+      ast: {
+        ...ast.renameTable,
+        fromSchema: 'one',
+      },
+    });
+
+    testWritten(`import { BaseTable } from '../baseTable';
+
+export class Table extends BaseTable {
+  table = 'renamedTable';
+  columns = this.setColumns((t) => ({}));
+}`);
+  });
+
+  it('should add `schema` property', async () => {
+    asMock(fs.readFile)
+      .mockResolvedValue(`import { BaseTable } from '../baseTable';
+
+export class Table extends BaseTable {
+  table = 'table';
+  columns = this.setColumns((t) => ({}));
+}`);
+
+    await updateTableFile({
+      ...params,
+      ast: {
+        ...ast.renameTable,
+        toSchema: 'schema',
+      },
+    });
+
+    testWritten(`import { BaseTable } from '../baseTable';
+
+export class Table extends BaseTable {
+  schema = 'schema';
+  table = 'renamedTable';
+  columns = this.setColumns((t) => ({}));
+}`);
+  });
+});

package/src/{appCodeUpdater → codegen}/updateTableFile/renameTable.ts

@@ -19,12 +19,27 @@ export const renameTable = async ({
   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 (node.name?.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,
+        `\n  schema = ${singleQuote(ast.toSchema)};`,
+      );
+    }
+
     for (const member of node.members) {
       const name = ts.prop.getName(member);
-      if (name !== 'table') continue;
+
+      if (name !== 'table' && !(changeSchema && name === 'schema')) continue;
 
       const { initializer: value } = member as unknown as {
         initializer?: Expression;
@@ -32,7 +47,19 @@ export const renameTable = async ({
 
       if (!value) continue;
 
-      changes.replace(value.pos, value.end, ` ${singleQuote(ast.to)}`);
+      if (name === 'schema') {
+        if (ast.toSchema) {
+          changes.replace(
+            value.pos,
+            value.end,
+            ` ${singleQuote(ast.toSchema)}`,
+          );
+        } else {
+          changes.remove(member.pos, member.end);
+        }
+      } else {
+        changes.replace(value.pos, value.end, ` ${singleQuote(ast.to)}`);
+      }
     }
   }
 

package/src/index.ts

@@ -1,4 +1,4 @@
 export * from './table';
 export * from './orm';
 export * from './repo';
-export * from './appCodeUpdater/appCodeUpdater';
+export * from './codegen/appCodeUpdater';

package/src/appCodeUpdater/updateTableFile/renameTable.test.ts

@@ -1,43 +0,0 @@
-import { updateTableFile } from './updateTableFile';
-import { asMock, ast, makeTestWritten, tablePath } from '../testUtils';
-import path from 'path';
-import fs from 'fs/promises';
-
-jest.mock('fs/promises', () => ({
-  readFile: jest.fn(),
-  writeFile: jest.fn(),
-}));
-
-const baseTablePath = path.resolve('baseTable.ts');
-const baseTableName = 'BaseTable';
-const params = { baseTablePath, baseTableName, tablePath };
-
-const testWritten = makeTestWritten(tablePath('renamedTable'));
-
-describe('renameTable', () => {
-  beforeEach(() => {
-    jest.resetAllMocks();
-  });
-
-  it('should only change `table` property', async () => {
-    asMock(fs.readFile)
-      .mockResolvedValue(`import { BaseTable } from '../baseTable';
-
-export class Table extends BaseTable {
-  table = 'table';
-  columns = this.setColumns((t) => ({}));
-}`);
-
-    await updateTableFile({
-      ...params,
-      ast: ast.renameTable,
-    });
-
-    testWritten(`import { BaseTable } from '../baseTable';
-
-export class Table extends BaseTable {
-  table = 'renamedTable';
-  columns = this.setColumns((t) => ({}));
-}`);
-  });
-});