119 lines
3.9 KiB
PHP
119 lines
3.9 KiB
PHP
<?php
|
|
// +----------------------------------------------------------------------
|
|
// | ThinkPHP [ WE CAN DO IT JUST THINK ]
|
|
// +----------------------------------------------------------------------
|
|
// | Copyright (c) 2006~2015 http://thinkphp.cn All rights reserved.
|
|
// +----------------------------------------------------------------------
|
|
// | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )
|
|
// +----------------------------------------------------------------------
|
|
// | Author: yunwuxin <448901948@qq.com>
|
|
// +----------------------------------------------------------------------
|
|
|
|
namespace think\console\helper;
|
|
|
|
use think\console\Output;
|
|
use think\Process as ThinkProcess;
|
|
use think\process\Builder as ProcessBuilder;
|
|
use think\process\exception\Failed as ProcessFailedException;
|
|
|
|
class Process extends Helper
|
|
{
|
|
|
|
/**
|
|
* 运行一个外部进程。
|
|
* @param Output $output 一个Output实例
|
|
* @param string|array|ThinkProcess $cmd 指令
|
|
* @param string|null $error 错误信息
|
|
* @param callable|null $callback 回调
|
|
* @param int $verbosity
|
|
* @return ThinkProcess
|
|
*/
|
|
public function run(Output $output, $cmd, $error = null, $callback = null, $verbosity = Output::VERBOSITY_VERY_VERBOSE)
|
|
{
|
|
/** @var Debug $formatter */
|
|
$formatter = $this->getHelperSet()->get('debug_formatter');
|
|
|
|
if (is_array($cmd)) {
|
|
$process = ProcessBuilder::create($cmd)->getProcess();
|
|
} elseif ($cmd instanceof ThinkProcess) {
|
|
$process = $cmd;
|
|
} else {
|
|
$process = new ThinkProcess($cmd);
|
|
}
|
|
|
|
if ($verbosity <= $output->getVerbosity()) {
|
|
$output->write($formatter->start(spl_object_hash($process), $this->escapeString($process->getCommandLine())));
|
|
}
|
|
|
|
if ($output->isDebug()) {
|
|
$callback = $this->wrapCallback($output, $process, $callback);
|
|
}
|
|
|
|
$process->run($callback);
|
|
|
|
if ($verbosity <= $output->getVerbosity()) {
|
|
$message = $process->isSuccessful() ? 'Command ran successfully' : sprintf('%s Command did not run successfully', $process->getExitCode());
|
|
$output->write($formatter->stop(spl_object_hash($process), $message, $process->isSuccessful()));
|
|
}
|
|
|
|
if (!$process->isSuccessful() && null !== $error) {
|
|
$output->writeln(sprintf('<error>%s</error>', $this->escapeString($error)));
|
|
}
|
|
|
|
return $process;
|
|
}
|
|
|
|
/**
|
|
* 运行指令
|
|
* @param Output $output
|
|
* @param string|ThinkProcess $cmd
|
|
* @param string|null $error
|
|
* @param callable|null $callback
|
|
* @return ThinkProcess
|
|
*/
|
|
public function mustRun(Output $output, $cmd, $error = null, $callback = null)
|
|
{
|
|
$process = $this->run($output, $cmd, $error, $callback);
|
|
|
|
if (!$process->isSuccessful()) {
|
|
throw new ProcessFailedException($process);
|
|
}
|
|
|
|
return $process;
|
|
}
|
|
|
|
/**
|
|
* 包装过程回调来添加调试输出
|
|
* @param Output $output
|
|
* @param ThinkProcess $process
|
|
* @param callable|null $callback
|
|
* @return callable
|
|
*/
|
|
public function wrapCallback(Output $output, ThinkProcess $process, $callback = null)
|
|
{
|
|
/** @var Debug $formatter */
|
|
$formatter = $this->getHelperSet()->get('debug_formatter');
|
|
|
|
return function ($type, $buffer) use ($output, $process, $callback, $formatter) {
|
|
$output->write($formatter->progress(spl_object_hash($process), $this->escapeString($buffer), ThinkProcess::ERR === $type));
|
|
|
|
if (null !== $callback) {
|
|
call_user_func($callback, $type, $buffer);
|
|
}
|
|
};
|
|
}
|
|
|
|
private function escapeString($str)
|
|
{
|
|
return str_replace('<', '\\<', $str);
|
|
}
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
public function getName()
|
|
{
|
|
return 'process';
|
|
}
|
|
}
|