feat: add new formatter for pretty JSON render

Author: inhere

Diff for: src/Component/Formatter/JSONPretty.php

@@ -9,52 +9,109 @@
 
 namespace Inhere\Console\Component\Formatter;
 
-use Inhere\Console\Component\MessageFormatter;
-use Toolkit\Stdlib\Str;
+use JsonException;
+use Toolkit\Cli\Color\ColorTag;
+use Toolkit\Stdlib\Helper\JsonHelper;
+use Toolkit\Stdlib\Obj\AbstractObj;
+use Toolkit\Stdlib\Str\StrBuffer;
+use function array_merge;
+use function explode;
+use function is_numeric;
 use function json_decode;
+use function preg_replace_callback;
+use function rtrim;
+use function str_contains;
+use function str_ends_with;
+use function trim;
 
 /**
  * class JSONPretty
  */
-class JSONPretty extends MessageFormatter
+class JSONPretty extends AbstractObj
 {
+    public const DEFAULT_THEME = [
+        'keyName' => 'mga',
+        'strVal'  => 'info',
+        'intVal'  => 'cyan',
+        'boolVal' => 'red',
+    ];
+
+    /**
+     * @var array{keyName: string, strVal: string, intVal: string, boolVal: string}
+     */
+    protected array $theme = self::DEFAULT_THEME;
+
     /**
-     * @var array|iterable
+     * @var int
      */
-    public $data;
+    public int $maxDepth = 10;
 
     /**
      * @param string $json
      *
-     * @return static
+     * @return string
+     * @throws JsonException
      */
-    public static function newFromString(string $json): self
+    public function render(string $json): string
     {
-        $self = new self();
-        $self->setData((array)json_decode($json, true));
+        $data = json_decode($json, true, 512, JSON_THROW_ON_ERROR);
 
-        return $self;
+        return $this->renderData($data);
     }
 
     /**
+     * @param array $data
+     *
      * @return string
      */
-    public function format(): string
+    public function renderData(array $data): string
     {
-        $buf = Str\StrBuffer::new();
+        $buf  = StrBuffer::new();
+        $json = JsonHelper::prettyJSON($data);
 
-        foreach ($this->data as $key => $item) {
-            // TODO
+        foreach (explode("\n", $json) as $line) {
+            $trimmed = trim($line);
+            // start or end chars. eg: {} []
+            if (!str_contains($trimmed, ': ')) {
+                $buf->writeln($line);
+                continue;
+            }
+
+            [$key, $val] = explode(': ', $line);
+
+            // format key name.
+            if ($keyTag = $this->theme['keyName']) {
+                $key = preg_replace_callback('/"[\w-]+"/', static function ($m) use ($keyTag) {
+                    return ColorTag::wrap($m[0], $keyTag);
+                }, $key);
+            }
+
+            // has end ',' clear it.
+            if ($hasEndComma = str_ends_with($val, ',')) {
+                $val = rtrim($val, ',');
+            }
+
+            // bool val
+            if ($val === 'true' || $val === 'false') {
+                $val = ColorTag::wrap($val, $this->theme['boolVal']);
+            } elseif (is_numeric($val)) { // number
+                $val = ColorTag::wrap($val, $this->theme['intVal']);
+            } else { // string
+                $val = ColorTag::wrap($val, $this->theme['strVal']);
+            }
+
+            $buf->writeln($key . ': ' . $val . ($hasEndComma ?  ',' : ''));
         }
 
-        return $buf->toString();
+        return $buf->getAndClear();
     }
 
     /**
-     * @param array|iterable $data
+     * @param array $theme
      */
-    public function setData($data): void
+    public function setTheme(array $theme): void
     {
-        $this->data = $data;
+        $this->theme = array_merge($this->theme, $theme);
     }
 }
+

Diff for: src/IO/Input.php

@@ -38,7 +38,7 @@ class Input extends AbstractInput
      *
      * @var resource
      */
-    protected $inputStream;
+    protected $stream;
 
     /**
      * Input constructor.
@@ -51,15 +51,10 @@ public function __construct(array $args = null)
             $args = $_SERVER['argv'];
         }
 
-        $this->inputStream = Cli::getInputStream();
+        $this->stream = Cli::getInputStream();
         $this->collectInfo($args);
     }
 
-    public function resetInputStream(): void
-    {
-        $this->inputStream = Cli::getInputStream();
-    }
-
     /**
      * @return string
      */
@@ -88,6 +83,14 @@ protected function tokenEscape(string $token): string
         return $token;
     }
 
+    /**
+     * @return string
+     */
+    public function readAll(): string
+    {
+        return File::streamReadAll($this->stream);
+    }
+
     /**
      * Read input information
      *
@@ -99,10 +102,23 @@ protected function tokenEscape(string $token): string
     public function readln(string $question = '', bool $nl = false): string
     {
         if ($question) {
-            fwrite($this->inputStream, $question . ($nl ? "\n" : ''));
+            fwrite($this->stream, $question . ($nl ? "\n" : ''));
         }
 
-        return File::streamFgets($this->inputStream);
+        return File::streamFgets($this->stream);
+    }
+
+    /**
+     * @return resource
+     */
+    public function getInputStream()
+    {
+        return $this->stream;
+    }
+
+    public function resetInputStream(): void
+    {
+        $this->stream = Cli::getInputStream();
     }
 
     /***********************************************************************************
@@ -128,9 +144,9 @@ public function getFullCommand(): string
     /**
      * @return resource
      */
-    public function getInputStream()
+    public function getStream()
     {
-        return $this->inputStream;
+        return $this->stream;
     }
 
     /**

Diff for: src/IO/Output.php

@@ -13,6 +13,7 @@
 use Inhere\Console\Console;
 use Inhere\Console\Concern\FormatOutputAwareTrait;
 use Toolkit\Cli\Cli;
+use Toolkit\FsUtil\File;
 
 /**
  * Class Output
@@ -130,6 +131,14 @@ public function read(string $question = '', bool $nl = false): string
         return Console::read($question, $nl);
     }
 
+    /**
+     * @return string
+     */
+    public function readAll(): string
+    {
+        return File::streamReadAll($this->outputStream);
+    }
+
     /**
      * Write a message to standard error output stream.
      *