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

import type { WithPgUnionDatabase } from "./database.ts"
import { autoPgDatabaseTransaction } from "./database.ts"

export interface ExecuteRawSqlOptions<
  Schemas extends AnySchemas,
  Relations extends AnyRelations,
> extends WithPgUnionDatabase<Schemas, Relations> {
  sql: string
}
export type ExecuteRawSql = <Result, Schemas extends AnySchemas, Relations extends AnyRelations>(
  options: ExecuteRawSqlOptions<Schemas, Relations>,
) => Promise<Result>
export const executeRawSql: ExecuteRawSql = async <
  Result,
  Schemas extends AnySchemas,
  Relations extends AnyRelations,
>(
  options: ExecuteRawSqlOptions<Schemas, Relations>,
): Promise<Result> => {
  const { database, sql } = options
  const result = await autoPgDatabaseTransaction(database, async (tx) => {
    const sqls = splitMultiStatementsSql(sql)
    let result
    for (const sql of sqls) {
      result = await tx.execute(sql)
    }
    return result
  })
  return result as Result
}

export type GetExecuteRawSql<Schemas extends AnySchemas, Relations extends AnyRelations> = <Result>(
  options: ExecuteRawSqlOptions<Schemas, Relations>,
) => Promise<Result>
export const getExecuteRawSql = <Schemas extends AnySchemas, Relations extends AnyRelations>() => {
  return async <Result>(options: ExecuteRawSqlOptions<Schemas, Relations>): Promise<Result> => {
    return await executeRawSql(options)
  }
}