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

import { PgliteDatabase as RawDatabase, PgliteTransaction as RawTransaction } from "./pglite.ts"

export type PgliteNormalDatabase<
  Schemas extends AnySchemas,
  Relations extends AnyRelations,
> = RawDatabase<Schemas, Relations>
export type PgliteTransactionDatabase<
  Schemas extends AnySchemas,
  Relations extends AnyRelations,
  // oxlint-disable-next-line typescript/no-explicit-any
> = RawTransaction<Schemas, Relations, any>
export type PgliteUnionDatabase<Schemas extends AnySchemas, Relations extends AnyRelations> =
  | PgliteNormalDatabase<Schemas, Relations>
  | PgliteTransactionDatabase<Schemas, Relations>
export interface WithPgliteUnionDatabase<
  Schemas extends AnySchemas,
  Relations extends AnyRelations,
> {
  database: PgliteUnionDatabase<Schemas, Relations>
}
export interface WithOptionalPgliteUnionDatabase<
  Schemas extends AnySchemas,
  Relations extends AnyRelations,
> {
  database?: PgliteUnionDatabase<Schemas, Relations> | undefined
}
export type WithoutPgliteUnionDatabase<T> = Omit<T, "database">

export const isPgliteNormalDatabase = <Schemas extends AnySchemas, Relations extends AnyRelations>(
  database: unknown,
): database is PgliteNormalDatabase<Schemas, Relations> => {
  return database instanceof RawDatabase
}
export type GetIsPgliteNormalDatabase<
  Schemas extends AnySchemas,
  Relations extends AnyRelations,
> = (database: unknown) => database is PgliteNormalDatabase<Schemas, Relations>
export const getIsPgliteNormalDatabase = <
  Schemas extends AnySchemas,
  Relations extends AnyRelations,
>(): GetIsPgliteNormalDatabase<Schemas, Relations> => {
  return isPgliteNormalDatabase<Schemas, Relations>
}
export const isPgliteTransactionDatabase = <
  Schemas extends AnySchemas,
  Relations extends AnyRelations,
>(
  database: unknown,
): database is PgliteTransactionDatabase<Schemas, Relations> => {
  return database instanceof RawTransaction
}
export type GetIsPgliteTransactionDatabase<
  Schemas extends AnySchemas,
  Relations extends AnyRelations,
> = (database: unknown) => database is PgliteTransactionDatabase<Schemas, Relations>
export const getIsPgliteTransactionDatabase = <
  Schemas extends AnySchemas,
  Relations extends AnyRelations,
>(): GetIsPgliteTransactionDatabase<Schemas, Relations> => {
  return isPgliteTransactionDatabase<Schemas, Relations>
}
export const isPgliteUnionDatabase = <Schemas extends AnySchemas, Relations extends AnyRelations>(
  database: unknown,
): database is PgliteUnionDatabase<Schemas, Relations> => {
  if (isPgliteTransactionDatabase(database) === true) {
    return true
  }
  if (isPgliteNormalDatabase(database) === true) {
    return true
  }
  return false
}
export type GetIsPgliteUnionDatabase<Schemas extends AnySchemas, Relations extends AnyRelations> = (
  database: unknown,
) => database is PgliteUnionDatabase<Schemas, Relations>
export const getIsPgliteUnionDatabase = <
  Schemas extends AnySchemas,
  Relations extends AnyRelations,
>(): GetIsPgliteUnionDatabase<Schemas, Relations> => {
  return isPgliteUnionDatabase<Schemas, Relations>
}

export type AutoPgliteDatabaseTransaction = <
  Result,
  Schemas extends AnySchemas,
  Relations extends AnyRelations,
>(
  database: PgliteUnionDatabase<Schemas, Relations>,
  run: (database: PgliteTransactionDatabase<Schemas, Relations>) => Promise<Result>,
) => Promise<Result>
export const autoPgliteDatabaseTransaction = async <
  Result,
  Schemas extends AnySchemas,
  Relations extends AnyRelations,
>(
  database: PgliteUnionDatabase<Schemas, Relations>,
  run: (database: PgliteTransactionDatabase<Schemas, Relations>) => Promise<Result>,
): Promise<Result> => {
  // detect transaction database first, then normal database
  // because transaction database is a subclass of normal database
  if (isPgliteTransactionDatabase(database)) {
    return await run(database)
  }
  if (isPgliteNormalDatabase(database)) {
    return await database.transaction(async (database) => {
      return await run(database)
    })
  }
  throw new Error("Invalid database instance")
}

export type GetAutoPgliteDatabaseTransaction<
  Schemas extends AnySchemas,
  Relations extends AnyRelations,
> = <Result>(
  database: PgliteUnionDatabase<Schemas, Relations>,
  run: (database: PgliteTransactionDatabase<Schemas, Relations>) => Promise<Result>,
) => Promise<Result>
export const getAutoPgliteDatabaseTransaction = <
  Schemas extends AnySchemas,
  Relations extends AnyRelations,
>() => {
  return async <Result>(
    database: PgliteUnionDatabase<Schemas, Relations>,
    run: (database: PgliteTransactionDatabase<Schemas, Relations>) => Promise<Result>,
  ): Promise<Result> => {
    return await autoPgliteDatabaseTransaction(database, run)
  }
}