/*
 * Copyright (C) 2015 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/>.
 *
 */


// 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)

#include "SessionsModel.h"
#include "SessionsModelPrivate.h"
#include <QtCore/QDir>
#include <QtCore/QString>

namespace QLightDM
{

SessionsModel::SessionsModel(QObject* parent) :
    QAbstractListModel(parent),
    d_ptr(new SessionsModelPrivate(this))
{
    Q_D(SessionsModel);
    m_roleNames = QAbstractListModel::roleNames();
    m_roleNames[KeyRole] = "key";
    m_roleNames[TypeRole] = "type";
}

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

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();
    }
}

QHash<int, QByteArray> SessionsModel::roleNames() const
{
    return m_roleNames;
}

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

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

int SessionsModel::numSessions() const
{
    Q_D(const SessionsModel);
    return d->numSessions;
}

int SessionsModel::numAvailableSessions() const
{
    Q_D(const SessionsModel);
    return d->numAvailableSessions();
}

QString SessionsModel::testScenario() const
{
    Q_D(const SessionsModel);
    return d->testScenario;
}

void SessionsModel::setNumSessions(int numSessions)
{
    Q_D(SessionsModel);

    if (d->numSessions != numSessions) {
        d->numSessions = numSessions;
        d->resetEntries();
    }
}

void SessionsModel::setTestScenario(QString testScenario)
{
    Q_D(SessionsModel);

    if (d->testScenario != testScenario) {
        d->testScenario = testScenario;
        d->resetEntries();
    }
}

} // namespace QLightDM
