825 lines
23 KiB
PHP
Executable File
825 lines
23 KiB
PHP
Executable File
<?php
|
|
// +----------------------------------------------------------------------
|
|
// | TopThink [ WE CAN DO IT JUST THINK IT ]
|
|
// +----------------------------------------------------------------------
|
|
// | Copyright (c) 2015 http://www.topthink.com All rights reserved.
|
|
// +----------------------------------------------------------------------
|
|
// | Author: zhangyajun <448901948@qq.com>
|
|
// +----------------------------------------------------------------------
|
|
|
|
namespace think;
|
|
|
|
use think\console\Command;
|
|
use think\console\command\Help as HelpCommand;
|
|
use think\console\Input;
|
|
use think\console\input\Argument as InputArgument;
|
|
use think\console\input\Definition as InputDefinition;
|
|
use think\console\input\Option as InputOption;
|
|
use think\console\Output;
|
|
use think\console\output\driver\Buffer;
|
|
|
|
class Console
|
|
{
|
|
|
|
private $name;
|
|
private $version;
|
|
|
|
/** @var Command[] */
|
|
private $commands = [];
|
|
|
|
private $wantHelps = false;
|
|
|
|
private $catchExceptions = true;
|
|
private $autoExit = true;
|
|
private $definition;
|
|
private $defaultCommand;
|
|
|
|
private static $defaultCommands = [
|
|
'help' => "think\\console\\command\\Help",
|
|
'list' => "think\\console\\command\\Lists",
|
|
'build' => "think\\console\\command\\Build",
|
|
'clear' => "think\\console\\command\\Clear",
|
|
'make:command' => "think\\console\\command\\make\\Command",
|
|
'make:controller' => "think\\console\\command\\make\\Controller",
|
|
'make:model' => "think\\console\\command\\make\\Model",
|
|
'make:middleware' => "think\\console\\command\\make\\Middleware",
|
|
'make:validate' => "think\\console\\command\\make\\Validate",
|
|
'optimize:autoload' => "think\\console\\command\\optimize\\Autoload",
|
|
'optimize:config' => "think\\console\\command\\optimize\\Config",
|
|
'optimize:schema' => "think\\console\\command\\optimize\\Schema",
|
|
'optimize:route' => "think\\console\\command\\optimize\\Route",
|
|
'run' => "think\\console\\command\\RunServer",
|
|
'version' => "think\\console\\command\\Version",
|
|
'route:list' => "think\\console\\command\\RouteList",
|
|
];
|
|
|
|
/**
|
|
* Console constructor.
|
|
* @access public
|
|
* @param string $name 名称
|
|
* @param string $version 版本
|
|
* @param null|string $user 执行用户
|
|
*/
|
|
public function __construct($name = 'UNKNOWN', $version = 'UNKNOWN', $user = null)
|
|
{
|
|
$this->name = $name;
|
|
$this->version = $version;
|
|
|
|
if ($user) {
|
|
$this->setUser($user);
|
|
}
|
|
|
|
$this->defaultCommand = 'list';
|
|
$this->definition = $this->getDefaultInputDefinition();
|
|
}
|
|
|
|
/**
|
|
* 设置执行用户
|
|
* @param $user
|
|
*/
|
|
public function setUser($user)
|
|
{
|
|
if (DIRECTORY_SEPARATOR == '\\') {
|
|
return;
|
|
}
|
|
|
|
$user = posix_getpwnam($user);
|
|
if ($user) {
|
|
posix_setuid($user['uid']);
|
|
posix_setgid($user['gid']);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* 初始化 Console
|
|
* @access public
|
|
* @param bool $run 是否运行 Console
|
|
* @return int|Console
|
|
*/
|
|
public static function init($run = true)
|
|
{
|
|
static $console;
|
|
|
|
if (!$console) {
|
|
$config = Container::get('config')->pull('console');
|
|
$console = new self($config['name'], $config['version'], $config['user']);
|
|
|
|
$commands = $console->getDefinedCommands($config);
|
|
|
|
// 添加指令集
|
|
$console->addCommands($commands);
|
|
}
|
|
|
|
if ($run) {
|
|
// 运行
|
|
return $console->run();
|
|
} else {
|
|
return $console;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* @access public
|
|
* @param array $config
|
|
* @return array
|
|
*/
|
|
public function getDefinedCommands(array $config = [])
|
|
{
|
|
$commands = self::$defaultCommands;
|
|
|
|
if (!empty($config['auto_path']) && is_dir($config['auto_path'])) {
|
|
// 自动加载指令类
|
|
$files = scandir($config['auto_path']);
|
|
|
|
if (count($files) > 2) {
|
|
$beforeClass = get_declared_classes();
|
|
|
|
foreach ($files as $file) {
|
|
if (pathinfo($file, PATHINFO_EXTENSION) == 'php') {
|
|
include $config['auto_path'] . $file;
|
|
}
|
|
}
|
|
|
|
$afterClass = get_declared_classes();
|
|
$commands = array_merge($commands, array_diff($afterClass, $beforeClass));
|
|
}
|
|
}
|
|
|
|
$file = Container::get('env')->get('app_path') . 'command.php';
|
|
|
|
if (is_file($file)) {
|
|
$appCommands = include $file;
|
|
|
|
if (is_array($appCommands)) {
|
|
$commands = array_merge($commands, $appCommands);
|
|
}
|
|
}
|
|
|
|
return $commands;
|
|
}
|
|
|
|
/**
|
|
* @access public
|
|
* @param string $command
|
|
* @param array $parameters
|
|
* @param string $driver
|
|
* @return Output|Buffer
|
|
*/
|
|
public static function call($command, array $parameters = [], $driver = 'buffer')
|
|
{
|
|
$console = self::init(false);
|
|
|
|
array_unshift($parameters, $command);
|
|
|
|
$input = new Input($parameters);
|
|
$output = new Output($driver);
|
|
|
|
$console->setCatchExceptions(false);
|
|
$console->find($command)->run($input, $output);
|
|
|
|
return $output;
|
|
}
|
|
|
|
/**
|
|
* 执行当前的指令
|
|
* @access public
|
|
* @return int
|
|
* @throws \Exception
|
|
* @api
|
|
*/
|
|
public function run()
|
|
{
|
|
$input = new Input();
|
|
$output = new Output();
|
|
|
|
$this->configureIO($input, $output);
|
|
|
|
try {
|
|
$exitCode = $this->doRun($input, $output);
|
|
} catch (\Exception $e) {
|
|
if (!$this->catchExceptions) {
|
|
throw $e;
|
|
}
|
|
|
|
$output->renderException($e);
|
|
|
|
$exitCode = $e->getCode();
|
|
if (is_numeric($exitCode)) {
|
|
$exitCode = (int) $exitCode;
|
|
if (0 === $exitCode) {
|
|
$exitCode = 1;
|
|
}
|
|
} else {
|
|
$exitCode = 1;
|
|
}
|
|
}
|
|
|
|
if ($this->autoExit) {
|
|
if ($exitCode > 255) {
|
|
$exitCode = 255;
|
|
}
|
|
|
|
exit($exitCode);
|
|
}
|
|
|
|
return $exitCode;
|
|
}
|
|
|
|
/**
|
|
* 执行指令
|
|
* @access public
|
|
* @param Input $input
|
|
* @param Output $output
|
|
* @return int
|
|
*/
|
|
public function doRun(Input $input, Output $output)
|
|
{
|
|
if (true === $input->hasParameterOption(['--version', '-V'])) {
|
|
$output->writeln($this->getLongVersion());
|
|
|
|
return 0;
|
|
}
|
|
|
|
$name = $this->getCommandName($input);
|
|
|
|
if (true === $input->hasParameterOption(['--help', '-h'])) {
|
|
if (!$name) {
|
|
$name = 'help';
|
|
$input = new Input(['help']);
|
|
} else {
|
|
$this->wantHelps = true;
|
|
}
|
|
}
|
|
|
|
if (!$name) {
|
|
$name = $this->defaultCommand;
|
|
$input = new Input([$this->defaultCommand]);
|
|
}
|
|
|
|
$command = $this->find($name);
|
|
|
|
$exitCode = $this->doRunCommand($command, $input, $output);
|
|
|
|
return $exitCode;
|
|
}
|
|
|
|
/**
|
|
* 设置输入参数定义
|
|
* @access public
|
|
* @param InputDefinition $definition
|
|
*/
|
|
public function setDefinition(InputDefinition $definition)
|
|
{
|
|
$this->definition = $definition;
|
|
}
|
|
|
|
/**
|
|
* 获取输入参数定义
|
|
* @access public
|
|
* @return InputDefinition The InputDefinition instance
|
|
*/
|
|
public function getDefinition()
|
|
{
|
|
return $this->definition;
|
|
}
|
|
|
|
/**
|
|
* Gets the help message.
|
|
* @access public
|
|
* @return string A help message.
|
|
*/
|
|
public function getHelp()
|
|
{
|
|
return $this->getLongVersion();
|
|
}
|
|
|
|
/**
|
|
* 是否捕获异常
|
|
* @access public
|
|
* @param bool $boolean
|
|
* @api
|
|
*/
|
|
public function setCatchExceptions($boolean)
|
|
{
|
|
$this->catchExceptions = (bool) $boolean;
|
|
}
|
|
|
|
/**
|
|
* 是否自动退出
|
|
* @access public
|
|
* @param bool $boolean
|
|
* @api
|
|
*/
|
|
public function setAutoExit($boolean)
|
|
{
|
|
$this->autoExit = (bool) $boolean;
|
|
}
|
|
|
|
/**
|
|
* 获取名称
|
|
* @access public
|
|
* @return string
|
|
*/
|
|
public function getName()
|
|
{
|
|
return $this->name;
|
|
}
|
|
|
|
/**
|
|
* 设置名称
|
|
* @access public
|
|
* @param string $name
|
|
*/
|
|
public function setName($name)
|
|
{
|
|
$this->name = $name;
|
|
}
|
|
|
|
/**
|
|
* 获取版本
|
|
* @access public
|
|
* @return string
|
|
* @api
|
|
*/
|
|
public function getVersion()
|
|
{
|
|
return $this->version;
|
|
}
|
|
|
|
/**
|
|
* 设置版本
|
|
* @access public
|
|
* @param string $version
|
|
*/
|
|
public function setVersion($version)
|
|
{
|
|
$this->version = $version;
|
|
}
|
|
|
|
/**
|
|
* 获取完整的版本号
|
|
* @access public
|
|
* @return string
|
|
*/
|
|
public function getLongVersion()
|
|
{
|
|
if ('UNKNOWN' !== $this->getName() && 'UNKNOWN' !== $this->getVersion()) {
|
|
return sprintf('<info>%s</info> version <comment>%s</comment>', $this->getName(), $this->getVersion());
|
|
}
|
|
|
|
return '<info>Console Tool</info>';
|
|
}
|
|
|
|
/**
|
|
* 注册一个指令 （便于动态创建指令）
|
|
* @access public
|
|
* @param string $name 指令名
|
|
* @return Command
|
|
*/
|
|
public function register($name)
|
|
{
|
|
return $this->add(new Command($name));
|
|
}
|
|
|
|
/**
|
|
* 添加指令集
|
|
* @access public
|
|
* @param array $commands
|
|
*/
|
|
public function addCommands(array $commands)
|
|
{
|
|
foreach ($commands as $key => $command) {
|
|