#include "XMPPThread/internal.h"

#include <Cytoplasm/Memory.h>
#include <Cytoplasm/Str.h>
#include <Cytoplasm/Log.h>

#include <Matrix.h>
#include <AS.h>

#include <string.h>

bool
MessageStanza(ParseeData *args, XMLElement *stanza, XMPPThread *thr)
{
    XMPPComponent *jabber = args->jabber;

    XMLElement *reactions = NULL;
    XMLElement *body = NULL;
    XMLElement *data = NULL;

    char *to, *room, *from, *from_matrix, *decode_from;
    char *mroom_id = NULL;
    char *replaced  = XMPPGetReplacedID(stanza);
    char *retracted = XMPPGetRetractedID(stanza);
    char *reply_to  = XMPPGetReply(stanza);
    char *moderated = XMPPGetModeration(stanza);
    bool chat       = StrEquals(HashMapGet(stanza->attrs, "type"), "chat");
    size_t i;

    to = NULL;
    from = NULL;
    decode_from = NULL;
    from_matrix = NULL;

    from = HashMapGet(stanza->attrs, "from");
    if (ParseeManageBan(args, from, NULL))
    {
        XMLFreeElement(stanza);
        return false;
    }

    if (moderated)
    {
        /* TODO: Parsee MUST check if it is a valid MUC */
        char *resource = ParseeGetResource(from);
        char *chat_id = ParseeGetFromMUCID(args, from);
        char *event_id  = NULL;
        char *encoded = ParseeMXID(args);
        mroom_id = ParseeGetBridgedRoom(args, stanza);
        if (!resource && chat_id)
        {
            event_id = ParseeGetEventFromID(args, stanza, moderated);
            ASRedact(args->config, mroom_id, NULL, event_id);
            ParseePushAllStanza(args, stanza, event_id);
            pthread_mutex_unlock(&thr->info->chk_lock);

            Free(event_id);
        }
        Free(mroom_id);
        Free(resource);
        Free(encoded);
        Free(chat_id);
    }
    body = XMLookForUnique(stanza, "body");

    PEPManagerHandle(thr->info->pep_manager, stanza);

#define CHAT_STATES "http://jabber.org/protocol/chatstates"
    if (XMLookForTKV(stanza, "composing", "xmlns", CHAT_STATES))
    {
        decode_from = ParseeLookupJID(from);
        from_matrix = ParseeEncodeJID(args->config, decode_from, true);
        mroom_id = ParseeGetBridgedRoom(args, stanza);
        
        ASType(args->config, from_matrix, mroom_id, true);

        Free(decode_from);
        Free(from_matrix);
        Free(mroom_id);
        mroom_id = NULL;
        decode_from = NULL;
        from_matrix = NULL;
    }
    if (XMLookForTKV(stanza, "active", "xmlns", CHAT_STATES))
    {
        char *latest = NULL;
        decode_from = ParseeLookupJID(from);
        from_matrix = ParseeEncodeJID(args->config, decode_from, true);
        mroom_id = ParseeGetBridgedRoom(args, stanza);

        latest = ParseeLookupHead(mroom_id);

        ASType(args->config, from_matrix, mroom_id, false);
        ASPresence(args->config, from_matrix, mroom_id, latest);

        Free(decode_from);
        Free(from_matrix);
        Free(latest);
        Free(mroom_id);
        mroom_id = NULL;
        decode_from = NULL;
        from_matrix = NULL;
    }
    if (XMLookForTKV(stanza, "paused", "xmlns", CHAT_STATES) ||
        XMLookForTKV(stanza, "received", "xmlns", "urn:xmpp:receipts") ||
        XMLookForTKV(stanza, "displayed", "xmlns", "urn:xmpp:chat-markers:0"))
    {
        /* TODO: Use stanza ID if possible */
        char *latest = NULL;
        decode_from = ParseeLookupJID(from);
        from_matrix = ParseeEncodeJID(args->config, decode_from, true);
        mroom_id = ParseeGetBridgedRoom(args, stanza);

        latest = ParseeLookupHead(mroom_id);

        ASPresence(args->config, from_matrix, mroom_id, latest);

        Free(decode_from);
        Free(from_matrix);
        Free(latest);
        Free(mroom_id);
        mroom_id = NULL;
        decode_from = NULL;
        from_matrix = NULL;

    }
#undef CHAT_STATES
    if (!body)
    {
        XMLFreeElement(stanza);
        return false;
    }

    to = ParseeDecodeMXID(HashMapGet(stanza->attrs, "to"));
    decode_from = ParseeLookupJID(from);
    from_matrix = ParseeEncodeJID(args->config, decode_from, true);
    room = ParseeFindDMRoom(args, to, from);
    data = ArrayGet(body->children, 0);

    /* TODO: CLEAN THAT UP */
    mroom_id = ParseeGetBridgedRoom(args, stanza);
    if (!mroom_id && !room && !XMPPIsParseeStanza(stanza) &&
        to && *to == '@')
    {
        DbRef *room_ref;
        HashMap *room_json;
        char *from = HashMapGet(stanza->attrs, "from");

        ASRegisterUser(args->config, from_matrix);
        room = ASCreateDM(args->config, from_matrix, to);
        mroom_id = StrDuplicate(room);
        if (room)
        {
            room_ref = DbCreate(args->db, 3, "rooms", room, "data");
            room_json = DbJson(room_ref);
            HashMapSet(room_json, "is_direct", JsonValueBoolean(true));
            HashMapSet(room_json, "xmpp_user", JsonValueString(from));
            DbUnlock(args->db, room_ref);
            ParseePushDMRoom(args, to, from, room);
        }
    }

    /* TODO: THIS IS A HACK. THIS CODE IGNORES ALL MUC MESSAGES EVER 
     * SENT TO NON-PARSEE PUPPETS, AS A "FIX" TO THE MULTITHREADED 
     * ISSUE. 
     *
     * I HATE THIS. I NEED TO FIND A BETTER WAY. */
    if (!chat && strncmp(HashMapGet(stanza->attrs, "to"), "parsee@", 7))
    {
        goto end;
    }

    if (mroom_id && !XMPPIsParseeStanza(stanza))
    {
        char *res = ParseeGetResource(from);
        char *encoded   = ParseeGetBridgedUser(args, stanza);
        char *event_id  = NULL;

        {
            char *parsee = ParseeJID(args);
            char *trim = ParseeTrimJID(from);
            XMLElement *ps;

            /* Subscribe to the sender's metadata node. */
            ps = CreatePubsubRequest(
                parsee, decode_from, "urn:xmpp:avatar:metadata"
            );
            pthread_mutex_lock(&jabber->write_lock);
            XMLEncode(jabber->stream, ps);
            StreamFlush(jabber->stream);
            pthread_mutex_unlock(&jabber->write_lock);
            XMLFreeElement(ps);

            ps = CreatePubsubRequest(
                parsee, trim, "urn:xmpp:avatar:metadata"
            );
            pthread_mutex_lock(&jabber->write_lock);
            XMLEncode(jabber->stream, ps);
            StreamFlush(jabber->stream);
            pthread_mutex_unlock(&jabber->write_lock);
            XMLFreeElement(ps);

            Free(parsee);
            Free(trim);
        }

        pthread_mutex_lock(&thr->info->chk_lock);
        if (ParseeVerifyAllStanza(args, stanza) && !replaced)
        {
            XMLElement *oob, *oob_data;

            pthread_mutex_unlock(&thr->info->chk_lock);
            ASRegisterUser(args->config, encoded);
            if (!chat)
            {
                ASSetName(args->config, encoded, res);
            }
            ASInvite(args->config, mroom_id, encoded);
            Free(ASJoin(args->config, mroom_id, encoded));

            /* Check if it is a media link */
            oob = XMLookForTKV(stanza, "x", "xmlns", "jabber:x:oob");
            reactions = XMLookForTKV(stanza,
                "reactions", "xmlns", "urn:xmpp:reactions:0"
            );
            if (oob)
            {
                char *mxc, *mime = NULL;
                HashMap *content = NULL;
                oob_data = XMLookForUnique(oob, "url");
                oob_data = 
                    oob_data ? ArrayGet(oob_data->children, 0) : NULL;

                if (oob_data)
                {
                    mxc = ASReupload(args->config, oob_data->data, &mime);
                    content = MatrixCreateMedia(mxc, data->data, mime);

                    /* Yeah, no, I'm not modifying the media creation code. */
                    HashMapSet(content, 
                        "at.kappach.at.parsee.external", 
                        JsonValueString(oob_data->data)
                    );
                    ShoveStanza(content, stanza);

                    event_id = ASSend(
                        args->config, mroom_id, encoded, 
                        "m.room.message", content
                    );
                    Free(mime);
                    Free(mxc);
                }
            }
            else if (reactions)
            {
                Array *react_child = reactions->children;
                size_t reacts = ArraySize(react_child);
                event_id = ParseeGetReactedEvent(args, stanza);
                for (i = 0; i < reacts; i++)
                {
                    XMLElement *reaction, *react_data;
                    reaction = ArrayGet(react_child, i);
                    react_data = ArrayGet(reaction->children, 0);

                    /* TODO: We should manage removed reactions. */
                    Free(ASSend(
                        args->config, mroom_id, encoded, 
                        "m.reaction", 
                        ShoveStanza(
                            MatrixCreateReact(event_id, react_data->data),
                            stanza
                        )
                    ));
                }
                Free(event_id);
                event_id = NULL;
            }
            else if (retracted)
            {
                event_id = ParseeGetEventFromID(args, stanza, retracted);
                ASRedact(args->config, mroom_id, encoded, event_id);
                ParseePushAllStanza(args, stanza, event_id);
                pthread_mutex_unlock(&thr->info->chk_lock);

                Free(event_id);
                event_id = NULL;
            }
            else
            {
                /* TODO: Use HTML-formatted bodies, and respect the fallback 
                 * trims the stanza provides us if possible. Element does not 
                 * like raw bodies on replies too. Go figure. */
                size_t off = 
                    reply_to ? ParseeFindDatastart(data->data) : 0;
                HashMap *ev = MatrixCreateMessage(data->data + off);
                if (reply_to)
                {
                    char *reply_id = ParseeGetEventFromID(args, stanza, reply_to);
                    MatrixSetReply(ev, reply_id);
                    Free(reply_id);
                }
                ShoveStanza(ev, stanza);
                event_id = ASSend(
                    args->config, mroom_id, encoded, 
                    "m.room.message", ev
                );
            }
            pthread_mutex_lock(&thr->info->chk_lock);
            ParseePushAllStanza(args, stanza, event_id);
            Free(event_id);
            pthread_mutex_unlock(&thr->info->chk_lock);
        }
        else if (replaced)
        {
            event_id = ParseeGetEventFromID(args, stanza, replaced);
            Free(ASSend(
                args->config, mroom_id, encoded, 
                "m.room.message", 
                ShoveStanza(MatrixCreateReplace(event_id, data->data), stanza)
            ));
            ParseePushAllStanza(args, stanza, event_id);
            pthread_mutex_unlock(&thr->info->chk_lock);

            Free(event_id);
        }
        else
        {
            pthread_mutex_unlock(&thr->info->chk_lock);
        }
        Free(res);
        Free(encoded);
    }
    else
    {
        XMLElement *parsee = XMLookForUnique(stanza, "x-parsee");
        XMLElement *event = XMLookForUnique(parsee, "event-id");
        XMLElement *e_d = ArrayGet(event ? event->children : NULL, 0);

        if (e_d)
        {
            ParseePushAllStanza(args, stanza, e_d->data);
        }
    }

end:
    Free(mroom_id);
    mroom_id = NULL;
    Free(from_matrix);
    Free(decode_from);
    Free(room);
    Free(to);

    return true;
}