/**

* @file ssukickstarter.cpp

* @copyright 2013 Jolla Ltd.

* @author Bernd Wachter <bwachter@lart.info>

* @date 2013

*/

#include <QStringList>

#include <QRegExp>

#include <QDirIterator>

#include "ssukickstarter.h"

#include "../constants.h"

/* TODO:

* - commands from the command section should be verified

* - allow overriding brand key

*/

SsuKickstarter::SsuKickstarter()

{

SsuDeviceInfo deviceInfo;

deviceModel = deviceInfo.deviceModel();

if ((ssu.deviceMode() & Ssu::RndMode) == Ssu::RndMode)

rndMode = true;

else

rndMode = false;

}

QStringList SsuKickstarter::commands()

{

SsuDeviceInfo deviceInfo(deviceModel);

QStringList result;

QHash<QString, QString>::const_iterator it = h.constBegin();

while (it != h.constEnd()) {

result.append(it.key() + " " + it.value());

it++;

}

}

QStringList SsuKickstarter::commandSection(const QString &section, const QString &description)

{

QStringList result;

SsuDeviceInfo deviceInfo(deviceModel);

QString commandFile;

QFile part;

QDir dir(Sandbox::map(QString("/%1/kickstart/%2/")

.arg(SSU_DATA_DIR)

.arg(section)));

if (description.isEmpty())

result.append("## No suitable configuration found in " + dir.path());

else

result.append("## No configuration for " + description + " found.");

return result;

}

if (description.isEmpty())

result.append("### Commands from " + dir.path() + "/" + commandFile);

else

result.append("### " + description + " from " + commandFile);

if (file.open(QIODevice::ReadOnly | QIODevice::Text)) {

QTextStream in(&file);

while (!in.atEnd())

result.append(in.readLine());

}

}

QString SsuKickstarter::replaceSpaces(const QString &value)

{

QString retval = value;

return retval.replace(" ", "_");

}

QStringList SsuKickstarter::repos()

{

QStringList result;

SsuDeviceInfo deviceInfo(deviceModel);

SsuRepoManager repoManager;

QTextStream qerr(stderr);

QStringList repos = repoManager.repos(rndMode, deviceInfo, Ssu::BoardFilter);

foreach (const QString &repo, repos) {

QString repoUrl = ssu.repoUrl(repo, rndMode, QHash<QString, QString>(), repoOverride);

qerr << "Repository " << repo << " does not have an URL, ignoring" << endl;

continue;

}

// Adaptation repos need to have separate naming so that when images are done

// the repository caches will not be mixed with each other.

if (repo.startsWith("adaptation")) {

result.append(QString("repo --name=%1-%2-%3%4 --baseurl=%5")

.arg(repo)

.arg(replaceSpaces(deviceModel))

.arg((rndMode ? repoOverride.value("rndRelease")

.arg(repoUrl)

);

result.append(QString("repo --name=%1-%2%3 --baseurl=%4")

.arg(repo)

.arg((rndMode ? repoOverride.value("rndRelease")

.arg(repoUrl)

);

return result;

}

{

QStringList result;

if (name == "packages") {

// insert @vendor configuration device

QString configuration = QString("@%1 Configuration %2")

.arg(repoOverride.value("brand"))

.arg(deviceModel);

result.append(configuration);

result.sort();

result.removeDuplicates();

} else {

result = commandSection(name);

result.prepend("%" + name);

result.append("%end");

return result;

}

// we intentionally don't support device-specific post scriptlets

{

QStringList result;

QString path;

QDir dir;

if ((flags & NoChroot) == NoChroot)

path = Sandbox::map(QString("/%1/kickstart/%2_nochroot/")

.arg(SSU_DATA_DIR)

.arg(name));

path = Sandbox::map(QString("/%1/kickstart/%2/")

.arg(SSU_DATA_DIR)

.arg(name));

if ((flags & DeviceSpecific) == DeviceSpecific) {

if (dir.exists(path + "/" + replaceSpaces(deviceModel.toLower())))

path = path + "/" + replaceSpaces(deviceModel.toLower());

else

path = path + "/default";

}

dir.setPath(path);

QStringList scriptlets = dir.entryList(QDir::AllEntries | QDir::NoDot | QDir::NoDotDot,

QDir::Name);

foreach (const QString &scriptlet, scriptlets) {

QFile file(dir.filePath(scriptlet));

result.append("### begin " + scriptlet);

if (file.open(QIODevice::ReadOnly | QIODevice::Text)) {

QTextStream in(&file);

while (!in.atEnd())

result.append(in.readLine());

}

result.append("### end " + scriptlet);

}

if (!result.isEmpty()) {

result.prepend(QString("export SSU_RELEASE_TYPE=%1")

.arg(rndMode ? "rnd" : "release"));

if ((flags & NoChroot) == NoChroot)

result.prepend("%" + name + " --nochroot");

else

result.prepend("%" + name);

result.append("%end");

}

}

void SsuKickstarter::setRepoParameters(QHash<QString, QString> parameters)

{

repoOverride = parameters;

if (repoOverride.contains("model"))

deviceModel = repoOverride.value("model");

}

{

QTextStream qerr(stderr);

QStringList commandSections;

// initialize with default 'part' for compatibility, as partitions

// used to work without configuration. It'll get replaced with

// configuration values, if found

commandSections.append("part");

// rnd mode should not come from the defaults

if (repoOverride.contains("rnd")) {

if (repoOverride.value("rnd") == "true")

rndMode = true;

else if (repoOverride.value("rnd") == "false")

rndMode = false;

QHash<QString, QString> defaults;

// get generic repo variables; domain and adaptation specific bits are not interesting

// in the kickstart

repoManager.repoVariables(&defaults, rndMode);

// overwrite with kickstart defaults

deviceInfo.variableSection("kickstart-defaults", &defaults);

if (deviceInfo.variable("kickstart-defaults", "commandSections")

.canConvert(QMetaType::QStringList)) {

commandSections =

deviceInfo.variable("kickstart-defaults", "commandSections").toStringList();

}

QHash<QString, QString>::const_iterator it = defaults.constBegin();

while (it != defaults.constEnd()) {

if (!repoOverride.contains(it.key()))

repoOverride.insert(it.key(), it.value());

it++;

}

// in rnd mode both rndRelease an release should be the same,

// as the variable name used is %(release)

if (rndMode && repoOverride.contains("rndRelease"))

repoOverride.insert("release", repoOverride.value("rndRelease"));

// release mode variables should not contain flavourName

if (!rndMode && repoOverride.contains("flavourName"))

repoOverride.remove("flavourName");

//TODO: check for mandatory keys, brand, ..

if (!repoOverride.contains("deviceModel"))

repoOverride.insert("deviceModel", deviceInfo.deviceModel());

// do sanity checking on the model

if (deviceInfo.contains() == false) {

qerr << "Device model '" << deviceInfo.deviceModel() << "' does not exist" << endl;

if (repoOverride.value("force") != "true")

return false;

}

QRegExp regex(" {2,}", Qt::CaseSensitive, QRegExp::RegExp2);

if (regex.indexIn(deviceInfo.deviceModel(), 0) != -1) {

qerr << "Device model '" << deviceInfo.deviceModel()

<< "' contains multiple consecutive spaces." << endl;

if (deviceInfo.contains())

qerr << "Since the model exists it looks like your configuration is broken." << endl;

return false;

if (!repoOverride.contains("brand")) {

qerr << "No brand set. Check your configuration." << endl;

return false;

}

bool opened = false;

QString outputDir = repoOverride.value("outputdir");

if (!outputDir.isEmpty()) outputDir.append("/");

QFile ks;

SsuVariables var;

if (repoOverride.contains("filename")) {

QString fileName = QString("%1%2")

.arg(outputDir)

.arg(replaceSpaces(var.resolveString(repoOverride.value("filename"),

&repoOverride)));

ks.setFileName(fileName);

opened = ks.open(QIODevice::WriteOnly);

} else {

qerr << "No filename specified, and no default filename configured" << endl;

return false;

}

ks.setFileName(outputDir + kickstart);

opened = ks.open(QIODevice::WriteOnly);

}

if (!opened) {

qerr << "Unable to write output file " << ks.fileName() << ": " << ks.errorString() << endl;

return false;

QString displayName = QString("# DisplayName: %1 %2/%3 (%4) %5")

.arg(repoOverride.value("brand"))

.arg(deviceInfo.deviceModel())

.arg(repoOverride.value("arch"))

.arg((rndMode ? "rnd"

: "release"))

.arg(repoOverride.value("version"));

QStringList featuresList = deviceInfo.value("img-features").toStringList();

// Add developer-mode feature to rnd images by default

if (rndMode)

featuresList << "developer-mode";

QString suggestedFeatures;

// work around some idiotic JS list parsing on our side by terminating one-element list by comma

if (featuresList.count() == 1)

suggestedFeatures = QString("# SuggestedFeatures: %1,")

else if (featuresList.count() > 1)

suggestedFeatures = QString("# SuggestedFeatures: %1")

QString imageType = "fs";

if (!deviceInfo.value("img-type").toString().isEmpty())

imageType = deviceInfo.value("img-type").toString();

QString imageArchitecture = "armv7hl";

if (!deviceInfo.value("img-arch").toString().isEmpty())

imageArchitecture = deviceInfo.value("img-arch").toString();

QString kickstartType = QString("# KickstartType: %1")

.arg((rndMode ? "rnd" : "release"));

kout.setDevice(&ks);

kout << displayName << endl;

kout << kickstartType << endl;

if (!suggestedFeatures.isEmpty())

kout << suggestedFeatures << endl;

kout << "# SuggestedImageType: " << imageType << endl;

kout << "# SuggestedArchitecture: " << imageArchitecture << endl << endl;

kout << commands().join("\n") << endl << endl;

foreach (const QString &section, commandSections) {

kout << commandSection(section).join("\n") << endl << endl;

}

// this allows simple search and replace postprocessing of the repos section

// to overcome shortcomings of the "keep image creation simple token based"

// approach

// TODO: QHash looks messy in the config, provide tool to edit it

QString repoSection = repos().join("\n");

if (deviceInfo.variable("kickstart-defaults", "urlPostprocess")

.canConvert(QMetaType::QVariantHash)) {

QHash<QString, QVariant> postproc =

deviceInfo.variable("kickstart-defaults", "urlPostprocess").toHash();

QHash<QString, QVariant>::const_iterator it = postproc.constBegin();

while (it != postproc.constEnd()) {

QRegExp regex(it.key(), Qt::CaseSensitive, QRegExp::RegExp2);

repoSection.replace(regex, it.value().toString());

it++;

}

}

kout << repoSection << endl << endl;

kout << packagesSection("packages").join("\n") << endl << endl;

kout << packagesSection("attachment").join("\n") << endl << endl;

// TODO: now that extending scriptlet section is might make sense to make it configurable

kout << scriptletSection("pre", Chroot).join("\n") << endl << endl;

kout << scriptletSection("post", Chroot).join("\n") << endl << endl;

kout << scriptletSection("post", NoChroot).join("\n") << endl << endl;

kout << scriptletSection("pack", DeviceSpecific).join("\n") << endl << endl;

// POST, die-on-error

return true;