import string
import sys


from pythonosc import dispatcher
from pythonosc import dispatcher
from pythonosc import osc_server
import threading
import argparse


class OscInput(object):
    def __init__(self, root, message_handler, display, actions, data):
        self.root = root
        self.message_handler = message_handler
        self.display = display
        self.actions = actions
        self.data = data
        self.osc_mappings = data.osc_mappings
        self.osc_server = self.setup_osc_server()

    def setup_osc_server(self):
        server_parser = argparse.ArgumentParser()
        server_parser.add_argument("--ip", default="127.0.0.1", help="the ip")
        server_parser.add_argument("--port", type=int, default=5433, help="the port")

        server_args = server_parser.parse_args()

        this_dispatcher = dispatcher.Dispatcher()
        this_dispatcher.map("/keyboard/*", self.on_osc_input)
        this_dispatcher.map("/shaderparam0", self.on_param_osc_input)
        this_dispatcher.map("/shaderparam1", self.on_param_osc_input)
        this_dispatcher.map("/shaderparam2", self.on_param_osc_input)
        this_dispatcher.map("/shaderparam3", self.on_param_osc_input)
        this_dispatcher.map("/shutdown", self.exit_osc_server)
        
        server = osc_server.ThreadingOSCUDPServer((server_args.ip, server_args.port), this_dispatcher)
        server_thread = threading.Thread(target=server.serve_forever)
        server_thread.start()
        return server

    def exit_osc_server(self, unused_addr, args):
        self.osc_server.shutdown()

    def on_osc_input(self, addr, args):
        args = str(args)
        print("!!!!!!!!!!!!!!!!" + args)
        print("the address", addr)
        key = addr[-1]
        numpad = list(string.ascii_lowercase[0:19])

        if key in numpad:
            self.run_action_for_osc_channel(key)
        else:
            print('{} is not in keypad map'.format(args))

    def on_param_osc_input(self, addr, args):
        print("the address", addr)

        self.run_action_for_osc_channel(addr, param_value=args)
        


    def run_action_for_osc_channel(self, channel, param_value=None):
        this_mapping = self.osc_mappings[channel]
        if self.data.control_mode in this_mapping:
            mode = self.data.control_mode
        elif 'DEFAULT' in this_mapping:
            mode = 'DEFAULT'

        if self.data.function_on and len(this_mapping[mode]) > 1:
            print('the action being called is {}'.format(this_mapping[mode][1]))
            getattr(self.actions, this_mapping[mode][1])()
            if self.data.settings['sampler']['FUNC_GATED']['value'] == 'off':
                self.data.function_on = False
        else:
            print('the action being called is {}'.format(this_mapping[mode][0]))

            if param_value is not None:
                getattr(self.actions, this_mapping[mode][0])(param_value)
            else:
                getattr(self.actions, this_mapping[mode][0])()
                self.display.refresh_display()