import type { PglightClient } from "../../pglight/index.ts"
import type { AnySchemas, AnyRelations, DrizzleConfig } from "../orm/index.ts"

import type { PglightNormalDatabase } from "./database.ts"
import { isPglightNormalDatabase } from "./database.ts"
import { drizzle } from "./pglight.ts"

export interface CreatePglightDrizzleClientOptions<
  TableSchemas extends AnySchemas,
  Relations extends AnyRelations,
> extends DrizzleConfig<TableSchemas, Relations> {
  pglightClient: PglightClient
}
export interface PglightDrizzleClient<
  TableSchemas extends AnySchemas,
  Relations extends AnyRelations,
> extends PglightNormalDatabase<TableSchemas, Relations> {
  $client: PglightClient
}
export const createPglightDrizzleClient = <
  TableSchemas extends AnySchemas,
  Relations extends AnyRelations,
>(
  options: CreatePglightDrizzleClientOptions<TableSchemas, Relations>,
): PglightDrizzleClient<TableSchemas, Relations> => {
  const { pglightClient, ...drizzleConfig } = options
  const drizzleClient = drizzle({
    client: pglightClient,
    ...drizzleConfig,
  })
  return drizzleClient
}
export type GetCreatePglightDrizzleClient<
  TableSchemas extends AnySchemas,
  Relations extends AnyRelations,
> = (
  options: CreatePglightDrizzleClientOptions<TableSchemas, Relations>,
) => PglightDrizzleClient<TableSchemas, Relations>
export const getCreatePglightDrizzleClient = <
  TableSchemas extends AnySchemas,
  Relations extends AnyRelations,
>(): GetCreatePglightDrizzleClient<TableSchemas, Relations> => {
  return createPglightDrizzleClient<TableSchemas, Relations>
}

export type AnyPglightDrizzleClient = PglightDrizzleClient<AnySchemas, AnyRelations>

export const isPglightDrizzleClient = <
  TableSchemas extends AnySchemas,
  Relations extends AnyRelations,
>(
  target: unknown,
): target is PglightDrizzleClient<TableSchemas, Relations> => {
  if (isPglightNormalDatabase(target) === false) {
    return false
  }
  if ("$client" in target === false) {
    return false
  }
  return true
}