added support for PHP 8 attributes

Author: dg

readme.md

@@ -346,6 +346,44 @@ Result:
 fn ($a, $b) => $a + $b
 ```
 
+Attributes
+----------
+
+You can add PHP 8 attributes to all classes, methods, properties, constants, functions, closures and parameters.
+
+```php
+$class = new Nette\PhpGenerator\ClassType('Demo');
+$class->addAttribute('Deprecated');
+
+$class->addProperty('list')
+	->addAttribute('WithArguments', [1, 2]);
+
+$method = $class->addMethod('count')
+	->addAttribute('Foo\Cached', ['mode' => true]);
+
+$method->addParameter('items')
+	->addAttribute('Bar');
+
+echo $class;
+```
+
+Result:
+
+```php
+#[Deprecated]
+class Demo
+{
+	#[WithArguments(1, 2)]
+	public $list;
+
+
+	#[Foo\Cached(mode: true)]
+	public function count(#[Bar] $items)
+	{
+	}
+}
+```
+
 Method and Function Body Generator
 ----------------------------------
 
@@ -468,7 +506,7 @@ Class Names Resolving
 ---------------------
 
 **When the class is part of the namespace, it is rendered slightly differently**: all types (ie. type hints, return types, parent class name,
-implemented interfaces and used traits) are automatically *resolved* (unless you turn it off, see below).
+implemented interfaces, used traits and attributes) are automatically *resolved* (unless you turn it off, see below).
 It means that you have to **use full class names** in definitions and they will be replaced
 with aliases (according to the use-statements) or fully qualified names in the resulting code:
 

src/PhpGenerator/Attribute.php

@@ -0,0 +1,49 @@
+<?php
+
+/**
+ * This file is part of the Nette Framework (https://nette.org)
+ * Copyright (c) 2004 David Grudl (https://davidgrudl.com)
+ */
+
+declare(strict_types=1);
+
+namespace Nette\PhpGenerator;
+
+use Nette;
+
+
+/**
+ * PHP Attribute.
+ */
+final class Attribute
+{
+	use Nette\SmartObject;
+
+	/** @var string */
+	private $name;
+
+	/** @var array */
+	private $args;
+
+
+	public function __construct(string $name, array $args)
+	{
+		if (!Helpers::isNamespaceIdentifier($name)) {
+			throw new Nette\InvalidArgumentException("Value '$name' is not valid attribute name.");
+		}
+		$this->name = $name;
+		$this->args = $args;
+	}
+
+
+	public function getName(): string
+	{
+		return $this->name;
+	}
+
+
+	public function getArguments(): array
+	{
+		return $this->args;
+	}
+}

src/PhpGenerator/ClassType.php

@@ -22,6 +22,7 @@ final class ClassType
 {
 	use Nette\SmartObject;
 	use Traits\CommentAware;
+	use Traits\AttributeAware;
 
 	public const
 		TYPE_CLASS = 'class',

src/PhpGenerator/Closure.php

@@ -21,6 +21,7 @@ final class Closure
 {
 	use Nette\SmartObject;
 	use Traits\FunctionLike;
+	use Traits\AttributeAware;
 
 	/** @var Parameter[] */
 	private $uses = [];

src/PhpGenerator/Constant.php

@@ -21,6 +21,7 @@ final class Constant
 	use Traits\NameAware;
 	use Traits\VisibilityAware;
 	use Traits\CommentAware;
+	use Traits\AttributeAware;
 
 	/** @var mixed */
 	private $value;

src/PhpGenerator/GlobalFunction.php

@@ -23,6 +23,7 @@ final class GlobalFunction
 	use Traits\FunctionLike;
 	use Traits\NameAware;
 	use Traits\CommentAware;
+	use Traits\AttributeAware;
 
 	public static function from(string $function): self
 	{

src/PhpGenerator/Method.php

@@ -24,6 +24,7 @@ final class Method
 	use Traits\NameAware;
 	use Traits\VisibilityAware;
 	use Traits\CommentAware;
+	use Traits\AttributeAware;
 
 	/** @var string|null */
 	private $body = '';

src/PhpGenerator/Parameter.php

@@ -21,6 +21,7 @@ class Parameter
 {
 	use Nette\SmartObject;
 	use Traits\NameAware;
+	use Traits\AttributeAware;
 
 	/** @var bool */
 	private $reference = false;

src/PhpGenerator/Printer.php

@@ -39,6 +39,7 @@ class Printer
 	public function printFunction(GlobalFunction $function, PhpNamespace $namespace = null): string
 	{
 		return Helpers::formatDocComment($function->getComment() . "\n")
+			. self::printAttributes($function->getAttributes(), $namespace)
 			. 'function '
 			. ($function->getReturnReference() ? '&' : '')
 			. $function->getName()
@@ -58,7 +59,8 @@ public function printClosure(Closure $closure): string
 			? "\n" . $this->indentation . implode(",\n" . $this->indentation, $uses) . "\n"
 			: $tmp;
 
-		return 'function '
+		return self::printAttributes($closure->getAttributes(), null, true)
+			. 'function '
 			. ($closure->getReturnReference() ? '&' : '')
 			. $this->printParameters($closure, null)
 			. ($uses ? " use ($useStr)" : '')
@@ -75,7 +77,8 @@ public function printArrowFunction(Closure $closure): string
 			}
 		}
 
-		return 'fn '
+		return self::printAttributes($closure->getAttributes(), null)
+			. 'fn '
 			. ($closure->getReturnReference() ? '&' : '')
 			. $this->printParameters($closure, null)
 			. $this->printReturnType($closure, null)
@@ -87,6 +90,7 @@ public function printMethod(Method $method, PhpNamespace $namespace = null): str
 	{
 		$method->validate();
 		return Helpers::formatDocComment($method->getComment() . "\n")
+			. self::printAttributes($method->getAttributes(), $namespace)
 			. ($method->isAbstract() ? 'abstract ' : '')
 			. ($method->isFinal() ? 'final ' : '')
 			. ($method->getVisibility() ? $method->getVisibility() . ' ' : '')
@@ -122,6 +126,7 @@ public function printClass(ClassType $class, PhpNamespace $namespace = null): st
 		foreach ($class->getConstants() as $const) {
 			$def = ($const->getVisibility() ? $const->getVisibility() . ' ' : '') . 'const ' . $const->getName() . ' = ';
 			$consts[] = Helpers::formatDocComment((string) $const->getComment())
+				. self::printAttributes($const->getAttributes(), $namespace)
 				. $def
 				. $this->dump($const->getValue(), strlen($def)) . ";\n";
 		}
@@ -134,6 +139,7 @@ public function printClass(ClassType $class, PhpNamespace $namespace = null): st
 				. '$' . $property->getName());
 
 			$properties[] = Helpers::formatDocComment((string) $property->getComment())
+				. self::printAttributes($property->getAttributes(), $namespace)
 				. $def
 				. ($property->getValue() === null && !$property->isInitialized() ? '' : ' = ' . $this->dump($property->getValue(), strlen($def) + 3)) // 3 = ' = '
 				. ";\n";
@@ -154,6 +160,7 @@ public function printClass(ClassType $class, PhpNamespace $namespace = null): st
 
 		return Strings::normalize(
 			Helpers::formatDocComment($class->getComment() . "\n")
+			. self::printAttributes($class->getAttributes(), $namespace)
 			. ($class->isAbstract() ? 'abstract ' : '')
 			. ($class->isFinal() ? 'final ' : '')
 			. ($class->getName() ? $class->getType() . ' ' . $class->getName() . ' ' : '')
@@ -259,14 +266,15 @@ public function printParameters($function, PhpNamespace $namespace = null): stri
 			$promoted = $param instanceof PromotedParameter ? $param : null;
 			$params[] =
 				($promoted ? Helpers::formatDocComment((string) $promoted->getComment()) : '')
-				. ($promoted ? ($promoted->getVisibility() ?: 'public') . ' ' : '')
+				. ($attrs = self::printAttributes($param->getAttributes(), $namespace, true))
+				. ($promoted ? ($param->getVisibility() ?: 'public') . ' ' : '')
 				. ltrim($this->printType($type, $param->isNullable(), $namespace) . ' ')
 				. ($param->isReference() ? '&' : '')
 				. ($variadic ? '...' : '')
 				. '$' . $param->getName()
 				. ($param->hasDefaultValue() && !$variadic ? ' = ' . $this->dump($param->getDefaultValue()) : '');
 
-			$special = $special || $promoted;
+			$special = $special || $promoted || $attrs;
 		}
 
 		$line = implode(', ', $params);
@@ -296,6 +304,22 @@ private function printReturnType($function, ?PhpNamespace $namespace): string
 	}
 
 
+	private function printAttributes(array $attrs, ?PhpNamespace $namespace, bool $inline = false): string
+	{
+		if (!$attrs) {
+			return '';
+		}
+		$items = [];
+		foreach ($attrs as $attr) {
+			$args = (new Dumper)->format('...?', $attr->getArguments());
+			$items[] = $this->printType($attr->getName(), false, $namespace) . ($args ? "($args)" : '');
+		}
+		return $inline
+			? '#[' . implode(', ', $items) . '] '
+			: '#[' . implode("]\n#[", $items) . "]\n";
+	}
+
+
 	private function joinProperties(array $props)
 	{
 		return $this->linesBetweenProperties

src/PhpGenerator/Property.php

@@ -23,6 +23,7 @@ final class Property
 	use Traits\NameAware;
 	use Traits\VisibilityAware;
 	use Traits\CommentAware;
+	use Traits\AttributeAware;
 
 	/** @var mixed */
 	private $value;