DP-36788: Add correct cache contexts.

massgov · Jan 31, 2025 · 0cbdd48 · 0cbdd48
1 parent 8960b93
commit 0cbdd48
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/docroot/modules/custom/mass_utility/src/EntityViewBuildAlterGravityFormToken.php b/docroot/modules/custom/mass_utility/src/EntityViewBuildAlterGravityFormToken.php
@@ -41,6 +41,11 @@ public function alter(array &$build, EntityInterface $entity, EntityViewDisplayI
     if ($entity instanceof NodeInterface
       && $entity->bundle() == 'form_page'
       && $build['#view_mode'] == 'full') {
+
+      $build['#cache']['contexts'] = $build['#cache']['contexts'] ?? [];
+      $build['#cache']['contexts'][] = 'url.query_args:gf_token';
+      $build['#cache']['contexts'][] = 'user.roles';
+
       if (empty($build["field_form_url"][0]["#url"])) {
         return;
       }
@@ -50,16 +55,14 @@ public function alter(array &$build, EntityInterface $entity, EntityViewDisplayI
         return;
       }
 
-      $build["field_form_url"]["iframe_url"] = $iframe_url->toString();
-
+      $build["field_form_url"]["iframe_url"]['#markup'] = $iframe_url->toString();
       $gf_token = $this->requestStack->getCurrentRequest()->get('gf_token');
       if (empty($gf_token)) {
         return;
       }
 
       $iframe_url->setOption('query', ['gf_token' => $gf_token]);
-
-      $build["field_form_url"]["iframe_url"] = $iframe_url->toString();
+      $build["field_form_url"]["iframe_url"]['#markup'] = $iframe_url->toString();
     }
   }