--- /dev/null
+<?php
+
+namespace Drupal\big_pipe\Render;
+
+use Drupal\Core\Asset\AssetCollectionRendererInterface;
+use Drupal\Core\Asset\AssetResolverInterface;
+use Drupal\Core\Config\ConfigFactoryInterface;
+use Drupal\Core\Extension\ModuleHandlerInterface;
+use Drupal\Core\Form\EnforcedResponseException;
+use Drupal\Core\Render\AttachmentsInterface;
+use Drupal\Core\Render\AttachmentsResponseProcessorInterface;
+use Drupal\Core\Render\HtmlResponseAttachmentsProcessor;
+use Drupal\Core\Render\RendererInterface;
+use Symfony\Component\HttpFoundation\RequestStack;
+
+/**
+ * Processes attachments of HTML responses with BigPipe enabled.
+ *
+ * @see \Drupal\Core\Render\HtmlResponseAttachmentsProcessor
+ * @see \Drupal\big_pipe\Render\BigPipe
+ */
+class BigPipeResponseAttachmentsProcessor extends HtmlResponseAttachmentsProcessor {
+
+ /**
+ * The HTML response attachments processor service.
+ *
+ * @var \Drupal\Core\Render\AttachmentsResponseProcessorInterface
+ */
+ protected $htmlResponseAttachmentsProcessor;
+
+ /**
+ * Constructs a BigPipeResponseAttachmentsProcessor object.
+ *
+ * @param \Drupal\Core\Render\AttachmentsResponseProcessorInterface $html_response_attachments_processor
+ * The HTML response attachments processor service.
+ * @param \Drupal\Core\Asset\AssetResolverInterface $asset_resolver
+ * An asset resolver.
+ * @param \Drupal\Core\Config\ConfigFactoryInterface $config_factory
+ * A config factory for retrieving required config objects.
+ * @param \Drupal\Core\Asset\AssetCollectionRendererInterface $css_collection_renderer
+ * The CSS asset collection renderer.
+ * @param \Drupal\Core\Asset\AssetCollectionRendererInterface $js_collection_renderer
+ * The JS asset collection renderer.
+ * @param \Symfony\Component\HttpFoundation\RequestStack $request_stack
+ * The request stack.
+ * @param \Drupal\Core\Render\RendererInterface $renderer
+ * The renderer.
+ * @param \Drupal\Core\Extension\ModuleHandlerInterface $module_handler
+ * The module handler service.
+ */
+ public function __construct(AttachmentsResponseProcessorInterface $html_response_attachments_processor, AssetResolverInterface $asset_resolver, ConfigFactoryInterface $config_factory, AssetCollectionRendererInterface $css_collection_renderer, AssetCollectionRendererInterface $js_collection_renderer, RequestStack $request_stack, RendererInterface $renderer, ModuleHandlerInterface $module_handler) {
+ $this->htmlResponseAttachmentsProcessor = $html_response_attachments_processor;
+ parent::__construct($asset_resolver, $config_factory, $css_collection_renderer, $js_collection_renderer, $request_stack, $renderer, $module_handler);
+ }
+
+ /**
+ * {@inheritdoc}
+ */
+ public function processAttachments(AttachmentsInterface $response) {
+ assert('$response instanceof \Drupal\Core\Render\HtmlResponse');
+
+ // First, render the actual placeholders; this will cause the BigPipe
+ // placeholder strategy to generate BigPipe placeholders. We need those to
+ // exist already so that we can extract BigPipe placeholders. This is hence
+ // a bit of unfortunate but necessary duplication.
+ // @see \Drupal\big_pipe\Render\Placeholder\BigPipeStrategy
+ // (Note this is copied verbatim from
+ // \Drupal\Core\Render\HtmlResponseAttachmentsProcessor::processAttachments)
+ try {
+ $response = $this->renderPlaceholders($response);
+ }
+ catch (EnforcedResponseException $e) {
+ return $e->getResponse();
+ }
+
+ // Extract BigPipe placeholders; HtmlResponseAttachmentsProcessor does not
+ // know (nor need to know) how to process those.
+ $attachments = $response->getAttachments();
+ $big_pipe_placeholders = [];
+ $big_pipe_nojs_placeholders = [];
+ if (isset($attachments['big_pipe_placeholders'])) {
+ $big_pipe_placeholders = $attachments['big_pipe_placeholders'];
+ unset($attachments['big_pipe_placeholders']);
+ }
+ if (isset($attachments['big_pipe_nojs_placeholders'])) {
+ $big_pipe_nojs_placeholders = $attachments['big_pipe_nojs_placeholders'];
+ unset($attachments['big_pipe_nojs_placeholders']);
+ }
+ $html_response = clone $response;
+ $html_response->setAttachments($attachments);
+
+ // Call HtmlResponseAttachmentsProcessor to process all other attachments.
+ $processed_html_response = $this->htmlResponseAttachmentsProcessor->processAttachments($html_response);
+
+ // Restore BigPipe placeholders.
+ $attachments = $processed_html_response->getAttachments();
+ $big_pipe_response = clone $processed_html_response;
+ if (count($big_pipe_placeholders)) {
+ $attachments['big_pipe_placeholders'] = $big_pipe_placeholders;
+ }
+ if (count($big_pipe_nojs_placeholders)) {
+ $attachments['big_pipe_nojs_placeholders'] = $big_pipe_nojs_placeholders;
+ }
+ $big_pipe_response->setAttachments($attachments);
+
+ return $big_pipe_response;
+ }
+
+}