feat: add gen go struct code from text

inhere · Apr 3, 2022 · 7b99f23 · 7b99f23
1 parent 789be67
commit 7b99f23
Show file tree

Hide file tree

Showing 18 changed files with 733 additions and 187 deletions.
diff --git a/app/Common/GitLocal/GitFlow.php b/app/Common/GitLocal/GitFlow.php
@@ -0,0 +1,13 @@
+<?php declare(strict_types=1);
+
+namespace Inhere\Kite\Common\GitLocal;
+
+/**
+ * class Gitflow
+ *
+ * @author inhere
+ */
+class GitFlow extends AbstractGitLocal
+{
+
+}
diff --git a/app/Console/Attach/Golang/GenerateCmd.php b/app/Console/Attach/Golang/GenerateCmd.php
@@ -0,0 +1,43 @@
+<?php declare(strict_types=1);
+
+namespace Inhere\Kite\Console\Attach\Golang;
+
+use Inhere\Console\Command;
+use Inhere\Console\IO\Input;
+use Inhere\Console\IO\Output;
+
+/**
+ * class GenerateCmd
+ *
+ * @author inhere
+ */
+class GenerateCmd extends Command
+{
+    protected static string $name = 'generate';
+    protected static string $desc = 'generate golang source codes';
+
+    public static function aliases(): array
+    {
+        return ['gen'];
+    }
+
+    protected function subCommands(): array
+    {
+        return [
+            GenerateStructCmd::class,
+        ];
+    }
+
+    /**
+     * Do execute command
+     *
+     * @param Input $input
+     * @param Output $output
+     *
+     * @return mixed|void
+     */
+    protected function execute(Input $input, Output $output)
+    {
+        return $this->showHelp();
+    }
+}
diff --git a/app/Console/Attach/Golang/GenerateStructCmd.php b/app/Console/Attach/Golang/GenerateStructCmd.php
@@ -0,0 +1,150 @@
+<?php declare(strict_types=1);
+
+namespace Inhere\Kite\Console\Attach\Golang;
+
+use Inhere\Console\Command;
+use Inhere\Console\IO\Input;
+use Inhere\Console\IO\Output;
+use Inhere\Kite\Console\Component\ContentsAutoReader;
+use Inhere\Kite\Console\Component\ContentsAutoWriter;
+use Inhere\Kite\Kite;
+use Inhere\Kite\Lib\Generate\GenCodeFactory;
+use Inhere\Kite\Lib\Generate\Json\JsonField;
+use Inhere\Kite\Lib\Parser\Text\Json5ItemParser;
+use Inhere\Kite\Lib\Parser\Text\TextItemParser;
+use Inhere\Kite\Lib\Parser\Text\TextParser;
+use InvalidArgumentException;
+use Toolkit\PFlag\FlagsParser;
+use Toolkit\PFlag\FlagType;
+use Toolkit\PFlag\Validator\EnumValidator;
+use Toolkit\Stdlib\Str;
+use Toolkit\Stdlib\Util\Stream\ListStream;
+use function array_filter;
+use function array_merge;
+use function str_replace;
+
+/**
+ * class GenerateStructCmd
+ *
+ * @author inhere
+ */
+class GenerateStructCmd extends Command
+{
+    protected static string $name = 'struct';
+    protected static string $desc = 'quick generate an go struct by json, text and more';
+
+    public static function aliases(): array
+    {
+        return ['st'];
+    }
+
+    protected function configFlags(FlagsParser $fs): void
+    {
+        $fs->addOptsByRules([
+            '-n, --name'              => 'The struct name',
+            '--tpl-dir'               => 'The tpl dir',
+            '--tpl-file'              => 'The tpl file name',
+            '--cols, --fields'        => 'The field names string, split by ","',
+            '--get-cols, --only-cols' => 'Only get the provide index cols, start is 0. eg: 1,5',
+            '-o, --output'            => 'The output target. default is stdout.',
+            '--of, --out-fmt'         => 'The output format. allow: raw, md-table',
+            '--is, --item-sep'        => 'The item sep char. default is NL(newline).',
+            '--vn, --value-num'       => 'int;set the item value number(cols number), default get from first item.',
+            '--vs, --value-sep'       => 'The item value sep char for "space" parser. default is SPACE',
+            '--parser, --item-parser' => [
+                'type'      => FlagType::STRING,
+                'desc'      => 'The item parser name for difference data type.
+TYPE:
+  space, text -  parser substr by space, use for text data.
+  json, json5 -  parser json(5) line',
+                'default'   => 'text',
+                'validator' => EnumValidator::new(['json', 'json5', 'text', 'space'])
+            ]
+        ]);
+
+        $fs->addOptByRule('s, source', 'The source json,text contents. allow: FILEPATH, @clipboard, @stdin');
+    }
+
+    /**
+     * Do execute command
+     *
+     * @param Input $input
+     * @param Output $output
+     *
+     * @return void
+     */
+    protected function execute(Input $input, Output $output): void
+    {
+        $fs = $this->flags;
+
+        $source = $this->flags->getOpt('source');
+        $source = ContentsAutoReader::readFrom($source);
+
+        $indexes   = [];
+        $idxString = $fs->getOpt('only-cols');
+        if ($idxString && !$indexes = Str::toInts($idxString)) {
+            throw new InvalidArgumentException('please provide valid column index string.');
+        }
+
+        $p = TextParser::new($source);
+        $p->setItemSep($fs->getOpt('item-sep'));
+        $p->setFieldNum($fs->getOpt('value-num'));
+
+        if ($nameString = $fs->getOpt('fields')) {
+            $p->setFields(Str::explode($nameString, ','));
+        }
+
+        switch ($fs->getOpt('item-parser')) {
+            case 'json':
+            case 'json5':
+                $itemParser = new Json5ItemParser;
+                break;
+            case 'text':
+            case 'space':
+            default:
+                $valueSep   = $fs->getOpt('value-sep', ' ');
+                $itemParser = TextItemParser::new($valueSep, $indexes);
+                break;
+        }
+
+        $p->setItemParser($itemParser);
+        $p->parse();
+        $data = $p->getData(true);
+        // $output->aList($data, 'parsed field list');
+
+        $lang = GenCodeFactory::LANG_GO;
+
+        $config = Kite::cliApp()->getArrayParam('gen_code');
+        $tplDir = $fs->getOpt('tpl-dir', $config['tplDir'] ?? '');
+        $tplDir = str_replace('{type}', $lang, $tplDir);
+
+        $tplFile = $fs->getOpt('tpl-file');
+        if (!$tplFile) {
+            $tplFile = "$tplDir/struct.tpl";
+        }
+
+        $config = array_merge($config, array_filter([
+            'tplDir'  => $tplDir,
+            'tplFile' => $tplFile,
+        ]));
+
+        $output->aList($config);
+
+        $gen = GenCodeFactory::create($lang)
+            ->setClassName($fs->getOpt('name'))
+            ->addTplVar('genMark', $input->getFullScript(true))
+            ->configThis($config)
+            ->setFields(ListStream::new($data)->eachToArray(function (array $item) {
+                return JsonField::new($item);
+            }))
+            ->setPathResolver([Kite::class, 'resolve']);
+
+        $output->aList($gen->getFields(), 'field list');
+
+        $result  = $gen->generate();
+        $outFile = $fs->getOpt('output');
+        $output->colored('------------------ Generated Codes -------------------');
+
+        ContentsAutoWriter::writeTo($outFile, $result);
+    }
+}
diff --git a/app/Console/Controller/GoController.php b/app/Console/Controller/GoController.php
@@ -13,6 +13,7 @@
 use Inhere\Console\IO\Input;
 use Inhere\Console\IO\Output;
 use Inhere\Kite\Common\CmdRunner;
+use Inhere\Kite\Console\Attach\Golang\GenerateCmd;
 use Toolkit\PFlag\FlagsParser;
 use function file_get_contents;
 
@@ -36,6 +37,17 @@ protected static function commandAliases(): array
             'pkgUp'     => [
                 'up', 'pkgup'
             ],
+            'generate' => GenerateCmd::aliases(),
+        ];
+    }
+
+    /**
+     * @return array
+     */
+    protected function subCommands(): array
+    {
+        return [
+            GenerateCmd::class,
         ];
     }
 

diff --git a/app/Console/Controller/StringController.php b/app/Console/Controller/StringController.php
@@ -404,13 +404,13 @@ public function replaceCommand(FlagsParser $fs, Output $output): void
      *
      * @options
      *       --fields               The field names, split by ','
-     *       --get-cols             Only get the provide index cols, eg: 1,5
+     *       --get-cols             Only get the provide index cols, start is 0. eg: 1,5
      *   -o, --output               The output target. default is stdout.
      *  --of, --out-fmt             The output format. allow: raw, md-table
      *  --is, --item-sep            The item sep char. default is NL.
      *  --vn, --value-num           int;The item value number. default get from first line.
      *  --vs, --value-sep           The item value sep char for 'space' parser. default is SPACE
-     *  --parser, --item-parser     The item parser name. allow:
+     *  --parser, --item-parser     The item parser name for difference data type. allow:
      *                              space       -  parser substr by space
      *                              json, json5 -  parser json(5) line
      *
@@ -481,12 +481,20 @@ public function parseCommand(FlagsParser $fs, Output $output): void
     /**
      * decode query string and print data.
      *
+     * @options
+     *  --full        bool;The input query argument is full url
+     *
      * @arguments
      * query    The URI query string.
      */
     public function dequeryCommand(FlagsParser $fs, Output $output): void
     {
         $str = $fs->getArg('query');
+        $str = rawurlencode($str);
+
+        // if (!$fs->getOpt('full')) {
+        //     $str = 'http://abc.com?' . $str;
+        // }
 
         parse_str($str, $ret);