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

import { PglightDatabase as RawDatabase, PglightTransaction as RawTransaction } from "./pglight.ts"

export type PglightNormalDatabase<
  Schemas extends AnySchemas,
  Relations extends AnyRelations,
> = RawDatabase<Schemas, Relations>
export type PglightTransactionDatabase<
  Schemas extends AnySchemas,
  Relations extends AnyRelations,
  // oxlint-disable-next-line typescript/no-explicit-any
> = RawTransaction<Schemas, Relations, any>
export type PglightUnionDatabase<Schemas extends AnySchemas, Relations extends AnyRelations> =
  | PglightNormalDatabase<Schemas, Relations>
  | PglightTransactionDatabase<Schemas, Relations>
export interface WithPglightUnionDatabase<
  Schemas extends AnySchemas,
  Relations extends AnyRelations,
> {
  database: PglightUnionDatabase<Schemas, Relations>
}
export interface WithOptionalPglightUnionDatabase<
  Schemas extends AnySchemas,
  Relations extends AnyRelations,
> {
  database?: PglightUnionDatabase<Schemas, Relations> | undefined
}
export type WithoutPglightUnionDatabase<T> = Omit<T, "database">

export const isPglightNormalDatabase = <Schemas extends AnySchemas, Relations extends AnyRelations>(
  database: unknown,
): database is PglightNormalDatabase<Schemas, Relations> => {
  return database instanceof RawDatabase
}
export type GetIsPglightNormalDatabase<
  Schemas extends AnySchemas,
  Relations extends AnyRelations,
> = (database: unknown) => database is PglightNormalDatabase<Schemas, Relations>
export const getIsPglightNormalDatabase = <
  Schemas extends AnySchemas,
  Relations extends AnyRelations,
>(): GetIsPglightNormalDatabase<Schemas, Relations> => {
  return isPglightNormalDatabase<Schemas, Relations>
}
export const isPglightTransactionDatabase = <
  Schemas extends AnySchemas,
  Relations extends AnyRelations,
>(
  database: unknown,
): database is PglightTransactionDatabase<Schemas, Relations> => {
  return database instanceof RawTransaction
}
export type GetIsPglightTransactionDatabase<
  Schemas extends AnySchemas,
  Relations extends AnyRelations,
> = (database: unknown) => database is PglightTransactionDatabase<Schemas, Relations>
export const getIsPglightTransactionDatabase = <
  Schemas extends AnySchemas,
  Relations extends AnyRelations,
>(): GetIsPglightTransactionDatabase<Schemas, Relations> => {
  return isPglightTransactionDatabase<Schemas, Relations>
}
export const isPglightUnionDatabase = <Schemas extends AnySchemas, Relations extends AnyRelations>(
  database: unknown,
): database is PglightUnionDatabase<Schemas, Relations> => {
  if (isPglightTransactionDatabase(database) === true) {
    return true
  }
  if (isPglightNormalDatabase(database) === true) {
    return true
  }
  return false
}
export type GetIsPglightUnionDatabase<
  Schemas extends AnySchemas,
  Relations extends AnyRelations,
> = (database: unknown) => database is PglightUnionDatabase<Schemas, Relations>
export const getIsPglightUnionDatabase = <
  Schemas extends AnySchemas,
  Relations extends AnyRelations,
>(): GetIsPglightUnionDatabase<Schemas, Relations> => {
  return isPglightUnionDatabase<Schemas, Relations>
}

export type AutoPglightDatabaseTransaction = <
  Result,
  Schemas extends AnySchemas,
  Relations extends AnyRelations,
>(
  database: PglightUnionDatabase<Schemas, Relations>,
  run: (database: PglightTransactionDatabase<Schemas, Relations>) => Promise<Result>,
) => Promise<Result>
export const autoPglightDatabaseTransaction: AutoPglightDatabaseTransaction = async <
  Result,
  Schemas extends AnySchemas,
  Relations extends AnyRelations,
>(
  database: PglightUnionDatabase<Schemas, Relations>,
  run: (database: PglightTransactionDatabase<Schemas, Relations>) => Promise<Result>,
): Promise<Result> => {
  // detect transaction database first, then normal database
  // because transaction database is a subclass of normal database
  if (isPglightTransactionDatabase(database)) {
    return await run(database)
  }
  if (isPglightNormalDatabase(database)) {
    return await database.transaction(async (database) => {
      return await run(database)
    })
  }
  throw new Error("Invalid database instance")
}

export type GetAutoPglightDatabaseTransaction<
  Schemas extends AnySchemas,
  Relations extends AnyRelations,
> = <Result>(
  database: PglightUnionDatabase<Schemas, Relations>,
  run: (database: PglightTransactionDatabase<Schemas, Relations>) => Promise<Result>,
) => Promise<Result>
export const getAutoPglightDatabaseTransaction = <
  Schemas extends AnySchemas,
  Relations extends AnyRelations,
>() => {
  return async <Result>(
    database: PglightUnionDatabase<Schemas, Relations>,
    run: (database: PglightTransactionDatabase<Schemas, Relations>) => Promise<Result>,
  ): Promise<Result> => {
    return await autoPglightDatabaseTransaction(database, run)
  }
}