#!/usr/bin/env python3

import atuino as at
import tkinter as tk

class LedButton(object):
    '''A button that controls one LED connected to an Arduino
    while providing visual feedback of the current button state
    '''

    def __init__(self, window, arduino, pin_no):
        self.button= tk.Button(
            window,
            text='Led {}'.format(pin_no),
            command= self.on_pressed
        )

        # Get an instance of OutputPin
        # and turn it off
        self.led= at.OutputPin(arduino, pin_no)
        self.set_state(False)

        self.button.pack()

    def set_state(self, state):
        if state:
            self.led.turn_on()
            self.button.config(relief='sunken')
            self.state= True

        else:
            self.led.turn_off()
            self.button.config(relief='raised')
            self.state= False

    def on_pressed(self):
        self.set_state(not self.state)

class PortSelector(object):
    '''Display a dropdown menu and a button in a given window
    that lets you choose the serial port your Arduino
    is connected to.
    '''

    def __init__(self, window, on_connect_cb):
        '''Initialize the port selector menu.

        Upon pressing the connect button the selected
        port name is passed as first argument to the
        on_connect_cb callback function.
        '''

        self.window= window
        self.on_connect_cb= on_connect_cb

        self.setup_port_menu()
        self.setup_connect_button()

    def setup_port_menu(self):
        ports= at.enumerate_ports()

        self.port_current= tk.StringVar(self.window)
        self.port_current.set(ports[0])

        self.menu_widget= tk.OptionMenu(
            self.window,
            self.port_current,
            *ports
        )
        self.menu_widget.pack()

    def setup_connect_button(self):
        self.connect_button= tk.Button(
            self.window,
            text= 'connect',
            command= self.on_button_clicked
        )
        self.connect_button.pack()

    def on_button_clicked(self):
        port_name= self.port_current.get()

        self.on_connect_cb(port_name)

    def hide(self):
        'Remove the port selector widgets from the window'

        self.connect_button.pack_forget()
        self.menu_widget.pack_forget()

class LightCenterWindow(object):
    def __init__(self):
        '''Initialize the window and setup
        the initial content.

        To actually display the window run
        has to be called.
        '''

        self.arduino= None
        self.led_buttons= list()

        # Setup the empty window
        self.setup_window()

        # Setup the initial window content
        self.port_selector= PortSelector(
            self.window, self.on_connect
        )

    def setup_window(self):
        self.window= tk.Tk()
        self.window.title('Light Center')

    def run(self):
        'Execute the tkinter mainloop to display the Light center window'

        self.window.mainloop()

    def on_connect(self, port_name):
        # Connect to the Arduino ...
        self.arduino= at.Arduino(port_name)

        # ... hide the connection widgets ...
        self.port_selector.hide()

        # ... and create buttons for the LEDs
        for pin_no in (10, 11, 12, 13):
            self.led_buttons.append(
                LedButton(self.window, self.arduino, pin_no)
            )

if __name__ == '__main__':
    window= LightCenterWindow()
    window.run()