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

import type { PgQueryResultHKT } from "./pgcore.ts"
import { PgAsyncDatabase, PgAsyncTransaction } from "./pgcore.ts"

export type PgNormalDatabase<
  Schemas extends AnySchemas,
  Relations extends AnyRelations,
> = PgAsyncDatabase<PgQueryResultHKT, Schemas, Relations>
export type PgTransactionDatabase<
  Schemas extends AnySchemas,
  Relations extends AnyRelations,
  // oxlint-disable-next-line typescript/no-explicit-any
> = PgAsyncTransaction<PgQueryResultHKT, Schemas, Relations, any>
export type PgUnionDatabase<Schemas extends AnySchemas, Relations extends AnyRelations> =
  | PgNormalDatabase<Schemas, Relations>
  | PgTransactionDatabase<Schemas, Relations>
export interface WithPgUnionDatabase<Schemas extends AnySchemas, Relations extends AnyRelations> {
  database: PgUnionDatabase<Schemas, Relations>
}
export interface WithOptionalPgUnionDatabase<
  Schemas extends AnySchemas,
  Relations extends AnyRelations,
> {
  database?: PgUnionDatabase<Schemas, Relations> | undefined
}
export type WithoutPgUnionDatabase<T> = Omit<T, "database">

export const isPgNormalDatabase = <Schemas extends AnySchemas, Relations extends AnyRelations>(
  database: unknown,
): database is PgNormalDatabase<Schemas, Relations> => {
  return database instanceof PgAsyncDatabase
}
export type GetIsPgNormalDatabase<Schemas extends AnySchemas, Relations extends AnyRelations> = (
  database: unknown,
) => database is PgNormalDatabase<Schemas, Relations>
export const getIsPgNormalDatabase = <
  Schemas extends AnySchemas,
  Relations extends AnyRelations,
>(): GetIsPgNormalDatabase<Schemas, Relations> => {
  return isPgNormalDatabase<Schemas, Relations>
}
export const isPgTransactionDatabase = <Schemas extends AnySchemas, Relations extends AnyRelations>(
  database: unknown,
): database is PgTransactionDatabase<Schemas, Relations> => {
  return database instanceof PgAsyncTransaction
}
export type GetIsPgTransactionDatabase<
  Schemas extends AnySchemas,
  Relations extends AnyRelations,
> = (database: unknown) => database is PgTransactionDatabase<Schemas, Relations>
export const getIsPgTransactionDatabase = <
  Schemas extends AnySchemas,
  Relations extends AnyRelations,
>(): GetIsPgTransactionDatabase<Schemas, Relations> => {
  return isPgTransactionDatabase<Schemas, Relations>
}
export const isPgUnionDatabase = <Schemas extends AnySchemas, Relations extends AnyRelations>(
  database: unknown,
): database is PgUnionDatabase<Schemas, Relations> => {
  if (isPgTransactionDatabase(database) === true) {
    return true
  }
  if (isPgNormalDatabase(database) === true) {
    return true
  }
  return false
}
export type GetIsPgUnionDatabase<Schemas extends AnySchemas, Relations extends AnyRelations> = (
  database: unknown,
) => database is PgUnionDatabase<Schemas, Relations>
export const getIsPgUnionDatabase = <
  Schemas extends AnySchemas,
  Relations extends AnyRelations,
>(): GetIsPgUnionDatabase<Schemas, Relations> => {
  return isPgUnionDatabase<Schemas, Relations>
}

export type AutoPgDatabaseTransaction = <
  Result,
  Schemas extends AnySchemas,
  Relations extends AnyRelations,
>(
  database: PgUnionDatabase<Schemas, Relations>,
  run: (database: PgTransactionDatabase<Schemas, Relations>) => Promise<Result>,
) => Promise<Result>
export const autoPgDatabaseTransaction: AutoPgDatabaseTransaction = async <
  Result,
  Schemas extends AnySchemas,
  Relations extends AnyRelations,
>(
  database: PgUnionDatabase<Schemas, Relations>,
  run: (database: PgTransactionDatabase<Schemas, Relations>) => Promise<Result>,
): Promise<Result> => {
  // detect transaction database first, then normal database
  // because transaction database is a subclass of normal database
  if (isPgTransactionDatabase(database) === true) {
    return await run(database)
  }
  if (isPgNormalDatabase(database) === true) {
    return await database.transaction(async (database) => {
      return await run(database)
    })
  }
  throw new Error("Invalid database instance")
}

export type GetAutoPgDatabaseTransaction<
  Schemas extends AnySchemas,
  Relations extends AnyRelations,
> = <Result>(
  database: PgUnionDatabase<Schemas, Relations>,
  run: (database: PgTransactionDatabase<Schemas, Relations>) => Promise<Result>,
) => Promise<Result>
export const getAutoPgDatabaseTransaction = <
  Schemas extends AnySchemas,
  Relations extends AnyRelations,
>() => {
  return async <Result>(
    database: PgUnionDatabase<Schemas, Relations>,
    run: (database: PgTransactionDatabase<Schemas, Relations>) => Promise<Result>,
  ): Promise<Result> => {
    return await autoPgDatabaseTransaction(database, run)
  }
}