37 lines
1.5 KiB
Python
37 lines
1.5 KiB
Python
import os
|
|
from wptserve.utils import isomorphic_encode
|
|
from json import dumps, loads
|
|
|
|
def main(request, response):
|
|
key = request.GET.first(b"key").decode("utf8")
|
|
mode = request.GET.first(b"mode", "content")
|
|
status = int(request.GET.first(b"status", b"200"))
|
|
stash = request.server.stash
|
|
cors = request.GET.first(b"cors", "true")
|
|
if cors == "true" or mode == b"info":
|
|
response.headers.set(b"Access-Control-Allow-Origin", b"*")
|
|
|
|
response.status = status
|
|
with stash.lock:
|
|
requests = loads(stash.take(key) or '[]')
|
|
if mode == b"info":
|
|
response.headers.set(b"Content-Type", "application/json")
|
|
json_reqs = dumps(requests)
|
|
response.content = json_reqs
|
|
stash.put(key, json_reqs)
|
|
return
|
|
else:
|
|
headers = {}
|
|
for header, value in request.headers.items():
|
|
headers[header.decode("utf8")] = value[0].decode("utf8")
|
|
path = request.url
|
|
requests.append({"headers": headers, "url": request.url})
|
|
stash.put(key, dumps(requests))
|
|
|
|
response.headers.set(b"Content-Type", request.GET.first(b"type", "text/plain"))
|
|
response.headers.set(b"Cache-Control", request.GET.first(b"cache-control", b"max-age: 604800"))
|
|
if b"file" in request.GET:
|
|
path = os.path.join(os.path.dirname(isomorphic_encode(__file__)), os.path.basename(request.GET.first(b"file")))
|
|
response.content = open(path, mode=u'rb').read()
|
|
else:
|
|
return request.GET.first(b"content", "123")
|