up: add more test, add some new methods

Author: inhere

src/Str/StrBuffer.php

@@ -9,38 +9,89 @@
 
 namespace Toolkit\Stdlib\Str;
 
+use function array_unshift;
+use function implode;
+use function sprintf;
+
 /**
  * Class StrBuffer
+ *
  * @package Toolkit\Stdlib\Str
  */
-final class StrBuffer
+class StrBuffer
 {
     /**
-     * @var string
+     * @var self
      */
-    private $body;
+    private static $global;
 
     /**
-     * @param string $content
+     * @var string[]
+     */
+    private $parts = [];
+
+    /**
+     * @param string $str
+     *
+     * @return self
+     */
+    public static function global(string $str = ''): self
+    {
+        if (!self::$global) {
+            self::$global = new self($str);
+        }
+
+        return self::$global;
+    }
+
+    /**
+     * @param string $str
      *
      * @return static
      */
-    public static function new(string $content): self
+    public static function new(string $str = ''): self
+    {
+        return new self($str);
+    }
+
+    /**
+     * Class constructor.
+     *
+     * @param string $str
+     */
+    public function __construct(string $str = '')
     {
-        return new self($content);
+        $this->write($str);
     }
 
-    public function __construct(string $content = '')
+    public function reset(): void
     {
-        $this->body = $content;
+        $this->parts = [];
     }
 
     /**
      * @param string $content
      */
     public function write(string $content): void
     {
-        $this->body .= $content;
+        $this->parts[] = $content;
+    }
+
+    /**
+     * @param string $fmt
+     * @param mixed  ...$args
+     */
+    public function writef(string $fmt, ...$args): void
+    {
+        $this->parts[] = sprintf($fmt, ...$args);
+    }
+
+    /**
+     * @param string $content
+     */
+    public function writeln(string $content): void
+    {
+        $this->parts[] = $content . "\n";
     }
 
     /**
@@ -56,43 +107,35 @@ public function append(string $content): void
      */
     public function prepend(string $content): void
     {
-        $this->body = $content . $this->body;
+        array_unshift($this->parts, $content);
     }
 
     /**
      * clear data
      */
     public function clear(): string
     {
-        $string = $this->body;
+        $strings = $this->parts;
         // clear
-        $this->body = '';
-
-        return $string;
-    }
+        $this->parts = [];
 
-    /**
-     * @return string
-     */
-    public function getBody(): string
-    {
-        return $this->body;
+        return implode($strings);
     }
 
     /**
-     * @param string $body
+     * @return string[]
      */
-    public function setBody(string $body): void
+    public function getParts(): array
     {
-        $this->body = $body;
+        return $this->parts;
     }
 
     /**
      * @return string
      */
     public function toString(): string
     {
-        return $this->body;
+        return implode($this->parts);
     }
 
     /**

src/Str/StrObject.php

@@ -3,6 +3,7 @@
 namespace Toolkit\Stdlib\Str;
 
 use Toolkit\Stdlib\Str;
+use function in_array;
 use function trim;
 
 /**
@@ -155,6 +156,26 @@ public function hasSuffix(string $suffix): bool
         return Str::hasSuffix($this->string, $suffix);
     }
 
+    /**
+     * @param string $str
+     *
+     * @return bool
+     */
+    public function isSubstrOf(string $str): bool
+    {
+        return Str::strpos($str, $this->string) !== false;
+    }
+
+    /**
+     * @param string[] $arr
+     *
+     * @return bool
+     */
+    public function isOneOf(array $arr): bool
+    {
+        return in_array($this->string, $arr, true);
+    }
+
     // ------------------ convert ------------------
 
     /**

test/Str/StrBufferTest.php

@@ -0,0 +1,28 @@
+<?php declare(strict_types=1);
+
+namespace Toolkit\StdlibTest\Str;
+
+use PHPUnit\Framework\TestCase;
+use Toolkit\Stdlib\Str\StrBuffer;
+
+/**
+ * Class StrBufferTest
+ *
+ * @package Toolkit\StdlibTest\Str
+ */
+class StrBufferTest extends TestCase
+{
+    public function testBasic(): void
+    {
+        $buf = StrBuffer::new();
+
+        $buf->write('a');
+        $buf->writef(' %s an ', 'is');
+        $buf->writeln('alpha');
+
+        self::assertEquals("a is an alpha\n", $buf->toString());
+
+        $buf->reset();
+        self::assertEmpty($buf->toString());
+    }
+}