magento2/app/code/Magento/Csp/Model/SubresourceIntegrityRepository.php

<?php
/**
 * Copyright © Magento, Inc. All rights reserved.
 * See COPYING.txt for license details.
 */
declare(strict_types=1);

namespace Magento\Csp\Model;

use Magento\Framework\App\CacheInterface;
use Magento\Framework\Serialize\SerializerInterface;

/**
 * Class contains methods equivalent to repository design to manage SRI hashes in cache.
 */
class SubresourceIntegrityRepository
{
    /**
     * Cache prefix.
     *
     * @var string
     */
    private const CACHE_PREFIX = 'INTEGRITY';

    /**
     * @var array|null
     */
    private ?array $data = null;

    /**
     * @var string|null
     */
    private ?string $context;

    /**
     * @var CacheInterface
     */
    private CacheInterface $cache;

    /**
     * @var SerializerInterface
     */
    private SerializerInterface $serializer;

    /**
     * @var SubresourceIntegrityFactory
     */
    private SubresourceIntegrityFactory $integrityFactory;

    /**
     * @param CacheInterface $cache
     * @param SerializerInterface $serializer
     * @param SubresourceIntegrityFactory $integrityFactory
     * @param string|null $context
     */
    public function __construct(
        CacheInterface $cache,
        SerializerInterface $serializer,
        SubresourceIntegrityFactory $integrityFactory,
        ?string $context = null
    ) {
        $this->cache = $cache;
        $this->serializer = $serializer;
        $this->integrityFactory = $integrityFactory;
        $this->context = $context;
    }

    /**
     * Gets an Integrity object by path.
     *
     * @param string $path
     *
     * @return SubresourceIntegrity|null
     */
    public function getByPath(string $path): ?SubresourceIntegrity
    {
        $data = $this->getData();

        if (isset($data[$path])) {
            return $this->integrityFactory->create(
                [
                    "data" => [
                        "path" => $path,
                        "hash" => $data[$path]
                    ]
                ]
            );
        }

        return null;
    }

    /**
     * Gets all available Integrity objects.
     *
     * @return SubresourceIntegrity[]
     */
    public function getAll(): array
    {
        $result = [];

        foreach ($this->getData() as $path => $hash) {
            $result[] = $this->integrityFactory->create(
                [
                    "data" => [
                        "path" => $path,
                        "hash" => $hash
                    ]
                ]
            );
        }

        return $result;
    }

    /**
     * Saves Integrity object.
     *
     * @param SubresourceIntegrity $integrity
     *
     * @return bool
     */
    public function save(SubresourceIntegrity $integrity): bool
    {
        $data = $this->getData();

        $data[$integrity->getPath()] = $integrity->getHash();

        $this->data = $data;

        return $this->cache->save(
            $this->serializer->serialize($this->data),
            $this->getCacheKey(),
            [self::CACHE_PREFIX]
        );
    }

    /**
     * Saves a bunch of Integrity objects.
     *
     * @param SubresourceIntegrity[] $bunch
     *
     * @return bool
     */
    public function saveBunch(array $bunch): bool
    {
        $data = $this->getData();

        foreach ($bunch as $integrity) {
            $data[$integrity->getPath()] = $integrity->getHash();
        }

        $this->data = $data;

        return $this->cache->save(
            $this->serializer->serialize($this->data),
            $this->getCacheKey(),
            [self::CACHE_PREFIX]
        );
    }

    /**
     * Deletes all Integrity objects.
     *
     * @return bool
     */
    public function deleteAll(): bool
    {
        $this->data = null;

        return $this->cache->remove(
            $this->getCacheKey()
        );
    }

    /**
     * Loads integrity data from a storage.
     *
     * @return array
     */
    private function getData(): array
    {
        if ($this->data === null) {
            $cache = $this->cache->load($this->getCacheKey());

            $this->data = $cache ? $this->serializer->unserialize($cache) : [];
        }

        return $this->data;
    }

    /**
     * Gets a cache key based on current context.
     *
     * @return string
     */
    private function getCacheKey(): string
    {
        $cacheKey = self::CACHE_PREFIX;

        if ($this->context) {
            $cacheKey .= "_" . $this->context;
        }

        return $cacheKey;
    }
}