import { UI_REQUEST } from '../../constants/ui-request';
import { getScriptType, serializedPath, validatePath } from '../helpers/pathUtils';
import { BaseMethod } from '../BaseMethod';
import { validateParams, validateResult } from '../helpers/paramsValidator';
import { getCoinInfo } from './helpers/btcParamsUtils';
import {
  getBitcoinForkSupportedProtocols,
  getBitcoinForkVersionRange,
} from './helpers/versionLimit';

import type { BTCAddress, BTCGetAddressParams } from '../../types/api/btcGetAddress';
import type { GetAddress } from '@onekeyfe/hd-transport';

export default class BTCGetAddress extends BaseMethod<GetAddress[]> {
  hasBundle = false;

  getSupportedProtocols() {
    return getBitcoinForkSupportedProtocols(this.params?.map(param => param.coin_name) ?? []);
  }

  init() {
    this.checkDeviceId = true;
    this.allowDeviceMode = [...this.allowDeviceMode, UI_REQUEST.NOT_INITIALIZE];

    this.hasBundle = Object.prototype.hasOwnProperty.call(this.payload, 'bundle');
    const payload = this.hasBundle ? this.payload : { bundle: [this.payload] };

    validateParams(payload, [{ name: 'bundle', type: 'array' }]);

    this.params = [];
    payload.bundle.forEach((batch: BTCGetAddressParams) => {
      const addressN = validatePath(batch.path, 1);

      validateParams(batch, [
        { name: 'path', required: true },
        { name: 'coin', type: 'string' },
        { name: 'showOnOneKey', type: 'boolean' },
        { name: 'multisig', type: 'object' },
        { name: 'scriptType', type: 'string' },
      ]);

      const showOnOneKey = batch.showOnOneKey ?? true;

      const { multisig, coin } = batch;

      let { scriptType } = batch;
      if (!scriptType) {
        scriptType = getScriptType(addressN);
        if (scriptType === 'SPENDMULTISIG' && !multisig) {
          scriptType = 'SPENDADDRESS';
        }
      }

      const coinName = getCoinInfo(addressN, coin).name;

      this.params.push({
        address_n: addressN,
        show_display: showOnOneKey,
        coin_name: coinName,
        multisig,
        script_type: scriptType || 'SPENDADDRESS',
      });
    });
  }

  getVersionRange() {
    return getBitcoinForkVersionRange(this.params.map(param => param.coin_name));
  }

  async run() {
    const responses: BTCAddress[] = [];

    for (let i = 0; i < this.params.length; i++) {
      const param = this.params[i];

      const res = await this.device.commands.typedCall('GetAddress', 'Address', {
        ...param,
      });

      const path = serializedPath(param.address_n);
      responses.push({
        path,
        ...res.message,
      });

      this.postPreviousAddressMessage({
        address: res.message.address,
        path,
      });
    }

    validateResult(responses, ['address'], {
      expectedLength: this.params.length,
    });

    return Promise.resolve(this.hasBundle ? responses : responses[0]);
  }
}