31 lines
1.4 KiB
Python
31 lines
1.4 KiB
Python
def main(request, response):
|
|
"""
|
|
A handler that does either one of the following based on the provided
|
|
"action" parameter:
|
|
1) action="store-cookie": Stores the provided token and the request cookie
|
|
to the stash, and returns a regular module script content.
|
|
2) action="get-cookie": Retrieves and returns the content stored in the
|
|
stash at the provided token.
|
|
"""
|
|
token = request.GET[b"token"]
|
|
action = request.GET[b"action"]
|
|
|
|
response.status = 200
|
|
response.headers.append(b"Content-Type", b"text/javascript")
|
|
|
|
if b"access_control_allow_credentials_header" in request.GET:
|
|
response.headers.append(b"Access-Control-Allow-Credentials", request.GET[b"access_control_allow_credentials_header"])
|
|
|
|
if b"access_control_allow_origin_header" in request.GET:
|
|
response.headers.append(b"Access-Control-Allow-Origin", request.GET[b"access_control_allow_origin_header"])
|
|
|
|
if b"shared_storage_cross_origin_worklet_allowed_header" in request.GET:
|
|
response.headers.append(b"Shared-Storage-Cross-Origin-Worklet-Allowed", request.GET[b"shared_storage_cross_origin_worklet_allowed_header"])
|
|
|
|
if action == b"store-cookie":
|
|
cookie = request.headers.get(b"Cookie", b"NO_COOKIE_HEADER")
|
|
request.server.stash.put(token, cookie)
|
|
return b""
|
|
else:
|
|
assert action == b"get-cookie"
|
|
return request.server.stash.take(token)
|