Merge pull request #261 from thomas-huegel/main

add SeeOther among common HTTP responses

Author: saviorand

lightbug_http/__init__.mojo

@@ -1,4 +1,11 @@
-from lightbug_http.http import HTTPRequest, HTTPResponse, OK, NotFound, StatusCode
+from lightbug_http.http import (
+    HTTPRequest,
+    HTTPResponse,
+    OK,
+    NotFound,
+    SeeOther,
+    StatusCode,
+)
 from lightbug_http.uri import URI
 from lightbug_http.header import Header, Headers, HeaderKey
 from lightbug_http.cookie import Cookie, RequestCookieJar, ResponseCookieJar

lightbug_http/cookie/response_cookie_jar.mojo

@@ -24,7 +24,11 @@ struct ResponseCookieKey(KeyElement):
         return not (self == other)
 
     fn __eq__(self: Self, other: Self) -> Bool:
-        return self.name == other.name and self.domain == other.domain and self.path == other.path
+        return (
+            self.name == other.name
+            and self.domain == other.domain
+            and self.path == other.path
+        )
 
     fn __moveinit__(out self: Self, owned existing: Self):
         self.name = existing.name
@@ -41,7 +45,7 @@ struct ResponseCookieKey(KeyElement):
 
 
 @value
-struct ResponseCookieJar(Writable, Stringable, Copyable, Movable, Sized):
+struct ResponseCookieJar(Copyable, Movable, Sized, Stringable, Writable):
     var _inner: Dict[ResponseCookieKey, Cookie]
 
     fn __init__(out self):
@@ -52,6 +56,11 @@ struct ResponseCookieJar(Writable, Stringable, Copyable, Movable, Sized):
         for cookie in cookies:
             self.set_cookie(cookie)
 
+    fn __init__(out self, cookies: List[Cookie]):
+        self._inner = Dict[ResponseCookieKey, Cookie]()
+        for cookie in cookies:
+            self.set_cookie(cookie)
+
     @always_inline
     fn __setitem__(mut self, key: ResponseCookieKey, value: Cookie):
         self._inner[key] = value
@@ -81,7 +90,9 @@ struct ResponseCookieJar(Writable, Stringable, Copyable, Movable, Sized):
 
     @always_inline
     fn set_cookie(mut self, cookie: Cookie):
-        self[ResponseCookieKey(cookie.name, cookie.domain, cookie.path)] = cookie
+        self[
+            ResponseCookieKey(cookie.name, cookie.domain, cookie.path)
+        ] = cookie
 
     @always_inline
     fn empty(self) -> Bool:

lightbug_http/http/common_response.mojo

@@ -15,7 +15,9 @@ fn OK(body: Bytes, content_type: String = "text/plain") -> HTTPResponse:
     )
 
 
-fn OK(body: Bytes, content_type: String, content_encoding: String) -> HTTPResponse:
+fn OK(
+    body: Bytes, content_type: String, content_encoding: String
+) -> HTTPResponse:
     return HTTPResponse(
         headers=Headers(
             Header(HeaderKey.CONTENT_TYPE, content_type),
@@ -25,27 +27,43 @@ fn OK(body: Bytes, content_type: String, content_encoding: String) -> HTTPRespon
     )
 
 
+fn SeeOther(
+    location: String, content_type: String, owned cookies: List[Cookie] = []
+) -> HTTPResponse:
+    return HTTPResponse(
+        bytes("See Other"),
+        cookies=ResponseCookieJar(cookies^),
+        headers=Headers(
+            Header(HeaderKey.LOCATION, location),
+            Header(HeaderKey.CONTENT_TYPE, content_type),
+        ),
+        status_code=303,
+        status_text="See Other",
+    )
+
+
+fn BadRequest() -> HTTPResponse:
+    return HTTPResponse(
+        bytes("Bad Request"),
+        headers=Headers(Header(HeaderKey.CONTENT_TYPE, "text/plain")),
+        status_code=400,
+        status_text="Bad Request",
+    )
+
+
 fn NotFound(path: String) -> HTTPResponse:
     return HTTPResponse(
+        body_bytes=bytes("path " + path + " not found"),
+        headers=Headers(Header(HeaderKey.CONTENT_TYPE, "text/plain")),
         status_code=404,
         status_text="Not Found",
-        headers=Headers(Header(HeaderKey.CONTENT_TYPE, "text/plain")),
-        body_bytes=bytes("path " + path + " not found"),
     )
 
 
 fn InternalError() -> HTTPResponse:
     return HTTPResponse(
         bytes("Failed to process request"),
-        status_code=500,
         headers=Headers(Header(HeaderKey.CONTENT_TYPE, "text/plain")),
+        status_code=500,
         status_text="Internal Server Error",
     )
-
-fn BadRequest() -> HTTPResponse:
-    return HTTPResponse(
-        bytes("Bad Request"),
-        status_code=400,
-        headers=Headers(Header(HeaderKey.CONTENT_TYPE, "text/plain")),
-        status_text="Bad Request",
-    )