Files
framework/library/think/console/command/Make.php
2016-07-03 21:23:26 +08:00

123 lines
3.8 KiB
PHP

<?php
// +----------------------------------------------------------------------
// | ThinkPHP [ WE CAN DO IT JUST THINK ]
// +----------------------------------------------------------------------
// | Copyright (c) 2016 http://thinkphp.cn All rights reserved.
// +----------------------------------------------------------------------
// | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )
// +----------------------------------------------------------------------
// | Author: 刘志淳 <chun@engineer.com>
// +----------------------------------------------------------------------
namespace think\console\command;
use think\App;
use think\console\Input;
use think\console\input\Argument;
use think\console\input\Option;
use think\console\Output;
class Make extends Command
{
/**
* {@inheritdoc}
*/
protected function configure()
{
$this
->setName('make')
->setDescription('Create a new applcation class')
->addArgument('namespace', Argument::REQUIRED)
->addOption('layer', 'l', Option::VALUE_OPTIONAL, 'Layer Name', null)
->addOption('extend', 'e', Option::VALUE_OPTIONAL, 'Extend Base class', null);
}
protected function execute(Input $input, Output $output)
{
$namespace = $input->getArgument('namespace');
$extend = $input->getOption('extend');
if (!$layer = $input->getOption('layer')) {
// 自动识别layer
$item = explode('\\', $namespace);
$layer = basename(dirname(implode(DS, $item)));
}
$result = $this->getResult($layer, $namespace, '', $extend);
$output->writeln("output:" . $result);
}
// 创建文件
protected static function buildFile($file, $content)
{
if (is_file($file)) {
exception('file already exists');
}
if (!is_dir(dirname($file))) {
mkdir(strtolower(dirname($file)), 0777, true);
}
file_put_contents($file, $content);
}
// 生成类库文件
protected function build($namespace, $extend, $content = '')
{
$tpl = file_get_contents(THINK_PATH . 'tpl' . DS . 'make.tpl');
// comminute namespace
$namespace = explode('\\', $namespace);
$className = array_pop($namespace);
if ($extend) {
$extend = 'extends \\' . ltrim($extend, '\\');
}
// 处理内容
$content = str_replace(['{%extend%}', '{%className%}', '{%namespace%}', '{%content%}'],
[$extend, $className, implode('\\', $namespace), $content],
$tpl);
// 处理文件名
array_shift($namespace);
$file = APP_PATH . implode(DS, $namespace) . DS . $className . '.php';
// 生成类库文件
self::buildFile($file, $content);
return realpath($file);
}
protected function getResult($layer, $namespace, $module, $extend, $content = '')
{
// 处理命名空间
if (!empty($module)) {
$namespace = App::$namespace . "\\" . $module . "\\" . $layer . "\\" . $namespace;
}
// 处理继承
if (empty($extend)) {
switch ($layer) {
case 'model':
$extend = '\\think\\Model';
break;
case 'validate':
$extend = '\\think\Validate';
break;
case 'controller':
default:
$extend = '';
break;
}
} else {
if (!preg_match("/\\\/", $extend)) {
if (!empty($module)) {
$extend = "\\" . App::$namespace . "\\" . $module . "\\" . $layer . "\\" . $extend;
}
}
}
return $this->build($namespace, $extend, $content);
}
}