main.py

from machine import Pin
from time import sleep_ms,time
import json

from totp import totp

import password
import PicoOled13
import cryptor

from synchronised_time import create_synchronised_time, do_connect

lcd = PicoOled13.get()

# Unlock secrets
key=password.get(lcd)

codes = json.loads(cryptor.decrypt("codes.json.encoded",key))
selected_idx = 0
code = None

# Do all required init i.e. manual time setting vs ntp
synchronised_time = create_synchronised_time(lcd,key)

# Begin,End markers for timer bar
lcd.fill_rect(0,61,4,2,lcd.white)
lcd.fill_rect(124,61,4,2,lcd.white)
# Size of text area to be wiped first time: full screen minus timer bar height of 4
# (Set this correctly and let the text clearing code handle this instead
# of lcd.clear() or lcd.fill().)
loc=(lcd.width,lcd.height-4)

screenawake=1

while True:
    # Is a button pressed
    if lcd.is_pressed(lcd.KEY0):
        selected_idx = (selected_idx + 1) % len(codes)

    if lcd.is_pressed(lcd.KEY1):
        if screenawake:
            screenawake=0
        else:
            # Empty the timer bar in the framebuffer
            lcd.fill_rect(4,60,120,4,0x0000)
            screenawake=1

    if screenawake:
        # Did the selection change from last time
        if code != codes[selected_idx]:
            code = codes[selected_idx]
            (passwd, expiry) = totp(synchronised_time(),
                                      code['key'],
                                      step_secs=code['step'],
                                      digits=code['digits'])

        # is it time to recalculate things yet?
        s=time()%30
        if s == 0:
            (passwd, expiry) = totp(synchronised_time(),
                                      code['key'],
                                      step_secs=code['step'],
                                      digits=code['digits'])

        # Wipe previous text (first run wipes full screen except timer bar)
        lcd.fill_rect(0,0,128,loc[1],0x0000)
        loc=lcd.text(code['name'],0,0,0xffff)
        loc=lcd.text(passwd,0,loc[1],0xffff)

        # Time left bar
        if s == 0:
            lcd.fill_rect(4,60,120,4,0x0000)
        lcd.fill_rect(4,60,(s+1)<<2,4,0xffff)
        lcd.on()
        lcd.show()
    else:
        lcd.off()

    sleep_ms(250)