import { Transport, coin as COIN } from '@coolwallet/core';
import { sign } from './sign';
import {
  getTransferArgument,
  getCancelOrderArgument,
  getPlaceOrderArgument,
  getTokenArgument,
} from './utils/scriptUtil';
import { signType, signCancelOrderType, signPlaceOrderType } from './config/types';
import * as param from './config/param';
import * as txUtil from './utils/transactionUtil';
import * as Types from './config/types';
import { TOKEN_SIGS } from './config/tokenType';

export default class BNB extends COIN.ECDSACoin implements COIN.Coin {
  public Types: any;

  constructor() {
    super(param.COIN_TYPE);
    this.Types = Types;
  }

  /**
   * Get Binance address by index
   */
  async getAddress(transport: Transport, appPrivateKey: string, appId: string, addressIndex: number): Promise<string> {
    const publicKey = await this.getPublicKey(transport, appPrivateKey, appId, addressIndex);
    return txUtil.publicKeyToAddress(publicKey);
  }

  async getAddressByAccountKey(accPublicKey: string, accChainCode: string, addressIndex: number): Promise<string> {
    const publicKey = await this.getAddressPublicKey(accPublicKey, accChainCode, addressIndex);
    return txUtil.publicKeyToAddress(publicKey);
  }

  /**
   * Sign Binance transfer transaction.
   */
  async signTransaction(signData: signType, returnSig = false): Promise<string> {
    const denom = signData.signObj.msgs[0].inputs[0].coins[0].denom;

    if ('BNB' === denom) {
      return this.signTransferTransaction(signData, returnSig);
    }

    let tokenSignature;
    for (const tokenInfo of TOKEN_SIGS) {
      // get tokenSignature
      if (denom.toUpperCase() === tokenInfo.contractAddress) {
        tokenSignature = tokenInfo.signature;
        break;
      }
    }
    if (tokenSignature) {
      return await this.signTokenTransaction(signData, denom, tokenSignature, returnSig); // 內建
    } else {
      return await this.signTokenTransaction(signData, denom, '', returnSig); // 自建
    }
  }

  /**
   * Sign Binance transfer transaction.
   */
  async signTransferTransaction(signData: signType, returnSig = false): Promise<string> {
    const script = param.TRANSFER.script + param.TRANSFER.signature;
    const argument = await getTransferArgument(signData.signObj, signData.addressIndex);
    const signature = await sign(
      signData.transport,
      signData.appId,
      signData.appPrivateKey,
      script,
      argument,
      signData.confirmCB,
      signData.authorizedCB
    );
    if (returnSig) return signature;
    return txUtil.composeSignedTransacton(signData.signObj as Types.Transfer, 'BNB', signature, signData.signPublicKey);
  }

  /**
   * Sign Binance token transaction.
   */
  async signTokenTransaction(
    signData: signType,
    denom: string,
    tokenSignature = '',
    returnSig = false
  ): Promise<string> {
    const script = param.BEP2Token.script + param.BEP2Token.signature;
    const argument = await getTokenArgument(signData.signObj, denom, tokenSignature, signData.addressIndex);
    const signature = await sign(
      signData.transport,
      signData.appId,
      signData.appPrivateKey,
      script,
      argument,
      signData.confirmCB,
      signData.authorizedCB
    );
    if (returnSig) return signature;
    return txUtil.composeSignedTransacton(signData.signObj as Types.Transfer, denom, signature, signData.signPublicKey);
  }

  /**
   * Sign PlaceOrder Transaction
   */
  async signPlaceOrder(signData: signPlaceOrderType): Promise<string> {
    const script = param.PlaceOrder.script + param.PlaceOrder.signature;
    const argument = await getPlaceOrderArgument(signData.signObj, signData.addressIndex);
    const signature = await sign(
      signData.transport,
      signData.appId,
      signData.appPrivateKey,
      script,
      argument,
      signData.confirmCB,
      signData.authorizedCB
    );
    return signature;
  }

  /**
   * Sign CancelOrder Transaction
   */
  async signCancelOrder(signData: signCancelOrderType): Promise<string> {
    const script = param.CancelOrder.script + param.CancelOrder.signature;
    const argument = await getCancelOrderArgument(signData.signObj, signData.addressIndex);
    const signature = await sign(
      signData.transport,
      signData.appId,
      signData.appPrivateKey,
      script,
      argument,
      signData.confirmCB,
      signData.authorizedCB
    );
    return signature;
  }
}