// SPDX-FileCopyrightText: 2021 Nheko Contributors
//
// SPDX-License-Identifier: GPL-3.0-or-later

#include "ReadReceiptsModel.h"

#include <QLocale>

#include "Cache.h"
#include "Cache_p.h"
#include "Logging.h"
#include "Utils.h"

ReadReceiptsModel::ReadReceiptsModel(QString event_id, QString room_id, QObject *parent)
  : QAbstractListModel{parent}
  , event_id_{event_id}
  , room_id_{room_id}
{
        try {
                addUsers(cache::readReceipts(event_id_, room_id_));
        } catch (const lmdb::error &) {
                nhlog::db()->warn("failed to retrieve read receipts for {} {}",
                                  event_id_.toStdString(),
                                  room_id_.toStdString());

                return;
        }

        connect(cache::client(), &Cache::newReadReceipts, this, &ReadReceiptsModel::update);
}

void
ReadReceiptsModel::update()
{
        try {
                addUsers(cache::readReceipts(event_id_, room_id_));
        } catch (const lmdb::error &) {
                nhlog::db()->warn("failed to retrieve read receipts for {} {}",
                                  event_id_.toStdString(),
                                  room_id_.toStdString());

                return;
        }
}

QHash<int, QByteArray>
ReadReceiptsModel::roleNames() const
{
        // Note: RawTimestamp is purposely not included here
        return {
          {Mxid, "mxid"},
          {DisplayName, "displayName"},
          {AvatarUrl, "avatarUrl"},
          {Timestamp, "timestamp"},
        };
}

QVariant
ReadReceiptsModel::data(const QModelIndex &index, int role) const
{
        if (!index.isValid() || index.row() >= (int)readReceipts_.size() || index.row() < 0)
                return {};

        switch (role) {
        case Mxid:
                return readReceipts_[index.row()].first;
        case DisplayName:
                return cache::displayName(room_id_, readReceipts_[index.row()].first);
        case AvatarUrl:
                return cache::avatarUrl(room_id_, readReceipts_[index.row()].first);
        case Timestamp:
                return dateFormat(readReceipts_[index.row()].second);
        case RawTimestamp:
                return readReceipts_[index.row()].second;
        default:
                return {};
        }
}

void
ReadReceiptsModel::addUsers(
  const std::multimap<uint64_t, std::string, std::greater<uint64_t>> &users)
{
        auto newReceipts = users.size() - readReceipts_.size();

        if (newReceipts > 0) {
                beginInsertRows(
                  QModelIndex{}, readReceipts_.size(), readReceipts_.size() + newReceipts - 1);

                for (const auto &user : users) {
                        QPair<QString, QDateTime> item = {
                          QString::fromStdString(user.second),
                          QDateTime::fromMSecsSinceEpoch(user.first)};
                        if (!readReceipts_.contains(item))
                                readReceipts_.push_back(item);
                }

                endInsertRows();
        }
}

QString
ReadReceiptsModel::dateFormat(const QDateTime &then) const
{
        auto now  = QDateTime::currentDateTime();
        auto days = then.daysTo(now);

        if (days == 0)
                return tr("Today %1")
                  .arg(QLocale::system().toString(then.time(), QLocale::ShortFormat));
        else if (days < 2)
                return tr("Yesterday %1")
                  .arg(QLocale::system().toString(then.time(), QLocale::ShortFormat));
        else if (days < 7)
                return QString("%1 %2")
                  .arg(then.toString("dddd"))
                  .arg(QLocale::system().toString(then.time(), QLocale::ShortFormat));

        return QLocale::system().toString(then.time(), QLocale::ShortFormat);
}

ReadReceiptsProxy::ReadReceiptsProxy(QString event_id, QString room_id, QObject *parent)
  : QSortFilterProxyModel{parent}
  , model_{event_id, room_id, this}
{
        setSourceModel(&model_);
        setSortRole(ReadReceiptsModel::RawTimestamp);
        sort(0, Qt::DescendingOrder);
        setDynamicSortFilter(true);
}