Module spin_sdk.http
Module with helpers for wasi http
-
-
-Expand source code -
-"""Module with helpers for wasi http"""
-
-import traceback
-
-from spin_sdk.wit import exports
-from spin_sdk.wit.types import Ok, Err
-from spin_sdk.wit.imports import types, outgoing_handler
-from spin_sdk.wit.imports.types import (
- Method, Method_Get, Method_Head, Method_Post, Method_Put, Method_Delete, Method_Connect, Method_Options,
- Method_Trace, Method_Patch, Method_Other, IncomingRequest, IncomingBody, ResponseOutparam, OutgoingResponse,
- Fields, Scheme, Scheme_Http, Scheme_Https, Scheme_Other, OutgoingRequest, OutgoingBody
-)
-from spin_sdk.wit.imports.streams import StreamError_Closed
-from dataclasses import dataclass
-from collections.abc import Mapping
-from typing import Optional
-from urllib import parse
-
-@dataclass
-class Request:
- """An HTTP request"""
- method: str
- uri: str
- headers: Mapping[str, str]
- body: Optional[bytes]
-
-@dataclass
-class Response:
- """An HTTP response"""
- status: int
- headers: Mapping[str, str]
- body: Optional[bytes]
-
-class IncomingHandler(exports.IncomingHandler):
- """Simplified handler for incoming HTTP requests using blocking, buffered I/O."""
-
- def handle_request(self, request: Request) -> Response:
- """Handle an incoming HTTP request and return a response or raise an error"""
- raise NotImplementedError
-
- def handle(self, request: IncomingRequest, response_out: ResponseOutparam):
- method = request.method()
-
- if isinstance(method, Method_Get):
- method_str = "GET"
- elif isinstance(method, Method_Head):
- method_str = "HEAD"
- elif isinstance(method, Method_Post):
- method_str = "POST"
- elif isinstance(method, Method_Put):
- method_str = "PUT"
- elif isinstance(method, Method_Delete):
- method_str = "DELETE"
- elif isinstance(method, Method_Connect):
- method_str = "CONNECT"
- elif isinstance(method, Method_Options):
- method_str = "OPTIONS"
- elif isinstance(method, Method_Trace):
- method_str = "TRACE"
- elif isinstance(method, Method_Patch):
- method_str = "PATCH"
- elif isinstance(method, Method_Other):
- method_str = method.value
- else:
- raise AssertionError
-
- request_body = request.consume()
- request_stream = request_body.stream()
- body = bytearray()
- while True:
- try:
- body += request_stream.blocking_read(16 * 1024)
- except Err as e:
- if isinstance(e.value, StreamError_Closed):
- request_stream.__exit__()
- IncomingBody.finish(request_body)
- break
- else:
- raise e
-
- request_uri = request.path_with_query()
- if request_uri is None:
- uri = "/"
- else:
- uri = request_uri
-
- try:
- simple_response = self.handle_request(Request(
- method_str,
- uri,
- dict(map(lambda pair: (pair[0], str(pair[1], "utf-8")), request.headers().entries())),
- bytes(body)
- ))
- except:
- traceback.print_exc()
-
- response = OutgoingResponse(Fields())
- response.set_status_code(500)
- ResponseOutparam.set(response_out, Ok(response))
- return
-
- response = OutgoingResponse(Fields.from_list(list(map(
- lambda pair: (pair[0], bytes(pair[1], "utf-8")),
- simple_response.headers.items()
- ))))
- response_body = response.body()
- response.set_status_code(simple_response.status)
- ResponseOutparam.set(response_out, Ok(response))
- response_stream = response_body.write()
- if simple_response.body is not None:
- MAX_BLOCKING_WRITE_SIZE = 4096
- offset = 0
- while offset < len(simple_response.body):
- count = min(len(simple_response.body) - offset, MAX_BLOCKING_WRITE_SIZE)
- response_stream.blocking_write_and_flush(simple_response.body[offset:offset+count])
- offset += count
- response_stream.__exit__()
- OutgoingBody.finish(response_body, None)
-
-def send(request: Request) -> Response:
- """Send an HTTP request and return a response or raise an error"""
-
- match request.method:
- case "GET":
- method: Method = Method_Get()
- case "HEAD":
- method = Method_Head()
- case "POST":
- method = Method_Post()
- case "PUT":
- method = Method_Put()
- case "DELETE":
- method = Method_Delete()
- case "CONNECT":
- method = Method_Connect()
- case "OPTIONS":
- method = Method_Options()
- case "TRACE":
- method = Method_Trace()
- case "PATCH":
- method = Method_Patch()
- case _:
- method = Method_Other(request.method)
-
- url_parsed = parse.urlparse(request.uri)
-
- match url_parsed.scheme:
- case "http":
- scheme: Scheme = Scheme_Http()
- case "https":
- scheme = Scheme_Https()
- case _:
- scheme = Scheme_Other(url_parsed.scheme)
-
- outgoing_request = OutgoingRequest(Fields.from_list(list(map(
- lambda pair: (pair[0], bytes(pair[1], "utf-8")),
- request.headers.items()
- ))))
- outgoing_request.set_method(method)
- outgoing_request.set_scheme(scheme)
- outgoing_request.set_authority(url_parsed.netloc)
- outgoing_request.set_path_with_query(url_parsed.path)
-
- if request.body is not None:
- raise NotImplementedError("todo: handle outgoing request bodies")
-
- future = outgoing_handler.handle(outgoing_request, None)
- pollable = future.subscribe()
-
- while True:
- response = future.get()
- if response is None:
- pollable.block()
- else:
- pollable.__exit__()
- future.__exit__()
-
- if isinstance(response, Ok):
- if isinstance(response.value, Ok):
- response_value = response.value.value
- response_body = response_value.consume()
- response_stream = response_body.stream()
- body = bytearray()
- while True:
- try:
- body += response_stream.blocking_read(16 * 1024)
- except Err as e:
- if isinstance(e.value, StreamError_Closed):
- response_stream.__exit__()
- IncomingBody.finish(response_body)
- simple_response = Response(
- response_value.status(),
- dict(map(
- lambda pair: (pair[0], str(pair[1], "utf-8")),
- response_value.headers().entries()
- )),
- bytes(body)
- )
- response_value.__exit__()
- return simple_response
- else:
- raise e
- else:
- raise response.value
- else:
- raise responseSub-modules
@@ -254,98 +46,18 @@Functions
Send an HTTP request and return a response or raise an error
-
+-Expand source code -
-def send(request: Request) -> Response:
- """Send an HTTP request and return a response or raise an error"""
-
- match request.method:
- case "GET":
- method: Method = Method_Get()
- case "HEAD":
- method = Method_Head()
- case "POST":
- method = Method_Post()
- case "PUT":
- method = Method_Put()
- case "DELETE":
- method = Method_Delete()
- case "CONNECT":
- method = Method_Connect()
- case "OPTIONS":
- method = Method_Options()
- case "TRACE":
- method = Method_Trace()
- case "PATCH":
- method = Method_Patch()
- case _:
- method = Method_Other(request.method)
-
- url_parsed = parse.urlparse(request.uri)
-
- match url_parsed.scheme:
- case "http":
- scheme: Scheme = Scheme_Http()
- case "https":
- scheme = Scheme_Https()
- case _:
- scheme = Scheme_Other(url_parsed.scheme)
-
- outgoing_request = OutgoingRequest(Fields.from_list(list(map(
- lambda pair: (pair[0], bytes(pair[1], "utf-8")),
- request.headers.items()
- ))))
- outgoing_request.set_method(method)
- outgoing_request.set_scheme(scheme)
- outgoing_request.set_authority(url_parsed.netloc)
- outgoing_request.set_path_with_query(url_parsed.path)
-
- if request.body is not None:
- raise NotImplementedError("todo: handle outgoing request bodies")
-
- future = outgoing_handler.handle(outgoing_request, None)
- pollable = future.subscribe()
-
- while True:
- response = future.get()
- if response is None:
- pollable.block()
- else:
- pollable.__exit__()
- future.__exit__()
-
- if isinstance(response, Ok):
- if isinstance(response.value, Ok):
- response_value = response.value.value
- response_body = response_value.consume()
- response_stream = response_body.stream()
- body = bytearray()
- while True:
- try:
- body += response_stream.blocking_read(16 * 1024)
- except Err as e:
- if isinstance(e.value, StreamError_Closed):
- response_stream.__exit__()
- IncomingBody.finish(response_body)
- simple_response = Response(
- response_value.status(),
- dict(map(
- lambda pair: (pair[0], str(pair[1], "utf-8")),
- response_value.headers().entries()
- )),
- bytes(body)
- )
- response_value.__exit__()
- return simple_response
- else:
- raise e
- else:
- raise response.value
- else:
- raise response
+async def send_and_close(sink: Sink,
data: bytes)
+
+async def send_async(request: Request) ‑> Response
+Classes
body += request_stream.blocking_read(16 * 1024) except Err as e: if isinstance(e.value, StreamError_Closed): - request_stream.__exit__() + request_stream.__exit__(None, None, None) IncomingBody.finish(request_body) break else: @@ -430,6 +142,10 @@Classes
ResponseOutparam.set(response_out, Ok(response)) return + if simple_response.headers.get('content-length') is None: + content_length = len(simple_response.body) if simple_response.body is not None else 0 + simple_response.headers['content-length'] = str(content_length) + response = OutgoingResponse(Fields.from_list(list(map( lambda pair: (pair[0], bytes(pair[1], "utf-8")), simple_response.headers.items() @@ -445,7 +161,7 @@Classes
count = min(len(simple_response.body) - offset, MAX_BLOCKING_WRITE_SIZE) response_stream.blocking_write_and_flush(simple_response.body[offset:offset+count]) offset += count - response_stream.__exit__() + response_stream.__exit__(None, None, None) OutgoingBody.finish(response_body, None)Ancestors
@@ -457,18 +173,10 @@Ancestors
Methods
-def handle_request(self, request: Request) ‑> Response +def handle_request(self,
request: Request) ‑> Response-
-
Handle an incoming HTTP request and return a response or raise an error
--Expand source code -
-
-def handle_request(self, request: Request) -> Response: - """Handle an incoming HTTP request and return a response or raise an error""" - raise NotImplementedError
Inherited members
@@ -482,7 +190,7 @@Inherited members
class Request
-(method: str, uri: str, headers: collections.abc.Mapping[str, str], body: Optional[bytes])
+(method: str, uri: str, headers: MutableMapping[str, str], body: bytes | None)
An HTTP request
Inherited members
"""An HTTP request""" method: str uri: str - headers: Mapping[str, str] + headers: MutableMapping[str, str] body: Optional[bytes]Class variables
-
-
var body : Optional[bytes]
+var body : bytes | None- -
var headers : collections.abc.Mapping[str, str]
+var headers : MutableMapping[str, str]- @@ -520,7 +228,7 @@
Class variables
class Response
-(status: int, headers: collections.abc.Mapping[str, str], body: Optional[bytes])
+(status: int, headers: MutableMapping[str, str], body: bytes | None)
An HTTP response
Class variables
class Response: """An HTTP response""" status: int - headers: Mapping[str, str] + headers: MutableMapping[str, str] body: Optional[bytes]Class variables
-
-
var body : Optional[bytes]
+var body : bytes | None- -
var headers : collections.abc.Mapping[str, str]
+var headers : MutableMapping[str, str]- @@ -555,7 +263,6 @@