#include "chatlistmodel.h"
#include <QListIterator>

ChatListModel::ChatListModel(TDLibWrapper *tdLibWrapper)
{
    this->tdLibWrapper = tdLibWrapper;
    connect(this->tdLibWrapper, SIGNAL(newChatDiscovered(QString, QVariantMap)), this, SLOT(handleChatDiscovered(QString, QVariantMap)));
    connect(this->tdLibWrapper, SIGNAL(chatLastMessageUpdated(QString, QVariantMap)), this, SLOT(handleChatLastMessageUpdated(QString, QVariantMap)));
    connect(this->tdLibWrapper, SIGNAL(chatOrderUpdated(QString, QString)), this, SLOT(handleChatOrderUpdated(QString, QString)));
}

ChatListModel::~ChatListModel()
{

}

int ChatListModel::rowCount(const QModelIndex &) const
{
    return chatList.size();
}

QVariant ChatListModel::data(const QModelIndex &index, int role) const
{
    if(index.isValid() && role == Qt::DisplayRole) {
        return QVariant(chatList.value(index.row()));
    }
    return QVariant();
}

bool ChatListModel::insertRows(int row, int count, const QModelIndex &parent)
{
    qDebug() << "[ChatListModel] Inserting at " << row << ", row count: " << count;
    beginInsertRows(parent, rowCount(QModelIndex()), rowCount(QModelIndex()) + count - 1);
    this->chatList.append(this->chatToBeAdded);
    this->chatIndexMap.insert(this->chatToBeAdded.value("id").toString(), row);
    endInsertRows();
    return true;
}

bool compareChats(const QVariant &chat1, const QVariant &chat2)
{
    QVariantMap chatMap1 = chat1.toMap();
    QVariantMap chatMap2 = chat2.toMap();
    // Order comes first...
    if (chatMap1.value("order").toLongLong() > chatMap2.value("order").toLongLong()) {
        return true;
    }
    // ID comes next...
    if (chatMap1.value("order").toLongLong() == chatMap2.value("order").toLongLong() && chatMap1.value("id").toLongLong() > chatMap2.value("id").toLongLong()) {
        return true;
    } else {
        return false;
    }
}

void ChatListModel::updateSorting()
{
    this->chatListMutex.lock();
    qDebug() << "[ChatListModel] List sorting will be updated...";

    beginResetModel();
    std::sort(this->chatList.begin(), this->chatList.end(), compareChats);

    this->chatIndexMap.clear();
    for (int i = 0; i < this->chatList.length(); i++) {
        QString currentChatId = this->chatList.at(i).toMap().value("id").toString();
        this->chatIndexMap.insert(currentChatId, i);
    }
    endResetModel();
    this->chatListMutex.unlock();
}

void ChatListModel::handleChatDiscovered(const QString &chatId, const QVariantMap &chatInformation)
{
    this->chatListMutex.lock();
    qDebug() << "[ChatListModel] Adding new chat " << chatId;
    this->chatToBeAdded = chatInformation;
    insertRows(rowCount(QModelIndex()), 1);
    this->chatListMutex.unlock();
}

void ChatListModel::handleChatLastMessageUpdated(const QString &chatId, const QVariantMap &lastMessage)
{
    this->chatListMutex.lock();
    int chatIndex = this->chatIndexMap.value(chatId).toInt();
    qDebug() << "[ChatListModel] Updating last message for chat " << chatId << " at index " << chatIndex;
    QVariantMap currentChat = this->chatList.value(chatIndex).toMap();
    currentChat.insert("last_message", lastMessage);
    this->chatList.replace(chatIndex, currentChat);
    emit dataChanged(this->index(chatIndex), this->index(chatIndex));
    this->chatListMutex.unlock();
}

void ChatListModel::handleChatOrderUpdated(const QString &chatId, const QString &order)
{
    this->chatListMutex.lock();
    qDebug() << "[ChatListModel] Updating chat order because of " << chatId << " new order " << order;
    int currentChatIndex = this->chatIndexMap.value(chatId).toInt();
    QVariantMap currentChat = this->chatList.at(currentChatIndex).toMap();
    currentChat.insert("order", order);
    this->chatList.replace(currentChatIndex, currentChat);
    emit dataChanged(this->index(currentChatIndex), this->index(currentChatIndex));
    this->chatListMutex.unlock();
}