<?php

namespace Autogedal\TbiBank\Model;

use Magento\Csp\Api\PolicyCollectorInterface;
use Magento\Csp\Model\Policy\FetchPolicy;

class CspCollector implements PolicyCollectorInterface
{
    private Environment $environment;

    public function __construct(Environment $environment)
    {
        $this->environment = $environment;
    }

    public function collect(array $defaultPolicies = []): array
    {
        foreach (['frame-src', 'connect-src'] as $policy) {
            $defaultPolicies[] = new FetchPolicy($policy, false, [$this->environment->getOrigin()]);
        }
        return $defaultPolicies;
    }
}