import type { PgliteClient } from "../../pglite/index.ts"
import type { AnySchemas, AnyRelations, DrizzleConfig } from "../orm/index.ts"

import type { PgliteNormalDatabase } from "./database.ts"
import { isPgliteNormalDatabase } from "./database.ts"
import { drizzle } from "./pglite.ts"

export interface CreatePgliteDrizzleClientOptions<
  TableSchemas extends AnySchemas,
  Relations extends AnyRelations,
> extends DrizzleConfig<TableSchemas, Relations> {
  pgliteClient: PgliteClient
}
export interface PgliteDrizzleClient<
  TableSchemas extends AnySchemas,
  Relations extends AnyRelations,
> extends PgliteNormalDatabase<TableSchemas, Relations> {
  $client: PgliteClient
}
export const createPgliteDrizzleClient = <
  TableSchemas extends AnySchemas,
  Relations extends AnyRelations,
>(
  options: CreatePgliteDrizzleClientOptions<TableSchemas, Relations>,
): PgliteDrizzleClient<TableSchemas, Relations> => {
  const { pgliteClient, ...drizzleConfig } = options
  const drizzleClient = drizzle({
    client: pgliteClient,
    ...drizzleConfig,
  })
  return drizzleClient
}
export type GetCreatePgliteDrizzleClient<
  TableSchemas extends AnySchemas,
  Relations extends AnyRelations,
> = (
  options: CreatePgliteDrizzleClientOptions<TableSchemas, Relations>,
) => PgliteDrizzleClient<TableSchemas, Relations>
export const getCreatePgliteDrizzleClient = <
  TableSchemas extends AnySchemas,
  Relations extends AnyRelations,
>(): GetCreatePgliteDrizzleClient<TableSchemas, Relations> => {
  return createPgliteDrizzleClient<TableSchemas, Relations>
}

export type AnyPgliteDrizzleClient = PgliteDrizzleClient<AnySchemas, AnyRelations>

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