symfony · marek-pietrzak-tg · Nov 24, 2015 · Nov 24, 2015 · Nov 24, 2015 · Nov 24, 2015
diff --git a/src/Symfony/Component/Console/Helper/FormatterHelper.php b/src/Symfony/Component/Console/Helper/FormatterHelper.php
@@ -72,6 +72,26 @@ public function formatBlock($messages, $style, $large = false)
        return implode("\n", $messages);
    }

+    /**
+     * Truncates a message to the given length.
+     *
+     * @param string $message
+     * @param int    $length
+     * @param string $suffix
+     *
+     * @return string
+     */
+    public function truncate($message, $length, $suffix = '...')
+    {
+        $computedLength = $length - $this->strlen($suffix);
+
+        if ($computedLength > $this->strlen($message)) {
+            return $message;
+        }
+
+        return mb_substr($message, 0, $length).$suffix;
+    }
+
    /**
     * {@inheritdoc}
     */

diff --git a/src/Symfony/Component/Console/Tests/Helper/FormatterHelperTest.php b/src/Symfony/Component/Console/Tests/Helper/FormatterHelperTest.php
@@ -89,4 +89,40 @@ public function testFormatBlockLGEscaping()
            '::formatBlock() escapes \'<\' chars'
        );
    }
+
+    public function testTruncatingWithShorterLengthThanMessageWithSuffix()
+    {
+        $formatter = new FormatterHelper();
+        $message = 'testing truncate';
+
+        $this->assertSame('test...', $formatter->truncate($message, 4));
+        $this->assertSame('testing truncat...', $formatter->truncate($message, 15));
+        $this->assertSame('testing truncate...', $formatter->truncate($message, 16));
+        $this->assertSame('zażółć gęślą...', $formatter->truncate('zażółć gęślą jaźń', 12));
+    }
+
+    public function testTruncatingMessageWithCustomSuffix()
+    {
+        $formatter = new FormatterHelper();
+        $message = 'testing truncate';
+
+        $this->assertSame('test!', $formatter->truncate($message, 4, '!'));
+    }
+
+    public function testTruncatingWithLongerLengthThanMessageWithSuffix()
+    {
+        $formatter = new FormatterHelper();
+        $message = 'test';
+
+        $this->assertSame($message, $formatter->truncate($message, 10));
+    }
+
+    public function testTruncatingWithNegativeLength()
+    {
+        $formatter = new FormatterHelper();
+        $message = 'testing truncate';
+
+        $this->assertSame('testing tru...', $formatter->truncate($message, -5));
+        $this->assertSame('...', $formatter->truncate($message, -100));
+    }
 }