Merge pull request #286 from thatstoasty/examples

Add echo server and client example

Author: saviorand

README.md

@@ -32,11 +32,11 @@ Lightbug currently has the following features:
  - [x] Pure Mojo HTTP 1.1 server with no Python dependencies. Everything is fully typed, with no `def` functions used
  - [x] Cookie support
 
- ### Check Out These Mojo Libraries:
+### Check Out These Mojo Libraries:
 
-- Logging - [@toasty/stump](https://github.com/thatstoasty/stump)
-- CLI and Terminal - [@toasty/prism](https://github.com/thatstoasty/prism), [@toasty/mog](https://github.com/thatstoasty/mog)
-- Date/Time - [@mojoto/morrow](https://github.com/mojoto/morrow.mojo) and [@toasty/small_time](https://github.com/thatstoasty/small_time)
+- HTTP Client - [@thatstoasty/floki](https://github.com/thatstoasty/floki)
+- CLI and Terminal - [@thatstoasty/prism](https://github.com/thatstoasty/prism), [@thatstoasty/mog](https://github.com/thatstoasty/mog)
+- Date/Time - [@mojoto/morrow](https://github.com/mojoto/morrow.mojo) and [@thatstoasty/small_time](https://github.com/thatstoasty/small_time)
 
 <p align="right">(<a href="#readme-top">back to top</a>)</p>
 
@@ -176,6 +176,11 @@ struct Welcome(HTTPService):
         return NotFound(req.uri.path)
 ```
 
+<!-- Examples -->
+## Examples
+
+Check out the examples directory for more example services built with Lightbug, including an echo server and client implementation!
+
 <!-- ROADMAP -->
 ## Roadmap
 

client.mojo

@@ -0,0 +1,15 @@
+# Echo Server
+
+A simple echo server example using `lightbug_http` for the server and `floki` for the client. The server listens for incoming HTTP requests and responds with the same body and content type.
+
+To run the server, execute the following command in the terminal:
+
+```bash
+pixi run server
+```
+
+In a separate terminal, you can run the client to send a request to the server:
+
+```bash
+pixi run client
+```

examples/echo_server/README.md

@@ -0,0 +1,17 @@
+import floki
+
+
+fn main() raises:
+    var response = floki.post(
+        "localhost:8080",
+        headers={"Content-Type": "text/plain"},
+        data="Hello, Echo Server!".as_bytes()
+    )
+    print(response.body.as_string_slice())
+
+    response = floki.post(
+        "localhost:8080",
+        headers={"Content-Type": "application/json"},
+        data={"message": "Hello, Echo Server!"}
+    )
+    print(response.body.as_json())

examples/echo_server/client.mojo

@@ -0,0 +1,17 @@
+[workspace]
+authors = ["Mikhail Tavarez <miktavarez@gmail.com>"]
+channels = ["https://conda.modular.com/max", "https://repo.prefix.dev/modular-community", "https://repo.prefix.dev/mojo-community", "conda-forge"]
+name = "echo_server"
+platforms = ["osx-arm64", "linux-64", "linux-aarch64"]
+version = "0.1.0"
+preview = ["pixi-build"]
+
+[tasks]
+server = "mojo run server.mojo"
+client = "mojo run client.mojo"
+
+[dependencies]
+mojo = ">=0.26.1.0,<0.26.2.0"
+curl_wrapper = { git = "https://github.com/thatstoasty/mojo-curl.git", branch = "main", subdirectory = "shim" }
+floki = ">=0.1.0,<0.2"
+lightbug_http = ">=0.26.1.2,<0.26.2.0"

examples/echo_server/pixi.lock

@@ -0,0 +1,14 @@
+from lightbug_http import OK, HTTPRequest, HTTPResponse, HTTPService, Server
+
+
+@fieldwise_init
+struct EchoService(HTTPService):
+    fn func(mut self, req: HTTPRequest) raises -> HTTPResponse:
+        var content_type = req.headers.get("content-type")
+        return OK(req.body_raw, content_type.or_else("text/plain; charset=utf-8"))
+
+
+fn main() raises:
+    var server = Server()
+    var handler = EchoService()
+    server.listen_and_serve("localhost:8080", handler)

examples/echo_server/pixi.toml

@@ -426,8 +426,6 @@ fn is_ipv6(network: NetworkType) -> Bool:
     return network in (NetworkType.tcp6, NetworkType.udp6, NetworkType.ip6)
 
 
-
-
 @fieldwise_init
 @register_passable("trivial")
 struct ParseEmptyAddressError(CustomError):
@@ -548,8 +546,6 @@ struct ParseIPProtocolPortError(CustomError):
         return Self.message
 
 
-
-
 @fieldwise_init
 @register_passable("trivial")
 struct GetaddrinfoNullAddrError(CustomError):
@@ -574,7 +570,6 @@ struct GetaddrinfoError(CustomError):
         return Self.message
 
 
-
 @fieldwise_init
 struct GetIPAddressError(Movable, Stringable, Writable):
     """Typed error variant for get_ip_address() function."""

examples/echo_server/server.mojo

@@ -371,7 +371,7 @@ struct TCPConnection[network: NetworkType = NetworkType.tcp4]:
         """
         var total_sent: UInt = 0
         while total_sent < UInt(len(buf)):
-            var sent = self.socket.send(buf[Int(total_sent):])
+            var sent = self.socket.send(buf[Int(total_sent) :])
             total_sent += sent
         return total_sent
 

lightbug_http/address.mojo

@@ -367,7 +367,9 @@ fn encode_latin1_header_value(value: String) -> List[UInt8]:
                 var b4 = utf8[i + 3]
                 if b2 >= 0x80 and b2 <= 0xBF and b3 >= 0x80 and b3 <= 0xBF and b4 >= 0x80 and b4 <= 0xBF:
                     seq_len = 4
-                    codepoint = ((Int(b) & 0x07) << 18) | ((Int(b2) & 0x3F) << 12) | ((Int(b3) & 0x3F) << 6) | (Int(b4) & 0x3F)
+                    codepoint = (
+                        ((Int(b) & 0x07) << 18) | ((Int(b2) & 0x3F) << 12) | ((Int(b3) & 0x3F) << 6) | (Int(b4) & 0x3F)
+                    )
 
             if seq_len > 0 and codepoint <= 0xFF:
                 out.append(UInt8(codepoint))