import "../../_dnt.polyfills.js";
import {
  Chain,
  ChainRune,
  Era,
  hex,
  is,
  Rune,
  RunicArgs,
  SignatureData,
  ss58,
  ValueRune,
} from "../../mod.js"
import { SignatureProps } from "./polkadot.js"

type StatemintSignatureProps<C extends Chain> = SignatureProps<C> & {
  assetId?: number
}

export function signature<X, C extends Chain>(_props: RunicArgs<X, StatemintSignatureProps<C>>) {
  return <CU>(chain: ChainRune<C, CU>) => {
    const props = RunicArgs.resolve(_props)
    const addrPrefix = chain.addressPrefix()
    const versions = chain.pallet("System").constant("Version").decoded
      .unsafeAs<any>().into(ValueRune)
    const specVersion = versions.access("specVersion")
    const transactionVersion = versions.access("transactionVersion")
    // TODO: create union rune (with `matchTag` method) and utilize here
    // TODO: MultiAddress conversion utils
    const senderSs58 = Rune
      .tuple([addrPrefix, props.sender])
      .map(([addrPrefix, sender]) => {
        switch ((sender.address as any).type) {
          case "Id":
            return ss58.encode(addrPrefix as any, (sender.address as any).value)
          default:
            throw new Error("unimplemented")
        }
      })
      .throws(is(ss58.InvalidPayloadLengthError))
    const nonce = Rune.resolve(props.nonce)
      .handle(is(undefined), () => chain.connection.call("system_accountNextIndex", senderSs58))
    const genesisHashHex = chain.connection.call("chain_getBlockHash", 0).unsafeAs<string>()
      .into(ValueRune)
    const genesisHash = genesisHashHex.map(hex.decode)
    const checkpointHash = Rune.tuple([props.checkpoint, genesisHashHex]).map(([a, b]) => a ?? b)
      .map(hex.decode)
    const mortality = Rune.resolve(props.mortality).map((x) => x ?? Era.Immortal)
    const tip = Rune.resolve(props.tip).map((x) => x ?? 0n)
    return Rune.object({
      sender: props.sender,
      extra: Rune.object({
        CheckMortality: mortality,
        CheckNonce: nonce,
        ChargeTransactionPayment: tip,
        ChargeAssetTxPayment: Rune.object({
          assetId: props.assetId,
          tip: tip,
        }),
      }),
      additional: Rune.object({
        CheckSpecVersion: specVersion,
        CheckTxVersion: transactionVersion,
        CheckGenesis: genesisHash,
        CheckMortality: checkpointHash,
      }),
      // @ts-ignore: .
    }) satisfies Rune<SignatureData<C>, unknown>
  }
}