/*
 *  SPDX-FileCopyrightText: 2012 Alejandro Fiestas Olivares <afiestas@kde.org>
 *  SPDX-FileCopyrightText: 2020 Dan Leinir Turthra Jensen <admin@leinir.dk>
 *
 *  SPDX-License-Identifier: GPL-2.0-or-later
 */

#include "accountsmodel.h"

#include "core.h"
#include "servicesmodel.h"

#include <QDebug>

#include <QIcon>
#include <klocalizedstring.h>

#include <SignOn/Identity>

#include <Accounts/Account>
#include <Accounts/Manager>

class AccountsModel::Private : public QObject
{
public:
    Private(AccountsModel *model)
        : accountsManager(KAccounts::accountsManager())
        , q(model)
    {
        accountIDs = accountsManager->accountList();

        connect(accountsManager, &Accounts::Manager::accountCreated, q, [this](Accounts::AccountId accountId) {
            int row = accountIDs.count();
            q->beginInsertRows(QModelIndex(), row, row);
            accountIDs.insert(row, accountId);
            q->endInsertRows();
        });
        connect(accountsManager, &Accounts::Manager::accountRemoved, q, [this](Accounts::AccountId accountId) {
            q->beginRemoveRows(QModelIndex(), accountIDs.indexOf(accountId), accountIDs.indexOf(accountId));
            removeAccount(accountId);
            q->endRemoveRows();
        });
    };
    virtual ~Private()
    {
        qDeleteAll(accounts);
    };

    Accounts::Manager *accountsManager;
    Accounts::AccountIdList accountIDs;
    QHash<int, Accounts::Account *> accounts;
    QHash<Accounts::Account *, ServicesModel *> servicesModels;

    Accounts::Account *accountById(int id);
    void removeAccount(Accounts::AccountId accountId);

private:
    AccountsModel *q;
};

Accounts::Account *AccountsModel::Private::accountById(int id)
{
    if (accounts.contains(id)) {
        return accounts.value(id);
    }

    // If we don't yet have this account cached, get it and connect it up to the model
    Accounts::Account *account = accountsManager->account(id);
    if (!account) {
        qDebug() << "\t Failed to get the account from manager";
        return nullptr;
    }

    connect(account, &Accounts::Account::displayNameChanged, q, [this, account]() {
        QModelIndex accountIndex = q->index(accountIDs.indexOf(account->id()));
        Q_EMIT q->dataChanged(accountIndex, accountIndex, QVector<int>() << AccountsModel::DisplayNameRole);
    });

    accounts[id] = account;
    return account;
}

void AccountsModel::Private::removeAccount(Accounts::AccountId accountId)
{
    accountIDs.removeOne(accountId);
    delete accounts.take(accountId);
}

AccountsModel::AccountsModel(QObject *parent)
    : QAbstractListModel(parent)
    , d(new AccountsModel::Private(this))
{
}

AccountsModel::~AccountsModel()
{
    delete d;
}

QHash<int, QByteArray> AccountsModel::roleNames() const
{
    static QHash<int, QByteArray> roles{
        {IdRole, "id"},
        {ServicesRole, "services"},
        {EnabledRole, "enabled"},
        {CredentialsIdRole, "credentialsId"},
        {DisplayNameRole, "displayName"},
        {ProviderNameRole, "providerName"},
        {IconNameRole, "iconName"},
        {DataObjectRole, "dataObject"},
        {ProviderDisplayNameRole, "providerDisplayName"},
    };
    return roles;
}

int AccountsModel::rowCount(const QModelIndex &parent) const
{
    if (parent.isValid()) {
        return 0;
    }

    return d->accountIDs.count();
}

QVariant AccountsModel::data(const QModelIndex &index, int role) const
{
    QVariant data;
    if (checkIndex(index)) {
        Accounts::AccountId accountId = d->accountIDs.value(index.row());
        Accounts::Account *account = d->accountById(accountId);
        if (account) {
            switch (role) {
            case IdRole:
                data.setValue(account->id());
                break;
            case ServicesRole: {
                ServicesModel *servicesModel{nullptr};
                if (d->servicesModels.contains(account)) {
                    servicesModel = d->servicesModels.value(account);
                } else {
                    // Not parenting to the account itself, so we can avoid it suddenly
                    // disappearing. Just to be on the safe side
                    servicesModel = new ServicesModel(d->accountsManager);
                    servicesModel->setAccount(account);
                    d->servicesModels[account] = servicesModel;
                }
                data.setValue(servicesModel);
                break;
            }
            case EnabledRole:
                data.setValue(account->enabled());
                break;
            case CredentialsIdRole:
                data.setValue(account->credentialsId());
                break;
            case DisplayNameRole:
                data.setValue(account->displayName());
                break;
            case ProviderNameRole:
                data.setValue(account->providerName());
                break;
            case ProviderDisplayNameRole:
                data.setValue(account->provider().displayName());
                break;
            case IconNameRole: {
                QString iconName = QStringLiteral("user-identity");
                if (account->provider().isValid() && !account->provider().iconName().isEmpty()) {
                    iconName = account->provider().iconName();
                }
                data.setValue(iconName);
                break;
            }
            case DataObjectRole:
                data.setValue<QObject *>(account);
                break;
            }
        }
    }

    return data;
}
