<?php
/**
 * Pimcore
 *
 * This source file is available under two different licenses:
 * - GNU General Public License version 3 (GPLv3)
 * - Pimcore Enterprise License (PEL)
 * Full copyright and license information is available in
 * LICENSE.md which is distributed with this source code.
 *
 * @copyright  Copyright (c) 2009-2016 pimcore GmbH (http://www.pimcore.org)
 * @license    http://www.pimcore.org/license     GPLv3 and PEL
 */

namespace Pimcore\Tool;

use Pimcore\Config;
use Symfony\Component\Process\Exception\ProcessFailedException;
use Symfony\Component\Process\Process;

class Console
{
    /**
     * @var string system environment
     */
    private static $systemEnvironment;

    /**
     * @var null|bool
     */
    protected static $timeoutKillAfterSupport = null;

    /**
     * @var array
     */
    protected static $executableCache = [];

    /**
     * @static
     * @return string "windows" or "unix"
     */
    public static function getSystemEnvironment()
    {
        if (self::$systemEnvironment == null) {
            if (stripos(php_uname("s"), "windows") !== false) {
                self::$systemEnvironment = 'windows';
            } else {
                self::$systemEnvironment = 'unix';
            }
        }

        return self::$systemEnvironment;
    }

    /**
     * @param $name
     * @param bool $throwException
     * @return bool|mixed|string
     * @throws \Exception
     */
    public static function getExecutable($name, $throwException = false)
    {
        if (isset(self::$executableCache[$name])) {
            return self::$executableCache[$name];
        }

        $pathVariable = Config::getSystemConfig()->general->path_variable;

        $paths = [];
        if ($pathVariable) {
            $paths = explode(":", $pathVariable);
        }

        array_unshift($paths, "");

        // allow custom setup routines for certain programs
        $customSetupMethod = "setup" . ucfirst($name);
        if (method_exists(__CLASS__, $customSetupMethod)) {
            self::$customSetupMethod();
        }

        // allow custom check routines for certain programs
        $customCheckMethod = "check" . ucfirst($name);
        if (!method_exists(__CLASS__, $customCheckMethod)) {
            $customCheckMethod = "checkDummy";
        }

        foreach ($paths as $path) {
            foreach (["--help", "-h", "-help"] as $option) {
                try {
                    $path = rtrim($path, "/\\ ");
                    if ($path) {
                        $executablePath = $path . DIRECTORY_SEPARATOR . $name;
                    } else {
                        $executablePath = $name;
                    }

                    $process = new Process($executablePath . " " . $option);
                    $process->run();

                    if ($process->isSuccessful() || self::$customCheckMethod($process)) {
                        if (empty($path) && self::getSystemEnvironment() == "unix") {
                            // get the full qualified path, seems to solve a lot of problems :)
                            // if not using the full path, timeout, nohup and nice will fail
                            $fullQualifiedPath = shell_exec("which " . $executablePath);
                            $fullQualifiedPath = trim($fullQualifiedPath);
                            if ($fullQualifiedPath) {
                                $executablePath = $fullQualifiedPath;
                            }
                        }

                        self::$executableCache[$name] = $executablePath;

                        return $executablePath;
                    }
                } catch (\Exception $e) {
                }
            }
        }

        self::$executableCache[$name] = false;

        if ($throwException) {
            throw new \Exception("No '$name' executable found, please install the application or add it to the PATH (in system settings or to your PATH environment variable");
        }

        return false;
    }

    /**
     *
     */
    protected static function setupComposer()
    {
        // composer needs either COMPOSER_HOME or HOME to be set
        if (!getenv("COMPOSER_HOME") && !getenv("HOME")) {
            $composerHome = PIMCORE_WEBSITE_VAR . "/composer";
            if (!is_dir($composerHome)) {
                mkdir($composerHome, 0777, true);
            }
            putenv("COMPOSER_HOME=" . $composerHome);
        }

        putenv("COMPOSER_DISABLE_XDEBUG_WARN=true");
    }

    /**
     * @param $process
     * @return bool
     */
    protected static function checkPngout($process)
    {
        if (strpos($process->getOutput() . $process->getErrorOutput(), "bitdepth") !== false) {
            return true;
        }

        return false;
    }

    /**
     * @param $process
     * @return bool
     */
    protected static function checkCjpeg($process)
    {
        if (strpos($process->getOutput() . $process->getErrorOutput(), "-optimize") !== false) {
            return true;
        }

        return false;
    }

    /**
     * @param $process
     * @return bool
     */
    protected static function checkDummy($process)
    {
        return false;
    }

    /**
     * @return mixed
     * @throws \Exception
     */
    public static function getPhpCli()
    {
        return self::getExecutable("php", true);
    }

    /**
     * @return bool|string
     */
    public static function getTimeoutBinary()
    {
        return self::getExecutable("timeout");
    }

    /**
     * @param $script
     * @param $arguments
     * @return string
     */
    protected static function buildPhpScriptCmd($script, $arguments)
    {
        $phpCli = Console::getPhpCli();

        $cmd = $phpCli . " " . $script;

        if (Config::getEnvironment()) {
            $cmd .= " --environment=" . Config::getEnvironment();
        }

        if (!empty($arguments)) {
            $cmd .= " " . $arguments;
        }

        return $cmd;
    }

    /**
     * @param $script
     * @param $arguments
     * @param $outputFile
     * @param $timeout
     * @return string
     */
    public static function runPhpScript($script, $arguments = "", $outputFile = null, $timeout = null)
    {
        $cmd = self::buildPhpScriptCmd($script, $arguments);
        $return = Console::exec($cmd, $outputFile, $timeout);

        return $return;
    }

    /**
     * @param $script
     * @param $arguments
     * @param $outputFile
     * @return string
     */
    public static function runPhpScriptInBackground($script, $arguments = "", $outputFile = null)
    {
        $cmd = self::buildPhpScriptCmd($script, $arguments);
        $return = Console::execInBackground($cmd, $outputFile);

        return $return;
    }

    /**
     * @param $cmd
     * @param null $outputFile
     * @param null $timeout
     * @return string
     */
    public static function exec($cmd, $outputFile = null, $timeout = null)
    {
        if ($timeout && self::getTimeoutBinary()) {

            // check if --kill-after flag is supported in timeout
            if (self::$timeoutKillAfterSupport === null) {
                $out = self::exec(self::getTimeoutBinary() . " --help");
                if (strpos($out, "--kill-after")) {
                    self::$timeoutKillAfterSupport = true;
                } else {
                    self::$timeoutKillAfterSupport = false;
                }
            }

            $killAfter = "";
            if (self::$timeoutKillAfterSupport) {
                $killAfter = " -k 1m";
            }

            $cmd = self::getTimeoutBinary() . $killAfter . " " . $timeout . "s " . $cmd;
        } elseif ($timeout) {
            \Logger::warn("timeout binary not found, executing command without timeout");
        }

        if ($outputFile) {
            $cmd = $cmd . " > ". $outputFile ." 2>&1";
        } else {
            // send stderr to /dev/null otherwise this goes to the apache error log and can fill it up pretty quickly
            if (self::getSystemEnvironment() != 'windows') {
                $cmd .= " 2> /dev/null";
            }
        }

        \Logger::debug("Executing command `" . $cmd . "` on the current shell");
        $return = shell_exec($cmd);

        return $return;
    }

    /**
     * @static
     * @param string $cmd
     * @param null|string $outputFile
     * @return int
     */
    public static function execInBackground($cmd, $outputFile = null)
    {

        // windows systems
        if (self::getSystemEnvironment() == 'windows') {
            return self::execInBackgroundWindows($cmd, $outputFile);
        } else {
            return self::execInBackgroundUnix($cmd, $outputFile);
        }
    }

    /**
     * @static
     * @param string $cmd
     * @param string $outputFile
     * @return int
     */
    protected static function execInBackgroundUnix($cmd, $outputFile)
    {
        if (!$outputFile) {
            $outputFile = "/dev/null";
        }

        $nice = (string) self::getExecutable("nice");
        if ($nice) {
            $nice .= " -n 19 ";
        }

        $nohup = (string) self::getExecutable("nohup");
        if ($nohup) {
            $nohup .= " ";
        }

        $commandWrapped = $nohup . $nice . $cmd . " > ". $outputFile ." 2>&1 & echo $!";
        \Logger::debug("Executing command `" . $commandWrapped . "´ on the current shell in background");
        $pid = shell_exec($commandWrapped);

        \Logger::debug("Process started with PID " . $pid);

        return $pid;
    }

    /**
     * @static
     * @param string $cmd
     * @param string $outputFile
     * @return int
     */
    protected static function execInBackgroundWindows($cmd, $outputFile)
    {
        if (!$outputFile) {
            $outputFile = "NUL";
        }

        $commandWrapped = "cmd /c " . $cmd . " > ". $outputFile . " 2>&1";
        \Logger::debug("Executing command `" . $commandWrapped . "´ on the current shell in background");

        $WshShell = new \COM("WScript.Shell");
        $WshShell->Run($commandWrapped, 0, false);
        \Logger::debug("Process started - returning the PID is not supported on Windows Systems");

        return 0;
    }

    /**
     * Returns a hash with all options passed to a cli script
     *
     * @return array
     */
    public static function getOptions($onlyFullNotationArgs = false)
    {
        global $argv;
        $options = [];
        $tmpOptions = $argv;
        array_shift($tmpOptions);

        foreach ($tmpOptions as $optionString) {
            if ($onlyFullNotationArgs && substr($optionString, 0, 2) != '--') {
                continue;
            }
            $exploded = explode("=", $optionString, 2);
            $options[str_replace('-', '', $exploded[0])] =  $exploded[1];
        }

        return $options;
    }

    /**
     * @param $options
     * @param string $concatenator
     * @param string $arrayConcatenator
     * @return string
     */
    public static function getOptionString($options, $concatenator = '=', $arrayConcatenator = ',')
    {
        $string = '';

        foreach ($options as $key => $value) {
            $string .= '--' . $key;
            if ($value) {
                if (is_array($value)) {
                    $value = implode($arrayConcatenator, $value);
                }
                $string .= $concatenator . "'" . $value . "'";
            }
            $string .= ' ';
        }

        return $string;
    }

    /**
     * @param array $allowedUsers
     * @throws \Exception
     */
    public static function checkExecutingUser($allowedUsers = [])
    {
        $configFile = \Pimcore\Config::locateConfigFile("system.php");
        $owner = fileowner($configFile);
        if ($owner == false) {
            throw new \Exception("Couldn't get user from file " . $configFile);
        }
        $userData = posix_getpwuid($owner);
        $allowedUsers[] = $userData['name'];

        $scriptExecutingUserData = posix_getpwuid(posix_geteuid());
        $scriptExecutingUser = $scriptExecutingUserData['name'];

        if (!in_array($scriptExecutingUser, $allowedUsers)) {
            throw new \Exception("The current system user is not allowed to execute this script. Allowed users: '" . implode(',', $allowedUsers) ."' Executing user: '$scriptExecutingUser'.");
        }
    }

    /**
     * @throws \Exception
     */
    public static function checkCliExecution()
    {
        if (php_sapi_name() != 'cli') {
            throw new \Exception("Script execution is restricted to CLI");
        }
    }
}
