Factory: supports attributes in PHP 8

Author: dg

src/PhpGenerator/Factory.php

@@ -40,6 +40,7 @@ public function fromClassReflection(\ReflectionClass $from, bool $withBodies = f
 		$class->setImplements($ifaces);
 
 		$class->setComment(Helpers::unformatDocComment((string) $from->getDocComment()));
+		$class->setAttributes(self::getAttributes($from));
 		if ($from->getParentClass()) {
 			$class->setExtends($from->getParentClass()->name);
 			$class->setImplements(array_diff($class->getImplements(), $from->getParentClass()->getInterfaceNames()));
@@ -99,6 +100,7 @@ public function fromMethodReflection(\ReflectionMethod $from): Method
 		$method->setReturnReference($from->returnsReference());
 		$method->setVariadic($from->isVariadic());
 		$method->setComment(Helpers::unformatDocComment((string) $from->getDocComment()));
+		$method->setAttributes(self::getAttributes($from));
 		if ($from->getReturnType() instanceof \ReflectionNamedType) {
 			$method->setReturnType($from->getReturnType()->getName());
 			$method->setReturnNullable($from->getReturnType()->allowsNull());
@@ -117,6 +119,7 @@ public function fromFunctionReflection(\ReflectionFunction $from, bool $withBody
 		if (!$from->isClosure()) {
 			$function->setComment(Helpers::unformatDocComment((string) $from->getDocComment()));
 		}
+		$function->setAttributes(self::getAttributes($from));
 		if ($from->getReturnType() instanceof \ReflectionNamedType) {
 			$function->setReturnType($from->getReturnType()->getName());
 			$function->setReturnNullable($from->getReturnType()->allowsNull());
@@ -150,6 +153,7 @@ public function fromParameterReflection(\ReflectionParameter $from): Parameter
 				: $from->getDefaultValue());
 			$param->setNullable($param->isNullable() && $param->getDefaultValue() !== null);
 		}
+		$param->setAttributes(self::getAttributes($from));
 		return $param;
 	}
 
@@ -164,6 +168,7 @@ public function fromConstantReflection(\ReflectionClassConstant $from): Constant
 				: ($from->isProtected() ? ClassType::VISIBILITY_PROTECTED : ClassType::VISIBILITY_PUBLIC)
 		);
 		$const->setComment(Helpers::unformatDocComment((string) $from->getDocComment()));
+		$const->setAttributes(self::getAttributes($from));
 		return $const;
 	}
 
@@ -185,6 +190,7 @@ public function fromPropertyReflection(\ReflectionProperty $from): Property
 			$prop->setInitialized(array_key_exists($prop->getName(), $defaults));
 		}
 		$prop->setComment(Helpers::unformatDocComment((string) $from->getDocComment()));
+		$prop->setAttributes(self::getAttributes($from));
 		return $prop;
 	}
 
@@ -327,4 +333,17 @@ private function parse($from): array
 
 		return [$code, $stmts];
 	}
+
+
+	private function getAttributes($from): array
+	{
+		if (PHP_VERSION_ID < 80000) {
+			return [];
+		}
+		$res = [];
+		foreach ($from->getAttributes() as $attr) {
+			$res[] = new Attribute($attr->getName(), $attr->getArguments());
+		}
+		return $res;
+	}
 }

tests/PhpGenerator/ClassType.from.80.phpt

@@ -13,5 +13,6 @@ require __DIR__ . '/../bootstrap.php';
 require __DIR__ . '/fixtures/classes.php80';
 
 $res[] = ClassType::from(new Abc\Class8(null));
+$res[] = ClassType::from(new Abc\Class9);
 
 sameFile(__DIR__ . '/expected/ClassType.from.80.expect', implode("\n", $res));

tests/PhpGenerator/Closure.attributes.80.phpt

@@ -0,0 +1,22 @@
+<?php
+
+/**
+ * @phpVersion 8.0
+ */
+
+declare(strict_types=1);
+
+use Nette\PhpGenerator\Closure;
+
+
+require __DIR__ . '/../bootstrap.php';
+
+
+$closure = #[ExampleAttribute] function (stdClass $a, $b = null) {};
+
+$function = Closure::from($closure);
+same(
+	'#[ExampleAttribute] function (stdClass $a, $b = null) {
+}',
+	(string) $function
+);

tests/PhpGenerator/GlobalFunction.attributes.80.phpt

@@ -0,0 +1,32 @@
+<?php
+
+/**
+ * @phpVersion 8.0
+ */
+
+declare(strict_types=1);
+
+use Nette\PhpGenerator\GlobalFunction;
+
+
+require __DIR__ . '/../bootstrap.php';
+
+
+#[ExampleAttribute]
+
+
+function func(stdClass $a, $b = null)
+{
+	return 1;
+}
+
+
+$function = GlobalFunction::from('func');
+same(
+	'#[ExampleAttribute]
+function func(stdClass $a, $b = null)
+{
+}
+',
+	(string) $function
+);

tests/PhpGenerator/expected/ClassType.from.80.expect

@@ -7,3 +7,29 @@ class Class8
 	) {
 	}
 }
+
+/**
+ * Description of class.
+ */
+#[\ExampleAttribute]
+#[NamedArguments(foo: 'bar', bar: [1, 2, 3])]
+class Class9
+{
+	/** Commented */
+	#[ExampleAttribute]
+	#[WithArguments(true)]
+	public const FOO = 123;
+
+	/** @var resource */
+	#[ExampleAttribute]
+	public $handle;
+
+
+	/**
+	 * Returns file handle
+	 */
+	#[ExampleAttribute]
+	public function getHandle(#[WithArguments(123)] $mode)
+	{
+	}
+}

tests/PhpGenerator/fixtures/classes.php80

@@ -13,3 +13,28 @@ class Class8
 	) {
 	}
 }
+
+
+/**
+ * Description of class.
+ */
+#[\ExampleAttribute]
+#[NamedArguments(foo: 'bar', bar: [1, 2, 3])]
+class Class9
+{
+	/** Commented */
+	#[ExampleAttribute]
+	#[WithArguments(true)]
+	const FOO = 123;
+
+	/** @var resource */
+	#[ExampleAttribute]
+	public $handle;
+
+
+	/** Returns file handle */
+	#[ExampleAttribute]
+	public function getHandle(#[WithArguments(123)] $mode)
+	{
+	}
+}