import { DeviceRebootType, DeviceSessionPinType } from '@onekeyfe/hd-transport';

import { BaseMethod } from '../BaseMethod';

import type { RebootToBootloader } from '@onekeyfe/hd-transport';

// Reboot BootLoader
export default class DeviceRebootToBootloader extends BaseMethod<RebootToBootloader> {
  getSupportedProtocols() {
    return ['V1', 'V2'] as const;
  }

  init() {
    this.useDevicePassphraseState = false;
    this.skipForceUpdateCheck = true;
    this.unlockPolicy = 'unlock-before-run';
    this.protocolV2PreUnlockPinType = DeviceSessionPinType.Any;
  }

  getVersionRange() {
    return {
      classic: {
        min: '2.1.11',
      },
      mini: {
        min: '2.1.11',
      },
    };
  }

  async run() {
    // Protocol V2 submodule firmware uses DeviceReboot with reboot_type enum.
    if (this.device.isProtocolV2()) {
      const res = await this.device.commands.typedCall('DeviceReboot', 'Success', {
        reboot_type: DeviceRebootType.Bootloader,
      });
      this.device.markProtocolV2Reboot(DeviceRebootType.Bootloader);
      return Promise.resolve(res.message);
    }

    const res = await this.device.commands.typedCall('RebootToBootloader', 'Success');
    return Promise.resolve(res.message);
  }
}