/*
 * Copyright (C) 2015-2016 Canonical Ltd.
 *
 * This program 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; version 3.
 *
 * 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/>.
 *
 */


#ifdef LIGHTDM_COMPAT_QT4
// LightDM currently is Qt4 compatible, and so doesn't define setRoleNames.
// To use the same method of setting role name that it does, we
// set our compatibility to Qt4 here too.
#define QT_DISABLE_DEPRECATED_BEFORE QT_VERSION_CHECK(4, 0, 0)
#endif // LIGHTDM_COMPAT_QT4

#include "MockController.h"
#include "MockSessionsModel.h"

namespace QLightDM
{

class SessionsModelPrivate
{
public:
#ifdef LIGHTDM_COMPAT_QT4
    QHash<int, QByteArray> roleNames;
#endif // LIGHTDM_COMPAT_QT4
    QList<MockController::SessionItem> sessionItems;
};

SessionsModel::SessionsModel(QObject* parent)
    : QAbstractListModel(parent)
    , d_ptr(new SessionsModelPrivate)
{
    Q_D(SessionsModel);

#ifdef LIGHTDM_COMPAT_QT4
    auto roleNames = QAbstractListModel::roleNames();
    roleNames[KeyRole] = "key";
    roleNames[TypeRole] = "type";

    // We can't reimplement roleNames() because actual liblightdm doesn't
    // implement them either.
#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Wdeprecated-declarations"
    setRoleNames(roleNames);
#pragma GCC diagnostic pop
#endif // LIGHTDM_COMPAT_QT4

    connect(MockController::instance(), &MockController::sessionModeChanged,
            this, &SessionsModel::resetEntries);
    connect(MockController::instance(), &MockController::numSessionsChanged,
            this, &SessionsModel::resetEntries);
    resetEntries();
}

SessionsModel::~SessionsModel()
{
    delete d_ptr;
}

#ifndef LIGHTDM_COMPAT_QT4
QHash<int, QByteArray> SessionsModel::roleNames() const
{
    QHash<int, QByteArray> roles;
    roles[Qt::DisplayRole] = "display";
    roles[Qt::ToolTipRole] = "toolTip";
    roles[KeyRole] = "key";
    roles[TypeRole] = "type";

    return roles;
}
#endif // !LIGHTDM_COMPAT_QT4

QVariant SessionsModel::data(const QModelIndex& index, int role) const
{
    Q_D(const SessionsModel);

    if(!index.isValid()) {
        return QVariant();
    }

    int row = index.row();

    switch (role) {
        case QLightDM::SessionsModel::KeyRole:
            return d->sessionItems[row].key;
        case Qt::DisplayRole:
            return d->sessionItems[row].name;
        default:
            return QVariant();
    }
}

int SessionsModel::rowCount(const QModelIndex& parent) const
{
    Q_D(const SessionsModel);

    if (parent.isValid()) {
        return 0;
    } else { // parent is root
        return d->sessionItems.size();
    }
}

void SessionsModel::resetEntries()
{
    Q_D(SessionsModel);

    beginResetModel();

    QString sessionMode = MockController::instance()->sessionMode();

    if (sessionMode == "full") {
        d->sessionItems = MockController::instance()->fullSessionItems();
        d->sessionItems = d->sessionItems.mid(0, MockController::instance()->numSessions());
    } else if (sessionMode == "single") {
        d->sessionItems = {MockController::instance()->fullSessionItems()[0]};
    } else {
        d->sessionItems = {};
    }

    endResetModel();
}

} // namespace QLightDM
