RedPanda-CPP/RedPandaIDE/compiler/executablerunner.cpp

#include "executablerunner.h"

#include <QProcess>
#include <windows.h>
#include <QDebug>
#include "compilermanager.h"
#include "../settings.h"
#include "../systemconsts.h"

ExecutableRunner::ExecutableRunner(const QString &filename, const QString &arguments, const QString &workDir):
    QThread(),
    mFilename(filename),
    mArguments(arguments),
    mWorkDir(workDir),
    mStop(false),
    mRedirectConsoleProgram(false)
{

}

void ExecutableRunner::stop()
{
    mStop = true;
}

bool ExecutableRunner::redirectConsoleProgram() const
{
    return mRedirectConsoleProgram;
}

void ExecutableRunner::setRedirectConsoleProgram(bool newRedirectConsoleProgram)
{
    mRedirectConsoleProgram = newRedirectConsoleProgram;
}

const QString &ExecutableRunner::redirectInputFilename() const
{
    return mRedirectInputFilename;
}

void ExecutableRunner::setRedirectInputFilename(const QString &newDataFile)
{
    mRedirectInputFilename = newDataFile;
}

void ExecutableRunner::run()
{
    emit started();
    QProcess process;
    mStop = false;
    bool errorOccurred = false;

    process.setProgram(mFilename);
    process.setArguments(QProcess::splitCommand(mArguments));
    process.setWorkingDirectory(mWorkDir);
    QProcessEnvironment env = QProcessEnvironment::systemEnvironment();
    QString path = env.value("PATH");
    QStringList pathAdded;
    if (pSettings->compilerSets().defaultSet()) {
        foreach(const QString& dir, pSettings->compilerSets().defaultSet()->binDirs()) {
            pathAdded.append(dir);
        }
    }
    pathAdded.append(pSettings->dirs().app());
    if (!path.isEmpty()) {
        path+= PATH_SEPARATOR + pathAdded.join(PATH_SEPARATOR);
    } else {
        path = pathAdded.join(PATH_SEPARATOR);
    }
    env.insert("PATH",path);
    process.setProcessEnvironment(env);
    if (redirectConsoleProgram()) {
        process.setCreateProcessArgumentsModifier([](QProcess::CreateProcessArguments * args){
            args->flags |= CREATE_NEW_CONSOLE;
        });
    } else {
        process.setCreateProcessArgumentsModifier([](QProcess::CreateProcessArguments * args){
            args->flags |= CREATE_NEW_CONSOLE;
            args->startupInfo -> dwFlags &= ~STARTF_USESTDHANDLES;
        });
    }
    process.connect(&process, &QProcess::errorOccurred,
                    [&](){
                        errorOccurred= true;
                    });
//    qDebug() << mFilename;
//    qDebug() << QProcess::splitCommand(mArguments);
    if (!redirectConsoleProgram()) {
        process.closeWriteChannel();
    }
    process.start();
    process.waitForStarted(5000);
    if (process.state()==QProcess::Running && redirectConsoleProgram()) {
        process.write(ReadFileToByteArray(redirectInputFilename()));
        process.closeWriteChannel();
    }
    while (true) {
        process.waitForFinished(1000);
        if (process.state()!=QProcess::Running) {
            break;
        }
        if (mStop) {
            process.closeReadChannel(QProcess::StandardOutput);
            process.closeReadChannel(QProcess::StandardError);
            process.closeWriteChannel();
            process.terminate();
            process.kill();
            break;
        }
        if (errorOccurred)
            break;
    }
    if (errorOccurred) {
        //qDebug()<<"process error:"<<process.error();
        switch (process.error()) {
        case QProcess::FailedToStart:
            emit runErrorOccurred(tr("The runner process '%1' failed to start.").arg(mFilename));
            break;
//        case QProcess::Crashed:
//            if (!mStop)
//                emit runErrorOccurred(tr("The runner process crashed after starting successfully."));
//            break;
        case QProcess::Timedout:
            emit runErrorOccurred(tr("The last waitFor...() function timed out."));
            break;
        case QProcess::WriteError:
            emit runErrorOccurred(tr("An error occurred when attempting to write to the runner process."));
            break;
        case QProcess::ReadError:
            emit runErrorOccurred(tr("An error occurred when attempting to read from the runner process."));
            break;
        default:
            break;
        }
    }
    emit terminated();
}