import machine
import time
from utime import sleep

## Pin interrupt handler
#
# A function to handle pin interrupts
# Expects a machine.Pin type parameter
def interrupt_callback(pin):
    pass

def main():
    # a tiny sleep to allow the first print to be displayed
    sleep(0.01)
    print('Program starting')

    # code below here

s0 = machine.Pin(28, machine.Pin.OUT)
s1 = machine.Pin(27, machine.Pin.OUT)
mux_in = machine.Pin(26, machine.Pin.IN, machine.Pin.PULL_DOWN)

BUTTON_COUNT = 3
LED_COUNT = 9
INPUT_COUNT = 4

BUTTON_START_ID = 16
LED_GPIO_START = 7
last_button_time_stamp = 0
key_presses = []

# Extract the numeric pin id from the passed in Pin instance
def PinId(pin):
    # Pin(GPIO17, mode=IN, pull=PULL_DOWN)
    # the pin value is in the format above
    # and the pin id, i.e. 7 or 17, can be single or double digit
    # So, we filter the characters 8-10, 
    # which would be '7,' or '17' depending on the pin,
    # then, we remove any ',' character at the end
    # and we are left with the numeric value of the pin, i.e. 7 or 17
    return int(str(pin)[8:11].rstrip(","))

def interrupt_callback(pin):
    global last_button_time_stamp

    cur_button_ts = time.ticks_ms()
    button_press_delta = cur_button_ts - last_button_time_stamp
    if button_press_delta > 200:
        last_button_time_stamp = cur_button_ts
        key_presses.append(pin)
        # Call the PinId method to get the numeric pin value
        print(f'key press: {PinId(pin) - BUTTON_START_ID}')

def main():
    global key_presses
    global last_button_time_stamp
    PASSCODE_LENGTH = 0



    buttons = []
    for btn_idx in range(0, BUTTON_COUNT):
        buttons.append(machine.Pin(BUTTON_START_ID + btn_idx, machine.Pin.IN, machine.Pin.PULL_DOWN))
        buttons[-1].irq(trigger=machine.Pin.IRQ_FALLING, handler=interrupt_callback)

    PASS_CODE = [buttons[0], buttons[2], buttons[1]]
    PASSCODE_LENGTH = len(PASS_CODE)

    out_pins = []
    for out_id in range(0, LED_COUNT):
        out_pins.append(machine.Pin(LED_GPIO_START + out_id, machine.Pin.OUT))

    last_decimal_val = -1
    while True:
        decimal_val = 0
        bit_number = 0
        for selector0_val in range(2):
            s0.value(selector0_val)
            for selector1_val in range(2):
                s1.value(selector1_val)
                decimal_val += (pow(2, bit_number) * mux_in.value())
                bit_number += 1
                sleep(0.02)

        if last_decimal_val != decimal_val:
            last_decimal_val = decimal_val
            print(f'selected output: {last_decimal_val}')
        sleep(0.1)

        if len(key_presses) >= PASSCODE_LENGTH:
            if key_presses[:PASSCODE_LENGTH] == PASS_CODE:
                print('correct passcode')
                if decimal_val < LED_COUNT:
                    print(f'toggling: {decimal_val}')
                    out_pins[decimal_val].toggle()
                else:
                    print(f'invalid output: {decimal_val}, ' + \
                    f'valid range: 0-{len(out_pins) - 1}, doing nothing')
            else:
                print('wrong passcode')
            print('')
            key_presses = key_presses[PASSCODE_LENGTH:]    

if __name__ == "__main__":
    main()