初始化项目

core/library/think/console/helper/Process.php

@@ -0,0 +1,118 @@
+<?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';
+    }
+}