#include <thread>
#include <sstream>

#include "DeviceMonitor.h"
#include "util.h"
#include "backend/hidpp10/Error.h"
#include "backend/dj/Receiver.h"

#define NON_WIRELESS_DEV(index) (index) == HIDPP::DefaultDevice ? "default" : "corded"

using namespace logid;
using namespace logid::backend;

/*
void stopAndDeleteConnectedDevice (ConnectedDevice &connected_device)
{
    if(!connected_device.device->waiting_for_receiver)
        log_printf(INFO, "%s (Device %d on %s) disconnected", connected_device.device->name.c_str(),
            connected_device.device->index, connected_device.device->path.c_str());
    connected_device.device->stop();
    connected_device.associatedThread.join();
    delete(connected_device.device);
}

DeviceMonitor::~DeviceMonitor()
{
    this->devices_mutex.lock();
        for (auto it = this->devices.begin(); it != this->devices.end(); it++) {
            for (auto jt = it->second.begin(); jt != it->second.end(); jt++) {
                stopAndDeleteConnectedDevice(jt->second);
            }
        }
    this->devices_mutex.unlock();
}

Device* DeviceMonitor::insertNewDevice(const std::string &path, HIDPP::DeviceIndex index)
{
    auto device = new Device(path, index);
    device->init();

    this->devices_mutex.lock();
        log_printf(INFO, "%s detected: device %d on %s", device->name.c_str(), index, path.c_str());
        auto path_bucket = this->devices.emplace(path, std::map<HIDPP::DeviceIndex, ConnectedDevice>()).first;
        path_bucket->second.emplace(index, ConnectedDevice{
            device,
            std::thread([device]() {
                device->start();
            })
        });
    this->devices_mutex.unlock();

    return device;
}

Device* DeviceMonitor::insertNewReceiverDevice(const std::string &path, HIDPP::DeviceIndex index)
{
    auto *device = new Device(path, index);
    this->devices_mutex.lock();
    auto path_bucket = this->devices.emplace(path, std::map<HIDPP::DeviceIndex, ConnectedDevice>()).first;
    path_bucket->second.emplace(index, ConnectedDevice{
            device,
            std::thread([device]() {
                device->waitForReceiver();
            })
    });
    this->devices_mutex.unlock();

    return device;
}

void DeviceMonitor::stopAndDeleteAllDevicesIn (const std::string &path)
{
    this->devices_mutex.lock();
        auto path_bucket = this->devices.find(path);
        if (path_bucket != this->devices.end())
        {
            for (auto& index_bucket : path_bucket->second) {
                stopAndDeleteConnectedDevice(index_bucket.second);
            }
            this->devices.erase(path_bucket);
        }
    this->devices_mutex.unlock();
}

void DeviceMonitor::stopAndDeleteDevice (const std::string &path, HIDPP::DeviceIndex index)
{
    this->devices_mutex.lock();
        auto path_bucket = this->devices.find(path);
        if (path_bucket != this->devices.end())
        {
            auto index_bucket = path_bucket->second.find(index);
            if (index_bucket != path_bucket->second.end())
            {
                stopAndDeleteConnectedDevice(index_bucket->second);
                path_bucket->second.erase(index_bucket);
            }
        }
    this->devices_mutex.unlock();

    log_printf(WARN, "Attempted to disconnect not previously connected device %d on %s", index, path.c_str());
}
*/

void DeviceMonitor::addDevice(std::string path)
{
    try {
        std::tuple<uint8_t, uint8_t> version;
        try
        {
            hidpp::Device device(path, hidpp::DefaultDevice);

            log_printf(DEBUG, "Detected HID++ device at %s", path.c_str());

            version = device.version();
            log_printf(DEBUG, "HID++ version: %d.%d", std::get<0>(version), std::get<1>(version));
        } catch(std::system_error &e) { }

        if(version == std::make_tuple(1, 0))
        {
            // This is a receiver
            dj::Receiver receiver(path);
            receiver.enumerate();
            receiver.listen();
            while(true) {}
        }
        /*
        auto eventHandler = std::make_shared<backend::hidpp::EventHandler>();
        eventHandler->condition = [device](backend::hidpp::Report& report)->bool
        {
            return true;
        };
        eventHandler->callback = [device](backend::hidpp::Report& report)->void
        {
            log_printf(DEBUG, "Event on %s:%d", device->devicePath().c_str(),
                    device->deviceIndex());
            for(auto& i : report.rawReport())
                    printf("%02x ", i);
            printf("\n");
        };

        device->addEventHandler("MONITOR_ALL", eventHandler);

        devices.push_back(device);*/

        //std::thread([device]() { device->listen(); }).detach();
    }
    catch(hidpp10::Error &e)
    {
        if(e.code() == hidpp10::Error::UnknownDevice) {}
        else
            throw;
    }
    catch(hidpp::Device::InvalidDevice &e)
    {
        log_printf(DEBUG, "Detected device at %s but %s", path.c_str(), e.what());
    }
    /*
    catch(std::system_error &e)
    {
        log_printf(WARN, "Failed to open %s: %s", path.c_str(), e.what());
    } */
}

void DeviceMonitor::removeDevice(std::string path)
{
    log_printf(DEBUG, "Device %s disconnected", path.c_str());
    /*
    ipc_server->removeReceiver(path);
    this->stopAndDeleteAllDevicesIn(std::string(path));
     */
}