import { MsgDelegate as BaseMsgDelegate } from '@routerprotocol/chain-api/cosmos/staking/v1beta1/tx_pb'
import { Coin } from '@routerprotocol/chain-api/cosmos/base/v1beta1/coin_pb'
import { MsgBase } from '../../MsgBase'

export declare namespace MsgDelegate {
  export interface Params {
    amount: {
      denom: string;
      amount: string;
    };
    validatorAddress: string;
    routerAddress: string;
  }
  export interface DirectSign {
    type: '/cosmos.staking.v1beta1.MsgDelegate'
    message: BaseMsgDelegate
  }

  export interface Data extends BaseMsgDelegate.AsObject {
    '@type': '/cosmos.staking.v1beta1.MsgDelegate'
  }

  export interface Amino extends BaseMsgDelegate.AsObject {
    type: 'cosmos-sdk/MsgDelegate'
  }

  export interface Web3 extends BaseMsgDelegate.AsObject {
    '@type': '/cosmos.authz.v1beta1.MsgDelegate'
  }

  export type Proto = BaseMsgDelegate
}

/**
 * @group Message
 */
export default class MsgDelegate extends MsgBase<
  MsgDelegate.Params,
  MsgDelegate.Data,
  MsgDelegate.Proto,
  MsgDelegate.Amino,
  MsgDelegate.DirectSign
> {
  static fromJSON(params: MsgDelegate.Params): MsgDelegate {
    return new MsgDelegate(params)
  }

  public toProto(): MsgDelegate.Proto {
    const { params } = this

    const coinAmount = new Coin()
    coinAmount.setDenom(params.amount.denom)
    coinAmount.setAmount(params.amount.amount)

    const message = new BaseMsgDelegate()
    message.setAmount(coinAmount)
    message.setDelegatorAddress(params.routerAddress);
    message.setValidatorAddress(params.validatorAddress)

    return message
  }

  public toData(): MsgDelegate.Data {
    const proto = this.toProto()

    return {
      '@type': '/cosmos.staking.v1beta1.MsgDelegate',
      ...proto.toObject(),
    }
  }

  public toAmino(): MsgDelegate.Amino {
    const proto = this.toProto()

    return {
      type: 'cosmos-sdk/MsgDelegate',
      ...proto.toObject(),
    }
  }

  public toWeb3(): MsgDelegate.Web3 {
    const amino = this.toAmino()
    const { type, ...rest } = amino

    return {
      '@type': '/cosmos.staking.v1beta1.MsgDelegate',
      ...rest,
    } as unknown as MsgDelegate.Web3
  }

  public toDirectSign(): MsgDelegate.DirectSign {
    const proto = this.toProto()

    return {
      type: '/cosmos.staking.v1beta1.MsgDelegate',
      message: proto,
    }
  }
}