"""Regression tests for enforced CSP alignment with report-only policy (#1909)."""

from __future__ import annotations

from http.server import BaseHTTPRequestHandler

from api.helpers import _security_headers
from server import Handler


class _HeaderCapture:
    def __init__(self):
        self.sent_headers = []

    def send_header(self, key, value):
        self.sent_headers.append((key, value))


def _headers_from_security_helper():
    handler = _HeaderCapture()
    _security_headers(handler)
    return dict(handler.sent_headers)


def _directives(policy: str):
    directives = {}
    for entry in policy.split(";"):
        entry = entry.strip()
        if not entry:
            continue
        key, _, value = entry.partition(" ")
        directives[key] = value.strip()
    return directives


def test_security_helper_sends_enforcing_csp_with_hardening_directives(monkeypatch):
    monkeypatch.delenv("HERMES_WEBUI_CSP_CONNECT_EXTRA", raising=False)

    headers = _headers_from_security_helper()

    policy = headers["Content-Security-Policy"]
    assert "default-src 'self' https://*.cloudflareaccess.com" in policy
    assert "base-uri 'self'" in policy
    assert "form-action 'self'" in policy
    assert "manifest-src 'self' https://*.cloudflareaccess.com" in policy
    assert "script-src 'self' 'unsafe-inline' https://cdn.jsdelivr.net https://static.cloudflareinsights.com blob:" in policy
    assert "style-src 'self' 'unsafe-inline' https://cdn.jsdelivr.net https://fonts.googleapis.com" in policy
    assert "worker-src blob: 'self' https://cdn.jsdelivr.net" in policy
    assert "font-src 'self' data: https://fonts.gstatic.com" in policy
    assert "object-src 'none'" in policy
    assert "frame-ancestors 'none'" in policy
    assert "media-src 'self' data: blob:" in policy
    assert "connect-src 'self' http://127.0.0.1:* http://localhost:* http://ipc.localhost https://127.0.0.1:* https://localhost:* ws://127.0.0.1:* ws://localhost:* https://cdn.jsdelivr.net" in policy


def test_enforcing_csp_honors_valid_extra_connect_origins(monkeypatch):
    monkeypatch.setenv(
        "HERMES_WEBUI_CSP_CONNECT_EXTRA",
        "https://metrics.example.com wss://events.example.com:443",
    )

    headers = _headers_from_security_helper()

    policy = headers["Content-Security-Policy"]
    assert (
        "connect-src 'self' http://127.0.0.1:* http://localhost:* "
        "http://ipc.localhost https://127.0.0.1:* https://localhost:* "
        "ws://127.0.0.1:* ws://localhost:* https://cdn.jsdelivr.net "
        "https://metrics.example.com wss://events.example.com:443; "
    ) in policy


def test_enforcing_csp_allows_trusted_loopback_sidecars_by_default(monkeypatch):
    monkeypatch.delenv("HERMES_WEBUI_CSP_CONNECT_EXTRA", raising=False)

    policy = _headers_from_security_helper()["Content-Security-Policy"]

    assert "connect-src" in policy
    assert "http://127.0.0.1:*" in policy
    assert "http://localhost:*" in policy
    assert "http://ipc.localhost" in policy
    assert "ws://127.0.0.1:*" in policy
    assert "ws://localhost:*" in policy
    assert "http://127.0.0.1:17787" not in policy


def test_enforcing_and_report_only_csp_share_validated_connect_extra(monkeypatch):
    monkeypatch.setenv("HERMES_WEBUI_CSP_CONNECT_EXTRA", "https://metrics.example.com")

    enforced = _headers_from_security_helper()["Content-Security-Policy"]
    report_only = Handler.csp_report_only_policy()

    assert "https://metrics.example.com" in enforced
    assert "https://metrics.example.com" in report_only


def test_report_only_policy_tracks_enforced_directives(monkeypatch):
    monkeypatch.delenv("HERMES_WEBUI_CSP_CONNECT_EXTRA", raising=False)

    enforced = _directives(_headers_from_security_helper()["Content-Security-Policy"])
    report_only = _directives(Handler.csp_report_only_policy())

    assert report_only.pop("report-uri") == "/api/csp-report"
    assert report_only.pop("report-to") == "csp-endpoint"
    assert report_only == enforced


def test_report_only_csp_headers_still_point_to_collector(monkeypatch):
    sent_headers = []
    handler = Handler.__new__(Handler)
    handler.send_header = lambda key, value: sent_headers.append((key, value))
    monkeypatch.setattr(BaseHTTPRequestHandler, "end_headers", lambda self: None)

    Handler.end_headers(handler)

    headers = dict(sent_headers)
    assert "Content-Security-Policy-Report-Only" in headers
    assert headers["Report-To"] == (
        '{"group":"csp-endpoint","max_age":10886400,'
        '"endpoints":[{"url":"/api/csp-report"}]}'
    )
    assert "report-uri /api/csp-report" in headers["Content-Security-Policy-Report-Only"]
    assert "report-to csp-endpoint" in headers["Content-Security-Policy-Report-Only"]


def test_end_headers_reuses_cached_extra_connect_validation(monkeypatch, caplog):
    monkeypatch.setenv(
        "HERMES_WEBUI_CSP_CONNECT_EXTRA",
        "https://metrics.example.com; script-src *",
    )

    sent_headers = []
    handler = Handler.__new__(Handler)
    handler.send_header = lambda key, value: sent_headers.append((key, value))
    monkeypatch.setattr(BaseHTTPRequestHandler, "end_headers", lambda self: None)

    _security_headers(handler)
    Handler.end_headers(handler)

    assert caplog.text.count("Ignoring invalid HERMES_WEBUI_CSP_CONNECT_EXTRA value") == 1