from wptserve.utils import isomorphic_decode

def main(request, response):
    token = request.GET.first(b"token", None)
    if b"querystate" in request.GET:
        from json import JSONEncoder
        response.headers.set(b"Content-Type", b"text/plain")
        return JSONEncoder().encode(request.server.stash.take(token))
    content = request.GET.first(b"content", None)
    tag = request.GET.first(b"tag", None)
    date = request.GET.first(b"date", None)
    expires = request.GET.first(b"expires", None)
    vary = request.GET.first(b"vary", None)
    cc = request.GET.first(b"cache_control", None)
    redirect = request.GET.first(b"redirect", None)
    inm = request.headers.get(b"If-None-Match", None)
    ims = request.headers.get(b"If-Modified-Since", None)
    pragma = request.headers.get(b"Pragma", None)
    cache_control = request.headers.get(b"Cache-Control", None)
    ignore = b"ignore" in request.GET

    if tag:
        tag = b'"%s"' % tag

    server_state = request.server.stash.take(token)
    if not server_state:
        server_state = []
    state = dict()
    if not ignore:
        if inm:
            state[u"If-None-Match"] = isomorphic_decode(inm)
        if ims:
            state[u"If-Modified-Since"] = isomorphic_decode(ims)
        if pragma:
            state[u"Pragma"] = isomorphic_decode(pragma)
        if cache_control:
            state[u"Cache-Control"] = isomorphic_decode(cache_control)
    server_state.append(state)
    request.server.stash.put(token, server_state)

    if tag:
        response.headers.set(b"ETag", b'%s' % tag)
    elif date:
        response.headers.set(b"Last-Modified", date)
    if expires:
        response.headers.set(b"Expires", expires)
    if vary:
        response.headers.set(b"Vary", vary)
    if cc:
        response.headers.set(b"Cache-Control", cc)

    # The only-if-cached redirect tests wants CORS to be okay, the other tests
    # are all same-origin anyways and don't care.
    response.headers.set(b"Access-Control-Allow-Origin", b"*")

    if redirect:
        response.headers.set(b"Location", redirect)
        response.status = (302, b"Redirect")
        return b""
    elif ((inm is not None and inm == tag) or
          (ims is not None and ims == date)):
        response.status = (304, b"Not Modified")
        return b""
    else:
        response.status = (200, b"OK")
        response.headers.set(b"Content-Type", b"text/plain")
        return content