refactor(cli): add skip when exists option (#2172)

packages/cli/src/commands/database/seed/index.ts

@@ -12,6 +12,7 @@ import { z } from 'zod';
 import { createPoolAndDatabaseIfNeeded, insertInto } from '../../../database';
 import {
   getRowsByKeys,
+  isConfigsTableExists,
   updateDatabaseTimestamp,
   updateValueByKey,
 } from '../../../queries/logto-config';
@@ -126,19 +127,32 @@ export const seedByPool = async (pool: DatabasePool, type: SeedChoice) => {
   });
 };
 
-const seed: CommandModule<Record<string, unknown>, { type: string }> = {
+const seed: CommandModule<Record<string, unknown>, { type: string; swe?: boolean }> = {
   command: 'seed [type]',
   describe: 'Create database then seed tables and data',
   builder: (yargs) =>
-    yargs.positional('type', {
+    yargs
+      .option('swe', {
+        describe: 'Skip the seeding process when Logto configs table exists',
+        alias: 'skip-when-exists',
+        type: 'boolean',
+      })
+      .positional('type', {
         describe: 'Optional seed type',
         type: 'string',
         choices: seedChoices,
         default: 'all',
       }),
-  handler: async ({ type }) => {
+  handler: async ({ type, swe }) => {
     const pool = await createPoolAndDatabaseIfNeeded();
 
+    if (swe && (await isConfigsTableExists(pool))) {
+      log.info('Seeding skipped');
+      await pool.end();
+
+      return;
+    }
+
     try {
       // Cannot avoid `as` since the official type definition of `yargs` doesn't work.
       // The value of `type` can be ensured, so it's safe to use `as` here.

packages/cli/src/queries/logto-config.ts

@@ -7,11 +7,20 @@ import {
   AlterationStateKey,
 } from '@logto/schemas';
 import { convertToIdentifiers } from '@logto/shared';
+import { Nullable } from '@silverhand/essentials';
 import { DatabasePool, DatabaseTransactionConnection, sql } from 'slonik';
 import { z } from 'zod';
 
 const { table, fields } = convertToIdentifiers(LogtoConfigs);
 
+export const isConfigsTableExists = async (pool: DatabasePool) => {
+  const { rows } = await pool.query<Nullable<string>>(
+    sql`select to_regclass(${LogtoConfigs.table})`
+  );
+
+  return Boolean(rows[0]);
+};
+
 export const getRowsByKeys = async (
   pool: DatabasePool | DatabaseTransactionConnection,
   keys: LogtoConfigKey[]