Dumper::format() array wrapping counts with code length

Author: dg

src/PhpGenerator/Dumper.php

@@ -179,19 +179,13 @@ public function format(string $statement, ...$args): string
 			} elseif (!$args) {
 				throw new Nette\InvalidArgumentException('Insufficient number of arguments.');
 			} elseif ($token === '?') {
-				$res .= $this->dump(array_shift($args));
+				$res .= $this->dump(array_shift($args), strlen($res) - strrpos($res, "\n"));
 			} elseif ($token === '...?' || $token === '?*') {
 				$arg = array_shift($args);
 				if (!is_array($arg)) {
 					throw new Nette\InvalidArgumentException('Argument must be an array.');
 				}
-				$items = [];
-				foreach ($arg as $tmp) {
-					$items[] = $this->dump($tmp);
-				}
-				$res .= strlen($tmp = implode(', ', $items)) > $this->wrapLength && count($items) > 1
-					? "\n" . Nette\Utils\Strings::indent(implode(",\n", $items)) . "\n"
-					: $tmp;
+				$res .= $this->dumpArguments($arg, strlen($res) - strrpos($res, "\n"));
 
 			} else { // $  ->  ::
 				$arg = array_shift($args);
@@ -208,6 +202,22 @@ public function format(string $statement, ...$args): string
 	}
 
 
+	private function dumpArguments(array &$var, int $column): string
+	{
+		$outInline = $outWrapped = '';
+
+		foreach ($var as &$v) {
+			$outInline .= $outInline === '' ? '' : ', ';
+			$outInline .= $this->dumpVar($v, [$var], 0, $column + strlen($outInline));
+			$outWrapped .= ($outWrapped === '' ? '' : ',') . "\n\t" . $this->dumpVar($v, [$var], 1);
+		}
+
+		return count($var) > 1 && (strpos($outInline, "\n") !== false || $column + strlen($outInline) > $this->wrapLength)
+			? $outWrapped . "\n"
+			: $outInline;
+	}
+
+
 	/**
 	 * @return object
 	 * @internal

tests/PhpGenerator/Dumper.format().wrap.phpt

@@ -0,0 +1,50 @@
+<?php
+
+/**
+ * Test: Nette\PhpGenerator\Dumper::format()
+ */
+
+declare(strict_types=1);
+
+use Nette\PhpGenerator\Dumper;
+use Tester\Assert;
+
+
+require __DIR__ . '/../bootstrap.php';
+
+
+$dumper = new Dumper;
+$dumper->wrapLength = 100;
+
+Assert::same('func([1, 2, 3])', $dumper->format('func(?)', [1, 2, 3]));
+
+same('loooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooong([
+	1,
+	2,
+	3,
+])', $dumper->format('loooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooong(?)', [1, 2, 3]));
+
+
+same('looooooooooooooooooooooooooooooooooooooooo([1, 2, 3]) + ooooooooooooooooooooooooooooooooooooooooooooooong([
+	1,
+	2,
+	3,
+])', $dumper->format('looooooooooooooooooooooooooooooooooooooooo(?) + ooooooooooooooooooooooooooooooooooooooooooooooong(?)', [1, 2, 3], [1, 2, 3]));
+
+
+// variadics
+Assert::same('func(1, 2, 3)', $dumper->format('func(...?)', [1, 2, 3]));
+
+
+same('loooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooong(
+	1,
+	2,
+	3
+)', $dumper->format('loooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooong(...?)', [1, 2, 3]));
+
+
+same('looooooooooooooooooooooooooooooooooooooooo(1, 2, 3) + ooooooooooooooooooooooooooooooooooooooooooooooong(
+	1,
+	2,
+	3
+)', $dumper->format('looooooooooooooooooooooooooooooooooooooooo(...?) + ooooooooooooooooooooooooooooooooooooooooooooooong(...?)', [1, 2, 3], [1, 2, 3]));