import { BigNumber, ethers } from "ethers"
import { arrayify } from "ethers/lib/utils"

export enum RelayerPayloadId {
  Delivery = 1,
  Redelivery = 2,
  // DeliveryStatus = 3,
}

export interface DeliveryInstructionsContainer {
  payloadId: number // 1
  sufficientlyFunded: boolean
  instructions: DeliveryInstruction[]
}

export interface DeliveryInstruction {
  targetChain: number
  targetAddress: Buffer
  refundAddress: Buffer
  maximumRefundTarget: BigNumber
  receiverValueTarget: BigNumber
  executionParameters: ExecutionParameters
}

export interface ExecutionParameters {
  version: number
  gasLimit: number
  providerDeliveryAddress: Buffer
}

export interface RedeliveryByTxHashInstruction {
  payloadId: number //2
  sourceChain: number
  sourceTxHash: Buffer
  sourceNonce: BigNumber
  targetChain: number
  deliveryIndex: number
  multisendIndex: number
  newMaximumRefundTarget: BigNumber
  newReceiverValueTarget: BigNumber
  executionParameters: ExecutionParameters
}

export function parsePayloadType(
  stringPayload: string | Buffer | Uint8Array
): RelayerPayloadId {
  const payload =
    typeof stringPayload === "string" ? arrayify(stringPayload) : stringPayload
  if (payload[0] == 0 || payload[0] >= 3) {
    throw new Error("Unrecogned payload type " + payload[0])
  }
  return payload[0]
}

export function parseDeliveryInstructionsContainer(
  bytes: Buffer
): DeliveryInstructionsContainer {
  let idx = 0
  const payloadId = bytes.readUInt8(idx)
  if (payloadId !== RelayerPayloadId.Delivery) {
    throw new Error(
      `Expected Delivery payload type (${RelayerPayloadId.Delivery}), found: ${payloadId}`
    )
  }
  idx += 1

  const sufficientlyFunded = Boolean(bytes.readUInt8(idx))
  idx += 1

  const numInstructions = bytes.readUInt8(idx)
  idx += 1
  let instructions = [] as DeliveryInstruction[]
  for (let i = 0; i < numInstructions; ++i) {
    const targetChain = bytes.readUInt16BE(idx)
    idx += 2
    const targetAddress = bytes.slice(idx, idx + 32)
    idx += 32
    const refundAddress = bytes.slice(idx, idx + 32)
    idx += 32
    const maximumRefundTarget = ethers.BigNumber.from(
      Uint8Array.prototype.subarray.call(bytes, idx, idx + 32)
    )
    idx += 32
    const receiverValueTarget = ethers.BigNumber.from(
      Uint8Array.prototype.subarray.call(bytes, idx, idx + 32)
    )
    idx += 32
    const executionParameters = parseExecutionParameters(bytes, idx)
    idx += 37
    instructions.push(
      // dumb typechain format
      {
        targetChain,
        targetAddress,
        refundAddress,
        maximumRefundTarget,
        receiverValueTarget,
        executionParameters,
      }
    )
  }
  return {
    payloadId,
    sufficientlyFunded,
    instructions,
  }
}

export function parseRedeliveryByTxHashInstruction(
  bytes: Buffer
): RedeliveryByTxHashInstruction {
  let idx = 0
  const payloadId = bytes.readUInt8(idx)
  if (payloadId !== RelayerPayloadId.Redelivery) {
    throw new Error(
      `Expected Delivery payload type (${RelayerPayloadId.Redelivery}), found: ${payloadId}`
    )
  }
  idx += 1

  const sourceChain = bytes.readUInt16BE(idx)
  idx += 2

  const sourceTxHash = bytes.slice(idx, idx + 32)
  idx += 32

  const sourceNonce = BigNumber.from(bytes.slice(idx, idx + 4))
  idx += 4

  const targetChain = bytes.readUInt16BE(idx)
  idx += 2

  const deliveryIndex = bytes.readUint8(idx)
  idx += 1

  const multisendIndex = bytes.readUint8(idx)
  idx += 1

  // note: confirmed that BigNumber.from(<Buffer>) assumes big endian
  const newMaximumRefundTarget = BigNumber.from(bytes.slice(idx, idx + 32))
  idx += 32

  const newReceiverValueTarget = BigNumber.from(bytes.slice(idx, idx + 32))
  idx += 32

  const executionParameters = parseExecutionParameters(bytes, idx)
  idx += 37
  return {
    payloadId,
    sourceChain,
    sourceTxHash,
    sourceNonce,
    targetChain,
    deliveryIndex,
    multisendIndex,
    newMaximumRefundTarget,
    newReceiverValueTarget,
    executionParameters,
  }
}

function parseExecutionParameters(bytes: Buffer, idx: number = 0): ExecutionParameters {
  const version = bytes.readUInt8(idx)
  idx += 1
  const gasLimit = bytes.readUint32BE(idx)
  idx += 4
  const providerDeliveryAddress = bytes.slice(idx, idx + 32)
  idx += 32
  return { version, gasLimit, providerDeliveryAddress }
}

/*
 * Helpers
 */

export function dbg<T>(x: T, msg?: string): T {
  if (msg) {
    console.log("[DEBUG] " + msg)
  }
  console.log(x)
  return x
}