#include "server.h"
#include "utils.h"

#include "strmanip.h"

ImmsServer::ImmsServer()
    : SocketServer(get_imms_root() + "socket")
{
    conn = 0;
}

string consume(list<string> &l)
{
    string res;
    while (!l.empty())
    {
        res += l.front() + " ";
        l.pop_front();
    }
    return res;
}

string getnum(list<string> &l)
{
    int size = l.size();
    if (size < 1 || l.size() > 2)
        return "";

    string sign, num;
    if (size == 2)
    {
        sign = l.front();
        l.pop_front();
        num = l.front();
        l.pop_front();
    }
    else
    {
        sign = l.front().substr(0, 1);
        num = l.front().substr(1);
        l.pop_front();
    }

    if (sign != ">" && sign != "<" && sign != "=")
        return "";

    if (!atoi(num.c_str()))
        return "";

    return sign + " '" + num + "'";
}

void ImmsServer::do_events()
{
    int fd = accept();
    if (fd > 0)
    {
        delete conn;
        conn = new Socket(fd);
        *conn << "Welcome to " << PACKAGE_STRING << "\n";
        *conn << "This feature is highly experimental, "
            "so quit complaining" << "\n";
        *conn << "Say 'help' if you are lost and hungry" << "\n";
    }

    if (conn && !conn->isok())
    {
        delete conn;
        conn = 0;
    }

    if (!conn)
        return;

    string command = conn->read();
    command = string_tolower(string_delete(command, "\n"));
    if (command == "")
        return;

    string predicate = "";
    list<string> parsed;
    string_split(parsed, command, " ");

    while (1)
    {
        string str = parsed.front();
        parsed.pop_front();

        if (str == "or" || str == "and")
        {
            predicate = str;
            continue;
        }

        if (str == "help")
        {
            *conn << "known commands:" << "\n";
            *conn << "  show" << "\n";
            *conn << "  clear" << "\n";
            *conn << "  [or|and] artist <str>" << "\n";
            *conn << "  [or|and] rating =|>|< <int>" << "\n";
            *conn << "  sql <str>" << "\n";

            break;
        }
        if (str == "show" || str == "ls" || str == "filter")
        {
            *conn << "filter: " << filter << "\n";
            break;
        }
        if (str == "clear" || str == "reset")
        {
            filter = "";
            ImmsDb::install_filter(filter);
            reset_selection();
            break;
        }
        if (str == "artist" || str == "sql" || str == "rating")
        {
            string params;
            if (str == "artist" || str == "sql")
            {
                if (parsed.front() == "=")
                    parsed.pop_front();
                params = consume(parsed);
            }
            else
                params = getnum(parsed);

            if (params == "")
            {
                *conn << str << ": parse error" << "\n";
                return;
            }

            if (predicate == "")
                filter = "";
            if (filter != "" && predicate == "and")
                filter = "(" + filter + ") AND ";
            else if (filter != "" && predicate == "or")
                filter += "OR ";

            if (str == "sql")
                filter = params;
            else if (str == "artist")
                filter += "similar(Info.artist, '"
                    + string_normalize(params) + "') ";
            else if (str == "rating")
                filter += "Rating.rating " + params + " ";
            reset_selection();

            int n = ImmsDb::install_filter(filter);
            *conn << itos(n) << " hits" << "\n";
            break;
        }

        *conn << "parse error at " << str << "\n";
        return;
    }

    if (parsed.size())
        *conn << "warning: ignored after " << parsed.front() << "\n";
}

ImmsServer::~ImmsServer()
{
    delete conn;
    close();
    unlink(string(getenv("HOME")).append("/.imms/socket").c_str());
}