import type { MigrationArgs } from '../lib/args'
import { createPostgresDb, createSqliteDb } from '../lib/db'
import { runMigration } from '../lib/migrate'

export const runPostgresToSqlite = async (args: MigrationArgs) => {
  const { db: sourceDb, pool } = createPostgresDb(args.source)
  const { db: targetDb, sqlite } = createSqliteDb(args.target)

  try {
    await runMigration({
      from: 'postgres',
      to: 'sqlite',
      sourceDb,
      targetDb,
      batchSize: args.batchSize,
    })
  } finally {
    await pool.end()
    sqlite.close()
  }
}