PythonExecutor.cpp

/**
 * @file
 * Copyright (c) 2023 - Tidaly
 * Authors:
 *   - Philippe CHEYPE <philippe.cheype@epitech.eu>
 *   - Quentin ROUVIER <quentin.rouvier@epitech.eu>
 * NOTICE: All information contained herein is, and remains
 * the property of Tidaly. Dissemination of this information
 * or reproduction of this material is strictly forbidden
 * unless prior written permission is obtained from Tidaly.
 */

#include "Core/PythonExecutor/PythonExecutor.hpp"

std::string const PyExecutor::executeScript(std::string const& scriptPath, std::string const& imagePath, int lastValue) {
    int pipefd[2];
    // if (pipe(pipefd) == -1) {
    //     THROW_EXCEPTION(PythonExecutorError, "Failed to create the pipe.");
    // }

    pid_t pid = fork();
    // if (pid == -1) {
    //     THROW_EXCEPTION(PythonExecutorError, "Failed to fork the process.");
    // }

    if (pid == 0) {
        // Child process
        close(pipefd[0]);
        // if (dup2(pipefd[1], STDOUT_FILENO) == -1) {
        //     THROW_EXCEPTION(PythonExecutorError, "Failed to redirect stdout.");
        // }
        signal(SIGALRM, [](int) { exit(1); });
        alarm(60); // 60 seconds timeout
        execlp("python3", "python3", scriptPath.c_str(), imagePath.c_str(), std::to_string(lastValue).c_str(), nullptr);
        // THROW_EXCEPTION(PythonExecutorError, "Failed to execute the python script.");
        return "";
    } else {
        // Parent process
        close(pipefd[1]);
        std::array<char, 128> buffer;
        std::string result;

        while (read(pipefd[0], buffer.data(), buffer.size()) > 0) {
            result += buffer.data();
        }
        int status;
        waitpid(pid, &status, 0);
        this->_handleExitStatus(status);
        return result;
    }
}

void PyExecutor::_handleExitStatus(int returnCode) {
    if (returnCode != 0) {
        if (WIFEXITED(returnCode)) {
            int statusCode = WEXITSTATUS(returnCode);
            std::cerr << "Command exited with status " << statusCode << std::endl;
            THROW_EXCEPTION(PythonExecutorError, "Command execution failed with status code.");
        // } else if (WIFSIGNALED(returnCode)) {
        //     int signalNumber = WTERMSIG(returnCode);
        //     if (signalNumber == SIGALRM) {
        //         throw std::runtime_error("PythonExecutor: Command execution timed out.");
        //     }
        //     std::cerr << "Command terminated by signal " << signalNumber << std::endl;
        //     THROW_EXCEPTION(PythonExecutorError, "Command terminated unexpectedly by a signal.");
        // } else {
        //     std::cerr << "Command failed with return code " << returnCode << std::endl;
        //     THROW_EXCEPTION(PythonExecutorError, "Command execution failed.");
        }
    }
}