ExportedNodeResolver - Add support for attributes

src/Dependency/ExportedNode/ExportedAttributeNode.php

@@ -0,0 +1,85 @@
+<?php declare(strict_types = 1);
+
+namespace PHPStan\Dependency\ExportedNode;
+
+use JsonSerializable;
+use PHPStan\Dependency\ExportedNode;
+use ReturnTypeWillChange;
+use function count;
+
+class ExportedAttributeNode implements ExportedNode, JsonSerializable
+{
+
+	/**
+	 * @param array<int|string, string> $args argument name or index(string|int) => value expression (string)
+	 */
+	public function __construct(
+		private string $name,
+		private array $args,
+	)
+	{
+	}
+
+	public function equals(ExportedNode $node): bool
+	{
+		if (!$node instanceof self) {
+			return false;
+		}
+
+		if ($this->name !== $node->name) {
+			return false;
+		}
+
+		if (count($this->args) !== count($node->args)) {
+			return false;
+		}
+
+		foreach ($this->args as $argName => $argValue) {
+			if (!isset($node->args[$argName]) || $argValue !== $node->args[$argName]) {
+				return false;
+			}
+		}
+
+		return true;
+	}
+
+	/**
+	 * @param mixed[] $properties
+	 * @return self
+	 */
+	public static function __set_state(array $properties): ExportedNode
+	{
+		return new self(
+			$properties['name'],
+			$properties['args'],
+		);
+	}
+
+	/**
+	 * @return mixed
+	 */
+	#[ReturnTypeWillChange]
+	public function jsonSerialize()
+	{
+		return [
+			'type' => self::class,
+			'data' => [
+				'name' => $this->name,
+				'args' => $this->args,
+			],
+		];
+	}
+
+	/**
+	 * @param mixed[] $data
+	 * @return self
+	 */
+	public static function decode(array $data): ExportedNode
+	{
+		return new self(
+			$data['name'],
+			$data['args'],
+		);
+	}
+
+}

src/Dependency/ExportedNode/ExportedClassConstantNode.php

@@ -4,12 +4,22 @@
 
 use JsonSerializable;
 use PHPStan\Dependency\ExportedNode;
+use PHPStan\ShouldNotHappenException;
 use ReturnTypeWillChange;
+use function array_map;
+use function count;
 
 class ExportedClassConstantNode implements ExportedNode, JsonSerializable
 {
 
-	public function __construct(private string $name, private string $value)
+	/**
+	 * @param ExportedAttributeNode[] $attributes
+	 */
+	public function __construct(
+		private string $name,
+		private string $value,
+		private array $attributes,
+	)
 	{
 	}
 
@@ -19,6 +29,16 @@ public function equals(ExportedNode $node): bool
 			return false;
 		}
 
+		if (count($this->attributes) !== count($node->attributes)) {
+			return false;
+		}
+
+		foreach ($this->attributes as $i => $attribute) {
+			if (!$attribute->equals($node->attributes[$i])) {
+				return false;
+			}
+		}
+
 		return $this->name === $node->name
 			&& $this->value === $node->value;
 	}
@@ -32,6 +52,7 @@ public static function __set_state(array $properties): ExportedNode
 		return new self(
 			$properties['name'],
 			$properties['value'],
+			$properties['attributes'],
 		);
 	}
 
@@ -44,6 +65,12 @@ public static function decode(array $data): ExportedNode
 		return new self(
 			$data['name'],
 			$data['value'],
+			array_map(static function (array $attributeData): ExportedAttributeNode {
+				if ($attributeData['type'] !== ExportedAttributeNode::class) {
+					throw new ShouldNotHappenException();
+				}
+				return ExportedAttributeNode::decode($attributeData['data']);
+			}, $data['attributes']),
 		);
 	}
 
@@ -58,6 +85,7 @@ public function jsonSerialize()
 			'data' => [
 				'name' => $this->name,
 				'value' => $this->value,
+				'attributes' => $this->attributes,
 			],
 		];
 	}

src/Dependency/ExportedNode/ExportedClassNode.php

@@ -17,6 +17,7 @@ class ExportedClassNode implements ExportedNode, JsonSerializable
 	 * @param string[] $usedTraits
 	 * @param ExportedTraitUseAdaptation[] $traitUseAdaptations
 	 * @param ExportedNode[] $statements
+	 * @param ExportedAttributeNode[] $attributes
 	 */
 	public function __construct(
 		private string $name,
@@ -28,6 +29,7 @@ public function __construct(
 		private array $usedTraits,
 		private array $traitUseAdaptations,
 		private array $statements,
+		private array $attributes,
 	)
 	{
 	}
@@ -50,6 +52,16 @@ public function equals(ExportedNode $node): bool
 			return false;
 		}
 
+		if (count($this->attributes) !== count($node->attributes)) {
+			return false;
+		}
+
+		foreach ($this->attributes as $i => $attribute) {
+			if (!$attribute->equals($node->attributes[$i])) {
+				return false;
+			}
+		}
+
 		if (count($this->traitUseAdaptations) !== count($node->traitUseAdaptations)) {
 			return false;
 		}
@@ -97,6 +109,7 @@ public static function __set_state(array $properties): ExportedNode
 			$properties['usedTraits'],
 			$properties['traitUseAdaptations'],
 			$properties['statements'],
+			$properties['attributes'],
 		);
 	}
 
@@ -118,6 +131,7 @@ public function jsonSerialize()
 				'usedTraits' => $this->usedTraits,
 				'traitUseAdaptations' => $this->traitUseAdaptations,
 				'statements' => $this->statements,
+				'attributes' => $this->attributes,
 			],
 		];
 	}
@@ -147,6 +161,12 @@ public static function decode(array $data): ExportedNode
 
 				return $nodeType::decode($node['data']);
 			}, $data['statements']),
+			array_map(static function (array $attributeData): ExportedAttributeNode {
+				if ($attributeData['type'] !== ExportedAttributeNode::class) {
+					throw new ShouldNotHappenException();
+				}
+				return ExportedAttributeNode::decode($attributeData['data']);
+			}, $data['attributes']),
 		);
 	}
 

src/Dependency/ExportedNode/ExportedEnumNode.php

@@ -4,6 +4,7 @@
 
 use JsonSerializable;
 use PHPStan\Dependency\ExportedNode;
+use PHPStan\ShouldNotHappenException;
 use ReturnTypeWillChange;
 use function array_map;
 use function count;
@@ -14,8 +15,16 @@ class ExportedEnumNode implements ExportedNode, JsonSerializable
 	/**
 	 * @param string[] $implements
 	 * @param ExportedNode[] $statements
+	 * @param ExportedAttributeNode[] $attributes
 	 */
-	public function __construct(private string $name, private ?string $scalarType, private ?ExportedPhpDocNode $phpDoc, private array $implements, private array $statements)
+	public function __construct(
+		private string $name,
+		private ?string $scalarType,
+		private ?ExportedPhpDocNode $phpDoc,
+		private array $implements,
+		private array $statements,
+		private array $attributes,
+	)
 	{
 	}
 
@@ -49,6 +58,16 @@ public function equals(ExportedNode $node): bool
 			return false;
 		}
 
+		if (count($this->attributes) !== count($node->attributes)) {
+			return false;
+		}
+
+		foreach ($this->attributes as $i => $attribute) {
+			if (!$attribute->equals($node->attributes[$i])) {
+				return false;
+			}
+		}
+
 		return $this->name === $node->name
 			&& $this->scalarType === $node->scalarType
 			&& $this->implements === $node->implements;
@@ -66,6 +85,7 @@ public static function __set_state(array $properties): ExportedNode
 			$properties['phpDoc'],
 			$properties['implements'],
 			$properties['statements'],
+			$properties['attributes'],
 		);
 	}
 
@@ -83,6 +103,7 @@ public function jsonSerialize()
 				'phpDoc' => $this->phpDoc,
 				'implements' => $this->implements,
 				'statements' => $this->statements,
+				'attributes' => $this->attributes,
 			],
 		];
 	}
@@ -103,6 +124,12 @@ public static function decode(array $data): ExportedNode
 
 				return $nodeType::decode($node['data']);
 			}, $data['statements']),
+			array_map(static function (array $attributeData): ExportedAttributeNode {
+				if ($attributeData['type'] !== ExportedAttributeNode::class) {
+					throw new ShouldNotHappenException();
+				}
+				return ExportedAttributeNode::decode($attributeData['data']);
+			}, $data['attributes']),
 		);
 	}
 

src/Dependency/ExportedNode/ExportedFunctionNode.php

@@ -14,13 +14,15 @@ class ExportedFunctionNode implements ExportedNode, JsonSerializable
 
 	/**
 	 * @param ExportedParameterNode[] $parameters
+	 * @param ExportedAttributeNode[] $attributes
 	 */
 	public function __construct(
 		private string $name,
 		private ?ExportedPhpDocNode $phpDoc,
 		private bool $byRef,
 		private ?string $returnType,
 		private array $parameters,
+		private array $attributes,
 	)
 	{
 	}
@@ -54,6 +56,16 @@ public function equals(ExportedNode $node): bool
 			return false;
 		}
 
+		if (count($this->attributes) !== count($node->attributes)) {
+			return false;
+		}
+
+		foreach ($this->attributes as $i => $attribute) {
+			if (!$attribute->equals($node->attributes[$i])) {
+				return false;
+			}
+		}
+
 		return $this->name === $node->name
 			&& $this->byRef === $node->byRef
 			&& $this->returnType === $node->returnType;
@@ -71,6 +83,7 @@ public static function __set_state(array $properties): ExportedNode
 			$properties['byRef'],
 			$properties['returnType'],
 			$properties['parameters'],
+			$properties['attributes'],
 		);
 	}
 
@@ -88,6 +101,7 @@ public function jsonSerialize()
 				'byRef' => $this->byRef,
 				'returnType' => $this->returnType,
 				'parameters' => $this->parameters,
+				'attributes' => $this->attributes,
 			],
 		];
 	}
@@ -109,6 +123,12 @@ public static function decode(array $data): ExportedNode
 				}
 				return ExportedParameterNode::decode($parameterData['data']);
 			}, $data['parameters']),
+			array_map(static function (array $attributeData): ExportedAttributeNode {
+				if ($attributeData['type'] !== ExportedAttributeNode::class) {
+					throw new ShouldNotHappenException();
+				}
+				return ExportedAttributeNode::decode($attributeData['data']);
+			}, $data['attributes']),
 		);
 	}