Merge pull request #131 from alexanderschnitzler/feature/add-datetimeaspect-get-dynamic-return-type-extension

[FEATURE] Add dynamic return type extension for DateTimeAspect::get()

Author: sabbelasichon

extension.neon

@@ -101,6 +101,10 @@ services:
         class: SaschaEgerer\PhpstanTypo3\Type\UserAspectGetDynamicReturnTypeExtension
         tags:
             - phpstan.broker.dynamicMethodReturnTypeExtension
+    -
+        class: SaschaEgerer\PhpstanTypo3\Type\DateTimeAspectGetDynamicReturnTypeExtension
+        tags:
+            - phpstan.broker.dynamicMethodReturnTypeExtension
 parameters:
     bootstrapFiles:
         - phpstan.bootstrap.php

src/Type/DateTimeAspectGetDynamicReturnTypeExtension.php

@@ -0,0 +1,59 @@
+<?php declare(strict_types = 1);
+
+namespace SaschaEgerer\PhpstanTypo3\Type;
+
+use PhpParser\Node\Arg;
+use PhpParser\Node\Expr\MethodCall;
+use PHPStan\Analyser\Scope;
+use PHPStan\Reflection\MethodReflection;
+use PHPStan\Type\Constant\ConstantStringType;
+use PHPStan\Type\DynamicMethodReturnTypeExtension;
+use PHPStan\Type\IntegerType;
+use PHPStan\Type\ObjectType;
+use PHPStan\Type\StringType;
+use TYPO3\CMS\Core\Context\DateTimeAspect;
+
+class DateTimeAspectGetDynamicReturnTypeExtension implements DynamicMethodReturnTypeExtension
+{
+
+	public function getClass(): string
+	{
+		return DateTimeAspect::class;
+	}
+
+	public function isMethodSupported(MethodReflection $methodReflection): bool
+	{
+		return $methodReflection->getName() === 'get';
+	}
+
+	public function getTypeFromMethodCall(
+		MethodReflection $methodReflection,
+		MethodCall $methodCall,
+		Scope $scope
+	): ?\PHPStan\Type\Type
+	{
+		$firstArgument = $methodCall->args[0];
+
+		if (!$firstArgument instanceof Arg) {
+			return null;
+		}
+
+		$argumentType = $scope->getType($firstArgument->value);
+
+		if ($argumentType instanceof ConstantStringType) {
+			switch ($argumentType->getValue()) {
+				case 'timestamp':
+				case 'accessTime':
+					return new IntegerType();
+				case 'iso':
+				case 'timezone':
+					return new StringType();
+				case 'full':
+					return new ObjectType(\DateTimeImmutable::class);
+			}
+		}
+
+		return null;
+	}
+
+}

tests/Unit/Type/DateTimeAspectGetDynamicReturnTypeExtension/DateTimeAspectGetDynamicReturnTypeExtensionTest.php

@@ -0,0 +1,41 @@
+<?php declare(strict_types = 1);
+
+namespace SaschaEgerer\PhpstanTypo3\Tests\Unit\Type\DateTimeAspectGetDynamicReturnTypeExtension;
+
+use PHPStan\Testing\TypeInferenceTestCase;
+
+final class DateTimeAspectGetDynamicReturnTypeExtensionTest extends TypeInferenceTestCase
+{
+
+	/**
+	 *
+	 * @return iterable<mixed>
+	 */
+	public function dataFileAsserts(): iterable
+	{
+		yield from $this->gatherAssertTypes(__DIR__ . '/data/datetime-aspect-get-return-types.php');
+	}
+
+	/**
+	 * @dataProvider dataFileAsserts
+	 * @param string $assertType
+	 * @param string $file
+	 * @param mixed ...$args
+	 */
+	public function testFileAsserts(
+		string $assertType,
+		string $file,
+		...$args
+	): void
+	{
+		$this->assertFileAsserts($assertType, $file, ...$args);
+	}
+
+	public static function getAdditionalConfigFiles(): array
+	{
+		return [
+			__DIR__ . '/../../../../extension.neon',
+		];
+	}
+
+}

tests/Unit/Type/DateTimeAspectGetDynamicReturnTypeExtension/data/datetime-aspect-get-return-types.php

@@ -0,0 +1,22 @@
+<?php declare(strict_types = 1);
+
+namespace SaschaEgerer\PhpstanTypo3\Tests\Unit\Type\DateTimeAspectGetDynamicReturnTypeExtension\data;
+
+use TYPO3\CMS\Core\Context\DateTimeAspect;
+
+use function PHPStan\Testing\assertType;
+
+// phpcs:ignore Squiz.Classes.ClassFileName.NoMatch
+class MyContext
+{
+
+	public function getTests(DateTimeAspect $context): void
+	{
+		assertType('int', $context->get('timestamp'));
+		assertType('string', $context->get('iso'));
+		assertType('string', $context->get('timezone'));
+		assertType(\DateTimeImmutable::class, $context->get('full'));
+		assertType('int', $context->get('accessTime'));
+	}
+
+}