Warm cache in batch

Author: tangrufus

src/MarkClosedPluginAsAbandoned.php

@@ -52,6 +52,28 @@ public function __construct(
         private IOInterface $io,
     ) {}
 
+    public function warmCache(CompletePackageInterface ...$packages): void
+    {
+        $slugs = array_map(
+            fn (CompletePackageInterface $package): ?string => $this->slug(
+                ...$package->getDistUrls(),
+                ...$package->getSourceUrls(),
+            ),
+            $packages,
+        );
+        $slugs = array_filter($slugs, static fn (?string $slug) => $slug !== null);
+
+        $slugCount = count($slugs);
+        $message = sprintf(
+            'Warming cache for WordPress.org closed plugin status for %d %s',
+            $slugCount,
+            $slugCount === 1 ? 'slug' : 'slugs',
+        );
+        $this->io->debug($message);
+
+        $this->client->warmCache(...$slugs);
+    }
+
     public function __invoke(CompletePackageInterface $package): void
     {
         $this->io->debug(

src/Plugin.php

@@ -48,8 +48,14 @@ public function uninstall(Composer $composer, IOInterface $io): void
     public static function getSubscribedEvents(): array
     {
         return [
-            PackageEvents::PRE_PACKAGE_INSTALL => ['markClosedAsAbandoned', PHP_INT_MAX - 1000],
-            PackageEvents::PRE_PACKAGE_UPDATE => ['markClosedAsAbandoned', PHP_INT_MAX - 1000],
+            PackageEvents::PRE_PACKAGE_INSTALL => [
+                ['warmCache', PHP_INT_MAX - 500],
+                ['markClosedAsAbandoned', PHP_INT_MAX - 1000],
+            ],
+            PackageEvents::PRE_PACKAGE_UPDATE => [
+                ['warmCache', PHP_INT_MAX - 500],
+                ['markClosedAsAbandoned', PHP_INT_MAX - 1000],
+            ],
 
             ScriptEvents::POST_INSTALL_CMD => ['markClosedLockedPackagesIfNotAlready', PHP_INT_MAX - 1000],
             ScriptEvents::POST_UPDATE_CMD => ['markClosedLockedPackagesIfNotAlready', PHP_INT_MAX - 1000],
@@ -58,6 +64,22 @@ public static function getSubscribedEvents(): array
         ];
     }
 
+    public function warmCache(PackageEvent $event): void
+    {
+        $packages = array_map(
+            static fn ($operation) => match (true) {
+                $operation instanceof InstallOperation => $operation->getPackage(),
+                $operation instanceof UpdateOperation => $operation->getTargetPackage(),
+                default => null,
+            },
+            $event->getOperations(),
+        );
+        $packages = array_filter($packages, static fn ($package) => $package instanceof CompletePackageInterface);
+        $packages = array_filter($packages, static fn ($package) => ! $package->isAbandoned());
+
+        $this->markClosedAsAbandoned->warmCache(...$packages);
+    }
+
     public function markClosedAsAbandoned(PackageEvent $event): void
     {
         $operation = $event->getOperation();
@@ -95,6 +117,8 @@ public function markClosedLockedPackagesIfNotAlready(ScriptEvent $event): void
             fn ($package) => ! in_array($package->getPrettyName(), $this->marked, true),
         );
 
+        $this->markClosedAsAbandoned->warmCache(...$packages);
+
         foreach ($packages as $package) {
             $this->marked[] = $package->getPrettyName();
             $this->markClosedAsAbandoned->__invoke($package);

src/WpOrg/Api/Client.php

@@ -18,6 +18,19 @@ public function __construct(
         private CacheInterface $cache,
     ) {}
 
+    public function warmCache(string ...$slugs): void
+    {
+        $slugs = array_map('trim', $slugs);
+        $slugs = array_filter($slugs, static fn (string $slug) => $slug !== '');
+
+        $promises = array_map(
+            fn (string $slug) => $this->isClosedAsync($slug),
+            $slugs,
+        );
+
+        $this->loop->wait($promises);
+    }
+
     public function isClosed(string $slug): bool
     {
         $slug = trim($slug);