PhpFile: added setStrictTypes()

Author: dg

readme.md

@@ -464,6 +464,7 @@ PHP files can contains multiple classes, namespaces and comments:
 ```php
 $file = new Nette\PhpGenerator\PhpFile;
 $file->addComment('This file is auto-generated.');
+$file->setStrictTypes(); // adds declare(strict_types=1)
 
 $namespace = $file->addNamespace('Foo');
 $class = $namespace->addClass('A');
@@ -484,6 +485,8 @@ Result:
  * This file is auto-generated.
  */
 
+declare(strict_types=1);
+
 namespace Foo;
 
 class A

src/PhpGenerator/PhpFile.php

@@ -28,6 +28,9 @@ final class PhpFile
 	/** @var PhpNamespace[] */
 	private $namespaces = [];
 
+	/** @var bool */
+	private $strictTypes = false;
+
 
 	public function addClass(string $name): ClassType
 	{
@@ -74,6 +77,23 @@ public function getNamespaces(): array
 	}
 
 
+	/**
+	 * Adds declare(strict_types=1) to output.
+	 * @return static
+	 */
+	public function setStrictTypes(bool $on = true): self
+	{
+		$this->strictTypes = $on;
+		return $this;
+	}
+
+
+	public function getStrictTypes(): bool
+	{
+		return $this->strictTypes;
+	}
+
+
 	public function __toString(): string
 	{
 		try {

src/PhpGenerator/Printer.php

@@ -176,6 +176,7 @@ public function printFile(PhpFile $file): string
 		return Strings::normalize(
 			"<?php\n"
 			. ($file->getComment() ? "\n" . Helpers::formatDocComment($file->getComment() . "\n") . "\n" : '')
+			. ($file->getStrictTypes() ? "declare(strict_types=1);\n\n" : '')
 			. implode("\n\n", $namespaces)
 		) . "\n";
 	}

tests/PhpGenerator/PhpFile.phpt

@@ -80,3 +80,10 @@ $file = new PhpFile;
 $file->addClass('A');
 
 sameFile(__DIR__ . '/expected/PhpFile.globalNamespace.expect', (string) $file);
+
+$file = new PhpFile;
+$file->addComment('This file is auto-generated. DO NOT EDIT!');
+$file->setStrictTypes();
+$file->addClass('A');
+
+sameFile(__DIR__ . '/expected/PhpFile.strictTypes.expect', (string) $file);

tests/PhpGenerator/expected/PhpFile.strictTypes.expect

@@ -0,0 +1,11 @@
+<?php
+
+/**
+ * This file is auto-generated. DO NOT EDIT!
+ */
+
+declare(strict_types=1);
+
+class A
+{
+}