#include <iostream>
#include <polyp/polyplib-error.h>

#include "ServerInfoManager.hh"
#include "paman.hh"

SinkInfo::SinkInfo(const struct pa_sink_info &i) :
    name(i.name),
    description(i.description),
    index(i.index),
    sample_spec(i.sample_spec),
    monitor_source(i.monitor_source),
    owner_module(i.owner_module),
    volume(i.volume),
    latency(i.latency),
    window(NULL) {
}

SinkInfo::~SinkInfo() {
    if (window)
        delete window;
}
    
void SinkInfo::update(const struct pa_sink_info &i) {
    name = Glib::ustring(i.name);
    description = i.description;
    index = i.index;
    sample_spec = i.sample_spec;
    monitor_source = i.monitor_source;
    owner_module = i.owner_module;
    volume = i.volume;
    latency = i.latency;

    if (window)
        window->updateInfo(*this);
    g_assert(mainWindow);
    mainWindow->updateInfo(*this);
}

void SinkInfo::showWindow() {
    if (window)
        window->present();
    else {
        window = SinkWindow::create();
        window->updateInfo(*this);
        window->show();
    }
}

SourceInfo::SourceInfo(const struct pa_source_info &i) :
    name(i.name),
    description(i.description),
    index(i.index),
    sample_spec(i.sample_spec),
    owner_module(i.owner_module),
    monitor_of_sink(i.monitor_of_sink),
    window(NULL) {
}

SourceInfo::~SourceInfo() {
    if (window)
        delete window;
}

void SourceInfo::update(const struct pa_source_info &i) {
    name = i.name;
    description = i.description;
    index = i.index;
    sample_spec = i.sample_spec;
    owner_module = i.owner_module;
    monitor_of_sink = i.monitor_of_sink;

    if (window)
        window->updateInfo(*this);
    g_assert(mainWindow);
    mainWindow->updateInfo(*this);
}

void SourceInfo::showWindow() {
    if (window)
        window->present();
    else {
        window = SourceWindow::create();
        window->updateInfo(*this);
        window->show();
    }
}

ClientInfo::ClientInfo(const struct pa_client_info &i) :
    index(i.index),
    name(i.name),
    protocol_name(i.protocol_name),
    owner_module(i.owner_module),
    window(NULL) {
}

ClientInfo::~ClientInfo() {
    if (window)
        delete window;
}

void ClientInfo::update(const struct pa_client_info &i) {
    name = i.name;
    protocol_name = i.protocol_name;
    index = i.index;
    owner_module = i.owner_module;

    if (window)
        window->updateInfo(*this);

    g_assert(mainWindow);
    mainWindow->updateInfo(*this);
}

void ClientInfo::showWindow() {
    if (window)
        window->present();
    else {
        window = ClientWindow::create();
        window->updateInfo(*this);
        window->show();
    }
}

ModuleInfo::ModuleInfo(const struct pa_module_info &i) :
    index(i.index),
    name(i.name),
    argument(i.argument),
    autoloaded(i.auto_unload),
    used(i.n_used),
    window(NULL) {
}

ModuleInfo::~ModuleInfo() {
    if (window)
        delete window;
}

void ModuleInfo::update(const struct pa_module_info &i) {
    name = i.name;
    argument = i.argument;
    index = i.index;
    autoloaded = i.auto_unload;
    used = i.n_used;

    if (window)
        window->updateInfo(*this);
    g_assert(mainWindow);
    mainWindow->updateInfo(*this);
}

void ModuleInfo::showWindow() {
    if (window)
        window->present();
    else {
        window = ModuleWindow::create();
        window->updateInfo(*this);
        window->show();
    }
}

SinkInputInfo::SinkInputInfo(const struct pa_sink_input_info &i) :
    name(i.name),
    index(i.index),
    sample_spec(i.sample_spec),
    sink(i.sink),
    client(i.client),
    owner_module(i.owner_module),
    volume(i.volume),
    buffer_usec(i.buffer_usec),
    sink_usec(i.sink_usec),
    window(NULL) {
}

SinkInputInfo::~SinkInputInfo() {
    if (window)
        delete window;
}

void SinkInputInfo::update(const struct pa_sink_input_info &i) {
    name = i.name;
    index = i.index;
    sample_spec = i.sample_spec;
    sink = i.sink;
    client = i.client;
    owner_module = i.owner_module;
    volume = i.volume;
    buffer_usec = i.buffer_usec;
    sink_usec = i.sink_usec;

    if (window)
        window->updateInfo(*this);
    g_assert(mainWindow);
    mainWindow->updateInfo(*this);
}

void SinkInputInfo::showWindow() {
    if (window)
        window->present();
    else {
        window = SinkInputWindow::create();
        window->updateInfo(*this);
        window->show();
    }
}

SourceOutputInfo::SourceOutputInfo(const struct pa_source_output_info &i) :
    name(i.name),
    index(i.index),
    sample_spec(i.sample_spec),
    source(i.source),
    client(i.client),
    owner_module(i.owner_module),
    window(NULL) {
}

SourceOutputInfo::~SourceOutputInfo() {
    if (window)
        delete window;
}

void SourceOutputInfo::update(const struct pa_source_output_info &i) {
    name = i.name;
    index = i.index;
    sample_spec = i.sample_spec;
    source = i.source;
    client = i.client;
    owner_module = i.owner_module;

    if (window)
        window->updateInfo(*this);
    g_assert(mainWindow);
    mainWindow->updateInfo(*this);
}

void SourceOutputInfo::showWindow() {
    if (window)
        window->present();
    else {
        window = SourceOutputWindow::create();
        window->updateInfo(*this);
        window->show();
    }
}

SampleInfo::SampleInfo(const struct pa_sample_info &i) :
    index(i.index),
    name(i.name),
    sample_spec(i.sample_spec),
    volume(i.volume),
    duration(i.duration),
    bytes(i.bytes),
    window(NULL) {
}

SampleInfo::~SampleInfo() {
    if (window)
        delete window;
}

void SampleInfo::update(const struct pa_sample_info &i) {
    index = i.index;
    name = i.name;
    sample_spec = i.sample_spec;
    volume = i.volume;
    duration = i.duration;
    bytes = i.bytes;

    if (window)
        window->updateInfo(*this);
    g_assert(mainWindow);
    mainWindow->updateInfo(*this);
}

void SampleInfo::showWindow() {
    if (window)
        window->present();
    else {
        window = SampleWindow::create();
        window->updateInfo(*this);
        window->show();
    }
}

extern "C" {
    static void server_info_callback(struct pa_context *c, const struct pa_server_info *i, void *userdata);
    static void sink_info_callback(struct pa_context *c, const struct pa_sink_info *i, int is_last, void *userdata);
    static void source_info_callback(struct pa_context *c, const struct pa_source_info *i, int is_last, void *userdata);
    static void client_info_callback(struct pa_context *c, const struct pa_client_info *i, int is_last, void *userdata);
    static void module_info_callback(struct pa_context *c, const struct pa_module_info *i, int is_last, void *userdata);
    static void sink_input_info_callback(struct pa_context *c, const struct pa_sink_input_info *i, int is_last, void *userdata);
    static void source_output_info_callback(struct pa_context *c, const struct pa_source_output_info *i, int is_last, void *userdata);
    static void sample_info_callback(struct pa_context *c, const struct pa_sample_info *i, int is_last, void *userdata);
    static void subscribe_callback(struct pa_context *c, enum pa_subscription_event_type t, uint32_t index, void *userdata);
};

static void server_info_callback(struct pa_context *c, const struct pa_server_info *i, void *userdata) {
    ServerInfoManager *si = (ServerInfoManager*) userdata;

    if (!i) {
        mainWindow->showFailure(pa_strerror(pa_context_errno(c)));
        return;
    }

    mainWindow->updateInfo(*i);
}

static void sink_info_callback(struct pa_context *c, const struct pa_sink_info *i, int is_last, void *userdata) {
    ServerInfoManager *si = (ServerInfoManager*) userdata;
    if (i) si->updateInfo(*i);
}

static void source_info_callback(struct pa_context *c, const struct pa_source_info *i, int is_last, void *userdata) {
    ServerInfoManager *si = (ServerInfoManager*) userdata;
    if (i) si->updateInfo(*i);
}

static void client_info_callback(struct pa_context *c, const struct pa_client_info *i, int is_last, void *userdata) {
    ServerInfoManager *si = (ServerInfoManager*) userdata;
    if (i) si->updateInfo(*i);
}

static void module_info_callback(struct pa_context *c, const struct pa_module_info *i, int is_last, void *userdata) {
    ServerInfoManager *si = (ServerInfoManager*) userdata;
    if (i) si->updateInfo(*i);
}

static void sink_input_info_callback(struct pa_context *c, const struct pa_sink_input_info *i, int is_last, void *userdata) {
    ServerInfoManager *si = (ServerInfoManager*) userdata;
    if (i) si->updateInfo(*i);
}

static void source_output_info_callback(struct pa_context *c, const struct pa_source_output_info *i, int is_last, void *userdata) {
    ServerInfoManager *si = (ServerInfoManager*) userdata;
    if (i) si->updateInfo(*i);
}

static void sample_info_callback(struct pa_context *c, const struct pa_sample_info *i, int is_last, void *userdata) {
    ServerInfoManager *si = (ServerInfoManager*) userdata;
    if (i) si->updateInfo(*i);
}

static void subscribe_callback(struct pa_context *c, enum pa_subscription_event_type t, uint32_t index, void *userdata) {
    ServerInfoManager *si = (ServerInfoManager*) userdata;

//    fprintf(stderr, "EV: %u %u\n", t, index);
    
    switch (t & PA_SUBSCRIPTION_EVENT_FACILITY_MASK) {
        case PA_SUBSCRIPTION_EVENT_SINK:
            if ((t & PA_SUBSCRIPTION_EVENT_TYPE_MASK) == PA_SUBSCRIPTION_EVENT_REMOVE)
                si->removeSinkInfo(index);
            else
                pa_operation_unref(pa_context_get_sink_info_by_index(c, index, sink_info_callback, si));
            break;
        case PA_SUBSCRIPTION_EVENT_SOURCE:
            if ((t & PA_SUBSCRIPTION_EVENT_TYPE_MASK) == PA_SUBSCRIPTION_EVENT_REMOVE)
                si->removeSourceInfo(index);
            else
                pa_operation_unref(pa_context_get_source_info_by_index(c, index, source_info_callback, si));
            break;
        case PA_SUBSCRIPTION_EVENT_MODULE:
            if ((t & PA_SUBSCRIPTION_EVENT_TYPE_MASK) == PA_SUBSCRIPTION_EVENT_REMOVE)
                si->removeModuleInfo(index);
            else
                pa_operation_unref(pa_context_get_module_info(c, index, module_info_callback, si));
            break;
        case PA_SUBSCRIPTION_EVENT_CLIENT:
            if ((t & PA_SUBSCRIPTION_EVENT_TYPE_MASK) == PA_SUBSCRIPTION_EVENT_REMOVE)
                si->removeClientInfo(index);
            else
                pa_operation_unref(pa_context_get_client_info(c, index, client_info_callback, si));
            break;
        case PA_SUBSCRIPTION_EVENT_SINK_INPUT:
            if ((t & PA_SUBSCRIPTION_EVENT_TYPE_MASK) == PA_SUBSCRIPTION_EVENT_REMOVE)
                si->removeSinkInputInfo(index);
            else
                pa_operation_unref(pa_context_get_sink_input_info(c, index, sink_input_info_callback, si));
            break;
        case PA_SUBSCRIPTION_EVENT_SOURCE_OUTPUT:
            if ((t & PA_SUBSCRIPTION_EVENT_TYPE_MASK) == PA_SUBSCRIPTION_EVENT_REMOVE)
                si->removeSourceOutputInfo(index);
            else
                pa_operation_unref(pa_context_get_source_output_info(c, index, source_output_info_callback, si));
            break;
        case PA_SUBSCRIPTION_EVENT_SAMPLE_CACHE:
            if ((t & PA_SUBSCRIPTION_EVENT_TYPE_MASK) == PA_SUBSCRIPTION_EVENT_REMOVE)
                si->removeSampleInfo(index);
            else
                pa_operation_unref(pa_context_get_sample_info_by_index(c, index, sample_info_callback, si));
            break;
        case PA_SUBSCRIPTION_EVENT_SERVER:
            pa_operation_unref(pa_context_get_server_info(c, server_info_callback, si));
            break;
            
        default:
            fprintf(stderr, "OTHER EVENT\n");
            break;
    }            
}

ServerInfoManager::ServerInfoManager(struct pa_context &c) :
    context(c),
    statWindow(NULL) {
    pa_operation_unref(pa_context_get_server_info(&c, server_info_callback, this));
    pa_operation_unref(pa_context_get_sink_info_list(&c, sink_info_callback, this));
    pa_operation_unref(pa_context_get_source_info_list(&c, source_info_callback, this));
    pa_operation_unref(pa_context_get_module_info_list(&c, module_info_callback, this));
    pa_operation_unref(pa_context_get_client_info_list(&c, client_info_callback, this));
    pa_operation_unref(pa_context_get_sink_input_info_list(&c, sink_input_info_callback, this));
    pa_operation_unref(pa_context_get_source_output_info_list(&c, source_output_info_callback, this));
    pa_operation_unref(pa_context_get_sample_info_list(&c, sample_info_callback, this));
    
    pa_context_set_subscribe_callback(&c, subscribe_callback, this);

    pa_operation_unref(pa_context_subscribe(&c, (enum pa_subscription_mask)
                                            (PA_SUBSCRIPTION_MASK_SINK|
                                             PA_SUBSCRIPTION_MASK_SOURCE|
                                             PA_SUBSCRIPTION_MASK_MODULE|
                                             PA_SUBSCRIPTION_MASK_SINK_INPUT|
                                             PA_SUBSCRIPTION_MASK_SOURCE_OUTPUT|
                                             PA_SUBSCRIPTION_MASK_CLIENT|
                                             PA_SUBSCRIPTION_MASK_SAMPLE_CACHE|
                                             PA_SUBSCRIPTION_MASK_SERVER), NULL, NULL));
}


ServerInfoManager::~ServerInfoManager() {
    for (std::map<int, SinkInfo*>::iterator i = sinks.begin(); i != sinks.end(); i++)
        delete i->second;

    for (std::map<int, SourceInfo*>::iterator i = sources.begin(); i != sources.end(); i++)
        delete i->second;

    for (std::map<int, ClientInfo*>::iterator i = clients.begin(); i != clients.end(); i++)
        delete i->second;

    for (std::map<int, ModuleInfo*>::iterator i = modules.begin(); i != modules.end(); i++)
        delete i->second;

    for (std::map<int, SinkInputInfo*>::iterator i = sinkInputs.begin(); i != sinkInputs.end(); i++)
        delete i->second;

    for (std::map<int, SourceOutputInfo*>::iterator i = sourceOutputs.begin(); i != sourceOutputs.end(); i++)
        delete i->second;

    for (std::map<int, SampleInfo*>::iterator i = samples.begin(); i != samples.end(); i++)
        delete i->second;
    
    if (statWindow)
        delete statWindow;
}

void ServerInfoManager::updateInfo(const struct pa_sink_info &i) {
    SinkInfo *si;
    if ((si = sinks[i.index]))
        si->update(i);
    else {
        SinkInfo *n = new SinkInfo(i);
        sinks[i.index] = n;
        mainWindow->updateInfo(*n);
    }
}

void ServerInfoManager::updateInfo(const struct pa_source_info &i) {
    SourceInfo *si;
    
    if ((si = sources[i.index]))
        si->update(i);
    else {
        SourceInfo *n = new SourceInfo(i);
        sources[i.index] = n;
        mainWindow->updateInfo(*n);
    }
}

void ServerInfoManager::updateInfo(const struct pa_client_info &i) {
    ClientInfo *ci;

    if ((ci = clients[i.index]))
        ci->update(i);
    else {
        ClientInfo *n = new ClientInfo(i);
        clients[i.index] = n;
        mainWindow->updateInfo(*n);
    }
}
void ServerInfoManager::updateInfo(const struct pa_module_info &i) {
    ModuleInfo *si;
    if ((si = modules[i.index]))
        si->update(i);
    else {
        ModuleInfo *n = new ModuleInfo(i);
        modules[i.index] = n;
        mainWindow->updateInfo(*n);
    }
}

void ServerInfoManager::updateInfo(const struct pa_sink_input_info &i) {
    SinkInputInfo *si;
    if ((si = sinkInputs[i.index]))
        si->update(i);
    else {
        SinkInputInfo *n = new SinkInputInfo(i);
        sinkInputs[i.index] = n;
        mainWindow->updateInfo(*n);
    }
}

void ServerInfoManager::updateInfo(const struct pa_source_output_info &i) {
    SourceOutputInfo *si;
    if ((si = sourceOutputs[i.index]))
        si->update(i);
    else {
        SourceOutputInfo *n = new SourceOutputInfo(i);
        sourceOutputs[i.index] = n;
        mainWindow->updateInfo(*n);
    }
}

void ServerInfoManager::updateInfo(const struct pa_sample_info &i) {
    SampleInfo *si;
    if ((si = samples[i.index]))
        si->update(i);
    else {
        SampleInfo *n = new SampleInfo(i);
        samples[i.index] = n;
        mainWindow->updateInfo(*n);
    }
}

void ServerInfoManager::showSinkWindow(uint32_t index) {
    SinkInfo *i;

    if ((i = sinks[index]))
        i->showWindow();
}

void ServerInfoManager::showSourceWindow(uint32_t index) {
    SourceInfo *i;

    if ((i = sources[index]))
        i->showWindow();
}

void ServerInfoManager::showClientWindow(uint32_t index) {
    ClientInfo *i;

    if ((i = clients[index]))
        i->showWindow();
}

void ServerInfoManager::showModuleWindow(uint32_t index) {
    ModuleInfo *i;

    if ((i = modules[index]))
        i->showWindow();
}

void ServerInfoManager::showSinkInputWindow(uint32_t index) {
    SinkInputInfo *i;

    if ((i = sinkInputs[index]))
        i->showWindow();
}

void ServerInfoManager::showSourceOutputWindow(uint32_t index) {
    SourceOutputInfo *i;

    if ((i = sourceOutputs[index]))
        i->showWindow();
}

void ServerInfoManager::showSampleWindow(uint32_t index) {
    SampleInfo *i;

    if ((i = samples[index]))
        i->showWindow();
}


SourceInfo* ServerInfoManager::getSourceInfo(uint32_t index) {
    return sources[index];
}

SinkInfo* ServerInfoManager::getSinkInfo(uint32_t index) {
    return sinks[index];
}

ClientInfo* ServerInfoManager::getClientInfo(uint32_t index) {
    return clients[index];
}

ModuleInfo* ServerInfoManager::getModuleInfo(uint32_t index) {
    return modules[index];
}

SampleInfo* ServerInfoManager::getSampleInfo(uint32_t index) {
    return samples[index];
}


void ServerInfoManager::removeSinkInfo(uint32_t index) {
    SinkInfo *i;

    if ((i = sinks[index])) {
        sinks.erase(index);
        mainWindow->removeInfo(*i);
        delete i;
    }
}

void ServerInfoManager::removeSourceInfo(uint32_t index) {
    SourceInfo *i;
    if ((i = sources[index])) {
        sources.erase(index);
        mainWindow->removeInfo(*i);
        delete i;
    }
}

void ServerInfoManager::removeClientInfo(uint32_t index) {
    ClientInfo *i;

    if ((i = clients[index])) {
        clients.erase(index);
        mainWindow->removeInfo(*i);
        delete i;
    }
}

void ServerInfoManager::removeModuleInfo(uint32_t index) {
    ModuleInfo *i;
    if ((i = modules[index])) {
        modules.erase(index);
        mainWindow->removeInfo(*i);
        delete i;
    }
}

void ServerInfoManager::removeSinkInputInfo(uint32_t index) {
    SinkInputInfo *i;
    if ((i = sinkInputs[index])) {
        sinkInputs.erase(index);
        mainWindow->removeInfo(*i);
        delete i;
    }
}

void ServerInfoManager::removeSourceOutputInfo(uint32_t index) {
    SourceOutputInfo *i;
    if ((i = sourceOutputs[index])) {
        sourceOutputs.erase(index);
        mainWindow->removeInfo(*i);
        delete i;
    }
}

void ServerInfoManager::removeSampleInfo(uint32_t index) {
    SampleInfo *i;
    if ((i = samples[index])) {
        samples.erase(index);
        mainWindow->removeInfo(*i);
        delete i;
    }
}

void ServerInfoManager::setSinkVolume(uint32_t index, uint32_t volume) {
    pa_operation_unref(pa_context_set_sink_volume_by_index(&context, index, volume, NULL, NULL));
}

void ServerInfoManager::setSinkInputVolume(uint32_t index, uint32_t volume) {
    pa_operation_unref(pa_context_set_sink_input_volume(&context, index, volume, NULL, NULL));
}

void ServerInfoManager::showStatWindow() {
    if (statWindow)
        statWindow->present();
    else {
        statWindow = StatWindow::create();
        statWindow->show();
    }
        
}

void ServerInfoManager::playSample(uint32_t sample, uint32_t sink) {
    SinkInfo *sinki = getSinkInfo(sink);
    SampleInfo *samplei = getSampleInfo(sample);

    if (!sinki || !samplei)
        return;

    pa_operation_unref(pa_context_play_sample(&context, samplei->name.c_str(), sinki->name.c_str(), PA_VOLUME_NORM, NULL, NULL));
}