trezor-firmware/src/trezor/ui/pin.py

from micropython import const
from trezor import ui, res
from trezor.crypto import random
from trezor.ui import display
from trezor.ui.button import Button, BTN_CLICKED, CLEAR_BUTTON, CLEAR_BUTTON_ACTIVE


def digit_area(i):
    width = const(80)
    height = const(48)
    x = (i % 3) * width
    y = (i // 3) * height
    # 48px is offset of input line / -1px is due to corner bug of overlaying
    # elements
    return (x, y + 48, width - 1, height - 1)


def generate_digits():
    digits = list(range(1, 10))  # 1-9
    random.shuffle(digits)
    return digits


class PinMatrix():

    def __init__(self, label, pin=''):
        self.label = label
        self.pin = pin
        self.digits = generate_digits()

        # we lay out the buttons top-left to bottom-right, but the order of the
        # digits is defined as bottom-left to top-right (on numpad)
        reordered_digits = self.digits[6:] + self.digits[3:6] + self.digits[:3]
        self.pin_buttons = [Button(digit_area(i), str(d))
                            for i, d in enumerate(reordered_digits)]

        self.clear_button = Button((240 - 35, 5, 30, 30),
                                   res.load('trezor/res/pin_close.toig'),
                                   normal_style=CLEAR_BUTTON,
                                   active_style=CLEAR_BUTTON_ACTIVE)

    def render(self):

        header = '*' * len(self.pin) if self.pin else self.label

        # clear canvas under input line
        display.bar(0, 0, 205, 48, ui.BLACK)

        # input line with a header
        display.text_center(120, 30, header, ui.BOLD, ui.GREY, ui.BLACK)

        # render clear button
        if self.pin:
            self.clear_button.render()
        else:
            display.bar(240 - 48, 0, 48, 42, ui.BLACK)

        # pin matrix buttons
        for btn in self.pin_buttons:
            btn.render()

        # vertical border bars
        # display.bar(79, 48, 2, 143, ui.blend(ui.BLACK, ui.WHITE, 0.25))
        # display.bar(158, 48, 2, 143, ui.blend(ui.BLACK, ui.WHITE, 0.25))

        # horizontal border bars
        # display.bar(0, 95, 240, 2, ui.blend(ui.BLACK, ui.WHITE, 0.25))
        # display.bar(0, 142, 240, 2, ui.blend(ui.BLACK, ui.WHITE, 0.25))

    def send(self, event, pos):
        if self.clear_button.send(event, pos) == BTN_CLICKED:
            self.pin = ''
        for btn in self.pin_buttons:
            if btn.send(event, pos) == BTN_CLICKED:
                if len(self.pin) < 9:
                    self.pin += btn.content