--- /dev/null
+<?php
+
+namespace Drupal\http2_server_push\Asset;
+
+use Drupal\Core\Asset\AssetCollectionRendererInterface;
+use Symfony\Component\HttpFoundation\RequestStack;
+
+/**
+ * Decorates the JS collection renderer service, adds Server Push.
+ *
+ * @see \Drupal\Core\Asset\JsCollectionRenderer
+ * @see \Drupal\http2_server_push\Render\HtmlResponseAttachmentsProcessor
+ */
+class JsCollectionRenderer implements AssetCollectionRendererInterface {
+
+ use AssetHtmlTagRenderElementTrait;
+
+ /**
+ * The decorated JS collection renderer.
+ *
+ * @var \Drupal\Core\Asset\AssetCollectionRendererInterface
+ */
+ protected $jsCollectionRenderer;
+
+ /**
+ * The request stack.
+ *
+ * @var \Symfony\Component\HttpFoundation\RequestStack
+ */
+ protected $requestStack;
+
+ /**
+ * Constructs a JsCollectionRenderer.
+ *
+ * @param \Drupal\Core\Asset\AssetCollectionRendererInterface $js_collection_renderer
+ * The decorated JS collection renderer.
+ * @param \Symfony\Component\HttpFoundation\RequestStack $request_stack
+ * The request stack.
+ */
+ public function __construct(AssetCollectionRendererInterface $js_collection_renderer, RequestStack $request_stack) {
+ $this->jsCollectionRenderer = $js_collection_renderer;
+ $this->requestStack = $request_stack;
+ }
+
+ /**
+ * {@inheritdoc}
+ */
+ public function render(array $js_assets) {
+ $elements = $this->jsCollectionRenderer->render($js_assets);
+
+ $request = $this->requestStack->getCurrentRequest();
+ $link_headers = $request->attributes->get('http2_server_push_link_headers', []);
+ foreach ($elements as &$element) {
+ if (!static::isScript($element)) {
+ continue;
+ }
+
+ // Locally served JS files that are sent to all browsers can be pushed.
+ if (isset($element['#attributes']['src']) && static::hasRootRelativeUrl($element, 'src') && static::isUnconditional($element)) {
+ $link_header_value = '<' . $element['#attributes']['src'] . '>; rel=preload; as=script';
+ $link_headers[] = $link_header_value;
+
+ // @todo When this is moved into Drupal core, consider allowing bubbling
+ // of bubbleable metadata from the return value of
+ // \Drupal\Core\Render\HtmlResponseAttachmentsProcessor::processAssetLibraries
+ // so this line of code can work, which would mean we would no longer
+ // need to use request attributes. (The problem is that a key assumption
+ // is that rendering set of asset libraries to HTML that loads CSS/JS,
+ // no further attachment bubbling happens. That's a fine assumption, but
+ // this module is the first and sole exception.)
+ //$element['#attached']['http_header'][] = ['Link', $link_header_value];
+ }
+ }
+ $request->attributes->set('http2_server_push_link_headers', $link_headers);
+ return $elements;
+ }
+
+}