spider-mane / leonidas

src/Console/Command/MakeModelCommand.php

<?php

namespace Leonidas\Console\Command;

use DirectoryIterator;
use Leonidas\Console\Command\Abstracts\HopliteCommand;
use Leonidas\Console\Library\Printer\Model\ModelComponentFactory;
use Leonidas\Console\Library\Printer\Model\PsrPrinterFactory;
use Nette\PhpGenerator\PhpFile;
use Symfony\Component\Console\Input\InputArgument;
use Symfony\Component\Console\Input\InputOption;

class MakeModelCommand extends HopliteCommand
{
    public const VALID_TEMPLATES = [
        'post',
        'post:h',
        'attachment',
        'term',
        'term:h',
        'user',
        // 'comment',
    ];

    protected const CORE_FILE_METHODS = [
        'model' => 'makeModelFiles',
        'collection' => 'makeCollectionFiles',
        'repository' => 'makeRepositoryFiles',
    ];

    protected const SUPPORT_FILE_METHODS = [
        'factories' => 'makeFactoryFiles',
        'access' => 'makeAccessProviderFiles',
        'facade' => 'makeFacadeFiles',
    ];

    protected const EXTRA_FILE_METHODS = [
        'registration' => 'updateRegistrationClass',
    ];

    protected static $defaultName = 'make:model';

    protected function configure()
    {
        $this
            ->addArgument('model', InputArgument::REQUIRED, '')
            ->addArgument('entity', InputArgument::REQUIRED, '')
            ->addArgument('single', InputArgument::REQUIRED, '')
            ->addArgument('plural', InputArgument::REQUIRED, '')
            ->addArgument('components', InputArgument::OPTIONAL | InputArgument::IS_ARRAY, '')
            ->addOption('template', 't', InputOption::VALUE_REQUIRED, '', 'post')
            ->addOption('namespace', 's', InputOption::VALUE_REQUIRED, '')
            ->addOption('contracts', 'c', InputOption::VALUE_REQUIRED, '')
            ->addOption('facades', 'x', InputOption::VALUE_REQUIRED, '')
            ->addOption('abstracts', 'a', InputOption::VALUE_NONE, '')
            ->addOption('bootstrap', 'r', InputOption::VALUE_REQUIRED, '')
            ->addOption('design', 'd', InputOption::VALUE_NONE, 'Generate interfaces only')
            ->addOption('build', 'b', InputOption::VALUE_NONE, 'Generate classes using interfaces as guide')
            ->addOption('omit', 'o', InputOption::VALUE_REQUIRED | InputOption::VALUE_IS_ARRAY, '')
            ->addOption('force', 'f', InputOption::VALUE_NONE, '');
    }

    protected function handle(): int
    {
        $template = $this->input->getOption('template');

        if ($this->isValidTemplate($template)) {
            $status = $this->makeFiles($template);
        } else {
            $this->output->error("Value \"{$template}\" is not an accepted value for template");

            $status = self::FAILURE;
        }

        return $status;
    }

    protected function isValidTemplate(string $template): bool
    {
        return in_array($template, static::VALID_TEMPLATES);
    }

    protected function makeFiles(string $template): int
    {
        $model = $this->convert($this->input->getArgument('model'))->toPascal();
        $namespace = $this->configurableOption('namespace', 'make.model.namespace');
        $contracts = $this->configurableOption('contracts', 'make.model.contracts');
        $facades = $this->configurableOption('facades', 'facades');

        $factory = $this->getComponentFactory($model, $namespace, $contracts, $facades, $template);

It seems like $namespace can also be of type null; however, parameter $namespace of Leonidas\Console\Command\MakeModelCommand::getComponentFactory() does only seem to accept string, maybe add an additional type check?

It seems like $contracts can also be of type null; however, parameter $contracts of Leonidas\Console\Command\MakeModelCommand::getComponentFactory() does only seem to accept string, maybe add an additional type check?

It seems like $facades can also be of type null; however, parameter $facades of Leonidas\Console\Command\MakeModelCommand::getComponentFactory() does only seem to accept string, maybe add an additional type check?

        $paths = $this->getOutputPaths($model, $namespace, $contracts, $facades);

It seems like $contracts can also be of type null; however, parameter $contracts of Leonidas\Console\Command\MakeModelCommand::getOutputPaths() does only seem to accept string, maybe add an additional type check?

It seems like $namespace can also be of type null; however, parameter $namespace of Leonidas\Console\Command\MakeModelCommand::getOutputPaths() does only seem to accept string, maybe add an additional type check?

It seems like $facades can also be of type null; however, parameter $facades of Leonidas\Console\Command\MakeModelCommand::getOutputPaths() does only seem to accept string, maybe add an additional type check?

        $action = $this->resolveRequestedAction();

        $force = $this->input->getOption('force');

        foreach ($this->resolveComponents('core') as $method) {
            $status = $this->$method($factory, $action, $paths, $force);

            if (self::SUCCESS !== $status) {
                return $status;
            }
        }

        if ('interfaces' !== $action) {
            foreach ($this->resolveComponents('support') as $method) {
                $status = $this->$method($factory, $paths, $force);

                if (self::SUCCESS !== $status) {
                    return $status;
                }
            }

            foreach ($this->resolveComponents('extra') as $method) {
                $status = $this->$method($factory, $force);

                if (self::SUCCESS !== $status) {
                    return $status;
                }
            }
        }

        $this->output->success("Successfully created model files");

        return self::SUCCESS;
    }

    protected function resolveComponents(string $set): array
    {
        $set = constant(
            sprintf('%s::%s_FILE_METHODS', static::class, strtoupper($set))
        );

        $selected = $this->input->getArgument('components') ?: array_keys($set);
        $omitted = $this->input->getOption('omit') ?? [];

        $resolved = [];

        foreach ($set as $component => $method) {
            if (in_array($component, $selected) && !in_array($component, $omitted)) {
                $resolved[] = $method;
            }
        }

        return $resolved;
    }

    protected function getComponentFactory(
        string $model,
        string $namespace,
        string $contracts,
        string $facades,
        string $template
    ): ModelComponentFactory {
        $entity = $this->input->getArgument('entity');
        $single = $this->input->getArgument('single');
        $plural = $this->input->getArgument('plural');

        $namespace = $this->pathToNamespace($namespace, $model);
        $contracts = $this->pathToNamespace($contracts, $model);
        $abstracts = $this->resolveAbstractNamespace($namespace);
        $facades = $this->pathToNamespace($facades);

        return ModelComponentFactory::build([
            'model' => $model,
            'namespace' => $namespace,
            'contracts' => $contracts,
            'abstracts' => $abstracts,
            'facades' => $facades,
            'entity' => $entity,
            'single' => $single,
            'plural' => $plural,
            'template' => $template,
        ]);
    }

    protected function getOutputPaths(string $model, string $namespace, string $contracts, string $facades): array
    {
        return [
            'interfaces' => $contracts . DIRECTORY_SEPARATOR . $model,
            'classes' => $namespace = $namespace . DIRECTORY_SEPARATOR . $model,
            'abstracts' => $this->resolveAbstractDir($namespace),
            'facades' => $facades,
        ];
    }

    protected function setupTestDir(): array
    {
        $playground = $this->external('/.playground/model');

        if ($this->filesystem->exists($playground)) {
            foreach (new DirectoryIterator($playground) as $file) {
                if (!$file->isFile()) {
                    continue;
                }

                $isInterface = str_ends_with($file->getBasename(), 'Interface.php');
                $isRegistrar = $file->getBasename() === 'RegisterModelServices.php';

                if (!($isInterface || $isRegistrar)) {
                    $this->filesystem->remove($file->getPathname());
                }

                if ($isInterface) {
                    require $file->getPathname();
                }
            }

        // $this->filesystem->remove($playground);
        } else {
            $this->filesystem->mkdir($playground);
        }

        $registrar = 'RegisterModelServices.php';

        $this->filesystem->copy(
            $this->external("/src/Framework/Bootstrap/{$registrar}"),
            "{$playground}/{$registrar}"
        );

        foreach (['Contracts', 'Library'] as $namespace) {
            $this->filesystem->remove(
                $this->external('/src/' . $namespace . '/System/Model/Test')
            );
        }

        return [
            'interfaces' => $playground,
            'classes' => $playground,
            'abstracts' => $playground,
            'facades' => $playground,
        ];
    }

    protected function resolveRequestedAction(): string
    {
        if ($this->input->getOption('design')) {
            return 'interfaces';
        } elseif ($this->input->getOption('build')) {
            return 'classes';
        } else {
            return 'complete';
        }
    }

    protected function makeModelFiles(ModelComponentFactory $factory, string $action, array $paths, bool $force): int
    {
        $interface = $factory->getModelInterfacePrinter();
        $class = $factory->getModelPrinter();

        $interfaceFile = $this->phpFile($paths['interfaces'], $interface->getClass());
        $classFile = $this->phpFile($paths['classes'], $class->getClass());

        if ('interfaces' === $action || 'complete' === $action) {
            $this->writeFile($interfaceFile, $interface->printFile(), $force);
        } elseif ('classes' === $action) {
            if (!interface_exists($interfaceFqn = $interface->getClassFqn())) {
                $this->output->error("Interface {$interfaceFqn} does not exist");

                return self::INVALID;
            }

            $this->writeFile($classFile, $class->printFromType(), $force);
        }

        if ('complete' === $action) {
            $this->writeFile($classFile, $class->printFile(), $force);
        }

        return self::SUCCESS;
    }

    protected function makeCollectionFiles(ModelComponentFactory $factory, string $action, array $paths, bool $force): int
    {
        $isPost = $factory->isPostTemplate();

        $interface = $factory->getCollectionInterfacePrinter();
        $collection = $isPost
            ? $factory->getChildCollectionPrinter()
            : $factory->getCollectionPrinter();
        $abstract = $factory->getAbstractCollectionPrinter();
        $query = $factory->getChildQueryPrinter();

        $interfaceFile = $this->phpFile($paths['interfaces'], $interface->getClass());
        $collectionFile = $this->phpFile($paths['classes'], $collection->getClass());
        $abstractFile = $this->phpFile($paths['abstracts'], $abstract->getClass());
        $queryFile = $this->phpFile($paths['classes'], $query->getClass());

        if ('interfaces' === $action || 'complete' === $action) {
            $this->writeFile($interfaceFile, $interface->printFile(), $force);
        } elseif ('classes' === $action) {
            if (!interface_exists($interface->getClassFqn())) {
                $this->output->error("Interface {$interface->getClassFqn()} does not exist");

                return self::INVALID;
            }

            if ($isPost) {
                $this->writeFile($abstractFile, $abstract->printFromType(), $force);
                $this->writeFile($collectionFile, $collection->printFile(), $force);
                $this->writeFile($queryFile, $query->printFile(), $force);
            } else {
                $this->writeFile($collectionFile, $collection->printFromType(), $force);

The method printFromType() does not exist on Leonidas\Console\Library\Printer\Model\ModelCollectionAsChildPrinter.

            }
        }

        if ('complete' === $action) {
            $this->writeFile($collectionFile, $collection->printFile(), $force);

            if ($isPost) {
                $this->writeFile($abstractFile, $abstract->printFile(), $force);
                $this->writeFile($queryFile, $query->printFile(), $force);
            }
        }

        return self::SUCCESS;
    }

    protected function makeRepositoryFiles(ModelComponentFactory $factory, string $action, array $paths, bool $force): int
    {
        $interface = $factory->getRepositoryInterfacePrinter();
        $class = $factory->getRepositoryPrinter();

        $interfaceFile = $this->phpFile($paths['interfaces'], $interface->getClass());
        $classFile = $this->phpFile($paths['classes'], $class->getClass());

        if ('interfaces' === $action || 'complete' === $action) {
            $this->writeFile($interfaceFile, $interface->printFile(), $force);
        } elseif ('classes' === $action) {
            if (!interface_exists($interfaceFqn = $interface->getClassFqn())) {
                $this->output->error("Interface {$interfaceFqn} does not exist");

                return self::INVALID;
            }

            $this->writeFile($classFile, $class->printFromType(), $force);
        }

        if ('complete' === $action) {
            $this->writeFile($classFile, $class->printFile(), $force);
        }

        return self::SUCCESS;
    }

    protected function makeFactoryFiles(ModelComponentFactory $factory, array $paths, bool $force): int
    {
        $model = $factory->getModelConverterPrinter();
        $collection = $factory->getCollectionFactoryPrinter();

        $modelFile = $this->phpFile($paths['classes'], $model->getClass());
        $collectionFile = $this->phpFile($paths['classes'], $collection->getClass());

        $this->writeFile($modelFile, $model->printFile(), $force);
        $this->writeFile($collectionFile, $collection->printFile(), $force);

        if ($factory->isPostTemplate()) {
            $query = $factory->getQueryFactoryPrinter();
            $queryFile = $this->phpFile($paths['classes'], $query->getClass());

            $this->writeFile($queryFile, $query->printFile(), $force);
        }

        return self::SUCCESS;
    }

    protected function makeAccessProviderFiles(ModelComponentFactory $factory, array $paths, bool $force): int
    {
        $get = $factory->getGetAccessProviderPrinter();
        $set = $factory->getSetAccessProviderPrinter();

        $getFile = $this->phpFile($paths['classes'], $get->getClass());
        $setFile = $this->phpFile($paths['classes'], $set->getClass());

        $this->writeFile($getFile, $get->printFile(), $force);
        $this->writeFile($setFile, $set->printFile(), $force);

        if ($factory->isPostTemplate()) {
            $tag = $factory->getTagAccessProviderPrinter();
            $tagFile = $this->phpFile($paths['classes'], $tag->getClass());

            $this->writeFile($tagFile, $tag->printFile(), $force);
        }

        return self::SUCCESS;
    }

    protected function makeFacadeFiles(ModelComponentFactory $factory, array $paths, bool $force): int
    {
        $facade = $factory->getRepositoryFacadePrinter();
        $facadeFile = $this->phpFile($paths['facades'], $facade->getClass());

        $this->writeFile($facadeFile, $facade->printFromType(), $force);

        return self::SUCCESS;
    }

    protected function updateRegistrationClass(ModelComponentFactory $factory, bool $force): int
    {
        $name = $factory->getSingle();
        $model = $factory->getModelPrinter();
        $schema = [
            'post' => 'post',
            'post:h' => 'post',
            'attachment' => 'attachment',
            'term' => 'term',
            'term:h' => 'term',
            'user' => 'user',
            // 'comment' => 'comment',
        ][$factory->getTemplate()];

        $registrar = $this->configurableOption('bootstrap', 'make.model.bootstrap');
        $file = PhpFile::fromCode(file_get_contents($registrar));

It seems like $registrar can also be of type null; however, parameter $filename of file_get_contents() does only seem to accept string, maybe add an additional type check?

        $namespaces = $file->getNamespaces();
        $namespace = reset($namespaces);

        $namespace->addUse($model->getClassFqn());

        $classes = $namespace->getClasses();
        $class = reset($classes);

        $class->addMethod($name . 'Services')
            ->setReturnType('void')
            ->setBody(sprintf(
                "\$this->register(%s::class, '%s', '%s');",
                $model->getClass(),
                $factory->getEntity(),
                $schema
            ));

        $this->writeFile(
            $registrar,

It seems like $registrar can also be of type null; however, parameter $path of Leonidas\Console\Command\Abstracts\HopliteCommand::writeFile() does only seem to accept string, maybe add an additional type check?

            PsrPrinterFactory::create()->printFile($file),
            true
        );

        return self::SUCCESS;
    }
}