catchAdmin/extend/catcher/generate/Generator.php

<?php
namespace catcher\generate;


use catcher\exceptions\FailedException;
use catcher\generate\factory\Controller;
use catcher\generate\factory\Migration;
use catcher\generate\factory\Model;
use catcher\generate\factory\Route;
use catcher\generate\factory\SQL;
use catcher\generate\support\Table;
use catcher\library\Composer;
use catcher\Utils;
use think\facade\Db;

class Generator
{

    const NEED_PACKAGE = 'nikic/php-parser';

    /**
     * generate
     *
     * @time 2020年04月29日
     * @param $params
     * @return array
     * @throws \think\db\exception\DbException
     * @throws \think\db\exception\ModelNotFoundException
     * @throws \think\db\exception\DataNotFoundException
     */
    public function done($params): array
    {
        // 判断是否安装了扩展包
        if (!(new Composer)->hasPackage(self::NEED_PACKAGE)) {
            throw new FailedException(
                sprintf('you must use [ composer require --dev %s]', self::NEED_PACKAGE)
            );
        }

        $params = \json_decode($params['data'], true);

        [$controller, $model] = $this->parseParams($params);

        $message = [];

        $files = [];
        $migration = '';
        $table = null;

        try {
            if ($params['create_controller']) {
                $files[] = (new Controller)->done($controller);
                array_push($message, 'controller created successfully');
            }

            if ($params['create_table']) {
                (new SQL)->done($model);
                array_push($message, 'table created successfully');
            }

            if ($params['create_model']) {
                $files[] = (new Model)->done($model);
                array_push($message, 'model created successfully');
            }

            if ($params['create_migration']) {
                $migration = (new Migration)->done([$controller['module'], $model['table']]);
                array_push($message, 'migration created successfully');
            }

            // 只有创建了 Controller 最后成功才写入 route
            if ($params['create_controller']) {
                (new Route)->controller($controller['controller'])
                    ->restful($controller['restful'])
                    ->done();
            }
        } catch (\Throwable $exception) {
            if (!$exception instanceof TableExistException) {
                $this->rollback($files, $migration);
            }

            throw new FailedException($exception->getMessage());
        }

        return $message;
    }

    /**
     * preview
     *
     * @time 2020年04月29日
     * @param $params
     * @return bool|string|string[]
     */
    public function preview($params)
    {
        $type = $params['type'];

        $params = \json_decode($params['data'], true);

        [$controller, $model] = $this->parseParams($params);

        switch ($type) {
            case 'controller':
                return (new Controller)->getContent($controller);
            case 'model':
                return (new Model)->getContent($model);
            default:
                break;
        }
    }


    /**
     * parse params
     *
     * @time 2020年04月28日
     * @param $params
     * @return array[]
     */
    protected function parseParams($params): array
    {
        $module = $params['controller']['module'] ?? false;

        if (!$module) {
            throw new FailedException('请设置模块');
        }

        $controller = [
            'module' => $module,
            'model'  => $params['controller']['model'] ?? '',
            'controller' => $params['controller']['controller'] ?? '',
            'restful' => $params['controller']['restful'],
        ];

        $table = $params['controller']['table'] ?? '';

        if ($table) {
            $table = Utils::tableWithPrefix($table);
        }

        $model = [
            'table' => $table,
            'model' => $params['controller']['model'] ?? '',
            'sql'   => $params['table_fields'],
            'extra' => $params['table_extra'],
        ];


        return [$controller, $model];
    }


    /**
     * 回滚
     *
     * @param $files
     * @param $migration
     * @throws \think\db\exception\DataNotFoundException
     * @throws \think\db\exception\DbException
     * @throws \think\db\exception\ModelNotFoundException
     */
    protected function rollback($files, $migration)
    {
        if (Table::exist()) {
           Table::drop();
        }

        foreach ($files as $file) {
            unlink($file);
        }

        if ($migration && unlink($migration)) {
            $model = new class extends \think\Model {
                protected $name = 'migrations';
            };

            $migration = $model->order('version', 'desc')->find();
            $model->where('version', $migration->version)->delete();
        }
    }
}