BookStack Code Mirror - bookstack/blob - app/Theming/CustomHtmlHeadContentProvider.php

   1 <?php
   2
   3 namespace BookStack\Theming;
   4
   5 use BookStack\Util\CspService;
   6 use BookStack\Util\HtmlContentFilter;
   7 use BookStack\Util\HtmlNonceApplicator;
   8 use Illuminate\Contracts\Cache\Repository as Cache;
   9
  10 class CustomHtmlHeadContentProvider
  11 {
  12     /**
  13      * @var CspService
  14      */
  15     protected $cspService;
  16
  17     /**
  18      * @var Cache
  19      */
  20     protected $cache;
  21
  22     public function __construct(CspService $cspService, Cache $cache)
  23     {
  24         $this->cspService = $cspService;
  25         $this->cache = $cache;
  26     }
  27
  28     /**
  29      * Fetch our custom HTML head content prepared for use on web pages.
  30      * Content has a nonce applied for CSP.
  31      */
  32     public function forWeb(): string
  33     {
  34         $content = $this->getSourceContent();
  35         $hash = md5($content);
  36         $html = $this->cache->remember('custom-head-web:' . $hash, 86400, function () use ($content) {
  37             return HtmlNonceApplicator::prepare($content);
  38         });
  39
  40         return HtmlNonceApplicator::apply($html, $this->cspService->getNonce());
  41     }
  42
  43     /**
  44      * Fetch our custom HTML head content prepared for use in export formats.
  45      * Scripts are stripped to avoid potential issues.
  46      */
  47     public function forExport(): string
  48     {
  49         $content = $this->getSourceContent();
  50         $hash = md5($content);
  51
  52         return $this->cache->remember('custom-head-export:' . $hash, 86400, function () use ($content) {
  53             return HtmlContentFilter::removeScriptsFromHtmlString($content);
  54         });
  55     }
  56
  57     /**
  58      * Get the original custom head content to use.
  59      */
  60     protected function getSourceContent(): string
  61     {
  62         return setting('app-custom-head', '');
  63     }
  64 }