56 lines
2.1 KiB
Python
56 lines
2.1 KiB
Python
# An HTTP/2 handler for testing Early Hints. Used as an entry point of Early
|
|
# Hints related tests to inject Early Hints response. See comments in
|
|
# `early-hints-helpers.sub.js`.
|
|
|
|
import json
|
|
import os
|
|
import time
|
|
|
|
|
|
def _remove_relative_resources_prefix(path):
|
|
if path.startswith("resources/"):
|
|
return path[len("resources/"):]
|
|
return path
|
|
|
|
|
|
def handle_headers(frame, request, response):
|
|
preload_headers = []
|
|
for encoded_preload in request.GET.get_list(b"preloads"):
|
|
preload = json.loads(encoded_preload.decode("utf-8"))
|
|
header = "<{}>; rel=preload; as={}".format(preload["url"], preload["as_attr"])
|
|
if "crossorigin_attr" in preload:
|
|
crossorigin = preload["crossorigin_attr"]
|
|
if crossorigin:
|
|
header += "; crossorigin={}".format(crossorigin)
|
|
else:
|
|
header += "; crossorigin"
|
|
if "fetchpriority_attr" in preload:
|
|
fetchpriority = preload["fetchpriority_attr"]
|
|
if fetchpriority:
|
|
header += "; fetchpriority={}".format(fetchpriority)
|
|
preload_headers.append(header.encode())
|
|
|
|
# Send a 103 response.
|
|
early_hints = [(b":status", b"103")]
|
|
for header in preload_headers:
|
|
early_hints.append((b"link", header))
|
|
response.writer.write_raw_header_frame(headers=early_hints,
|
|
end_headers=True)
|
|
|
|
# Simulate the response generation is taking time.
|
|
time.sleep(0.2)
|
|
response.status = 200
|
|
response.headers[b"content-type"] = "text/html"
|
|
if request.GET[b"exclude_preloads_from_ok_response"].decode("utf-8") != "true":
|
|
for header in preload_headers:
|
|
response.headers.append(b"link", header)
|
|
response.write_status_headers()
|
|
|
|
|
|
def main(request, response):
|
|
test_path = _remove_relative_resources_prefix(
|
|
request.GET[b"test_url"].decode("utf-8"))
|
|
current_dir = os.path.dirname(os.path.realpath(__file__))
|
|
file_path = os.path.join(current_dir, test_path)
|
|
test_content = open(file_path, "r").read()
|
|
response.writer.write_data(item=test_content, last=True)
|