drizzle.seed.ts

import 'dotenv/config';
import { ConfigService } from '@nestjs/config';
import { Pool } from 'pg';
import { dbSchema } from './src/infra/db/schema';
import { drizzle } from 'drizzle-orm/node-postgres';
import { sql } from 'drizzle-orm';
import { IS_PRODUCTION } from './src/shared/utils/is-production';
import { seedUiRepeatTimer } from './test/e2e/helpers/db/ui/repeat-timer.helper';
import { seedCarouselReview } from './test/e2e/helpers/db/ui/carousel-review.helper';
import {
  seedAdminUser,
  seedPgUsers,
} from './test/e2e/helpers/db/lms/user.helper';
import {
  seedPgFirstCourse,
  seedPgFirstCourseProduct,
  seedPgSecondCourse,
  seedPgSecondCourseProduct,
  seedPgTeacher,
  seedPgThirdCourseProduct,
} from './test/e2e/helpers/db/lms/course-product.helper';
import { clearDatabase } from './src/shared/helpers/db';
import {
  seedCommunityCategories,
  seedNavbarCategories,
} from './test/e2e/helpers/db/lms/post-category.helper';
import { seedCarouselMainBanner } from './test/e2e/helpers/db/ui/carousel-main-banner.helper';
import { seedSignupTerms } from './test/e2e/helpers/db/lms/term.helper';

const env = new ConfigService();

async function seed() {
  if (IS_PRODUCTION) {
    throw new Error('Do not run seed script in production!');
  }

  const connectionString = env.get('DATABASE_URL');
  if (!connectionString) {
    throw new Error(
      'DATABASE_URL is not provided. Check your environment variables.',
    );
  }

  const pool = new Pool({ connectionString });
  const db = drizzle(pool, { schema: dbSchema });
  const confirmDatabaseReady = await db.execute(sql`SELECT 1`);

  console.debug(`[Test container ready]`, confirmDatabaseReady.rowCount);
  console.debug('[Test container uri]', connectionString);

  await clearDatabase(db);

  // Seed data
  try {
    // await seedCourseCategoriesWithChildren({ count: 3 }, db);
    await seedUiRepeatTimer({ count: 2 }, db, '/');
    await seedCarouselReview({ count: 4 }, db, '/');
    await seedCarouselMainBanner({ count: 1 }, db, '/');
    await seedAdminUser(
      {
        email: env.get('ADMIN_EMAIL'),
        password: env.get('ADMIN_PASSWORD'),
      },
      db,
    );
    // 토스페이먼츠, 카카오페이 pg 심사 계정
    const [testPgUser] = await seedPgUsers(db);
    const pgTeacherUser = await seedPgTeacher(db);
    const pgFirstCourse = await seedPgFirstCourse(
      { ...pgTeacherUser.teacher, account: pgTeacherUser.user },
      db,
    );
    const pgSecondCourse = await seedPgSecondCourse(
      { ...pgTeacherUser.teacher, account: pgTeacherUser.user },
      db,
    );
    const pgFirstCourseProduct = await seedPgFirstCourseProduct(
      pgFirstCourse,
      db,
    );
    const pgSecondCourseProduct = await seedPgSecondCourseProduct(
      pgSecondCourse,
      db,
    );
    const pgThirdCourseProduct = await seedPgThirdCourseProduct(
      pgSecondCourse,
      db,
    );
    await seedNavbarCategories(db);
    const { freeCategory, discussionCategory, profitCertificationCategory } =
      await seedCommunityCategories(db);
    await seedSignupTerms({ count: 3 }, db);
  } catch (e) {
    console.error(e);
    await clearDatabase(db);
    throw new Error('Seed failed. Database is cleared.');
  } finally {
    await pool.end();
  }
}

seed()
  .then(() => {
    console.log('🌱 Seed completed.');
    process.exit(0);
  })
  .catch((e) => {
    console.error('❗️ Seed failed. maybe too many connection pool.');
    console.error(e);
    process.exit(1);
  });