import { createRequire } from "node:module";
import path from "node:path";

export interface SqliteStatement {
  run: (...params: unknown[]) => unknown;
  get: (...params: unknown[]) => unknown;
  all: (...params: unknown[]) => unknown[];
}

export interface SqliteDatabase {
  exec: (sql: string) => void;
  prepare: (sql: string) => SqliteStatement;
  close: () => void;
}

interface NodeSqliteModule {
  DatabaseSync: new (filename: string) => {
    exec: (sql: string) => void;
    prepare: (sql: string) => SqliteStatement;
    close: () => void;
  };
}

interface BunSqliteStatement {
  run: (...params: unknown[]) => unknown;
  get: (...params: unknown[]) => unknown;
  all: (...params: unknown[]) => unknown[];
}

interface BunSqliteModule {
  Database: new (filename: string) => {
    exec: (sql: string) => unknown;
    query: (sql: string) => BunSqliteStatement;
    close: () => void;
  };
}

function openNodeSqliteDatabase(
  requireFn: ReturnType<typeof createRequire>,
  dbPath: string,
): SqliteDatabase {
  const nodeSqlite = requireFn("node:sqlite") as NodeSqliteModule;
  const db = new nodeSqlite.DatabaseSync(dbPath);
  return {
    exec: (sql) => {
      db.exec(sql);
    },
    prepare: (sql) => db.prepare(sql),
    close: () => {
      db.close();
    },
  };
}

function openBunSqliteDatabase(
  requireFn: ReturnType<typeof createRequire>,
  dbPath: string,
): SqliteDatabase {
  const bunSqlite = requireFn("bun:sqlite") as BunSqliteModule;
  const db = new bunSqlite.Database(dbPath);
  return {
    exec: (sql) => {
      db.exec(sql);
    },
    prepare: (sql) => {
      const statement = db.query(sql);
      return {
        run: (...params: unknown[]) => statement.run(...params),
        get: (...params: unknown[]) => statement.get(...params),
        all: (...params: unknown[]) => statement.all(...params),
      };
    },
    close: () => {
      db.close();
    },
  };
}

export function openSqliteDatabase(dbPath: string): SqliteDatabase {
  const requireFn = createRequire(path.join(process.cwd(), "t3code-sqlite-adapter.cjs"));

  if (process.versions.bun) {
    // Development mode runs in Bun, so use Bun's SQLite adapter
    return openBunSqliteDatabase(requireFn, dbPath);
  }

  return openNodeSqliteDatabase(requireFn, dbPath);
}