chromium/third_party/blink/web_tests/wpt_internal/trust-tokens/resources/trust_token_issuance.py

"""A Python file handler for WPT that handles token issuance requests."""

import importlib
import os
import sys

wpt_internal_dir = os.path.dirname(
    os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
if wpt_internal_dir not in sys.path:
    sys.path.insert(0, wpt_internal_dir)

tt = importlib.import_module("trust-tokens.resources.trust_token_voprf")
issuer = tt.create_trust_token_issuer()


def main(request, response):
    request_data = request.headers.get("Sec-Private-State-Token").decode(
        "utf-8")
    try:
        issuance_response = tt.issue_trust_token(issuer=issuer,
                                                 request_data=request_data,
                                                 key_id=0)
        response.headers.set("Sec-Private-State-Token",
                             issuance_response.to_string())
        response.status = 200
        # Add a response body for the iframe E2E test to read
        response.content = "Trust token issuance succeeded."
    except Exception:
        response.status = 500
        response.content = "Trust token issuance failed."