eeparser.cpp

#include "eeparser.h"

#include <QFile>
#include <QDebug>
#include <QRegularExpression>
#include <QDateTime>
#include <QFileSystemWatcher>
#include <QFileInfo>
#include <QDir>
#include <QTimer>
#include <QThread>


#include "filewatcher.h"

namespace Yate {

EEParser::EEParser(QString logFilename, bool isLive, QObject *parent):
    QObject(parent), filename_(logFilename), liveParsing_(isLive),
    currentPosition_(0), evtId_(0),
    lineParseRegex_("(\\d+\\.\\d+)\\s+(\\w+)\\s+\\[(\\w+)\\]\\s*\\:\\s*(.*)"),
    logDoesNotExist_(true), hostJustUnloaded_(true)
{

}

bool EEParser::liveParsing() const
{
    return liveParsing_;
}

void EEParser::setLiveParsing(bool newLiveParsing)
{
    liveParsing_ = newLiveParsing;
}

const QString &EEParser::filename() const
{
    return filename_;
}

void EEParser::setFilename(const QString &newFilename)
{
    filename_ = newFilename;
}

int EEParser::currentPosition() const
{
    return currentPosition_;
}

void EEParser::setCurrentPosition(int newCurrentPosition)
{
    currentPosition_ = newCurrentPosition;
}

void EEParser::reset()
{
    setCurrentPosition(0);
    evtId_ = 0;
    hostJustUnloaded_ = true;
}

void Yate::EEParser::parseLine(QString &line)
{
    line = line.trimmed();
    if (!line.length()) {
        return;
    }

    auto it = lineParseRegex_.globalMatch(line);
    bool matched = false;
    float timestamp;
    QString msgText;
    if (it.hasNext()) {
        matched = true;
        auto match = it.next();
        timestamp = match.captured(1).toFloat();
        msgText = match.captured(4).trimmed();
    }

    if(matched) {
        int val;
        QString strVal;
        auto evtType = EEParser::msgToEventType(msgText, val, strVal);
        if (evtType != LogEventType::Invalid) {

            if (hostJustUnloaded_) {
                if (evtType != LogEventType::TeralystSpawn && evtType != LogEventType::NightBegin && evtType != LogEventType::HostJoin
                        && evtType != LogEventType::DoorOpening && evtType != LogEventType::DoorOpened) {
                    return;
                }
            }

            LogEvent evt(evtId_++, evtType, timestamp, val, strVal);
            if (evtType == LogEventType::HostUnload) {
                hostJustUnloaded_ = true;
            } else {
                hostJustUnloaded_ = false;
            }
            emit logEvent(evt);
        }
    }
}

void EEParser::stopParsing()
{
    qDebug() << "Stopping live feedback parser.";
    emit stopWatcher();
    emit parsingFinished();
}

void EEParser::startOffline()
{
    qDebug() << "Offline parsing started.";
    emit parsingStarted();
    QFile logFile(filename());
    QString fileContent;

    if(logFile.open(QIODevice::ReadOnly)) {
        fileContent =  QString(logFile.readAll());
        auto lines = fileContent.split(QRegularExpression("[\r\n]"), Qt::SkipEmptyParts);
        for(int i = 0; i < lines.size(); i++) {
            parseLine(lines[i]);
        }
        logFile.close();
    } else {
         qDebug() << "Offline parsing file error: " << logFile.errorString();
        emit parsingError(logFile.errorString());
    }

    qDebug() << "Offline parsing finshed.";
    emit parsingFinished();

}


void EEParser::processFileContent(QFile &logFile)
{
    QString fileContent =  QString(logFile.readAll());
    auto lines = fileContent.split(QRegularExpression("[\r\n]"), Qt::SkipEmptyParts);

    for(int i = 0; i < lines.size() - 2; i++) {
        parseLine(lines[i]);
    }
    int currPos = currentPosition() + fileContent.length();
    if (lines.size() > 0) {
        currPos -= (lines[lines.size() - 1].size() + 2);
        if (lines.size() > 1) {
            currPos -= (lines[lines.size() - 2].size() + 2);
        }
    }
    setCurrentPosition(currPos);
}

void EEParser::startLive()
{
    qDebug() << "Live parsing started.";
    emit parsingStarted();
    QFile logFile(filename());
    QThread *watcherThread = new QThread;

    watcher_ = new FileWatcher(nullptr, filename());
    watcher_->moveToThread(watcherThread);

    connect( watcherThread, &QThread::started, watcher_, &FileWatcher::start);
    connect( this, &EEParser::parsingFinished, watcher_, &FileWatcher::stop, Qt::DirectConnection);
    connect( watcherThread, &QThread::finished, watcher_, &QThread::deleteLater);
    connect( watcherThread, &QThread::finished, watcherThread, &QThread::deleteLater);


    connect(watcher_, &FileWatcher::fileChanged, this, &EEParser::onFileChanged);

    QTimer *tmr = new QTimer(this);
    tmr->setInterval(1000);
    connect(tmr, &QTimer::timeout, [&]() {if(QFileInfo::exists(filename())) { QFileInfo(filename()).lastModified();}});
    tmr->start();

    QString fileContent;


    if(logFile.open(QIODevice::ReadOnly)) {
        setCurrentPosition(0);
        processFileContent(logFile);
        logFile.close();
    } else {
        qDebug() << "Live parsing file error: " << logFile.errorString();
        emit parsingError(logFile.errorString());
    }
    watcherThread->start();

}

LogEventType EEParser::msgToEventType(QString msg, int &val, QString &strVal)
{
    QMap<QString, LogEventType> msgEvtMap{
        {"TeralystEncounter.lua: It's nighttime!", LogEventType::NightBegin},
        {"TeralystEncounter.lua: It's daytime!", LogEventType::DayBegin},
        {"TeralystAvatarScript.lua: Weakpoint Destroyed", LogEventType::LimbBreak},
        {"TeralystAvatarScript.lua: Teralyst Captured", LogEventType::EidolonCapture},
        {"TeralystAvatarScript.lua: Teralyst Killed", LogEventType::EidolonKill},
        {"SnapPickupToGround.lua: Snapping pickup to ground (DefaultArcanePickup)", LogEventType::LootDrop},
        {"TeralystEncounter.lua: Shrine enabled", LogEventType::ShrineEnable},
        {"TeralystEncounter.lua: Shrine disabled", LogEventType::ShrineDisable},
        {"EidolonMP.lua: EIDOLONMP: Finalize Eidolon transition", LogEventType::TeralystSpawn},
        {"TeralystEncounter.lua:      Eidolon spawning SUCCESS", LogEventType::EidolonSpawn},
        {"TeralystAvatarScript.lua: Wailing Song Complete - Teleporting", LogEventType::EidolonTeleport},
        {"EidolonMP.lua: EIDOLONMP: Level fully destroyed", LogEventType::HostUnload},
        {"TeralystEncounter.lua: Teralyst Escape complete. All Teralysts should be gone now", LogEventType::EidolonDespawn},
        {"EidolonMP.lua: EIDOLONMP: TownTriggerFirstTouched", LogEventType::DoorOpening},
        {"EidolonMP.lua: EIDOLONMP: EnableDoor(WORLD,true)", LogEventType::DoorOpened}

    };
    val = -1;


    if (msgEvtMap.contains(msg)) {
        return msgEvtMap[msg];
    } else if (msg.startsWith("Starting load for")) {
        const QRegularExpression rx("Starting load for\\s+(.+)");
        auto match = rx.match(msg);
        if (!match.isValid()) {
            return LogEventType::Invalid;
        }
        strVal = match.captured(1);
        if (strVal == "Player") {
            return LogEventType::Invalid;
        }
        return LogEventType::SquadJoin;


    } else if (msg.startsWith("Logged in ")) {
        const QRegularExpression rx("Logged in\\s+(.+)\\s+\\(");
        auto match = rx.match(msg);
        if (!match.isValid()) {
            return LogEventType::Invalid;
        }

        strVal = match.captured(1);
        if (strVal == "Player") {
            return LogEventType::Invalid;
        }
        return LogEventType::HostJoin;
    } else if (msg.startsWith("TeralystEncounter.lua: A shard has been put in the Eidolon Shrine.")) {
        const QRegularExpression rx("[^0-9]+");
        const auto&& parts = msg.split(rx, Qt::SkipEmptyParts);

        val = parts[0].toInt();
        return LogEventType::ShardInsert;
    } else if (msg.startsWith("TeralystEncounter.lua: A shard has been removed from the Eidolon Shrine.")) {
        return LogEventType::ShardRemove;
    } else {
        return LogEventType::Invalid;
    }

}

void EEParser::onFileChanged(bool exists)
{
    if (!exists) {
        setCurrentPosition(0);
        emit parsingReset();
    } else {
        QFile logFile(filename());
        QString fileContent;
        if(logFile.open(QIODevice::ReadOnly)) {
            logFile.seek(currentPosition());
            processFileContent(logFile);
            logFile.close();
        } else {
            emit parsingError(logFile.errorString());
        }
    }

}

}