import { Address, Hex, Signature, Bytes } from 'ox'
import * as Client from './client/guard.gen.js'
import * as Types from './types.js'

export class Guard implements Types.Guard {
  private readonly guard?: Client.Guard
  public readonly address: Address.Address

  constructor(hostname: string, address: Address.Address, fetch?: Client.Fetch) {
    if (hostname && address) {
      this.guard = new Client.Guard(hostname, fetch ?? window.fetch)
    }
    this.address = address
  }

  async signPayload(
    wallet: Address.Address,
    chainId: number,
    type: Client.PayloadType,
    digest: Bytes.Bytes,
    message: Bytes.Bytes,
    signatures?: Client.Signature[],
    token?: Client.AuthToken,
  ) {
    if (!this.guard || !this.address) {
      throw new Error('Guard not initialized')
    }

    try {
      const res = await this.guard.signWith({
        signer: this.address,
        request: {
          chainId: chainId,
          msg: Hex.fromBytes(digest),
          wallet,
          payloadType: type,
          payloadData: Hex.fromBytes(message),
          signatures,
        },
        token,
      })

      Hex.assert(res.sig)
      return Signature.fromHex(res.sig)
    } catch (error) {
      if (error instanceof Client.RequiresTOTPError) {
        throw new Types.AuthRequiredError('TOTP')
      }
      if (error instanceof Client.RequiresPINError) {
        throw new Types.AuthRequiredError('PIN')
      }
      console.error(error)
      throw new Error('Error signing with guard', { cause: error })
    }
  }
}