import type { AnyDrizzleClient } from "./drizzle-client.ts"

export interface DatabaseClientExecuteRawSqlOptions {
  sql: string
}
export interface DatabaseClientOnceOptions<
  DrizzleClient extends AnyDrizzleClient = AnyDrizzleClient,
  Result = unknown,
> {
  run: (drizzleClient: DrizzleClient) => Promise<Result>
}

export interface DatabaseClientOptions<DrizzleClient extends AnyDrizzleClient = AnyDrizzleClient> {
  drizzleClient: DrizzleClient
}
export abstract class DatabaseClient<DrizzleClient extends AnyDrizzleClient = AnyDrizzleClient> {
  protected managedDrizzleClient: DrizzleClient

  constructor(options: DatabaseClientOptions<DrizzleClient>) {
    this.managedDrizzleClient = options.drizzleClient
  }

  getDrizzleClient(): DrizzleClient {
    return this.managedDrizzleClient
  }

  abstract executeRawSql<Result>(options: DatabaseClientExecuteRawSqlOptions): Promise<Result>

  // 底层数据库实例的关闭方法不同，如 `postgres.js` 使用 `client.end()`，
  // `@electric-sql/pglite` 使用 `client.close()`，因此在这里定义一个抽
  // 象方法，由具体的客户端实现类来实现。
  abstract close(): Promise<void>

  async once<Result>(options: DatabaseClientOnceOptions<DrizzleClient, Result>): Promise<Result> {
    const { run } = options
    const drizzleClient = this.getDrizzleClient()
    try {
      const result = await run(drizzleClient)
      return result
    } finally {
      await this.close()
    }
  }
}

export const isDatabaseClient = <DrizzleClient extends AnyDrizzleClient = AnyDrizzleClient>(
  target: unknown,
): target is DatabaseClient<DrizzleClient> => {
  return target instanceof DatabaseClient
}