1 files changed, 228 insertions, 0 deletions
diff --git a/src/b2qt-flashing-wizard/scriptwriter.cpp b/src/b2qt-flashing-wizard/scriptwriter.cpp
new file mode 100644
index 0000000..59996ed
--- /dev/null
+++ b/src/b2qt-flashing-wizard/scriptwriter.cpp
@@ -0,0 +1,228 @@
+/****************************************************************************
+**
+** Copyright (C) 2014 Digia Plc
+** All rights reserved.
+** For any questions to Digia, please use the contact form at
+** http://qt.digia.com/
+**
+** This file is part of Qt Enterprise Embedded.
+**
+** Licensees holding valid Qt Enterprise licenses may use this file in
+** accordance with the Qt Enterprise License Agreement provided with the
+** Software or, alternatively, in accordance with the terms contained in
+** a written agreement between you and Digia.
+**
+** If you have questions regarding the use of this file, please use
+** the contact form at http://qt.digia.com/
+**
+****************************************************************************/
+
+#include "scriptwriter.h"
+#include <QPair>
+#include <QStringList>
+#include <QFileInfo>
+#include <QDebug>
+#include <QTimer>
+#include <QProcessEnvironment>
+#include <QElapsedTimer>
+#include <sys/stat.h>
+#include "elevator.h"
+#include "common.h"
+
+ScriptWriter::ScriptWriter(QObject *parent)
+    : Actor(parent)
+    , mElapsed(new QElapsedTimer)
+    , mDebug(false)
+{
+    mProcess.setProcessChannelMode(QProcess::MergedChannels);
+    mDebug = qEnvironmentVariableIsSet("DEBUG");
+}
+
+ScriptWriter::~ScriptWriter()
+{
+    delete mElapsed;
+}
+
+void ScriptWriter::setScriptFile(const QString &fileName)
+{
+    mScriptName = fileName;
+}
+
+bool ScriptWriter::ready(QString &error) const
+{
+    if (mScriptName.isEmpty()) {
+        error = "File name is empty";
+        return false;
+    }
+
+    QFileInfo file(mScriptName);
+    if (!file.exists()) {
+        error = "File does not exist: " + mScriptName;
+        return false;
+    }
+    if (!file.isFile()) {
+        error = "File " + mScriptName + " is not a file";
+        return false;
+    }
+    if (!file.isReadable()) {
+        error = "File is not readable";
+        return false;
+    }
+    if (!file.isExecutable()) {
+        error = "File is not executable";
+        return false;
+    }
+    if (file.size() == 0) {
+        error = "File is empty";
+        return false;
+    }
+
+    return true;
+}
+
+void ScriptWriter::start()
+{
+    connect(&mProcess, &QProcess::readyReadStandardOutput, this, &ScriptWriter::readOutput);
+    connect(&mProcess, (void (QProcess::*)(QProcess::ProcessError))&QProcess::error, this, &ScriptWriter::processError);
+    connect(&mProcess, (void (QProcess::*)(int, QProcess::ExitStatus))&QProcess::finished, this, &ScriptWriter::processFinished);
+
+    // Due to some random convenience output in the deploy scripts "set -x" has to be used to synchronize
+    QStringList args = elevate() << "/bin/sh" << "-x" << mScriptName << mAdditionalArgs;
+    qDebug() << "Executing" << args;
+
+/*    QProcessEnvironment pe ;
+    QString var = qgetenv("XDG_SESSION_COOKIE");
+    pe.insert("XDG_SESSION_COOKIE", var);
+    var = qgetenv("HOME");
+    pe.insert("HOME", var);
+    mProcess.setProcessEnvironment(pe);
+    */
+    mProcess.setProcessEnvironment(QProcessEnvironment::systemEnvironment());
+
+    mProcess.start(args.takeFirst(), args);
+    if (!mProcess.waitForStarted())
+        qFatal("Failed to start script");
+    mElapsed->start();
+    mProcess.write("y\n");
+}
+
+void ScriptWriter::readOutput()
+{
+    QByteArray ba = mProcess.readAllStandardOutput();
+    QList<QByteArray> baList = ba.split('\n');
+
+    static unsigned int currentProgress = 0;
+    unsigned int nextProgress = 0;
+    unsigned int milliseconds = 0;
+
+    unsigned int whole = 0;
+    if (!mTimings.isEmpty())
+        whole = mTimings.last().second;
+
+    foreach (const QByteArray &line, baList) {
+        if (line.startsWith("+")) {
+            // Command line
+            while (!mTimings.isEmpty() && !mTimings.first().first)
+                mTimings.removeFirst();
+
+            // Should not happen
+            if (mTimings.isEmpty())
+                break;
+
+            currentProgress = mTimings.takeFirst().second;
+        } else {
+            // Output line
+            if (!mTimings.isEmpty() && !mTimings.first().first)
+                currentProgress = mTimings.takeFirst().second;
+        }
+    }
+
+    if (!mTimings.isEmpty()) {
+        nextProgress = mTimings.first().second;
+        milliseconds = mTimings.last().second / 100 * (nextProgress - currentProgress);
+    } else {
+        nextProgress = currentProgress;
+    }
+
+    double slowness = ((double)currentProgress / mElapsed->elapsed()) * (whole/100.0);
+//    if (slowness == 0)
+        slowness = 1;
+
+    if (mDebug) {
+        qDebug() << ba;
+        qDebug() << "slowness:" << slowness << "currentProgress:" << currentProgress << "elapsed:" << mElapsed->elapsed() << "milliseconds:" << milliseconds << "nextProgress:" << nextProgress << "whole:" << whole << "stack:" << mTimings.size();
+    }
+
+    if (currentProgress > 100)
+        currentProgress = 100;
+    if (nextProgress > 100)
+        nextProgress = 100;
+
+    if (currentProgress || nextProgress)
+        emit progress(currentProgress, nextProgress, milliseconds/slowness);
+
+    emit details(ba);
+}
+
+void ScriptWriter::processFinished(int exitCode, QProcess::ExitStatus exitStatus)
+{
+    if (exitStatus != QProcess::NormalExit) {
+        qWarning("process crashed");
+        return;
+    }
+
+    if (exitCode != 0) {
+        qWarning("process failed");
+        return;
+    }
+    emit finished();
+}
+
+void ScriptWriter::processError(QProcess::ProcessError error)
+{
+    qWarning() << "processError" << error;
+}
+
+void ScriptWriter::setEnvironment(const QString &key, const QString &value)
+{
+    QProcessEnvironment env = mProcess.processEnvironment();
+    env.insert(key, value);
+    mProcess.setProcessEnvironment(env);
+}
+
+void ScriptWriter::setProgressFile(const QString &fileName)
+{
+    mTimings.clear();
+
+    QFile f(fileName);
+    if (!f.open(QFile::ReadOnly)) {
+        qWarning() << "Could not open progress file";
+        return;
+    }
+
+    bool ok;
+
+    while (!f.atEnd()) {
+        QByteArray line = f.readLine().simplified();
+
+        int i = line.toInt(&ok);
+        if (!ok) {
+            qWarning() << "Could not parse progress file:" << line;
+            mTimings.clear();
+            return;
+        }
+        if (i < 0)
+            mTimings += qMakePair(true, -i);
+        else if (i == 0)
+            mTimings += qMakePair(line.startsWith('-'), 0);
+        else
+            mTimings += qMakePair(false, i);
+    }
+
+    qDebug() << "steps" << mTimings.size();
+}
+
+void ScriptWriter::setAdditionalArgs(const QStringList &args)
+{
+    mAdditionalArgs = args;
+}