webNeat / lumen-generators

src/Commands/MigrationCommand.php

<?php namespace Wn\Generators\Commands;


class MigrationCommand extends BaseCommand {

    protected $signature = 'wn:migration
        {table : The table name.}
        {--schema= : the schema.}
        {--add= : specifies additional columns like timestamps, softDeletes, rememberToken and nullableTimestamps.}
        {--keys= : foreign keys.}
        {--file= : name of the migration file (to use only for testing purpose).}
        {--parsed : tells the command that arguments have been already parsed. To use when calling the command from an other command and passing the parsed arguments and options}
        {--force= : override the existing files}
    ';
    // {action : One of create, add, remove or drop options.}
    // The action is only create for the moment

    protected $description = 'Generates a migration to create a table with schema';

    public function handle()
    {
        $table = $this->argument('table');
        $name = 'Create' . ucwords(\Illuminate\Support\Str::camel($table));

It seems like $table defined by $this->argument('table') on line 22 can also be of type array; however, Illuminate\Support\Str::camel() does only seem to accept string, maybe add an additional type check?

        $snakeName = \Illuminate\Support\Str::snake($name);

        $content = $this->getTemplate('migration')
            ->with([
                'table' => $table,
                'name' => $name,
                'schema' => $this->getSchema(),
                'additionals' => $this->getAdditionals(),
                'constraints' => $this->getConstraints()
            ])
            ->get();

        $file = $this->option('file');
        if(! $file){
            $file = date('Y_m_d_His_') . $snakeName . '_table';
            $this->deleteOldMigration($snakeName);
        }else{
            $this->deleteOldMigration($file);
        }

        $this->save($content, "./database/migrations/{$file}.php", "{$table} migration");
    }

    protected function deleteOldMigration($fileName)
    {
        foreach (new \DirectoryIterator("./database/migrations/") as $fileInfo){
            if($fileInfo->isDot()) continue;

            if(strpos($fileInfo->getFilename(), $fileName) !== FALSE){
                unlink($fileInfo->getPathname());
            }
        }
    }

    protected function getSchema()

This method seems to be duplicated in your project.

    {
        $schema = $this->option('schema');
        if(! $schema){
            return $this->spaces(12) . "// Schema declaration";
        }

        $items = $schema;
        if( ! $this->option('parsed')){
            $items = $this->getArgumentParser('schema')->parse($schema);
        }

        $fields = [];
        foreach ($items as $item) {
            $fields[] = $this->getFieldDeclaration($item);
        }

        return implode(PHP_EOL, $fields);
    }

    protected function getAdditionals()
    {
        $additionals = $this->option('add');
        if (empty($additionals)) {
            return '';
        }

        $additionals = explode(',', $additionals);
        $lines = [];
        foreach ($additionals as $add) {
            $add = trim($add);
            $lines[] = $this->spaces(12) . "\$table->{$add}();";
        }

        return implode(PHP_EOL, $lines);
    }

    protected function getFieldDeclaration($parts)
    {
        $name = $parts[0]['name'];
        $parts[1]['args'] = array_merge(["'{$name}'"], $parts[1]['args']);
        unset($parts[0]);
        $parts = array_map(function($part){
            return '->' . $part['name'] . '(' . implode(', ', $part['args']) . ')';
        }, $parts);
        return "            \$table" . implode('', $parts) . ';';
    }

    protected function getConstraints()

This method seems to be duplicated in your project.

    {
        $keys = $this->option('keys');
        if(! $keys){
            return $this->spaces(12) . "// Constraints declaration";
        }

        $items = $keys;
        if(! $this->option('parsed')){
            $items = $this->getArgumentParser('foreign-keys')->parse($keys);
        }

        $constraints = [];
        foreach ($items as $item) {
            $constraints[] = $this->getConstraintDeclaration($item);
        }

        return implode(PHP_EOL, $constraints);
    }

    protected function getConstraintDeclaration($key)
    {
        if(! $key['column']){
            $key['column'] = 'id';
        }
        if(! $key['table']){
            $key['table'] = \Illuminate\Support\Str::plural(substr($key['name'], 0, strlen($key['name']) - 3));
        }

        $constraint = $this->getTemplate('migration/foreign-key')
            ->with([
                'name' => $key['name'],
                'table' => $key['table'],
                'column' => $key['column']
            ])
            ->get();

        if($key['on_delete']){

This code seems to be duplicated across your project.

            $constraint .= PHP_EOL . $this->getTemplate('migration/on-constraint')
                    ->with([
                        'event' => 'Delete',
                        'action' => $key['on_delete']
                    ])
                    ->get();
        }

        if($key['on_update']){

This code seems to be duplicated across your project.

            $constraint .= PHP_EOL . $this->getTemplate('migration/on-constraint')
                    ->with([
                        'event' => 'Update',
                        'action' => $key['on_update']
                    ])
                    ->get();
        }

        return $constraint . ';';
    }

}