/*
 * Squawk messenger. 
 * Copyright (C) 2019  Yury Gubich <blue@macaw.me>
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#include "messageline.h"
#include <QDebug>
#include <QCoreApplication>
#include <cmath>

MessageLine::MessageLine(bool p_room, QWidget* parent):
    QWidget(parent),
    messageIndex(),
    messageOrder(),
    myMessages(),
    palMessages(),
    uploadPaths(),
    palAvatars(),
    exPalAvatars(),
    layout(new QVBoxLayout(this)),
    myName(),
    myAvatarPath(),
    palNames(),
    uploading(),
    downloading(),
    room(p_room),
    busyShown(false),
    progress()
{
    setContentsMargins(0, 0, 0, 0);
    layout->setContentsMargins(0, 0, 0, 0);
    layout->setSpacing(0);
    layout->addStretch();
}

MessageLine::~MessageLine()
{
    for (Index::const_iterator itr = messageIndex.begin(), end = messageIndex.end(); itr != end; ++itr) {
        delete itr->second;
    }
}

MessageLine::Position MessageLine::message(const Shared::Message& msg, bool forceOutgoing)
{
    QString id = msg.getId();
    Index::iterator itr = messageIndex.find(id);
    if (itr != messageIndex.end()) {
        qDebug() << "received more then one message with the same id, skipping yet the new one";
        return invalid;
    }
    
    QString sender;
    QString aPath;
    bool outgoing;
    
    if (forceOutgoing) {
        sender = myName;
        aPath = myAvatarPath;
        outgoing = true;
    } else {
        if (room) {
            if (msg.getFromResource() == myName) {
                sender = myName;
                aPath = myAvatarPath;
                outgoing = true;
            } else {
                sender = msg.getFromResource();
                std::map<QString, QString>::iterator aItr = palAvatars.find(sender);
                if (aItr != palAvatars.end()) {
                    aPath = aItr->second;
                } else {
                    aItr = exPalAvatars.find(sender);
                    if (aItr != exPalAvatars.end()) {
                        aPath = aItr->second;
                    }
                }
                outgoing = false;
            }
        } else {
            if (msg.getOutgoing()) {
                sender = myName;
                aPath = myAvatarPath;
                outgoing = true;
            } else {
                QString jid = msg.getFromJid();
                std::map<QString, QString>::iterator itr = palNames.find(jid);
                if (itr != palNames.end()) {
                    sender = itr->second;
                } else {
                    sender = jid;
                }
                
                std::map<QString, QString>::iterator aItr = palAvatars.find(jid);
                if (aItr != palAvatars.end()) {
                    aPath = aItr->second;
                }
                
                outgoing = false;
            }
        }
    }
    
    Message* message = new Message(msg, outgoing, sender, aPath);
    
    std::pair<Order::const_iterator, bool> result = messageOrder.insert(std::make_pair(msg.getTime(), message));
    if (!result.second) {
        qDebug() << "Error appending a message into a message list - seems like the time of that message exactly matches the time of some other message, can't put them in order, skipping yet";
        delete message;
        return invalid;
    }
    if (outgoing) {
        myMessages.insert(std::make_pair(id, message));
    } else {
        QString senderId;
        if (room) {
            senderId = sender;
        } else {
            senderId = msg.getFromJid();
        }
        
        std::map<QString, Index>::iterator pItr = palMessages.find(senderId);
        if (pItr == palMessages.end()) {
            pItr = palMessages.insert(std::make_pair(senderId, Index())).first;
        }
        pItr->second.insert(std::make_pair(id, message));
    }
    messageIndex.insert(std::make_pair(id, message));
    unsigned long index = std::distance<Order::const_iterator>(messageOrder.begin(), result.first);   //need to make with binary indexed tree
    Position res = invalid;
    if (index == 0) {
        res = beggining;
    } else if (index == messageIndex.size() - 1) {
        res = end;
    } else {
        res = middle;
    }
    
    if (busyShown) {
        index += 1;
    }
    
        
    if (res == end) {
        layout->addWidget(message);
    } else {
        layout->insertWidget(index + 1, message);
    }
    
    if (msg.hasOutOfBandUrl()) {
        emit requestLocalFile(msg.getId(), msg.getOutOfBandUrl());
        connect(message, &Message::buttonClicked, this, &MessageLine::onDownload);
    }
    
    return res;
}

void MessageLine::changeMessage(const QString& id, const QMap<QString, QVariant>& data)
{
    Index::const_iterator itr = messageIndex.find(id);
    if (itr != messageIndex.end()) {
        Message* msg = itr->second;
        if (msg->change(data)) {                    //if ID changed (stanza in replace of another)
            QString newId = msg->getId();           //need to updated IDs of that message in all maps
            messageIndex.erase(itr);
            messageIndex.insert(std::make_pair(newId, msg));
            if (msg->outgoing) {
                QString senderId;
                if (room) {
                    senderId = msg->getSenderResource();
                } else {
                    senderId = msg->getSenderJid();
                }
                
                std::map<QString, Index>::iterator pItr = palMessages.find(senderId);
                if (pItr != palMessages.end()) {
                    Index::const_iterator sItr = pItr->second.find(id);
                    if (sItr != pItr->second.end()) {
                        pItr->second.erase(sItr);
                        pItr->second.insert(std::make_pair(newId, msg));
                    } else {
                        qDebug() << "Was trying to replace message in open conversations, couldn't find it among pal's messages, probably an error"; 
                    }
                } else {
                    qDebug() << "Was trying to replace message in open conversations, couldn't find pal messages, probably an error"; 
                }
            } else {
                Index::const_iterator mItr = myMessages.find(id);
                if (mItr != myMessages.end()) {
                    myMessages.erase(mItr);
                    myMessages.insert(std::make_pair(newId, msg));
                } else {
                    qDebug() << "Was trying to replace message in open conversations, couldn't find it among my messages, probably an error"; 
                }
            }
        }
    }
}

void MessageLine::onDownload()
{
    Message* msg = static_cast<Message*>(sender());
    QString messageId = msg->getId();
    Index::const_iterator itr = downloading.find(messageId);
    if (itr == downloading.end()) {
        downloading.insert(std::make_pair(messageId, msg));
        msg->setProgress(0);
        msg->showComment(tr("Downloading..."));
        emit downloadFile(messageId, msg->getFileUrl());
    } else {
        qDebug() << "An attempt to initiate download for already downloading file" << msg->getFileUrl() << ", skipping";
    }
}

void MessageLine::setMyName(const QString& name)
{
    myName = name;
    for (Index::const_iterator itr = myMessages.begin(), end = myMessages.end(); itr != end; ++itr) {
        itr->second->setSender(name);
    }
}

void MessageLine::setPalName(const QString& jid, const QString& name)
{
    std::map<QString, QString>::iterator itr = palNames.find(jid);
    if (itr == palNames.end()) {
        palNames.insert(std::make_pair(jid, name));
    } else {
        itr->second = name;
    }
    
    std::map<QString, Index>::iterator pItr = palMessages.find(jid);
    if (pItr != palMessages.end()) {
        for (Index::const_iterator itr = pItr->second.begin(), end = pItr->second.end(); itr != end; ++itr) {
            itr->second->setSender(name);
        }
    }
}

void MessageLine::setPalAvatar(const QString& jid, const QString& path)
{
    std::map<QString, QString>::iterator itr = palAvatars.find(jid);
    if (itr == palAvatars.end()) {
        palAvatars.insert(std::make_pair(jid, path));
        
        std::map<QString, QString>::const_iterator eitr = exPalAvatars.find(jid);
        if (eitr != exPalAvatars.end()) {
            exPalAvatars.erase(eitr);
        }
    } else {
        itr->second = path;
    }
    
    std::map<QString, Index>::iterator pItr = palMessages.find(jid);
    if (pItr != palMessages.end()) {
        for (Index::const_iterator itr = pItr->second.begin(), end = pItr->second.end(); itr != end; ++itr) {
            itr->second->setAvatarPath(path);
        }
    }
}

void MessageLine::dropPalAvatar(const QString& jid)
{
    std::map<QString, QString>::iterator itr = palAvatars.find(jid);
    if (itr != palAvatars.end()) {
        palAvatars.erase(itr);
    }
    
    std::map<QString, QString>::const_iterator eitr = exPalAvatars.find(jid);
    if (eitr != exPalAvatars.end()) {
        exPalAvatars.erase(eitr);
    }
    
    std::map<QString, Index>::iterator pItr = palMessages.find(jid);
    if (pItr != palMessages.end()) {
        for (Index::const_iterator itr = pItr->second.begin(), end = pItr->second.end(); itr != end; ++itr) {
            itr->second->setAvatarPath("");
        }
    }
}

void MessageLine::movePalAvatarToEx(const QString& name)
{
    std::map<QString, QString>::iterator itr = palAvatars.find(name);
    if (itr != palAvatars.end()) {
        std::map<QString, QString>::iterator eitr = exPalAvatars.find(name);
        if (eitr != exPalAvatars.end()) {
            eitr->second = itr->second;
        } else {
            exPalAvatars.insert(std::make_pair(name, itr->second));
        }
        
        palAvatars.erase(itr);
    }
}

void MessageLine::resizeEvent(QResizeEvent* event)
{
    QWidget::resizeEvent(event);
    emit resize(event->size().height() - event->oldSize().height());
}


QString MessageLine::firstMessageId() const
{
    if (messageOrder.size() == 0) {
        return "";
    } else {
        return messageOrder.begin()->second->getId();
    }
}

void MessageLine::showBusyIndicator()
{
    if (!busyShown)  {
        layout->insertWidget(0, &progress);
        progress.start();
        busyShown = true;
    }
}

void MessageLine::hideBusyIndicator()
{
    if (busyShown) {
        progress.stop();
        layout->removeWidget(&progress);
        busyShown = false;
    }
}

void MessageLine::fileProgress(const QString& messageId, qreal progress)
{
    Index::const_iterator itr = messageIndex.find(messageId);
    if (itr == messageIndex.end()) {
        //TODO may be some logging, that's not normal
    } else {
        itr->second->setProgress(progress);
    }
}

void MessageLine::responseLocalFile(const QString& messageId, const QString& path)
{
    Index::const_iterator itr = messageIndex.find(messageId);
    if (itr == messageIndex.end()) {
        
    } else {
        Index::const_iterator uItr = uploading.find(messageId);
        if (path.size() > 0) {
            Index::const_iterator dItr = downloading.find(messageId);
            if (dItr != downloading.end()) {
                downloading.erase(dItr);
                itr->second->showFile(path);
            } else {
                if (uItr != uploading.end()) {
                    uploading.erase(uItr);
                    std::map<QString, QString>::const_iterator muItr = uploadPaths.find(messageId);
                    if (muItr != uploadPaths.end()) {
                        uploadPaths.erase(muItr);
                    }
                    Shared::Message msg = itr->second->getMessage();
                    removeMessage(messageId);
                    msg.setCurrentTime();
                    message(msg);
                    itr = messageIndex.find(messageId);
                    itr->second->showFile(path);
                } else {
                    itr->second->showFile(path); //then it is already cached file
                }
            }
        } else {
            if (uItr == uploading.end()) {
                const Shared::Message& msg = itr->second->getMessage();
                itr->second->addButton(QIcon::fromTheme("download"), tr("Download"), "<a href=\"" + msg.getOutOfBandUrl() + "\">" + msg.getOutOfBandUrl() + "</a>");
                itr->second->showComment(tr("Push the button to download the file"));
            } else {
                qDebug() << "An unhandled state for file uploading - empty path";
            }
        }
    }
}

void MessageLine::removeMessage(const QString& messageId)
{
    Index::const_iterator itr = messageIndex.find(messageId);
    if (itr != messageIndex.end()) {
        Message* ui = itr->second;
        const Shared::Message& msg = ui->getMessage();
        messageIndex.erase(itr);
        Order::const_iterator oItr = messageOrder.find(msg.getTime());
        if (oItr != messageOrder.end()) {
            messageOrder.erase(oItr);
        } else {
            qDebug() << "An attempt to remove message from messageLine, but it wasn't found in order";
        }
        if (msg.getOutgoing()) {
            Index::const_iterator mItr = myMessages.find(messageId);
            if (mItr != myMessages.end()) {
                myMessages.erase(mItr);
            } else {
                qDebug() << "Error removing message: it seems to be outgoing yet it wasn't found in outgoing messages";
            }
        } else {
            if (room) {
            
            } else {
                QString jid = msg.getFromJid();
                std::map<QString, Index>::iterator pItr = palMessages.find(jid);
                if (pItr != palMessages.end()) {
                    Index& pMsgs = pItr->second;
                    Index::const_iterator pmitr = pMsgs.find(messageId);
                    if (pmitr != pMsgs.end()) {
                        pMsgs.erase(pmitr);
                    } else {
                        qDebug() << "Error removing message: it seems to be incoming yet it wasn't found among messages from that penpal";
                    }
                }
            }
        }
        ui->deleteLater();
        qDebug() << "message" << messageId << "has been removed";
    } else {
        qDebug() << "An attempt to remove non existing message from messageLine";
    }
}

void MessageLine::fileError(const QString& messageId, const QString& error)
{
    Index::const_iterator itr = downloading.find(messageId);
    if (itr == downloading.end()) {
        Index::const_iterator itr = uploading.find(messageId);
        if (itr == uploading.end()) {
            //TODO may be some logging, that's not normal
        } else {
            itr->second->showComment(tr("Error uploading file: %1\nYou can try again").arg(QCoreApplication::translate("NetworkErrors", error.toLatin1())), true);
            itr->second->addButton(QIcon::fromTheme("upload"), tr("Upload"));
        }
    } else {
        const Shared::Message& msg = itr->second->getMessage();
        itr->second->addButton(QIcon::fromTheme("download"), tr("Download"), "<a href=\"" + msg.getOutOfBandUrl() + "\">" + msg.getOutOfBandUrl() + "</a>");
        itr->second->showComment(tr("Error downloading file: %1\nYou can try again").arg(QCoreApplication::translate("NetworkErrors", error.toLatin1())), true);
    }
}

void MessageLine::appendMessageWithUpload(const Shared::Message& msg, const QString& path)
{
    appendMessageWithUploadNoSiganl(msg, path);
    emit uploadFile(msg, path);
}

void MessageLine::appendMessageWithUploadNoSiganl(const Shared::Message& msg, const QString& path)
{
    message(msg, true);
    QString id = msg.getId();
    Message* ui = messageIndex.find(id)->second;
    connect(ui, &Message::buttonClicked, this, &MessageLine::onUpload);     //this is in case of retry;
    ui->setProgress(0);
    ui->showComment(tr("Uploading..."));
    uploading.insert(std::make_pair(id, ui));
    uploadPaths.insert(std::make_pair(id, path));
}


void MessageLine::onUpload()
{
    //TODO retry
}

void MessageLine::setMyAvatarPath(const QString& p_path)
{
    if (myAvatarPath != p_path) {
        myAvatarPath = p_path;
        for (std::pair<QString, Message*> pair : myMessages) {
            pair.second->setAvatarPath(myAvatarPath);
        }
    }
}

void MessageLine::setExPalAvatars(const std::map<QString, QString>& data)
{
    exPalAvatars = data;
    
    for (const std::pair<QString, Index>& pair : palMessages) {
        if (palAvatars.find(pair.first) == palAvatars.end()) {
            std::map<QString, QString>::const_iterator eitr = exPalAvatars.find(pair.first);
            if (eitr != exPalAvatars.end()) {
                for (const std::pair<QString, Message*>& mp : pair.second) {
                    mp.second->setAvatarPath(eitr->second);
                }
            }
        }
    }
}