custom/plugins/SwagPlatformSecurity/src/Fixes/NEXT23562/PatchedEntityTemplateLoader.php line 53

Open in your IDE?
  1. <?php declare(strict_types=1);
  3. namespace Swag\Security\Fixes\NEXT23562;
  5. use Doctrine\DBAL\Connection;
  6. use Shopware\Core\Framework\Adapter\Twig\EntityTemplateLoader;
  7. use Shopware\Core\Framework\Feature;
  8. use Twig\Error\LoaderError;
  9. use Twig\Loader\LoaderInterface;
  10. use Twig\Source;
  13. class PatchedEntityTemplateLoader extends EntityTemplateLoader
  14. {
  15.     /**
  16.      * @var array<string, array<string, array{template: string, updatedAt: \DateTimeInterface|null}|null>>
  17.      */
  18.     private $databaseTemplateCache = [];
  20.     /**
  21.      * @var Connection
  22.      */
  23.     private $connection;
  25.     /**
  26.      * @var string
  27.      */
  28.     private $environment;
  30.     /**
  31.      * @var EntityTemplateLoader
  32.      */
  33.     private $original;
  35.     public function __construct(array $origArgs, EntityTemplateLoader $original, Connection $connection, string $environment)
  36.     {
  37.         parent::__construct(...$origArgs);
  38.         $this->original = $original;
  39.         $this->connection = $connection;
  40.         $this->environment = $environment;
  41.     }
  43.     public function clearInternalCache(): void
  44.     {
  45.         Feature::triggerDeprecationOrThrow(
  46.             'v6.5.0.0',
  47.             Feature::deprecatedMethodMessage(__CLASS__, __METHOD__, 'v6.5.0.0', 'reset()')
  48.         );
  50.         $this->reset();
  51.     }
  53.     public function reset(): void
  54.     {
  55.         $this->databaseTemplateCache = [];
  56.     }
  58.     public function getSourceContext(/* string */ $name): Source
  59.     {
  60.         $template = $this->findDatabaseTemplatePatched($name);
  62.         if (!$template) {
  63.             throw new LoaderError(sprintf('Template "%s" is not defined.', $name));
  64.         }
  66.         return new Source($template['template'], $name);
  67.     }
  69.     public function getCacheKey(/* string */ $name): string
  70.     {
  71.         return $name;
  72.     }
  74.     public function isFresh(/* string */ $name, /* int */ $time): bool
  75.     {
  76.         $template = $this->findDatabaseTemplatePatched($name);
  77.         if (!$template) {
  78.             return false;
  79.         }
  81.         return $template['updatedAt'] === null || $template['updatedAt']->getTimestamp() < $time;
  82.     }
  84.     /**
  85.      * @return bool
  86.      */
  87.     public function exists(/* string */ $name)
  88.     {
  89.         $template = $this->findDatabaseTemplatePatched($name);
  90.         if (!$template) {
  91.             return false;
  92.         }
  94.         return true;
  95.     }
  97.     /**
  98.      * @return array{template: string, updatedAt: \DateTimeInterface|null}|null
  99.      */
  100.     private function findDatabaseTemplatePatched(string $name): ?array
  101.     {
  102.         if (isset($_ENV['DISABLE_EXTENSIONS']) && $_ENV['DISABLE_EXTENSIONS'] === false) {
  103.             return null;
  104.         }
  106.         /*
  107.          * In dev env app templates are directly loaded over the filesystem
  108.          * @see TwigLoaderConfigCompilerPass::addAppTemplatePaths()
  109.          */
  110.         if ($this->environment === 'dev') {
  111.             return null;
  112.         }
  114.         $templateName = $this->splitTemplateName($name);
  115.         $namespace = $templateName['namespace'];
  116.         $path = $templateName['path'];
  118.         if (empty($this->databaseTemplateCache)) {
  119.             $query = 'SELECT
  120.                         `app_template`.`path` AS `path`,
  121.                         `app_template`.`template` AS `template`,
  122.                         `app_template`.`updated_at` AS `updatedAt`,
  123.                         `app`.`name` AS `namespace`
  124.                     FROM `app_template`
  125.                     INNER JOIN `app` ON `app_template`.`app_id` = `app`.`id`
  126.                     WHERE `app_template`.`active` = 1 AND `app`.`active` = 1';
  127.             if (is_callable(
  128.                 [
  129.                     $this->connection,
  130.                     'fetchAllAssociative'
  131.                 ]
  132.             )) {
  133.                 /** @var array<array{path: string, template: string, updatedAt: string|null, namespace: string}> $templates */
  134.                 $templates = $this->connection->fetchAllAssociative($query);
  135.             } else {
  136.                 $templates = $this->connection->fetchAll($query);
  137.             }
  139.             foreach ($templates as $template) {
  140.                 $this->databaseTemplateCache[$template['path']][$template['namespace']] = [
  141.                     'template' => $template['template'],
  142.                     'updatedAt' => $template['updatedAt'] ? new \DateTimeImmutable($template['updatedAt']) : null,
  143.                 ];
  144.             }
  145.         }
  147.         if (\array_key_exists($path, $this->databaseTemplateCache) && \array_key_exists($namespace, $this->databaseTemplateCache[$path])) {
  148.             return $this->databaseTemplateCache[$path][$namespace];
  149.         }
  151.         /** @deprecated tag:v6.5.0 - only for intermediate backwards compatibility */
  152.         if (
  153.             \array_key_exists('../' . $path, $this->databaseTemplateCache)
  154.             && \array_key_exists($namespace, $this->databaseTemplateCache['../' . $path])
  155.         ) {
  156.             return $this->databaseTemplateCache['../' . $path][$namespace];
  157.         }
  159.         // we have already loaded all DB templates
  160.         // if the namespace is not included return null
  161.         return $this->databaseTemplateCache[$path][$namespace] = null;
  162.     }
  164.     /**
  165.      * @return array{namespace: string, path: string}
  166.      */
  167.     private function splitTemplateName(string $template): array
  168.     {
  169.         // remove static template inheritance prefix
  170.         if (mb_strpos($template, '@') !== 0) {
  171.             return ['path' => $template, 'namespace' => ''];
  172.         }
  174.         // remove "@"
  175.         $template = mb_substr($template, 1);
  177.         $template = explode('/', $template);
  178.         $namespace = array_shift($template);
  179.         $template = implode('/', $template);
  181.         return ['path' => $template, 'namespace' => $namespace];
  182.     }
  183. }