Expose ClassMethod and Function_ on MethodReturnStatementsNode and FunctionReturnStatementsNode

src/Node/FunctionReturnStatementsNode.php

@@ -42,6 +42,11 @@ public function __construct(
 		parent::__construct($function->getAttributes());
 	}
 
+	public function getFunction(): Function_
+	{
+		return $this->function;
+	}
+
 	public function getReturnStatements(): array
 	{
 		return $this->returnStatements;

src/Node/MethodReturnStatementsNode.php

@@ -47,6 +47,11 @@ public function __construct(
 		$this->classMethod = $method;
 	}
 
+	public function getClassMethod(): ClassMethod
+	{
+		return $this->classMethod;
+	}
+
 	public function getReturnStatements(): array
 	{
 		return $this->returnStatements;

tests/PHPStan/Node/ReturnStatementsNodeParserNodeRule.php

@@ -0,0 +1,44 @@
+<?php declare(strict_types = 1);
+
+namespace PHPStan\Node;
+
+use PhpParser\Node;
+use PHPStan\Analyser\Scope;
+use PHPStan\Rules\Rule;
+use PHPStan\Rules\RuleErrorBuilder;
+use function sprintf;
+
+/**
+ * @implements Rule<ReturnStatementsNode>
+ */
+class ReturnStatementsNodeParserNodeRule implements Rule
+{
+
+	public function getNodeType(): string
+	{
+		return ReturnStatementsNode::class;
+	}
+
+	public function processNode(Node $node, Scope $scope): array
+	{
+		if ($node instanceof MethodReturnStatementsNode) {
+			$parserNode = $node->getClassMethod();
+		} elseif ($node instanceof FunctionReturnStatementsNode) {
+			$parserNode = $node->getFunction();
+		} else {
+			return [];
+		}
+
+		return [
+			RuleErrorBuilder::message(sprintf(
+				'%s %s(), lines %d-%d, return type node: %s',
+				$parserNode->getType(),
+				$parserNode->name->toString(),
+				$parserNode->getStartLine(),
+				$parserNode->getEndLine(),
+				$parserNode->returnType !== null ? $parserNode->returnType->getType() : 'none',
+			))->identifier('tests.returnStatementsNodeParserNode')->build(),
+		];
+	}
+
+}

tests/PHPStan/Node/ReturnStatementsNodeParserNodeRuleTest.php

@@ -0,0 +1,37 @@
+<?php declare(strict_types = 1);
+
+namespace PHPStan\Node;
+
+use PHPStan\Rules\Rule;
+use PHPStan\Testing\RuleTestCase;
+
+/**
+ * @extends RuleTestCase<ReturnStatementsNodeParserNodeRule>
+ */
+class ReturnStatementsNodeParserNodeRuleTest extends RuleTestCase
+{
+
+	protected function getRule(): Rule
+	{
+		return new ReturnStatementsNodeParserNodeRule();
+	}
+
+	public function testRule(): void
+	{
+		$this->analyse([__DIR__ . '/data/return-statements-node-parser-node.php'], [
+			[
+				'Stmt_ClassMethod doFoo(), lines 8-11, return type node: NullableType',
+				8,
+			],
+			[
+				'Stmt_ClassMethod doBar(), lines 13-16, return type node: none',
+				13,
+			],
+			[
+				'Stmt_Function doBaz(), lines 20-23, return type node: Identifier',
+				20,
+			],
+		]);
+	}
+
+}

tests/PHPStan/Node/data/return-statements-node-parser-node.php

@@ -0,0 +1,23 @@
+<?php
+
+namespace ReturnStatementsNodeParserNode;
+
+class Foo
+{
+
+	public function doFoo(): ?int
+	{
+		return 1;
+	}
+
+	public function doBar()
+	{
+		$f = static fn (): int => 1;
+	}
+
+}
+
+function doBaz(): int
+{
+	return 2;
+}