#!/usr/bin/env python3
"""
    Open Sound Control send/recieve daemon for Tascam Firewire control surface
    ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

    :copyright: Copyright (c) 2012-2014 Scott Bahling
        This program is free software; you can redistribute it and/or modify
        it under the terms of the GNU General Public License version 2 as
        published by the Free Software Foundation.

        This program is distributed in the hope that it will be useful,
        but WITHOUT ANY WARRANTY; without even the implied warranty of
        MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
        GNU General Public License for more details.

        You should have received a copy of the GNU General Public License
        along with this program (see the file COPYING); if not, write to the
        Free Software Foundation, Inc.,
        51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
    :license: GPL-2.0, see COPYING for details
"""

import re
import pyautogui
from osc import osc


keymap = {'CTRL': 'ctrl',
          'ALT': 'alt',
          'SHIFT': 'shift',
          }


osc_addrs = {'STOP': '/transport_stop',
             'PLAY': '/transport_play',
             'F.FWD': '/ffwd',
             'REW': '/rewind',
             'REC': '/rec_enable_toggle',
             }

re_strip_num = re.compile('.*([1-9])')

current_bank = 1
more_banks_up = False
more_banks_down = False


class Button():
    def __init__(self, console, name):
        self.name = name
        self.state = 0
        self.console = None

    def press(self):
        self.state = 1
        print('%s pressed' % self.name)

    def release(self):
        self.state = 0
        print('%s released' % self.name)

    @property
    def pressed(self):
        if self.state:
            return True
        return False


class EncoderButton(Button):
    def __init__(self, console, name):
        super().__init__(console, name)

    def press(self):

        encoder_mode = self.console.state.get('encoder_mode', None)
        print(encoder_mode)
        if encoder_mode == self.name:
            return

        self.state = 1
        if encoder_mode:
            self.console.unit.leds.turn_off(encoder_mode)
            self.console.state[encoder_mode] = 0
        self.console.unit.leds.turn_on(self.name)
        self.console.state['encoder_mode'] = self.name
        super().press()


class StripSelButton(Button):
    def __init__(self, console, name, strip):
        super().__init__(console, name)
        self.strip = strip
        self.strip.select = self

    def press(self):

        strip = self.strip
        print('handle_strip_sel', strip)
        if self.get('REC ENABLE', 0):
            if strip.recenable:
                osc.send_message('/strip/recenable', (strip.num, 0))
            else:
                osc.send_message('/strip/recenable', (strip.num, 1))
        super().press()


class StripMuteButton(Button):
    def __init__(self, console, name, strip):
        super().__init__(console, name)
        self.strip = strip
        self.strip.mute = self

    def press(self):

        strip = self.strip
        if strip.mute:
            strip.mute = False
            osc.send_message('/strip/mute', (strip.num, 0))
        else:
            strip.mute = True
            osc.send_message('/strip/mute', (strip.num, 1))
        super().press()


class StripSoloButton(Button):
    def __init__(self, console, name, strip):
        super().__init__(console, name)
        self.strip = strip
        self.strip.solo = self

    def press(self):

        strip = self.strip
        if strip.solo:
            strip.solo = False
            osc.send_message('/strip/solo', (strip.num, 0))
        else:
            strip.solo = True
            osc.send_message('/strip/solo', (strip.num, 1))
        super().press()


class ArrowButton(Button):
    def __init__(self, console, name):
        super().__init__(console, name)

    def press(self):
        key = keymap.get(self.name.lower(), self.name.lower())
        pyautogui.press(key)


class ModButton(Button):
    def __init__(self, console, name):
        super().__init__(console, name)

    def mod_button_press(self):
        key = keymap.get(self.name, None)
        if key is None:
            return

        pyautogui.keyDown(key)
        super().press()

    def release(self):
        key = keymap.get(self.name, None)
        if key is None:
            return

        pyautogui.keyUp(key)
        super().release()


class ComputerButton(Button):
    def __init__(self, console, name):
        super().__init__(console, name)

    def press(self):
        osc.send_message('/set_surface', (8, 7, 19, 1, 8, 11))
        super().press()


class ClrSoloButton(Button):
    def __init__(self, console, name):
        super().__init__(console, name)

    def press(self):
        osc.send_message('/cancel_all_solos', 1)
        super().press()


class LoopButton(Button):
    def __init__(self, console, name):
        super().__init__(console, name)

    def press(self, *args):
        if self.console.state.get('loop', 0):
            print('******* loop off')
            osc.send_message('/loop_toggle', 0)
        else:
            print('******* loop on')
            osc.send_message('/loop_toggle', 1)
        super().press()


class TransportButton(Button):
    def __init__(self, console, name):
        super().__init__(console, name)

    def press(self):
        addr = osc_addrs.get(self.name, None)
        if addr:
            osc.send_message(addr, 1)
        super().press()


class BankSwitchButton(Button):
    def __init__(self, console, name, direction):
        super().__init__(console, name)
        self.direction = direction

    def press(self):

        direction = self.direction
        print(direction, more_banks_up, more_banks_down)
        if direction > 0 and more_banks_up:
            print('calling /bank_up 1')
            osc.send_message('/bank_up', 1)
        elif direction < 0 and more_banks_down:
            print('calling /bank_down 1')
            osc.send_message('/bank_down', 1)
        super().press()


class NudgeButton(Button):
    def __init__(self, console, name, direction):
        super().__init__(console, name)
        self.direction = direction

    def nudge_press(self):
        direction = self.direction
        print(direction)
        if direction > 0:
            osc.send_message('/access_action', 'Common/nudge-next-forward')
        else:
            osc.send_message('/access_action', 'Common/nudge-next-backward')
        super().press()