orchid-orm 1.5.18 → 1.5.20

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "orchid-orm",
-  "version": "1.5.18",
+  "version": "1.5.20",
   "description": "Postgres ORM",
   "homepage": "https://orchid-orm.netlify.app/guide/orm-setup-and-overview.html",
   "repository": {
@@ -55,7 +55,7 @@
     "orchid-orm-schema-to-zod": "0.2.18",
     "pg": "^8.7.3",
     "pg-transactional-tests": "^1.0.7",
-    "rake-db": "2.3.17",
+    "rake-db": "2.3.18",
     "rimraf": "^3.0.2",
     "rollup": "^2.79.0",
     "rollup-plugin-dts": "^4.2.2",

package/src/bin/init.test.ts

@@ -773,6 +773,37 @@ export const seed = async () => {
 
   await db.$close();
 };
+`);
+    });
+
+    it('should create seed file with sample records when demoTables is set to true', async () => {
+      await initOrchidORM({
+        demoTables: true,
+      });
+
+      const [, content] = asMock(fs.writeFile).mock.calls.find(
+        ([to]) => to === seedPath,
+      );
+      expect(content).toBe(`import { db } from './db';
+
+export const seed = async () => {
+  await db.post.findBy({ title: 'Sample post' }).orCreate({
+    title: 'Post',
+    text: 'This is a text for a sample post. It contains words, spaces, and punctuation.',
+    comments: {
+      create: [
+        {
+          text: 'Nice post!',
+        },
+        {
+          text: \`Too long, didn't read\`,
+        },
+      ],
+    },
+  });
+
+  await db.$close();
+};
 `);
     });
   });

package/src/bin/init.ts

@@ -73,7 +73,7 @@ export const initOrchidORM = async (config: InitConfig) => {
   await setupMainDb(config);
   await setupMigrationScript(config);
   await createMigrations(config);
-  await createSeed();
+  await createSeed(config);
 
   greet();
 };
@@ -473,14 +473,35 @@ const makeFileTimeStamp = (now: Date) => {
     .join('');
 };
 
-const createSeed = async () => {
+const createSeed = async (config: InitConfig) => {
   const filePath = path.join(dirPath, 'seed.ts');
+
+  let content;
+  if (config.demoTables) {
+    content = `await db.post.findBy({ title: 'Sample post' }).orCreate({
+    title: 'Post',
+    text: 'This is a text for a sample post. It contains words, spaces, and punctuation.',
+    comments: {
+      create: [
+        {
+          text: 'Nice post!',
+        },
+        {
+          text: \`Too long, didn't read\`,
+        },
+      ],
+    },
+  });`;
+  } else {
+    content = `// create records here`;
+  }
+
   await fs.writeFile(
     filePath,
     `import { db } from './db';
 
 export const seed = async () => {
-  // create records here
+  ${content}
 
   await db.$close();
 };

package/src/codegen/appCodeUpdater.test.ts

@@ -30,10 +30,15 @@ describe('appCodeUpdater', () => {
   const fn = appCodeUpdater(params);
 
   it('should call table and file updaters with proper arguments', async () => {
-    await fn({ ast: ast.addTable, options: {}, cache: {} });
+    await fn({
+      ast: ast.addTable,
+      options: {},
+      basePath: __dirname,
+      cache: {},
+    });
 
-    const mainFilePath = path.resolve(params.mainFilePath);
-    const tablePath = path.resolve(params.tablePath('table'));
+    const mainFilePath = path.resolve(__dirname, params.mainFilePath);
+    const tablePath = path.resolve(__dirname, params.tablePath('table'));
 
     const main = asMock(updateMainFile).mock.calls[0];
     expect(main[0]).toBe(mainFilePath);
@@ -55,11 +60,11 @@ describe('appCodeUpdater', () => {
     const cache = {};
     expect(createBaseTableFile).not.toBeCalled();
 
-    await fn({ ast: ast.addTable, options: {}, cache });
+    await fn({ ast: ast.addTable, options: {}, basePath: __dirname, cache });
 
     expect(createBaseTableFile).toBeCalledTimes(1);
 
-    await fn({ ast: ast.addTable, options: {}, cache });
+    await fn({ ast: ast.addTable, options: {}, basePath: __dirname, cache });
 
     expect(createBaseTableFile).toBeCalledTimes(1);
   });

package/src/codegen/appCodeUpdater.ts

@@ -3,6 +3,7 @@ import * as path from 'path';
 import { updateMainFile } from './updateMainFile';
 import { updateTableFile } from './updateTableFile/updateTableFile';
 import { createBaseTableFile } from './createBaseTableFile';
+import { SetOptional } from 'pqb';
 
 export class AppCodeUpdaterError extends Error {}
 
@@ -14,15 +15,21 @@ export type AppCodeUpdaterConfig = {
 };
 
 export const appCodeUpdater = (
-  config: AppCodeUpdaterConfig,
+  config: SetOptional<AppCodeUpdaterConfig, 'baseTableName'>,
 ): AppCodeUpdater => {
-  const params = {
-    ...config,
-    tablePath: (name: string) => path.resolve(config.tablePath(name)),
-    mainFilePath: path.resolve(config.mainFilePath),
-  };
+  return async ({ ast, options, basePath, cache: cacheObject }) => {
+    const params: AppCodeUpdaterConfig = {
+      ...config,
+      baseTableName: config.baseTableName || 'BaseTable',
+      tablePath(name: string) {
+        const file = config.tablePath(name);
+        return path.isAbsolute(file) ? file : path.resolve(basePath, file);
+      },
+      mainFilePath: path.isAbsolute(config.mainFilePath)
+        ? config.mainFilePath
+        : path.resolve(basePath, config.mainFilePath),
+    };
 
-  return async ({ ast, options, cache: cacheObject }) => {
     const promises: Promise<void>[] = [
       updateMainFile(params.mainFilePath, params.tablePath, ast, options),
       updateTableFile({ ...params, ast }),

package/src/orm.test.ts

@@ -1,4 +1,4 @@
-import { orchidORM } from './orm';
+import { OrchidORM, orchidORM } from './orm';
 import {
   assertType,
   expectSql,
@@ -11,6 +11,14 @@ import { createBaseTable } from './table';
 describe('orm', () => {
   useTestDatabase();
 
+  let db:
+    | OrchidORM<{ user: typeof UserTable; profile: typeof ProfileTable }>
+    | undefined;
+
+  afterEach(async () => {
+    if (db) await db.$close();
+  });
+
   const BaseTable = createBaseTable();
 
   type User = UserTable['columns']['type'];
@@ -31,7 +39,7 @@ describe('orm', () => {
   }
 
   it('should return object with provided adapter, close and transaction method, tables', () => {
-    const db = orchidORM(pgConfig, {
+    db = orchidORM(pgConfig, {
       user: UserTable,
       profile: ProfileTable,
     });
@@ -45,7 +53,7 @@ describe('orm', () => {
   });
 
   it('should return table which is a queryable interface', async () => {
-    const db = orchidORM(pgConfig, {
+    db = orchidORM(pgConfig, {
       user: UserTable,
       profile: ProfileTable,
     });
@@ -71,7 +79,7 @@ describe('orm', () => {
   });
 
   it('should be able to turn on autoPreparedStatements', () => {
-    const db = orchidORM(
+    db = orchidORM(
       { ...pgConfig, autoPreparedStatements: true },
       {
         user: UserTable,

package/src/transaction.test.ts

@@ -4,6 +4,8 @@ import { Client } from 'pg';
 import { noop } from 'pqb';
 
 describe('transaction', () => {
+  afterAll(db.$close);
+
   it('should have override transaction method which implicitly connects tables with a single transaction', async () => {
     const spy = jest.spyOn(Client.prototype, 'query');