HP-2508 Working on typing for Psalm

src/product/BillingRegistryInterface.php

@@ -13,6 +13,9 @@ interface BillingRegistryInterface extends HasLockInterface
      */
     public function priceTypes(): Generator;
 
+    /**
+     * @return TariffTypeDefinitionInterface[]
+     */
     public function getTariffTypeDefinitions(): array;
 
     public function addTariffType(TariffTypeDefinitionInterface $tariffTypeDefinition): void;

src/product/TariffTypeDefinition.php

@@ -39,7 +39,7 @@ public function belongToTariffType(TariffTypeInterface $tariffType): bool
         return $this->tariffType->equals($tariffType);
     }
 
-    public function ofProduct(ProductInterface $product): TariffTypeDefinitionInterface
+    public function ofProduct(ProductInterface $product): static
     {
         $this->ensureNotLocked();
         $this->product = $product;
@@ -61,7 +61,7 @@ private function ensureProductExists(): void
         }
     }
 
-    public function setPricesSuggester(string $suggesterClass): TariffTypeDefinitionInterface
+    public function setPricesSuggester(string $suggesterClass): static
     {
         $this->ensureNotLocked();
 
@@ -76,7 +76,10 @@ public function withPrices(): PriceTypeDefinitionCollection
         return $this->prices;
     }
 
-    public function withBehaviors(): BehaviorTariffTypeCollection
+    /**
+     * @return BehaviorTariffTypeCollection<TariffTypeDefinition>
+     */
+    public function withBehaviors()
     {
         $this->ensureNotLocked();
 

src/product/TariffTypeDefinitionInterface.php

@@ -8,9 +8,9 @@
 use hiqdev\php\billing\product\trait\HasLockInterface;
 
 /**
- * @template T of PriceTypeDefinitionCollectionInterface
+ * @template-covariant T of PriceTypeDefinitionCollectionInterface
  */
 interface TariffTypeDefinitionInterface extends HasBehaviorsInterface, HasLockInterface
 {
     public function tariffType(): TariffTypeInterface;
 
@@ -22,17 +22,16 @@
      */
     public function belongToTariffType(TariffTypeInterface $tariffType): bool;
 
-    public function ofProduct(ProductInterface $product): self;
+    public function ofProduct(ProductInterface $product): static;
 
     public function getProduct(): ProductInterface;
 
-    public function setPricesSuggester(string $suggesterClass): self;
+    public function setPricesSuggester(string $suggesterClass): static;
 
     /**
-     * @return PriceTypeDefinitionCollectionInterface
-     * @psalm-return T
+     * @return T
      */
-    public function withPrices(): PriceTypeDefinitionCollectionInterface;
+    public function withPrices();
 
-    public function end(): self;
+    public function end();
 }

src/product/behavior/BehaviorCollection.php

@@ -5,7 +5,12 @@
 use hiqdev\php\billing\product\Domain\Model\TariffTypeInterface;
 use hiqdev\php\billing\product\trait\HasLock;
 
+/**
+ * @template TPriceDefinition
+ * @implements BehaviorCollectionInterface<TPriceDefinition>
+ * @psalm-consistent-templates
+ */
 abstract class BehaviorCollection implements BehaviorCollectionInterface
 {
     use HasLock;
 
@@ -21,7 +26,7 @@
         return new \ArrayIterator($this->behaviors);
     }
 
-    public function attach(BehaviorInterface $behavior): self
+    public function attach(BehaviorInterface $behavior): static
     {
         $this->ensureNotLocked();
 

src/product/behavior/BehaviorCollectionInterface.php

@@ -10,6 +10,7 @@
 
 /**
  * @extends IteratorAggregate<int, BehaviorInterface>
+ * @psalm-consistent-templates
  */
 interface BehaviorCollectionInterface extends IteratorAggregate, HasLockInterface
 {
@@ -18,10 +19,7 @@ interface BehaviorCollectionInterface extends IteratorAggregate, HasLockInterfac
      */
     public function getIterator(): Traversable;
 
-    public function attach(BehaviorInterface $behavior): self;
+    public function attach(BehaviorInterface $behavior): static;
 
-    /**
-     * @return TariffTypeDefinitionInterface|PriceTypeDefinitionInterface
-     */
-    public function end(): TariffTypeDefinitionInterface|PriceTypeDefinitionInterface;
+    public function end();
 }

src/product/behavior/BehaviorPriceTypeDefinitionCollection.php

@@ -5,14 +5,34 @@
 use hiqdev\php\billing\product\Domain\Model\TariffTypeInterface;
 use hiqdev\php\billing\product\price\PriceTypeDefinitionInterface;
 
+/**
+ * @template TPriceDefinition
+ * @extends BehaviorCollection<TPriceDefinition>
+ * @psalm-consistent-templates
+ */
 class BehaviorPriceTypeDefinitionCollection extends BehaviorCollection
 {
-    public function __construct(private readonly PriceTypeDefinitionInterface $parent, TariffTypeInterface $tariffType)
-    {
+    /**
+     * @psalm-var TPriceDefinition
+     */
+    private readonly PriceTypeDefinitionInterface $parent;
+
+    /**
+     * @psalm-param TPriceDefinition $parent
+     */
+    public function __construct(
+        PriceTypeDefinitionInterface $parent,
+        TariffTypeInterface $tariffType
+    ) {
+        $this->parent = $parent;
+
         parent::__construct($tariffType);
     }
 
-    public function end(): PriceTypeDefinitionInterface
+    /**
+     * @return TPriceDefinition
+     */
+    public function end()
     {
         return $this->parent;
     }

src/product/behavior/BehaviorTariffTypeCollection.php

@@ -5,13 +5,27 @@
 use hiqdev\php\billing\product\Domain\Model\TariffTypeInterface;
 use hiqdev\php\billing\product\TariffTypeDefinitionInterface;
 
+/**
+ * @template T
+ */
 class BehaviorTariffTypeCollection extends BehaviorCollection
 {
-    public function __construct(private readonly TariffTypeDefinitionInterface $parent, TariffTypeInterface $tariffType)
-    {
+    /**
+     * @psalm-param T $parent
+     */
+    public function __construct(
+        /**
+         * @var T
+         */
+        private readonly TariffTypeDefinitionInterface $parent,
+        TariffTypeInterface $tariffType
+    ) {
         parent::__construct($tariffType);
     }
 
+    /**
+     * @psalm-return T
+     */
     public function end(): TariffTypeDefinitionInterface
     {
         return $this->parent;

src/product/behavior/HasBehaviorsInterface.php

@@ -2,9 +2,16 @@
 
 namespace hiqdev\php\billing\product\behavior;
 
+/**
+ * @template TParentCollection
+ * @psalm-consistent-templates
+ */
 interface HasBehaviorsInterface
 {
-    public function withBehaviors(): BehaviorCollectionInterface;
+//    /**
+//     * @return BehaviorCollectionInterface<TParentCollection>
+//     */
+//    public function withBehaviors();
 
     public function hasBehavior(string $behaviorClassName): bool;
 }

src/product/behavior/TariffTypeBehaviorRegistry.php

@@ -23,16 +23,27 @@
  *  - To avoid code duplication of behavior-related methods in multiple classes.
  *  - To separate concerns by handling behavior-related logic in a dedicated class.
  *  - To improve maintainability and testability of tariff behavior handling.
+ *
+ * @template-covariant T of TariffTypeDefinitionInterface
  */
 final class TariffTypeBehaviorRegistry implements HasLockInterface
 {
+    /**
+     * @var BehaviorTariffTypeCollection<T>
+     */
     private BehaviorTariffTypeCollection $behaviorCollection;
 
+    /**
+     * @psalm-param T $tariffTypeDefinition
+     */
     public function __construct(TariffTypeDefinitionInterface $tariffTypeDefinition, TariffTypeInterface $tariffType)
     {
         $this->behaviorCollection = new BehaviorTariffTypeCollection($tariffTypeDefinition, $tariffType);
     }
 
+    /**
+     * @return BehaviorTariffTypeCollection<T>
+     */
     public function getBehaviors(): BehaviorTariffTypeCollection
     {
         return $this->behaviorCollection;

src/product/invoice/RepresentationCollection.php

@@ -9,8 +9,9 @@
 use Traversable;
 
 /**
- * @template T of PriceTypeDefinition
+ * @template T
  * @implements IteratorAggregate<int, RepresentationInterface>
+ * @psalm-consistent-templates
  */
 class RepresentationCollection implements IteratorAggregate, HasLockInterface
 {
@@ -20,8 +21,18 @@ class RepresentationCollection implements IteratorAggregate, HasLockInterface
 
     private RepresentationUniquenessGuard $uniquenessGuard;
 
-    public function __construct(private readonly PriceTypeDefinition $priceTypeDefinition)
-    {
+    /**
+     * @psalm-var T
+     */
+    private readonly PriceTypeDefinition $priceTypeDefinition;
+
+    /**
+     * @psalm-param T $priceTypeDefinition
+     */
+    public function __construct(
+        PriceTypeDefinition $priceTypeDefinition,
+    ) {
+        $this->priceTypeDefinition = $priceTypeDefinition;
         $this->uniquenessGuard = new RepresentationUniquenessGuard();
     }
 
@@ -33,7 +44,7 @@ public function getIterator(): Traversable
         return new \ArrayIterator($this->representations);
     }
 
-    public function attach(RepresentationInterface $representation): self
+    public function attach(RepresentationInterface $representation): static
     {
         $this->ensureNotLocked();
 
@@ -49,7 +60,7 @@ public function attach(RepresentationInterface $representation): self
     /**
      * @psalm-return T
      */
-    public function end(): PriceTypeDefinition
+    public function end()
     {
         return $this->priceTypeDefinition;
     }