// SPDX-License-Identifier: LGPL-3.0-or-later

/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
*
* Copyright (C) 2013 - 2024, nymea GmbH
* Copyright (C) 2024 - 2025, chargebyte austria GmbH
*
* This file is part of libnymea-app.
*
* libnymea-app is free software: you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public License
* as published by the Free Software Foundation, either version 3
* of the License, or (at your option) any later version.
*
* libnymea-app 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 Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with libnymea-app. If not, see <https://www.gnu.org/licenses/>.
*
* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */

#include "packages.h"
#include "package.h"

Packages::Packages(QObject *parent) : QAbstractListModel(parent)
{

}

int Packages::rowCount(const QModelIndex &parent) const
{
    Q_UNUSED(parent)
    return static_cast<int>(m_list.count());
}

QVariant Packages::data(const QModelIndex &index, int role) const
{
    switch (role) {
    case RoleId:
        return m_list.at(index.row())->id();
    case RoleDisplayName:
        return m_list.at(index.row())->displayName();
    case RoleSummary:
        return m_list.at(index.row())->summary();
    case RoleInstalledVersion:
        return m_list.at(index.row())->installedVersion();
    case RoleCandidateVersion:
        return m_list.at(index.row())->candidateVersion();
    case RoleChangelog:
        return m_list.at(index.row())->changelog();
    case RoleUpdateAvailable:
        return m_list.at(index.row())->updateAvailable();
    case RoleRollbackAvailable:
        return m_list.at(index.row())->rollbackAvailable();
    }
    return QVariant();
}

QHash<int, QByteArray> Packages::roleNames() const
{
    QHash<int, QByteArray> roles;
    roles.insert(RoleId, "id");
    roles.insert(RoleDisplayName, "displayName");
    roles.insert(RoleSummary, "summary");
    roles.insert(RoleInstalledVersion, "installedVersion");
    roles.insert(RoleCandidateVersion, "candidateVersion");
    roles.insert(RoleChangelog, "changelog");
    roles.insert(RoleUpdateAvailable, "updateAvailable");
    roles.insert(RoleRollbackAvailable, "rollbackAvailable");
    return roles;
}

void Packages::addPackage(Package *package)
{
    package->setParent(this);
    beginInsertRows(QModelIndex(), static_cast<int>(m_list.count()), static_cast<int>(m_list.count()));
    m_list.append(package);
    connect(package, &Package::summaryChanged, this, [this, package](){
        emit dataChanged(index(static_cast<int>(m_list.indexOf(package))), index(static_cast<int>(m_list.indexOf(package))), {RoleSummary});
        emit countChanged();
    });
    connect(package, &Package::installedVersionChanged, this, [this, package](){
        emit dataChanged(index(static_cast<int>(m_list.indexOf(package))), index(static_cast<int>(m_list.indexOf(package))), {RoleInstalledVersion});
        emit countChanged();
    });
    connect(package, &Package::candidateVersionChanged, this, [this, package](){
        emit dataChanged(index(static_cast<int>(m_list.indexOf(package))), index(static_cast<int>(m_list.indexOf(package))), {RoleCandidateVersion});
        emit countChanged();
    });
    connect(package, &Package::changelogChanged, this, [this, package](){
        emit dataChanged(index(static_cast<int>(m_list.indexOf(package))), index(static_cast<int>(m_list.indexOf(package))), {RoleChangelog});
        emit countChanged();
    });
    connect(package, &Package::updateAvailableChanged, this, [this, package](){
        emit dataChanged(index(static_cast<int>(m_list.indexOf(package))), index(static_cast<int>(m_list.indexOf(package))), {RoleUpdateAvailable});
        emit countChanged();
    });
    connect(package, &Package::rollbackAvailableChanged, this, [this, package](){
        emit dataChanged(index(static_cast<int>(m_list.indexOf(package))), index(static_cast<int>(m_list.indexOf(package))), {RoleRollbackAvailable});
        emit countChanged();
    });
    endInsertRows();
    emit countChanged();
}

void Packages::removePackage(const QString &packageId)
{
    int idx = -1;
    for (int i = 0; i < m_list.count(); i++) {
        if (m_list.at(i)->id() == packageId) {
            idx = i;
            break;
        }
    }
    if (idx < 0) {
        return;
    }

    beginRemoveRows(QModelIndex(), idx, idx);
    m_list.takeAt(idx)->deleteLater();
    endRemoveRows();
    emit countChanged();
}

Package *Packages::get(int index) const
{
    if (index >= 0 && index < m_list.count()) {
        return m_list.at(index);
    }
    return nullptr;
}

Package *Packages::getPackage(const QString &packageId)
{
    foreach (Package *p, m_list) {
        if (p->id() == packageId) {
            return p;
        }
    }
    return nullptr;
}

void Packages::clear()
{
    beginResetModel();
    foreach (Package *package, m_list)
        package->deleteLater();

    m_list.clear();
    endResetModel();
    emit countChanged();
}