Update PantheonContentPublisherViewController.php

Author: rkunjappan

src/Controller/PantheonContentPublisherViewController.php

@@ -19,32 +19,35 @@ class PantheonContentPublisherViewController extends EntityViewController {
 
   public function pantheonView(Request $request, $pantheon_id): array {
     $query = $request->query;
-    $is_preview = $query->get('publishingLevel') === 'REALTIME';
     $collection = $query->get('siteId') ?: array_key_first(PantheonDocumentCollection::loadMultiple());
+    if ($query->get('publishingLevel') === 'REALTIME') {
+      return $this->pantheonPreview($collection, $pantheon_id);
+    }
+    $document = $this->loadPantheonDocument($collection, $pantheon_id);
+    $page = $this->view($document);
+    $page['#cache']['contexts'][] = 'url.query_args:publishingLevel';
+    return $page;
+  }
+
+  public function pantheonPreview($collection, $pantheon_id) {
     try {
-      $document = PantheonDocument::load(PantheonDocumentStorage::getEntityId($collection, $pantheon_id));
+      $document = $this->loadPantheonDocument($collection, $pantheon_id);
     }
     catch (GraphQLException $e) {
-      if ($is_preview) {
-        $document = PantheonDocument::create(['collection' => $collection]);
-      }
-      else {
-        throw $e;
-      }
-    }
-    if ($is_preview) {
-      // PantheonTagsFormatter turns this into
-      // <div id="pantheon-content-publisher-preview"></div>
-      $document->get('content')->value = '{"attrs":{"id":"pantheon-content-publisher-preview"}}';
+      $document = PantheonDocument::create(['collection' => $collection]);
     }
+    // PantheonTagsFormatter turns this into
+    // <div id="pantheon-content-publisher-preview"></div>
+    $document->get('content')->value = '{"attrs":{"id":"pantheon-content-publisher-preview"}}';
     $page = $this->view($document);
-    if ($is_preview) {
-      $page['#attached']['library'][] = 'pantheon_content_publisher/preview';
-      $page['#attached']['drupalSettings']['pantheon_content_publisher']['site_id'] = $collection;
-      $page['#attached']['http_header'][] = [PantheonContentPublisherXFrameSubscriber::HEADER_NAME, ''];
-    }
+    $page['#attached']['library'][] = 'pantheon_content_publisher/preview';
+    $page['#attached']['drupalSettings']['pantheon_content_publisher']['site_id'] = $collection;
+    $page['#attached']['http_header'][] = [PantheonContentPublisherXFrameSubscriber::HEADER_NAME, ''];
     $page['#cache']['contexts'][] = 'url.query_args:publishingLevel';
     return $page;
   }
 
+  protected function loadPantheonDocument(float|bool|int|string|null $collection, $pantheon_id): \Drupal\Core\Entity\EntityBase|\Drupal\Core\Entity\EntityInterface|null|PantheonDocument {
+    return PantheonDocument::load(PantheonDocumentStorage::getEntityId($collection, $pantheon_id));
+  }
 }