Abstract deployment check

src/LtiMessageLaunch.php

@@ -178,7 +178,7 @@ public function validate(array $request = null)
     public function migrate()
     {
         if (!$this->shouldMigrate()) {
-            return $this;
+            return $this->ensureDeploymentExists();
         }
 
         if (!isset($this->jwt['body'][LtiConstants::LTI1P1]['oauth_consumer_key_sign'])) {
@@ -191,11 +191,7 @@ public function migrate()
 
         $this->deployment = $this->db->migrateFromLti1p1($this);
 
-        if (!isset($this->deployment)) {
-            throw new LtiException(static::ERR_NO_DEPLOYMENT);
-        }
-
-        return $this;
+        return $this->ensureDeploymentExists();
     }
 
     public function cacheLaunchData()
@@ -522,8 +518,8 @@ protected function validateDeployment(): self
         $client_id = $this->getAud();
         $this->deployment = $this->db->findDeployment($this->jwt['body']['iss'], $this->jwt['body'][LtiConstants::DEPLOYMENT_ID], $client_id);
 
-        if (!isset($this->deployment) && !$this->canMigrate()) {
-            throw new LtiException(static::ERR_NO_DEPLOYMENT);
+        if (!$this->canMigrate()) {
+            return $this->ensureDeploymentExists();
         }
 
         return $this;
@@ -573,6 +569,15 @@ private function getAud(): string
         }
     }
 
+    private function ensureDeploymentExists(): self
+    {
+        if (!isset($this->deployment)) {
+            throw new LtiException(static::ERR_NO_DEPLOYMENT);
+        }
+
+        return $this;
+    }
+
     private function canMigrate(): bool
     {
         return $this->db instanceof IMigrationDatabase;