from workers import Blob, FormData, Response, WorkerEntrypoint


class Default(WorkerEntrypoint):
    async def fetch(self, request):
        if request.url.endswith("/sub"):
            return Response("Hi there!")
        elif request.url.endswith("/sub_headers"):
            assert request.headers.get("Custom-Req-Header") == "123"
            return Response(
                "Hi there!", headers={"Custom-Header-That-Should-Passthrough": True}
            )
        elif request.url.endswith("/redirect"):
            return Response.redirect("https://example.com/sub", status=301)
        elif request.url.endswith("/formdata"):
            data = FormData({"field": "value"})
            return Response(data)
        elif request.url.endswith("/formdatablob"):
            data = FormData({"field": "value"})
            data["blob.py"] = Blob("print(42)", content_type="text/python")
            data.append(
                "metadata",
                Blob("{}", content_type="text/python"),
                filename="metadata.json",
            )
            return Response(data)
        elif request.url.endswith("/ignore"):
            # Just a test path that we ignore, used for requests that we don't care about
            # the result of.
            return Response("ignored")
        else:
            raise ValueError("Unexpected path " + request.url)

    def test(self):
        pass