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

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

import type { DeviceRebootParams } from './helpers';

export default class DeviceReboot extends BaseMethod<DeviceRebootParams> {
  getSupportedProtocols() {
    return ['V2'] as const;
  }

  init() {
    this.skipForceUpdateCheck = true;
    this.useDevicePassphraseState = false;
    this.unlockPolicy = 'unlock-before-run';
    // Device-management action: main PIN or Attach PIN may authorize reboot.
    this.protocolV2PreUnlockPinType = DeviceSessionPinType.Any;
    this.params = {
      rebootType: this.payload.rebootType,
      reboot_type: this.payload.reboot_type,
    };
  }

  async run() {
    const rebootType = normalizeRebootType(this.params.reboot_type ?? this.params.rebootType);
    const res = await this.device.commands.typedCall('DeviceReboot', 'Success', {
      reboot_type: rebootType,
    });
    this.device.markProtocolV2Reboot(rebootType);
    return Promise.resolve(res.message);
  }
}