mojo format all files

Author: saviorand

bench.mojo

@@ -11,26 +11,48 @@ from tests.utils import (
     FakeServer,
 )
 
-alias headers = bytes('''GET /index.html HTTP/1.1\r\nHost: example.com\r\nUser-Agent: Mozilla/5.0\r\nContent-Type: text/html\r\nContent-Length: 1234\r\nConnection: close\r\nTrailer: end-of-message\r\n\r\n''')
+alias headers = bytes(
+    """GET /index.html HTTP/1.1\r\nHost: example.com\r\nUser-Agent: Mozilla/5.0\r\nContent-Type: text/html\r\nContent-Length: 1234\r\nConnection: close\r\nTrailer: end-of-message\r\n\r\n"""
+)
 alias body = bytes(String("I am the body of an HTTP request") * 5)
-alias Request = bytes('''GET /index.html HTTP/1.1\r\nHost: example.com\r\nUser-Agent: Mozilla/5.0\r\nContent-Type: text/html\r\nContent-Length: 1234\r\nConnection: close\r\nTrailer: end-of-message\r\n\r\n''') + body
-alias Response = bytes("HTTP/1.1 200 OK\r\nserver: lightbug_http\r\ncontent-type: application/octet-stream\r\nconnection: keep-alive\r\ncontent-length: 13\r\ndate: 2024-06-02T13:41:50.766880+00:00\r\n\r\n") + body
+alias Request = bytes(
+    """GET /index.html HTTP/1.1\r\nHost: example.com\r\nUser-Agent: Mozilla/5.0\r\nContent-Type: text/html\r\nContent-Length: 1234\r\nConnection: close\r\nTrailer: end-of-message\r\n\r\n"""
+) + body
+alias Response = bytes(
+    "HTTP/1.1 200 OK\r\nserver: lightbug_http\r\ncontent-type:"
+    " application/octet-stream\r\nconnection: keep-alive\r\ncontent-length:"
+    " 13\r\ndate: 2024-06-02T13:41:50.766880+00:00\r\n\r\n"
+) + body
+
 
 fn main():
     run_benchmark()
 
+
 fn run_benchmark():
     try:
         var config = BenchConfig(warmup_iters=100)
         config.verbose_timing = True
         config.tabular_view = True
         var m = Bench(config)
-        m.bench_function[lightbug_benchmark_header_encode](BenchId("HeaderEncode"))
-        m.bench_function[lightbug_benchmark_header_parse](BenchId("HeaderParse"))
-        m.bench_function[lightbug_benchmark_request_encode](BenchId("RequestEncode"))
-        m.bench_function[lightbug_benchmark_request_parse](BenchId("RequestParse"))
-        m.bench_function[lightbug_benchmark_response_encode](BenchId("ResponseEncode"))
-        m.bench_function[lightbug_benchmark_response_parse](BenchId("ResponseParse"))
+        m.bench_function[lightbug_benchmark_header_encode](
+            BenchId("HeaderEncode")
+        )
+        m.bench_function[lightbug_benchmark_header_parse](
+            BenchId("HeaderParse")
+        )
+        m.bench_function[lightbug_benchmark_request_encode](
+            BenchId("RequestEncode")
+        )
+        m.bench_function[lightbug_benchmark_request_parse](
+            BenchId("RequestParse")
+        )
+        m.bench_function[lightbug_benchmark_response_encode](
+            BenchId("ResponseEncode")
+        )
+        m.bench_function[lightbug_benchmark_response_parse](
+            BenchId("ResponseParse")
+        )
         m.dump_report()
     except:
         print("failed to start benchmark")
@@ -41,18 +63,21 @@ var headers_struct = Headers(
     Header("Content-Length", "1234"),
     Header("Connection", "close"),
     Header("Date", "some-datetime"),
-    Header("SomeHeader", "SomeValue")
+    Header("SomeHeader", "SomeValue"),
 )
 
+
 @parameter
 fn lightbug_benchmark_response_encode(inout b: Bencher):
     @always_inline
     @parameter
     fn response_encode():
-        var res = HTTPResponse(body, headers = headers_struct)
+        var res = HTTPResponse(body, headers=headers_struct)
         _ = encode(res^)
+
     b.iter[response_encode]()
 
+
 @parameter
 fn lightbug_benchmark_response_parse(inout b: Bencher):
     @always_inline
@@ -63,8 +88,10 @@ fn lightbug_benchmark_response_parse(inout b: Bencher):
             _ = HTTPResponse.from_bytes(res^)
         except:
             pass
+
     b.iter[response_parse]()
 
+
 @parameter
 fn lightbug_benchmark_request_parse(inout b: Bencher):
     @always_inline
@@ -75,21 +102,25 @@ fn lightbug_benchmark_request_parse(inout b: Bencher):
             _ = HTTPRequest.from_bytes("127.0.0.1/path", 4096, r^)
         except:
             pass
+
     b.iter[request_parse]()
 
+
 @parameter
 fn lightbug_benchmark_request_encode(inout b: Bencher):
     @always_inline
     @parameter
     fn request_encode():
         var req = HTTPRequest(
             URI.parse("http://127.0.0.1:8080/some-path")[URI],
-            headers = headers_struct,
-            body = body
+            headers=headers_struct,
+            body=body,
         )
         _ = encode(req^)
+
     b.iter[request_encode]()
-        
+
+
 @parameter
 fn lightbug_benchmark_header_encode(inout b: Bencher):
     @always_inline
@@ -98,8 +129,10 @@ fn lightbug_benchmark_header_encode(inout b: Bencher):
         var b = ByteWriter()
         var h = headers_struct
         h.encode_to(b)
+
     b.iter[header_encode]()
 
+
 @parameter
 fn lightbug_benchmark_header_parse(inout b: Bencher):
     @always_inline
@@ -115,16 +148,21 @@ fn lightbug_benchmark_header_parse(inout b: Bencher):
 
     b.iter[header_parse]()
 
+
 fn lightbug_benchmark_server():
     var server_report = benchmark.run[run_fake_server](max_iters=1)
     print("Server: ")
     server_report.print(benchmark.Unit.ms)
 
 
 fn lightbug_benchmark_misc() -> None:
-    var direct_set_report = benchmark.run[init_test_and_set_a_direct](max_iters=1)
+    var direct_set_report = benchmark.run[init_test_and_set_a_direct](
+        max_iters=1
+    )
 
-    var recreating_set_report = benchmark.run[init_test_and_set_a_copy](max_iters=1)
+    var recreating_set_report = benchmark.run[init_test_and_set_a_copy](
+        max_iters=1
+    )
 
     print("Direct set: ")
     direct_set_report.print(benchmark.Unit.ms)
@@ -134,6 +172,7 @@ fn lightbug_benchmark_misc() -> None:
 
 var GetRequest = HTTPRequest(URI.parse("http://127.0.0.1/path")[URI])
 
+
 fn run_fake_server():
     var handler = FakeResponder()
     var listener = new_fake_listener(2, encode(GetRequest))

bench_server.mojo

@@ -1,11 +1,12 @@
 from lightbug_http.sys.server import SysServer
 from lightbug_http.service import TechEmpowerRouter
 
+
 def main():
     try:
         var server = SysServer(tcp_keep_alive=True)
         var handler = TechEmpowerRouter()
         server.listen_and_serve("0.0.0.0:8080", handler)
     except e:
         print("Error starting server: " + e.__str__())
-        return
+        return

client.mojo

@@ -1,10 +1,10 @@
 from lightbug_http import *
 from lightbug_http.sys.client import MojoClient
 
+
 fn test_request(inout client: MojoClient) raises -> None:
     var uri = URI.parse_raises("http://httpbin.org/status/404")
 
-
     var request = HTTPRequest(uri)
     var response = client.do(request^)
 
@@ -16,12 +16,14 @@ fn test_request(inout client: MojoClient) raises -> None:
     print("Content-Length", response.headers["Content-Length"])
     print("Server:", to_string(response.headers["Server"]))
 
-    print("Is connection set to connection-close? ", response.connection_close())
+    print(
+        "Is connection set to connection-close? ", response.connection_close()
+    )
 
     # print body
     print(to_string(response.body_raw))
 
 
 fn main() raises -> None:
     var client = MojoClient()
-    test_request(client)
+    test_request(client)

lightbug_http/__init__.mojo

@@ -4,6 +4,7 @@ from lightbug_http.service import HTTPService, Welcome
 from lightbug_http.sys.server import SysServer
 from lightbug_http.strings import to_string
 
+
 trait DefaultConstructible:
     fn __init__(inout self) raises:
         ...

lightbug_http/error.mojo

@@ -3,8 +3,9 @@ from lightbug_http.io.bytes import bytes
 
 alias TODO_MESSAGE = String("TODO").as_bytes()
 
+
 # TODO: Custom error handlers provided by the user
 @value
 struct ErrorHandler:
     fn Error(self) -> HTTPResponse:
-        return HTTPResponse(TODO_MESSAGE)
+        return HTTPResponse(TODO_MESSAGE)

lightbug_http/header.mojo

@@ -4,6 +4,7 @@ from collections import Dict
 from lightbug_http.utils import ByteReader, ByteWriter, is_newline, is_space
 from lightbug_http.strings import rChar, nChar, lineBreak, to_string
 
+
 struct HeaderKey:
     # TODO: Fill in more of these
     alias CONNECTION = "connection"
@@ -12,22 +13,26 @@ struct HeaderKey:
     alias CONTENT_ENCODING = "content-encoding"
     alias DATE = "date"
 
+
 @value
 struct Header:
     var key: String
     var value: String
 
+
 @always_inline
 fn write_header(inout writer: Formatter, key: String, value: String):
     writer.write(key + ": ", value, lineBreak)
 
+
 @always_inline
 fn write_header(inout writer: ByteWriter, key: String, inout value: String):
-    var  k = key + ": "
+    var k = key + ": "
     writer.write(k)
     writer.write(value)
     writer.write(lineBreak)
 
+
 @value
 struct Headers(Formattable, Stringable):
     """Represents the header key/values in an http request/response.
@@ -59,7 +64,7 @@ struct Headers(Formattable, Stringable):
             return self._inner[key.lower()]
         except:
             return String()
-        
+
     @always_inline
     fn __setitem__(inout self, key: String, value: String):
         self._inner[key.lower()] = value
@@ -72,11 +77,13 @@ struct Headers(Formattable, Stringable):
         except:
             return 0
 
-    fn parse_raw(inout self, inout r: ByteReader) raises -> (String, String, String):
+    fn parse_raw(
+        inout self, inout r: ByteReader
+    ) raises -> (String, String, String):
         var first_byte = r.peek()
         if not first_byte:
             raise Error("Failed to read first byte from response header")
-        
+
         var first = r.read_word()
         r.increment()
         var second = r.read_word()
@@ -91,7 +98,7 @@ struct Headers(Formattable, Stringable):
             # TODO (bgreni): Handle possible trailing whitespace
             var value = r.read_line()
             self._inner[to_string(key^).lower()] = to_string(value^)
-        
+
         return (to_string(first^), to_string(second^), to_string(third^))
 
     fn format_to(self, inout writer: Formatter):
@@ -104,4 +111,3 @@ struct Headers(Formattable, Stringable):
 
     fn __str__(self) -> String:
         return to_string(self)
-