Laravel 實現自定義生成檔案命令

Egfly發表於2019-03-16
Laravel

laravel實現自定義生成檔案命令

前言

現在的專案中有使用Entiy檔案，每次新增一張表的時候都要建立對應的Entiy檔案,同時還要設定變數，非常麻煩。所以想著能不能建立一個像make:controller那樣的命令生成Entity檔案。

laravel裡的 php artisan make:model這樣的建立檔案命令，都是繼承自Illuminate\Console\GeneratorCommand.而實現這樣的命令的原理其實很簡單：

先寫好一個範本檔案，然後定義關鍵字，最後在command裡將關鍵字替換成我們想要的內容.

模版檔案（stub）

拿make:command來看：他的模版是model.stub，位於vendor\laravel\framework\src\Illuminate\Foundation\Console\stubs\model.stub.

<?php

namespace DummyNamespace;

use Illuminate\Database\Eloquent\Model;

class DummyClass extends Model
{
    //
}

從model.stub來看，當我們利用php artisan make:model建立模型時，只是替換了名稱空間關鍵字DummyNamespace和類名關鍵字DummyClass.

GeneratorCommand

make:model命令對應的類是ModelMakeCommand,位於model.stub的父級目錄下。內容有點多，就不貼了，有興趣可以去看下。直接看它的父類Illuminate\Console\GeneratorCommand.

// 核心程式碼 
public function handle()
{
    // 拼接名稱空間
    $name = $this->qualifyClass($this->getNameInput());

    // 獲取檔案儲存位置
    $path = $this->getPath($name);

    // 判斷檔案是否存在，存在則丟擲錯誤
    if ((! $this->hasOption('force') || ! $this->option('force')) && $this->alreadyExists($this->getNameInput())) {
        $this->error($this->type.' already exists!');

        return false;
    }

    // 生成檔案
    $this->makeDirectory($path);

    // 替換模版中的關鍵字，如：命令空間和類名
    $this->files->put($path, $this->buildClass($name));

    $this->info($this->type.' created successfully.');
}

// 編輯模版內容
protected function buildClass($name)
{
    // 獲取模版檔案內容
    $stub = $this->files->get($this->getStub()); 

    return $this->replaceNamespace($stub, $name)->replaceClass($stub, $name);
}

// 替換名稱空間 DummyRootNamespace
protected function replaceNamespace(&$stub, $name)
{
    // 拼接名稱空間
    $stub = str_replace(
        ['DummyNamespace', 'DummyRootNamespace', 'NamespacedDummyUserModel'],
        [$this->getNamespace($name), $this->rootNamespace(), config('auth.providers.users.model')],
        $stub
    );

    return $this;
}

// 替換類名 DummyClass
protected function replaceClass($stub, $name)
{ 
    $class = str_replace($this->getNamespace($name).'\\', '', $name);

    return str_replace('DummyClass', $class, $stub);
}

通過原始碼可以看出，建立檔案就是替換模版中的關鍵字就行了。在這個基礎上我們可以實現一些命令來生成我們想要的檔案。

實現 create:entity 命令

1. 建立模版
<?php

namespace DummyNamespace;

class DummyClass extends Entity
{
// 自定義需要替換的內容關鍵字
AAA
}

2. 實現command

class CreateEntity extends GeneratorCommand
{
    protected $signature = 'create:entity {name}';

    protected $description = '自動生成Model Entity 例項';

    // 資料庫型別對照
    public static $dictionary = [
        'string' => ['char', 'text'],
        'int' => ['int','numeric'],
        'float' => ['double','float','decimal']
    ];

    protected $type = 'entity';

    /**
     * 設定模板地址
     * @return string
     */
    protected function getStub()
    {
        return __DIR__.'/stubs/Entity.stub';
    }

    /**
     * 設定名稱空間,以及檔案路徑
     * @param string $rootNamespace
     * @return string
     */
    protected function getDefaultNamespace($rootNamespace)
    {
        return $rootNamespace.'\Repositories\Entitys'; //偷懶、直接寫死
    }

    /**
     * 設定類名和自定義替換內容
     * @param string $stub
     * @param string $name
     * @return string
     */
    protected function replaceClass($stub, $name)
    {

        $stub = $this->replaceCustomizeSetting($stub); //替換自定義內容

        return parent::replaceClass($stub, $name);
    }

    /**
     * 替換自定義內容
     * @param $stub
     * @return mixed
     */
    protected function replaceCustomizeSetting($stub){

        //將輸入的類名處理成表名
        $name =  $this->getNameInput();
        $name = rtrim($name,'E');
        $tableName = strtolower(trim(preg_replace("/[A-Z]/", "_\\0",$name), "_")); //駝峰變成小寫加_

        $info = collect(\DB::select("desc rxt_".$tableName.";"))->toArray(); //獲取表欄位和型別列表
        $list = [];
        foreach ($info as $key => $value){ //轉成二維陣列
            $arr = collect($info[$key])->toArray();
            if($arr['Field'] == 'deleted_at') continue;
            array_push($list, $arr);
        }
        $fieldExample = "    /**  \r\n    * @var type isNull \r\n    */ \r\n    public \$fieldName;\r\n";
        $result = null;
        foreach ($list as $item){
            $result = $result.$fieldExample;
            foreach (static::$dictionary as $key => $value){
                foreach ($value as $a => $b){
                    if(strstr($item['Type'], $b)){
                        $result = str_replace('type', $key, $result);
                    }
                }
            }
            $isNull = $item['Null'] == 'YES' ? '|null' : '';
            $result = str_replace('isNull', $isNull, $result);
            $result = str_replace('fieldName', $item['Field'], $result);

        }

        return str_replace('AAA', $result, $stub);
    }

}
本作品採用《CC 協議》，轉載必須註明作者和本文連結

相關文章