#pragma once

#include "types.h"
#include "submitter.h"
#include "sender.h"

#include <rtmidi/RtMidi.h>

#include <string>

namespace midi_router
{

struct Device_Connection : public Sender
{
    Device_Connection(Device_Id const & source_id, std::string const & device_name, Submitter & submitter, RtMidiIn::RtMidiCallback callback):
        source_id(source_id),
        device_name(device_name),
        submitter(submitter) 
    {
        midi_in.setCallback(callback, this);
        midi_in.ignoreTypes(true, false, true);
    }

    Device_Id const & source_id;
    std::string const & device_name;
    Submitter & submitter;
    RtMidiIn midi_in {};
    RtMidiOut midi_out {};

    void
    send(std::vector<std::uint8_t> const & payload) override
    {
        midi_out.sendMessage(&payload);
    }

    std::string
    get_id() const override
    {
        return source_id;
    }

    void
    open()
    {
        if (!open_port(midi_in, device_name))
        {
            std::cerr << "Input port not found for device " << device_name << "\n";
        }

        if (!open_port(midi_out, device_name))
        {
            std::cerr << "Output port not found for device " << device_name << "\n";
        }
    }

    void
    close()
    {
        midi_in.closePort();
        midi_out.closePort();
    }

private:
    bool 
    open_port(RtMidi & midi, std::string const & name)
    {
        for (std::size_t i = 0; i < midi.getPortCount(); ++i)
        {
            if (midi.getPortName(i).contains(name))
            {
                midi.openPort(i);
                return true;
            }
        }
        return false;
    }
};

} // namespace midi_router