Printer: added support for printing attributes on PhpFile and TraitUse

Author: dg

src/PhpGenerator/PhpFile.php

@@ -24,6 +24,7 @@
 final class PhpFile
 {
 	use Traits\CommentAware;
+	use Traits\AttributeAware;
 
 	/** @var array<string, PhpNamespace> */
 	private array $namespaces = [];

src/PhpGenerator/Printer.php

@@ -155,6 +155,7 @@ public function printClass(
 				$resolutions = implode(";\n", $trait->getResolutions());
 				$resolutions = Helpers::simplifyTaggedNames($resolutions, $this->namespace);
 				$traits[] = $this->printDocComment($trait)
+					. $this->printAttributes($trait->getAttributes())
 					. 'use ' . $resolver($trait->getName())
 					. ($resolutions
 						? " {\n" . $this->indent($resolutions) . ";\n}\n"
@@ -284,6 +285,7 @@ public function printFile(PhpFile $file): string
 		return "<?php\n"
 			. ($file->getComment() ? "\n" . $this->printDocComment($file) : '')
 			. "\n"
+			. ($file->getAttributes() ? $this->printAttributes($file->getAttributes()) . "\n" : '')
 			. ($file->hasStrictTypes() ? "declare(strict_types=1);\n\n" : '')
 			. implode("\n\n", $namespaces);
 	}

src/PhpGenerator/TraitUse.php

@@ -19,6 +19,7 @@ final class TraitUse
 {
 	use Traits\NameAware;
 	use Traits\CommentAware;
+	use Traits\AttributeAware;
 
 	/** @var list<string> */
 	private array $resolutions = [];

tests/PhpGenerator/ClassType.attributes.phpt

@@ -25,6 +25,9 @@ $class
 		],
 	]);
 
+$class->addTrait('ExampleTrait')
+	->addAttribute('TraitAttribute');
+
 $class->addConstant('FOO', 123)
 	->addComment('Commented')
 	->addAttribute('ExampleAttribute')

tests/PhpGenerator/PhpFile.phpt

@@ -137,6 +137,58 @@ test('file with strict types declaration', function () {
 });
 
 
+test('file with attributes', function () {
+	$file = new PhpFile;
+	$file->addAttribute('FileAttribute');
+	$file->addAttribute('AnotherAttribute', ['key' => 'value']);
+	$file->addClass('A');
+
+	same(
+		<<<'XX'
+			<?php
+
+			#[FileAttribute]
+			#[AnotherAttribute(key: 'value')]
+
+			class A
+			{
+			}
+
+			XX,
+		(string) $file,
+	);
+});
+
+
+test('file with comment, attributes and strict types', function () {
+	$file = new PhpFile;
+	$file->addComment('This file is auto-generated.');
+	$file->addAttribute('FileAttribute');
+	$file->setStrictTypes();
+	$file->addClass('A');
+
+	same(
+		<<<'XX'
+			<?php
+
+			/**
+			 * This file is auto-generated.
+			 */
+
+			#[FileAttribute]
+
+			declare(strict_types=1);
+
+			class A
+			{
+			}
+
+			XX,
+		(string) $file,
+	);
+});
+
+
 test('fromCode extracts file from PHP code', function () {
 	$file = PhpFile::fromCode(file_get_contents(__DIR__ . '/fixtures/classes.php'));
 	Assert::type(PhpFile::class, $file);

tests/PhpGenerator/expected/ClassType.attributes.expect

@@ -6,6 +6,9 @@
 #[Table(name: 'user', constraints: [new UniqueConstraint(name: 'ean', columns: ['ean'])])]
 class Example
 {
+	#[TraitAttribute]
+	use ExampleTrait;
+
 	/** Commented */
 	#[ExampleAttribute]
 	#[WithArguments(true)]