squawk/core/contact.cpp

/*
 * <one line to give the program's name and a brief idea of what it does.>
 * 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 "contact.h"

Core::Contact::Contact(const QString& pJid, const QString& account, QObject* parent):
    QObject(parent),
    jid(pJid),
    name(),
    groups(),
    archiveState(empty),
    archive(new Archive(jid)),
    subscriptionState(Shared::unknown),
    syncronizing(false),
    requestedCount(0),
    requestedBefore(),
    hisoryCache(),
    appendCache(),
    responseCache(),
    requestCache()
{
    archive->open(account);
    if (archive->isFromTheBeginning()) {
        archiveState = beginning;
    } else {
        if (archive->size() != 0) {
            archiveState = chunk;
        }
    }
}

Core::Contact::~Contact()
{
    delete archive;
}

Core::Contact::ArchiveState Core::Contact::getArchiveState() const
{
    return archiveState;
}

QSet<QString> Core::Contact::getGroups() const
{
    return groups;
}

QString Core::Contact::getName() const
{
    return name;
}

void Core::Contact::setName(const QString& n)
{
    if (name != n) {
        name = n;
        emit nameChanged(name);
    }
}

Shared::SubscriptionState Core::Contact::getSubscriptionState() const
{
    return subscriptionState;
}

void Core::Contact::setGroups(const QSet<QString>& set)
{
    QSet<QString> toRemove = groups - set;
    QSet<QString> toAdd = set - groups;
    
    groups = set;
    
    for (QSet<QString>::iterator itr = toRemove.begin(), end = toRemove.end(); itr != end; ++itr) {
        emit groupRemoved(*itr);
    }
    
    for (QSet<QString>::iterator itr = toAdd.begin(), end = toAdd.end(); itr != end; ++itr) {
        emit groupAdded(*itr);
    }
}

void Core::Contact::setSubscriptionState(Shared::SubscriptionState state)
{
    if (subscriptionState != state) {
        subscriptionState = state;
        emit subscriptionStateChanged(subscriptionState);
    }
}

unsigned int Core::Contact::groupsCount() const
{
    return groups.size();
}

void Core::Contact::addMessageToArchive(const Shared::Message& msg)
{
    if (msg.getId().size() > 0 && msg.getBody().size() > 0) {
        hisoryCache.emplace_back(msg);
    }
}

void Core::Contact::requestHistory(int count, const QString& before)
{
    if (syncronizing) {
        requestCache.emplace_back(count, before);
    } else {
        performRequest(count, before);
    }
}

void Core::Contact::nextRequest()
{
    if (syncronizing) {
        if (requestedCount != -1) {
            emit historyResponse(responseCache);
        }
    }
    if (requestCache.size() > 0) {
        std::pair<int, QString> request = requestCache.front();
        requestCache.pop_front();
        performRequest(request.first, request.second);
    } else {
        syncronizing = false;
        requestedCount = 0;
        requestedBefore = "";
        hisoryCache.clear();
        responseCache.clear();
    }
}

void Core::Contact::performRequest(int count, const QString& before)
{
    syncronizing = true;
    requestedCount = count;
    requestedBefore = before;
    hisoryCache.clear();
    responseCache.clear();
    
    switch (archiveState) {
        case empty:
            emit needHistory(before, "", QDateTime(), QDateTime());
            break;
        case chunk:
        case beginning:
            if (count != -1) {
                requestCache.emplace_back(requestedCount, before);
                requestedCount = -1;
            }
            emit needHistory("", archive->newestId(), QDateTime(), QDateTime());
            break;
        case end: 
            if (count != -1) {
                bool found = requestFromArchive(before);
                if (found) {
                    int rSize = responseCache.size();
                    if (rSize < count) {
                        if (rSize != 0) {
                            emit needHistory(responseCache.front().getId(), "", QDateTime(), QDateTime());
                        } else {
                            emit needHistory(before, "", QDateTime(), QDateTime());
                        }
                    } else {
                        nextRequest();
                    }
                } else {
                    requestCache.emplace_back(requestedCount, before);
                    requestedCount = -1;
                    emit needHistory(archive->oldestId(), "", QDateTime(), QDateTime());
                }
            } else {
                emit needHistory(archive->oldestId(), "", QDateTime(), QDateTime());
            }
            break;
        case complete:
            if (!requestFromArchive(before)) {
                qDebug("requesting id hasn't been found in archive, skipping");
            }
            nextRequest();
            break;
    }
}

bool Core::Contact::requestFromArchive(const QString& before) {
    std::list<Shared::Message> arc;
    QString lBefore;
    if (responseCache.size() > 0) {
        lBefore = responseCache.front().getId();
    } else {
        lBefore = before;
    }
    if (requestedCount != -1) {
        try {
            arc = archive->getBefore(requestedCount - responseCache.size(), lBefore);
            responseCache.insert(responseCache.begin(), arc.begin(), arc.end());
            return true;
        } catch (Archive::NotFound e) {
            requestCache.emplace_back(requestedCount, before);
            requestedCount = -1;
            emit needHistory(archive->oldestId(), "", QDateTime(), QDateTime());
            return false;
        }
    } else {
        try {
            arc = archive->getBefore(1, lBefore);
            //just do nothing since response is not required
            //may be even it's a signal that the history is now complete?
            return true;
        } catch (Archive::NotFound e) {
            emit needHistory(archive->oldestId(), "", QDateTime(), QDateTime());
            return false;
        }
    }
}

void Core::Contact::appendMessageToArchive(const Shared::Message& msg)
{
    const QString& id = msg.getId(); 
    if (id.size() > 0) {
        if (msg.getBody().size() > 0) {
            switch (archiveState) {
                case empty:
                    if (archive->addElement(msg)) {
                        archiveState = end;
                    };
                    if (!syncronizing) {
                        requestHistory(-1, id);
                    }
                    break;
                case beginning:
                    appendCache.emplace_back(msg);
                    if (!syncronizing) {
                        requestHistory(-1, id);
                    }
                    break;
                case end:
                    archive->addElement(msg);
                    break;
                case chunk:
                    appendCache.emplace_back(msg);
                    if (!syncronizing) {
                        requestHistory(-1, id);
                    }
                    break;
                case complete:
                    archive->addElement(msg);
                    break;
            }
        } else if (!syncronizing && archiveState == empty) {
            requestHistory(-1, id);
        }
    }
}

void Core::Contact::flushMessagesToArchive(bool finished, const QString& firstId, const QString& lastId)
{
    if (hisoryCache.size() > 0) {
        archive->addElements(hisoryCache);
        hisoryCache.clear();
    }
    
    switch (archiveState) {
            break;
        case beginning:
            if (finished) {
                archiveState = complete;
                nextRequest();
            } else {
                emit needHistory("", lastId, QDateTime(), QDateTime());
            }
        case chunk:
            if (finished) {
                archiveState = end;
                nextRequest();
            } else {
                emit needHistory("", lastId, QDateTime(), QDateTime());
            }
            break;
        case empty:
            archiveState = end;
        case end:
            if (finished) {
                archiveState = complete;
                archive->setFromTheBeginning(true);
            }
            if (requestedCount != -1) {
                QString before;
                if (responseCache.size() > 0) {
                    before = responseCache.front().getId();
                } else {
                    before = requestedBefore;
                }
                if (!requestFromArchive(before)) {
                    qDebug("Something went terrible wrong flushing messages to the archive");
                }
                if (requestedCount < responseCache.size()) {
                    if (archiveState == complete) {
                        nextRequest();
                    } else {
                        emit needHistory(firstId, "", QDateTime(), QDateTime());
                    }
                } else {
                    nextRequest();
                }
            } else {
                nextRequest();
            }
            break;
        case complete:
            nextRequest();
            break;
    }
}