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

#include "UsersModelPrivate.h"
#include "UsersModel.h"

#include <QDir>

namespace QLightDM
{

UsersModelPrivate::UsersModelPrivate(UsersModel* parent)
  : mockMode("single")
  , q_ptr(parent)
{
    char *envMockMode = getenv("LIBLIGHTDM_MOCK_MODE");
    if (envMockMode) {
        mockMode = envMockMode;
    }
    resetEntries();
}

void UsersModelPrivate::resetEntries()
{
    Q_Q(UsersModel);

    q->beginResetModel();

    if (mockMode == "single") {
        resetEntries_single();
    } else if (mockMode == "single-passphrase") {
        resetEntries_singlePassphrase();
    } else if (mockMode == "single-pin") {
        resetEntries_singlePin();
    } else if (mockMode == "full") {
        resetEntries_full();
    }

    // Assign uids in a loop, just to avoid having to muck with them when
    // adding or removing test users.
    for (int i = 0; i < entries.size(); i++) {
        entries[i].uid = i + 1;
    }

    // Assign backgrounds
    QDir backgroundDir("/usr/share/backgrounds");
    QStringList backgrounds = backgroundDir.entryList(QDir::Files);
    if (!backgrounds.empty()) {
        for (int i = 0; i < entries.size(); i++) {
            if (entries[i].background.isNull()) {
                entries[i].background = backgroundDir.filePath(backgrounds[i % backgrounds.size()]);
            }
        }
    }

    q->endResetModel();
}

void UsersModelPrivate::resetEntries_single()
{
    entries =
    {
        { "single", "Single User", 0, 0, false, false, "ubuntu", 0, 0 },
    };
}

void UsersModelPrivate::resetEntries_singlePassphrase()
{
    entries =
    {
        { "single", "Single User", 0, 0, false, false, "ubuntu", 0, 0 },
    };
}

void UsersModelPrivate::resetEntries_singlePin()
{
    entries =
    {
        { "has-pin", "Has PIN", 0, 0, false, false, "ubuntu", 0, 0 },
    };
}

void UsersModelPrivate::resetEntries_full()
{
    entries =
    {
        { "has-password",      "Has Password", 0, 0, false, false, "ubuntu", 0, 0 },
        { "has-pin",           "Has PIN",      0, 0, false, false, "ubuntu", 0, 0 },
        { "different-prompt",  "Different Prompt", 0, 0, false, false, "ubuntu", 0, 0 },
        { "no-password",       "No Password", 0, 0, false, false, "ubuntu", 0, 0 },
        { "auth-error",        "Auth Error", 0, 0, false, false, "ubuntu", 0, 0 },
        { "two-factor",        "Two Factor", 0, 0, false, false, "ubuntu", 0, 0 },
        { "info-prompt",       "Info Prompt", 0, 0, false, false, "ubuntu", 0, 0 },
        { "html-info-prompt",  "HTML Info Prompt", 0, 0, false, false, "ubuntu", 0, 0 },
        { "long-info-prompt",  "Long Info Prompt", 0, 0, false, false, "ubuntu", 0, 0 },
        { "wide-info-prompt",  "Wide Info Prompt", 0, 0, false, false, "ubuntu", 0, 0 },
        { "multi-info-prompt", "Multi Info Prompt", 0, 0, false, false, "ubuntu", 0, 0 },
        { "long-name",         "Long name (far far too long to fit, seriously this would never fit on the screen, you will never see this part of the name)", 0, 0, false, false, "ubuntu", 0, 0 },
        { "color-background",  "Color Background", "#E95420", 0, false, false, "ubuntu", 0, 0 },
        // white and black are a bit redundant, but useful for manually testing if UI is still readable
        { "white-background",  "White Background", "#ffffff", 0, false, false, "ubuntu", 0, 0 },
        { "black-background",  "Black Background", "#000000", 0, false, false, "ubuntu", 0, 0 },
        { "no-background",     "No Background", "", 0, false, false, "ubuntu", 0, 0 },
        { "unicode",           "가나다라마", 0, 0, false, false, "ubuntu", 0, 0 },
        { "no-response",       "No Response", 0, 0, false, false, "ubuntu", 0, 0 },
        { "empty-name",        "", 0, 0, false, false, "ubuntu", 0, 0 },
        { "active",            "Active Account", 0, 0, true, false, "ubuntu", 0, 0 },
    };
}

} // namespace QLightDM