ostinato/client/logsmodel.cpp

/*
Copyright (C) 2018 Srivats P.

This file is part of "Ostinato"

This 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 "logsmodel.h"

#include <QBrush>
#include <QMimeData>

// XXX: Keep the enum in sync with it's string
enum {
    kTimeStamp,
    kLogLevel,
    kPort,
    kMessage,
    kFieldCount
};
static QStringList columnTitles = QStringList()
    << "Timestamp"
    << "Level"
    << "Port"
    << "Message";

static QStringList levelTitles = QStringList()
    << "Info"
    << "Warning"
    << "Error";

LogsModel::LogsModel(QObject *parent)
    : QAbstractTableModel(parent)
{
}

int LogsModel::rowCount(const QModelIndex &/*parent*/) const
{
    return logs_.size();
}

int LogsModel::columnCount(const QModelIndex &/*parent*/) const
{
    return kFieldCount;
}

QVariant LogsModel::headerData(
        int section, Qt::Orientation orientation, int role) const
{
    if (role != Qt::DisplayRole)
        return QVariant();

    switch (orientation) {
    case Qt::Horizontal: // Column Header
        return columnTitles.at(section);
    case Qt::Vertical:   // Row Header
        return QVariant();
    default:
        break;
    }
    return QVariant();
}

QVariant LogsModel::data(const QModelIndex &index, int role) const
{
    if (!index.isValid() || (index.row() >= logs_.size()))
            return QVariant();

    if (role == Qt::ForegroundRole) {
        switch(logs_.at(index.row()).logLevel) {
        case kError:
            return QBrush(QColor("darkred"));
        case kWarning:
            return QBrush(QColor("orangered"));
        case kInfo:
        default:
            return QVariant();
        }
    }

    if (role != Qt::DisplayRole)
        return QVariant();

    switch (index.column()) {
    case kTimeStamp:
        return logs_.at(index.row()).timeStamp.toString("hh:mm:ss.zzz");
    case kLogLevel:
        return levelTitles.at(logs_.at(index.row()).logLevel);
    case kPort:
        return logs_.at(index.row()).port;
    case kMessage:
        return logs_.at(index.row()).message;
    default:
        break;
    }

    return QVariant();
}

QStringList LogsModel::mimeTypes() const
{
    return QStringList() << "text/plain";
}

QMimeData* LogsModel::mimeData(const QModelIndexList &indexes) const
{
    int lastRow = -1;
    QString text;
    foreach(QModelIndex index, indexes) {
        if (index.row() != lastRow) {
            if (!text.isEmpty())
                text.append("\n");
        }
        else
            text.append("\t");
        text.append(data(index).toString());
        lastRow = index.row();
    }

    QMimeData *mimeData = new QMimeData();
    mimeData->setText(text);
    return mimeData; // caller is responsible for freeing!
}

// --------------------------------------------- //
// Slots
// --------------------------------------------- //
void LogsModel::clear()
{
    beginResetModel();
    logs_.clear();
    endResetModel();
}

void LogsModel::setLogLevel(int level)
{
    currentLevel_ = static_cast<LogLevel>(level % kLevelCount);
    log(currentLevel_, QString("--"),
        QString("Log level changed to %1 or higher")
            .arg(levelTitles.at(currentLevel_)));
}

void LogsModel::log(int logLevel,QString port, QString message)
{
    if (logLevel < currentLevel_)
        return;

    // TODO: discard logs older than some threshold

    //qDebug("adding log %u %s", logs_.size(), qPrintable(message));
    beginInsertRows(QModelIndex(), logs_.size(), logs_.size());
    Log l;
    logs_.append(l);
    logs_.last().timeStamp = QTime::currentTime();
    logs_.last().logLevel = logLevel;
    logs_.last().port = port;
    // XXX: QTableView does not honour newline unless we increase the
    // row height, so we replace newlines with semicolon for now
    logs_.last().message = message.trimmed().replace("\n", "; ");
    endInsertRows();
}