from midi_wrapper_mido import MidiWrapper

class MidiController:
    DEVICE_NAME = "Launchpad Mini MIDI 1"

    def __init__(self, soloTool, midiWrapperOverride=None):
        self._soloTool = soloTool
        if midiWrapperOverride is not None:
            self._midiWrapper = midiWrapperOverride
        else:
            self._midiWrapper = MidiWrapper()

        self._handlers = {
            96  : self._soloTool.stop,
            112 : self._playPause,
            101 : self._soloTool.jumpToA,
            102 : self._soloTool.previousStoredAbLimits,
            103 : self._soloTool.nextStoredAbLimits,
            118 : self._soloTool.previousSong,
            119 : self._soloTool.nextSong
        }
        
        for i in range(0, 8):
            volume = round(0.125 + i * 0.125, 3)
            self._handlers[i] = self._createSetPlaybackVolumeCallback(volume)

        for i in range(16, 24):
            rate = round(0.5 + (i - 16) * 0.1, 1)
            self._handlers[i] = self._createSetPlaybackRateCallback(rate)

    def connect(self):
        self._midiWrapper.connect(MidiController.DEVICE_NAME)
        self._midiWrapper.setCallback(self._callback)

    def _callback(self, msg):
        if msg.velocity < 127:
            return

        if msg.note in self._handlers:
            handler = self._handlers[msg.note]()

    def _playPause(self):
        if self._soloTool.isPlaying():
            self._soloTool.pause()
        else:
            self._soloTool.play()

    def _createSetPlaybackRateCallback(self, rate):
        def f():
            self._soloTool.setPlaybackRate(rate)
        return f

    def _createSetPlaybackVolumeCallback(self, volume):
        def f():
            self._soloTool.setPlaybackVolume(volume)
        return f