import os

from .device import Device


class TrezorOne(Device):
    def touch(self, location, action):
        self.now()
        print(
            "[hardware/trezor] Touching the {} button by {}...".format(location, action)
        )
        self.serial.write(("{} {}\n".format(location, action)).encode())

    def update_firmware(self, file=None):
        if file:
            unofficial = True
            trezorctlcmd = "trezorctl firmware-update -s -f {} &".format(file)
            print("[software/trezorctl] Updating the firmware to {}...".format(file))
        else:
            unofficial = False
            trezorctlcmd = "trezorctl firmware-update &"
            print("[software/trezorctl] Updating the firmware to latest...")
        self.wait(3)
        self._enter_bootloader()

        self.wait(3)
        os.system(trezorctlcmd)
        self.wait(3)
        self.touch("right", "click")
        self.wait(20)
        if unofficial:
            self.touch("right", "click")
        self.wait(10)
        self.power_off()
        self.power_on()
        if unofficial:
            self.touch("right", "click")
            self.wait(5)
            self.touch("right", "click")
        self.wait(5)
        os.system("trezorctl get-features|grep version")

    def _enter_bootloader(self):
        self.power_off()
        self.touch("all", "press")
        self.wait(2)
        self.power_on()
        self.wait(2)
        self.touch("all", "unpress")