from solo_tool import SoloTool
from midi_controller_launchpad_mini import MidiController

class SoloToolCLI:
    def __init__(self, sessionJson, soloToolOverride=None, midiOverride=None):
        self._soloTool = SoloTool() if soloToolOverride is None else soloToolOverride
        self._soloTool.loadSession(sessionJson)
        self._midiController = MidiController() if midiOverride is None else midiOverride
        self._commands = {
            "song" : self._song,
            "midi" : self._midi
        }

    def input(self, commandString):
        split = commandString.strip().split(" ")
        if split[0] in self._commands:
            self._commands[split[0]](split[1:])

    def _song(self, args):
        if len(args) > 0:
            self._soloTool.setSong(int(args[0]))
        else:
            songs = self._soloTool.getSongs()
            print("Songs:")
            for i, s in enumerate(songs):
                print(f"  {i} {s}")

    def _midi(self, args):
        if len(args) > 0 and args[0] == "connect":
            print("Connecting to MIDI device...")
            self._midiController.connect()
        else:
            print("Supported device: Novation Launchpad Mini MkII")