import * as Pgcore from "../pgcore/index.ts"

import type { AnyPglightDrizzleClient } from "./drizzle-client.ts"
import { executeRawSql } from "./execute.ts"

export interface DatabaseClientExecuteRawSqlOptions
  extends Pgcore.DatabaseClientExecuteRawSqlOptions {}
export interface DatabaseClientOnceOptions<
  DrizzleClient extends AnyPglightDrizzleClient = AnyPglightDrizzleClient,
  Result = unknown,
> extends Pgcore.DatabaseClientOnceOptions<DrizzleClient, Result> {}

export interface DatabaseClientOptions<
  DrizzleClient extends AnyPglightDrizzleClient = AnyPglightDrizzleClient,
> extends Pgcore.DatabaseClientOptions<DrizzleClient> {}
export class DatabaseClient<
  DrizzleClient extends AnyPglightDrizzleClient = AnyPglightDrizzleClient,
> extends Pgcore.DatabaseClient<DrizzleClient> {
  constructor(options: DatabaseClientOptions<DrizzleClient>) {
    super(options)
  }

  async executeRawSql<Result>(options: DatabaseClientExecuteRawSqlOptions): Promise<Result> {
    const { sql } = options
    const database = this.getDrizzleClient()
    const result = await executeRawSql({ database, sql })
    return result as Result
  }

  async close(): Promise<void> {
    await this.managedDrizzleClient.$client.end()
  }
}

export const createDatabaseClient = <DrizzleClient extends AnyPglightDrizzleClient>(
  options: DatabaseClientOptions<DrizzleClient>,
): DatabaseClient<DrizzleClient> => {
  const { drizzleClient } = options
  const client = new DatabaseClient({
    drizzleClient,
  })
  return client
}

export const isDatabaseClient = <
  DrizzleClient extends AnyPglightDrizzleClient = AnyPglightDrizzleClient,
>(
  target: unknown,
): target is DatabaseClient<DrizzleClient> => {
  return target instanceof DatabaseClient
}