FastAdmin/application/admin/command/Crud.php

<?php

namespace app\admin\command;

use fast\Form;
use think\Config;
use think\console\Command;
use think\console\Input;
use think\console\input\Option;
use think\console\Output;
use think\Db;
use think\Exception;
use think\Lang;

class Crud extends Command
{

    protected function configure()
    {
        $this
                ->setName('crud')
                ->addOption('table', 't', Option::VALUE_REQUIRED, 'table name without prefix', null)
                ->addOption('controller', 'c', Option::VALUE_OPTIONAL, 'controller name', null)
                ->addOption('model', 'm', Option::VALUE_OPTIONAL, 'model name', null)
                ->addOption('force', 'f', Option::VALUE_OPTIONAL, 'force override', null)
                ->addOption('local', 'l', Option::VALUE_OPTIONAL, 'local model', 1)
                ->setDescription('Build CRUD controller and model from table');
    }

    protected function execute(Input $input, Output $output)
    {
        $adminPath = dirname(__DIR__) . DS;
        //表名
        $table = $input->getOption('table') ? : '';
        //自定义控制器
        $controller = $input->getOption('controller');
        //自定义模型
        $model = $input->getOption('model');
        //强制覆盖
        $force = $input->getOption('force');
        //是否为本地model,为0时表示为全局model将会把model放在app/common/model中
        $local = $input->getOption('local');
        if (!$table)
        {
            throw new Exception('table name can\'t empty');
        }
        $dbname = Config::get('database.database');
        $prefix = Config::get('database.prefix');
        $tableName = $prefix . $table;
        $tableInfo = Db::query("SHOW TABLE STATUS LIKE '{$tableName}'", [], TRUE);
        if (!$tableInfo)
        {
            throw new Exception("table not found");
        }
        $tableInfo = $tableInfo[0];

        //根据表名匹配对应的Fontawesome图标
        $iconPath = ROOT_PATH . '/public/assets/libs/font-awesome/less/variables.less';
        $iconName = is_file($iconPath) && stripos(file_get_contents($iconPath), '@fa-var-' . $table . ':') ? $table : 'fa fa-circle-o';

        //控制器默认以表名进行处理,以下划线进行分隔,如果需要自定义则需要传入controller,格式为目录层级
        $controllerArr = !$controller ? explode('_', strtolower($table)) : explode('/', strtolower($controller));
        $controllerUrl = implode('/', $controllerArr);
        $controllerName = ucfirst(array_pop($controllerArr));
        $controllerDir = implode('/', $controllerArr);
        $controllerFile = ($controllerDir ? $controllerDir . '/' : '') . $controllerName . '.php';

        //非覆盖模式时如果存在控制器文件则报错
        if (is_file($controllerFile) && !$force)
        {
            throw new Exception('controller already exists');
        }

        //模型默认以表名进行处理,以下划线进行分隔,如果需要自定义则需要传入model,不支持目录层级
        if (!$model)
        {
            $modelarr = explode('_', strtolower($table));
            foreach ($modelarr as $k => &$v)
                $v = ucfirst($v);
            unset($v);
            $modelName = implode('', $modelarr);
        }
        else
        {
            $modelName = ucfirst($model);
        }
        $modelFile = ($local ? $adminPath : APP_PATH . 'common' . DS) . 'model' . DS . $modelName . '.php';

        //非覆盖模式时如果存在模型文件则报错
        if (is_file($modelFile) && !$force)
        {
            throw new Exception('model already exists');
        }

        //从数据库中获取表字段信息
        $columnList = Db::query("SELECT * FROM `information_schema`.`columns` WHERE TABLE_SCHEMA = ? AND table_name = ? ORDER BY ORDINAL_POSITION", [$dbname, $tableName]);
        $fields = [];
        foreach ($columnList as $k => $v)
        {
            $fields[] = $v['COLUMN_NAME'];
        }

        $addList = [];
        $editList = [];
        $javascriptList = [];
        $langList = [];
        $field = 'id';
        $order = 'id';

        //循环所有字段,开始构造视图的HTML和JS信息
        foreach ($columnList as $k => $v)
        {
            $field = $v['COLUMN_NAME'];
            $fieldLang = ucfirst($field);
            // 语言列表
            if ($v['COLUMN_COMMENT'] != '')
            {
                $langList[] = $this->getLangItem($field, $v['COLUMN_COMMENT']);
            }
            if ($v['COLUMN_KEY'] != 'PRI')
            {
                $inputType = 'text';
                $step = 0;
                switch ($v['DATA_TYPE'])
                {
                    case 'bigint':
                    case 'int':
                    case 'mediumint':
                    case 'smallint':
                    case 'tinyint':
                        $inputType = 'number';
                        break;
                    case 'enum':
                    case 'set':
                        $inputType = 'select';
                        break;
                    case 'decimal':
                    case 'double':
                    case 'float':
                        $inputType = 'number';
                        $step = "0." . str_repeat(0, $v['NUMERIC_SCALE'] - 1) . "1";
                    case 'text':
                        $inputType = 'textarea';
                    default:
                        break;
                }
                if (substr($field, -4) == 'time')
                {
                    $inputType = 'datetime';
                }

                if ($inputType == 'select')
                {
                    $itemlist = substr($v['COLUMN_TYPE'], strlen($v['DATA_TYPE']) + 1, -1);
                    $itemlist = str_replace("'", '', $itemlist);
                    $attr = "'id'=>'c-$field','class'=>'form-control selectpicker'";
                    if ($v['DATA_TYPE'] == 'enum')
                    {
                        $attr .= ",'multiple'=>''";
                    }
                    if ($v['COLUMN_DEFAULT'] == '')
                    {
                        $attr .= ",'required'=>''";
                    }
                    $formAddElement = "{:build_select('row[$field]', '{$itemlist}', null, [{$attr}])}";
                    $formEditElement = "{:build_select('row[$field]', '{$itemlist}', \$row['$field'], [{$attr}])}";
                }
                else
                {
                    //CSS类名
                    $cssClass = ['form-control'];
                    $cssClass[] = substr($field, -4) == 'time' ? 'datetimepicker' : '';
                    $cssClass[] = $v['DATA_TYPE'] == 'text' ? 'summernote' : '';
                    $cssClass[] = substr($field, -3) == '_id' ? 'typeahead' : '';
                    $cssClass[] = substr($field, -4) == '_ids' ? 'tagsinput' : '';
                    $cssClass = array_filter($cssClass);
                    //因为有自动完成可输入其它内容
                    if (array_intersect($cssClass, ['typeahead', 'tagsinput']))
                    {
                        $inputType = 'text';
                        $step = 0;
                    }
                    $attr = ['id' => "c-{$field}", 'class' => implode(' ', $cssClass)];
                    if ($step)
                    {
                        $attr['step'] = $step;
                    }
                    //如果是图片则额外附加
                    if (substr($field, -5) == 'image' || substr($field, -6) == 'avatar')
                    {
                        $attr['data-plupload-id'] = "plupload-{$field}-text";
                        $attr['size'] = 50;
                    }
                    $fieldFunc = substr($field, -4) == 'time' ? "|datetime" : "";
                    if ($inputType == 'textarea')
                    {
                        $formAddElement = Form::textarea("row[{$field}]", $v['COLUMN_DEFAULT'], $attr);
                        $formEditElement = Form::textarea("row[{$field}]", "{\$row.{$field}{$fieldFunc}}", $attr);
                    }
                    else
                    {
                        $formAddElement = Form::input($inputType, "row[{$field}]", $v['COLUMN_DEFAULT'], $attr);
                        $formEditElement = Form::input($inputType, "row[{$field}]", "{\$row.{$field}{$fieldFunc}}", $attr);
                    }

                    if (substr($field, -5) == 'image' || substr($field, -6) == 'avatar')
                    {
                        //如果是图片或头像
                        $formAddElement = $this->getImageUpload($field, $formAddElement);
                        $formEditElement = $this->getImageUpload($field, $formEditElement);
                    }
                    else if ($field == 'status')
                    {
                        //如果是状态字段
                        $formAddElement = "{:build_radios('row[status]', ['normal'=>__('Normal'), 'hidden'=>__('Hidden')])}";
                        $formEditElement = "{:build_radios('row[status]', ['normal'=>__('Normal'), 'hidden'=>__('Hidden')], \$row['status'])}";
                    }
                }
                //构造添加和编辑HTML信息
                $addList[] = $this->getFormGroup($field, $formAddElement);
                $editList[] = $this->getFormGroup($field, $formEditElement);
            }

            //过滤text类型字段
            if ($v['DATA_TYPE'] != 'text')
            {
                //主键
                if ($v['COLUMN_KEY'] == 'PRI')
                {
                    $javascriptList[] = "{field: 'state', checkbox: true}";
                }
                //构造JS列信息
                $javascriptList[] = $this->getJsColumn($field);
                //排序方式,如果有weigh则按weigh,否则按主键排序
                $order = $field == 'weigh' ? 'weigh' : $order;
            }
        }
        //JS最后一列加上操作列
        $javascriptList[] = str_repeat(" ", 24) . "{field: 'operate', title: __('Operate'), events: Table.api.events.operate, formatter: Table.api.formatter.operate}";
        $addList = implode("\n", array_filter($addList));
        $editList = implode("\n", array_filter($editList));
        $javascriptList = implode(",\n", array_filter($javascriptList));
        $langList = implode(",\n", array_filter($langList));

        //表注释
        $tableComment = $tableInfo['Comment'];
        $tableComment = mb_substr($tableComment, -1) == '表' ? mb_substr($tableComment, 0, -1) . '管理' : $tableComment;

        //最终将生成的文件路径
        $controllerFile = $adminPath . 'controller' . DS . $controllerFile;
        $javascriptFile = ROOT_PATH . 'public' . DS . 'assets' . DS . 'js' . DS . 'backend' . DS . $controllerUrl . '.js';
        $addFile = $adminPath . 'view' . DS . $controllerUrl . DS . 'add.html';
        $editFile = $adminPath . 'view' . DS . $controllerUrl . DS . 'edit.html';
        $indexFile = $adminPath . 'view' . DS . $controllerUrl . DS . 'index.html';
        $langFile = $adminPath . 'lang' . DS . Lang::detect() . DS . $controllerUrl . '.php';

        $appNamespace = Config::get('app_namespace');
        $moduleName = 'admin';
        $controllerNamespace = "{$appNamespace}\\{$moduleName}\\controller" . ($controllerDir ? "\\" : "") . str_replace('/', "\\", $controllerDir);
        $modelNamespace = "{$appNamespace}\\" . ($local ? $moduleName : "common") . "\\model";

        $data = [
            'controllerNamespace'     => $controllerNamespace,
            'modelNamespace'          => $modelNamespace,
            'controllerUrl'           => $controllerUrl,
            'controllerDir'           => $controllerDir,
            'controllerName'          => $controllerName,
            'modelName'               => $modelName,
            'tableComment'            => $tableComment,
            'iconName'                => $iconName,
            'order'                   => $order,
            'table'                   => $table,
            'tableName'               => $tableName,
            'addList'                 => $addList,
            'editList'                => $editList,
            'javascriptList'          => $javascriptList,
            'langList'                => $langList,
            'modelAutoWriteTimestamp' => in_array('createtime', $fields) || in_array('updatetime', $fields) ? "'int'" : 'false',
            'createTime'              => in_array('createtime', $fields) ? "'createtime'" : 'false',
            'updateTime'              => in_array('updatetime', $fields) ? "'updatetime'" : 'false',
        ];

        // 生成控制器文件
        $result = $this->writeToFile('controller', $data, $controllerFile);
        // 生成模型文件
        $result = $this->writeToFile('model', $data, $modelFile);
        // 生成视图文件
        $result = $this->writeToFile('add', $data, $addFile);
        $result = $this->writeToFile('edit', $data, $editFile);
        $result = $this->writeToFile('index', $data, $indexFile);
        // 生成JS文件
        $result = $this->writeToFile('javascript', $data, $javascriptFile);
        // 生成语言文件
        if ($langList)
        {
            $result = $this->writeToFile('lang', $data, $langFile);
        }
        $output->writeln("<info>Build Successed</info>");
    }

    /**
     * 写入到文件
     * @param string $name
     * @param array $data
     * @param string $pathname
     * @return mixed
     */
    protected function writeToFile($name, $data, $pathname)
    {
        $search = $replace = [];
        foreach ($data as $k => $v)
        {
            $search[] = "{%{$k}%}";
            $replace[] = $v;
        }
        $stub = file_get_contents($this->getStub($name));
        $content = str_replace($search, $replace, $stub);

        if (!is_dir(dirname($pathname)))
        {
            mkdir(strtolower(dirname($pathname)), 0755, true);
        }
        return file_put_contents($pathname, $content);
    }

    /**
     * 获取基础模板
     * @param string $name
     * @return string
     */
    protected function getStub($name)
    {
        return __DIR__ . '/Crud/stubs/' . $name . '.stub';
    }

    protected function getLangItem($field, $content)
    {
        if (!Lang::has($field))
        {
            return <<<EOD
    '{$field}'  =>  '{$content}'
EOD;
        }
        else
        {
            return '';
        }
    }

    /**
     * 获取表单分组数据
     * @param string $field
     * @param string $content
     * @return string
     */
    protected function getFormGroup($field, $content)
    {
        $langField = ucfirst($field);
        return<<<EOD
    <div class="form-group">
        <label for="c-{$field}" class="control-label col-xs-12 col-sm-2">{:__('{$langField}')}:</label>
        <div class="col-xs-12 col-sm-8">
            {$content}
        </div>
    </div>
EOD;
    }

    /**
     * 获取图片模板数据
     * @param string $field
     * @param string $content
     * @return array
     */
    protected function getImageUpload($field, $content)
    {
        return <<<EOD
<div class="form-inline">
                {$content}
                <span><button id="plupload-{$field}" class="btn btn-danger plupload" ><i class="fa fa-upload"></i> {:__('Upload')}</button></span>
            </div>
EOD;
    }

    /**
     * 获取JS列数据
     * @param string $field
     * @return string
     */
    protected function getJsColumn($field)
    {
        $lang = ucfirst($field);
        $html = str_repeat(" ", 24) . "{field: '{$field}', title: __('{$lang}')";
        $formatter = '';
        if ($field == 'status')
            $formatter = 'status';
        else if ($field == 'icon')
            $formatter = 'icon';
        else if ($field == 'flag')
            $formatter = 'flag';
        else if (substr($field, -4) == 'time')
            $formatter = 'datetime';
        else if (substr($field, -3) == 'url')
            $formatter = 'url';
        else if (substr($field, -5) == 'image')
            $formatter = 'image';
        if ($formatter)
            $html .= ", formatter: Table.api.formatter." . $formatter . "}";
        else
            $html .= "}";
        return $html;
    }

}