import type { MySqlDatabase } from 'drizzle-orm/mysql-core';
import { drizzle as drizzleMysql } from 'drizzle-orm/mysql2';
import { migrate as migrateMysql } from 'drizzle-orm/mysql2/migrator';
import { createConnection } from 'mysql2';
import { PHASE_PRODUCTION_BUILD } from 'next/dist/shared/lib/constants';

import * as schema from '@/models';

import { Env } from './Env';

let connection;
let drizzle: MySqlDatabase<any, any, any>;

if (
  process.env.NEXT_PHASE !== PHASE_PRODUCTION_BUILD &&
  process.env.NODE_ENV === 'production' &&
  Env.DATABASE_URL
) {
  connection = await createConnection({
    uri: Env.DATABASE_URL,
  });

  drizzle = drizzleMysql(connection, { schema, mode: 'default' });

  await migrateMysql(drizzle, { migrationsFolder: './migrations' });
} else {
  const global = globalThis as unknown as { connection: any };

  if (!global.connection) {
    global.connection = await createConnection({
      ...
    });
  }

  drizzle = drizzleMysql(global.connection, { schema, mode: 'default' });
  await migrateMysql(drizzle, { migrationsFolder: './migrations' });
}

export const db = drizzle;

import type { MySqlDatabase } from 'drizzle-orm/mysql-core';
import { drizzle as drizzleMysql } from 'drizzle-orm/mysql2';
import { migrate as migrateMysql } from 'drizzle-orm/mysql2/migrator';
import { createConnection } from 'mysql2';
import { PHASE_PRODUCTION_BUILD } from 'next/dist/shared/lib/constants';

import * as schema from '@/models';

import { Env } from './Env';

let connection;
let drizzle: MySqlDatabase<any, any, any>;

if (
  process.env.NEXT_PHASE !== PHASE_PRODUCTION_BUILD &&
  process.env.NODE_ENV === 'production' &&
  Env.DATABASE_URL
) {
  connection = await createConnection({
    uri: Env.DATABASE_URL,
  });

  drizzle = drizzleMysql(connection, { schema, mode: 'default' });

  await migrateMysql(drizzle, { migrationsFolder: './migrations' });
} else {
  const global = globalThis as unknown as { connection: any };

  if (!global.connection) {
    global.connection = await createConnection({
      ...
    });
  }

  drizzle = drizzleMysql(global.connection, { schema, mode: 'default' });
  await migrateMysql(drizzle, { migrationsFolder: './migrations' });
}

export const db = drizzle;