stage interface improvement for stages

src/Jobs/Import.php

@@ -5,6 +5,7 @@
 use Elastic\Elasticsearch\Client;
 use Illuminate\Bus\Queueable;
 use Illuminate\Support\Collection;
+use Matchish\ScoutElasticSearch\Jobs\Stages\StageInterface;
 use Matchish\ScoutElasticSearch\ProgressReportable;
 use Matchish\ScoutElasticSearch\Searchable\ImportSource;
 
@@ -40,6 +41,7 @@ public function handle(Client $elasticsearch): void
         $estimate = $stages->sum->estimate();
         $this->progressBar()->setMaxSteps($estimate);
         $stages->each(function ($stage) use ($elasticsearch) {
+            /** @var StageInterface $stage */
             $this->progressBar()->setMessage($stage->title());
             $stage->handle($elasticsearch);
             $this->progressBar()->advance($stage->estimate());

src/Jobs/Stages/CleanUp.php

@@ -11,7 +11,7 @@
 /**
  * @internal
  */
-final class CleanUp
+final class CleanUp implements StageInterface
 {
     /**
      * @var ImportSource

src/Jobs/Stages/CreateWriteIndex.php

@@ -13,7 +13,7 @@
 /**
  * @internal
  */
-final class CreateWriteIndex
+final class CreateWriteIndex implements StageInterface
 {
     /**
      * @var ImportSource

src/Jobs/Stages/PullFromSource.php

@@ -2,13 +2,14 @@
 
 namespace Matchish\ScoutElasticSearch\Jobs\Stages;
 
+use Elastic\Elasticsearch\Client;
 use Illuminate\Support\Collection;
 use Matchish\ScoutElasticSearch\Searchable\ImportSource;
 
 /**
  * @internal
  */
-final class PullFromSource
+final class PullFromSource implements StageInterface
 {
     /**
      * @var ImportSource
@@ -23,7 +24,7 @@ public function __construct(ImportSource $source)
         $this->source = $source;
     }
 
-    public function handle(): void
+    public function handle(Client $elasticsearch = null): void
     {
         $results = $this->source->get()->filter->shouldBeSearchable();
         if (! $results->isEmpty()) {

src/Jobs/Stages/RefreshIndex.php

@@ -9,7 +9,7 @@
 /**
  * @internal
  */
-final class RefreshIndex
+final class RefreshIndex implements StageInterface
 {
     /**
      * @var Index

src/Jobs/Stages/StageInterface.php

@@ -0,0 +1,12 @@
+<?php
+
+namespace Matchish\ScoutElasticSearch\Jobs\Stages;
+
+use Elastic\Elasticsearch\Client;
+
+interface StageInterface
+{
+    public function title(): string;
+    public function estimate(): int;
+    public function handle(Client $elasticsearch): void;
+}

src/Jobs/Stages/SwitchToNewAndRemoveOldIndex.php

@@ -11,7 +11,7 @@
 /**
  * @internal
  */
-final class SwitchToNewAndRemoveOldIndex
+final class SwitchToNewAndRemoveOldIndex implements StageInterface
 {
     /**
      * @var ImportSource