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)