<?php

namespace catcher\command;

use catcher\CatchAdmin;
use think\console\Command;
use think\console\Input;
use think\console\input\Argument;
use think\console\input\Option as InputOption;
use think\console\Output;
use think\facade\Db;
use think\helper\Str;

class ModelGeneratorCommand extends Command
{
    protected function configure()
    {
        $this->setName('create:model')
            ->addArgument('module', Argument::REQUIRED, 'module name')
            ->addArgument('model', Argument::REQUIRED, 'model name')
            ->setDescription('create model');
    }

    protected function execute(Input $input, Output $output)
    {
        $model = ucfirst($input->getArgument('model'));
        $module = strtolower($input->getArgument('module'));

        $table = Str::snake($model);

        $modelFile= CatchAdmin::getModuleModelDirectory($module) . $model . '.php';

        $asn = 'Y';
        if (file_exists($modelFile)) {
            $asn = $this->output->ask($this->input, "Model File {$model} already exists.Are you sure to overwrite, the content will be lost(Y/N)");
        }

        if (strtolower($asn) == 'n') {
            exit(0);
        }

        file_put_contents($modelFile, $this->replaceContent([
            $module, $model, $table, $this->generateFields($this->getTableFields($table))
        ]));

        if (file_exists($modelFile)) {
            $output->info(sprintf('%s Create Successfully!', $modelFile));
        } else {
            $output->error(sprintf('%s Create Failed!', $modelFile));
        }
    }


    private function getTableFields($table): array
    {
        $fields = Db::query('show full columns from ' .
            config('database.connections.mysql.prefix') . $table);

        $new = [];

        foreach ($fields as $field) {
            $new[$field['Field']] = $field['Comment'];
        }

        return $new;
    }

    private function generateFields($fields)
    {
        $f = '';
        foreach ($fields as $field => $comment) {
            $f .= sprintf("'%s', // %s" . "\r\n\t\t", $field, $comment);
        }

        return rtrim($f, "\r\n\t\t");
    }

    private function replaceContent(array $replace)
    {
        return str_replace([
            '{Module}', '{Class}', '{Name}', '{Field}'
        ], $replace, $this->content());
    }

    private function content()
    {
        return <<<EOT
<?php
namespace catchAdmin\{Module}\model;

use catcher\base\CatchModel;

class {Class} extends CatchModel
{
    protected \$name = '{Name}';
    
    protected \$field = [
        {Field}   
    ]; 
}
EOT;
    }
}