Method, Parameter: added support for PHP 7 type hints

Author: pavelkouril

src/PhpGenerator/Method.php

@@ -51,6 +51,9 @@ class Method extends Nette\Object
 	/** @var PhpNamespace|NULL */
 	private $namespace;
 
+	/** @var string|NULL */
+	private $returnType;
+
 
 	/**
 	 * @return self
@@ -79,6 +82,9 @@ public static function from($from)
 		$method->returnReference = $from->returnsReference();
 		$method->variadic = PHP_VERSION_ID >= 50600 && $from->isVariadic();
 		$method->documents = $from->getDocComment() ? array(preg_replace('#^\s*\* ?#m', '', trim($from->getDocComment(), "/* \r\n\t"))) : array();
+		if (PHP_VERSION_ID >= 70000 && $from->hasReturnType()) {
+			$method->returnType = (string) $from->getReturnType();
+		}
 		return $method;
 	}
 
@@ -123,6 +129,7 @@ public function __toString()
 			. ' ' . $this->name
 			. '(' . implode(', ', $parameters) . ')'
 			. ($this->uses ? ' use (' . implode(', ', $uses) . ')' : '')
+			. ($this->returnType ? ': ' . $namespace->unresolveName($this->returnType) : '')
 			. ($this->abstract || $this->body === FALSE ? ';'
 				: ($this->name ? "\n" : ' ') . "{\n" . Nette\Utils\Strings::indent(ltrim(rtrim($this->body) . "\n"), 1) . '}');
 	}
@@ -408,4 +415,24 @@ public function setNamespace(PhpNamespace $val = NULL)
 		return $this;
 	}
 
+
+	/**
+	 * @param  string|NULL
+	 * @return self
+	 */
+	public function setReturnType($val)
+	{
+		$this->returnType = $val ? (string) $val : NULL;
+		return $this;
+	}
+
+
+	/**
+	 * @return string|NULL
+	 */
+	public function getReturnType()
+	{
+		return $this->returnType;
+	}
+
 }

src/PhpGenerator/Parameter.php

@@ -38,7 +38,9 @@ public static function from(\ReflectionParameter $from)
 	{
 		$param = new static($from->getName());
 		$param->reference = $from->isPassedByReference();
-		if ($from->isArray()) {
+		if (PHP_VERSION_ID >= 70000) {
+			$param->typeHint = $from->hasType() ? (string) $from->getType() : NULL;
+		} elseif ($from->isArray()) {
 			$param->typeHint = 'array';
 		} elseif (PHP_VERSION_ID >= 50400 && $from->isCallable()) {
 			$param->typeHint = 'callable';

tests/PhpGenerator/Method.returnTypes.phpt

@@ -0,0 +1,48 @@
+<?php
+
+/**
+ * Test: Nette\PhpGenerator - PHP7 return type declarations
+ * @phpversion 7.0
+ */
+
+namespace A
+{
+	class Foo {}
+}
+
+namespace
+{
+	use Nette\PhpGenerator\Method;
+	use Tester\Assert;
+
+
+	require __DIR__ . '/../bootstrap.php';
+
+	// test from
+
+	interface A
+	{
+		function testClass(): \A\Foo;
+		function testScalar(): string;
+	}
+
+	$method = Method::from(A::class .'::testClass');
+	Assert::same('A\Foo', $method->getReturnType());
+
+	$method = Method::from(A::class .'::testScalar');
+	Assert::same('string', $method->getReturnType());
+
+	// generating methods with return type declarations
+
+	$method = (new Method('create'))
+		->setReturnType('Foo')
+		->setBody('return new Foo();');
+
+	Assert::match(
+		'function create(): Foo
+{
+	return new Foo();
+}
+', (string) $method);
+
+}

tests/PhpGenerator/Method.scalarParameters.phpt

@@ -0,0 +1,49 @@
+<?php
+
+/**
+ * Test: Nette\PhpGenerator - PHP7 scalar type hints
+ * @phpversion 7.0
+ */
+
+
+use Nette\PhpGenerator\Method;
+use Tester\Assert;
+
+
+require __DIR__ . '/../bootstrap.php';
+
+// test from
+
+
+interface Foo
+{
+	function scalars(string $a, bool $b, int $c, float $d);
+}
+
+$method = Method::from(Foo::class . '::scalars');
+Assert::same('string', $method->getParameters()['a']->getTypeHint());
+
+$method = Method::from(Foo::class . '::scalars');
+Assert::same('bool', $method->getParameters()['b']->getTypeHint());
+
+$method = Method::from(Foo::class . '::scalars');
+Assert::same('int', $method->getParameters()['c']->getTypeHint());
+
+$method = Method::from(Foo::class . '::scalars');
+Assert::same('float', $method->getParameters()['d']->getTypeHint());
+
+
+// generating methods with scalar type hints
+
+$method = (new Method)
+	->setName('create')
+	->setBody('return null;');
+$method->addParameter('a')->setTypeHint('string');
+$method->addParameter('b')->setTypeHint('bool');
+
+Assert::match(
+'function create(string $a, bool $b)
+{
+	return null;
+}
+', (string) $method);