From 192ea76dca70f303a5ddd39d55e377f95cfb6119 Mon Sep 17 00:00:00 2001
From: Guy Sartorelli <36352093+GuySartorelli@users.noreply.github.com>
Date: Fri, 19 Jan 2024 17:09:12 +1300
Subject: [PATCH] ENH Add generic types (#182)

---
 src/Contract/StaticPublishingTrigger.php          | 4 ++--
 src/Extension/Engine/SiteTreePublishingEngine.php | 4 ++--
 src/Extension/Publishable/PublishableSiteTree.php | 6 ++++--
 3 files changed, 8 insertions(+), 6 deletions(-)

diff --git a/src/Contract/StaticPublishingTrigger.php b/src/Contract/StaticPublishingTrigger.php
index 6472c882..03b290b1 100644
--- a/src/Contract/StaticPublishingTrigger.php
+++ b/src/Contract/StaticPublishingTrigger.php
@@ -13,7 +13,7 @@ interface StaticPublishingTrigger
      * Provides an SS_List of StaticallyPublishable objects which need to be regenerated.
      *
      * @param array $context An associative array with extra engine-specific information.
-     * @return array|SS_List
+     * @return iterable
      */
     public function objectsToUpdate($context);
 
@@ -21,7 +21,7 @@ public function objectsToUpdate($context);
      * Provides a SS_list of objects that need to be deleted.
      *
      * @param array $context An associative array with extra engine-specific information.
-     * @return array|SS_List
+     * @return iterable
      */
     public function objectsToDelete($context);
 }
diff --git a/src/Extension/Engine/SiteTreePublishingEngine.php b/src/Extension/Engine/SiteTreePublishingEngine.php
index 0b5891c9..582906b9 100644
--- a/src/Extension/Engine/SiteTreePublishingEngine.php
+++ b/src/Extension/Engine/SiteTreePublishingEngine.php
@@ -25,6 +25,8 @@
  * * action - name of the executed action: publish or unpublish
  *
  * @see PublishableSiteTree
+ *
+ * @extends DataExtension<SiteTree>
  */
 class SiteTreePublishingEngine extends SiteTreeExtension implements Resettable
 {
@@ -300,7 +302,6 @@ public function flushChanges()
         $urlsToDelete = $this->getUrlsToDelete();
 
         if ($urlsToUpdate) {
-            /** @var UrlBundleInterface $urlService */
             $urlService = Injector::inst()->create(UrlBundleInterface::class);
             $urlService->addUrls($urlsToUpdate);
 
@@ -314,7 +315,6 @@ public function flushChanges()
         }
 
         if ($urlsToDelete) {
-            /** @var UrlBundleInterface $urlService */
             $urlService = Injector::inst()->create(UrlBundleInterface::class);
             $urlService->addUrls($urlsToDelete);
 
diff --git a/src/Extension/Publishable/PublishableSiteTree.php b/src/Extension/Publishable/PublishableSiteTree.php
index 4b1956e3..ba44dcf7 100644
--- a/src/Extension/Publishable/PublishableSiteTree.php
+++ b/src/Extension/Publishable/PublishableSiteTree.php
@@ -19,6 +19,8 @@
  * an extension via the config system ordering (inject your extension "before" the PublishableSiteTree).
  *
  * @see SiteTreePublishingEngine
+ *
+ * @extends DataExtension<SiteTree>
  */
 class PublishableSiteTree extends DataExtension implements StaticallyPublishable, StaticPublishingTrigger
 {
@@ -36,7 +38,7 @@ public function getMyVirtualPages()
     }
 
     /**
-     * @return array|SS_List
+     * @return iterable<SiteTree>
      */
     public function objectsToUpdate($context)
     {
@@ -92,7 +94,7 @@ public function objectsToUpdate($context)
     /**
      * This method controls which caches will be purged
      *
-     * @return array|SS_List
+     * @return iterable<SiteTree>
      */
     public function objectsToDelete($context)
     {