clean up example

Author: saviorand

lightbug.mojo

@@ -1,18 +1,11 @@
-# lightbug_api showcase — metaprogramming ergonomics demo
+# lightbug_api — quick-start example
 # ─────────────────────────────────────────────────────────────────────────────
-# Run:  mojo lightbug.mojo
-# Test: curl http://localhost:8080/
-#       curl http://localhost:8080/items
+# Run:  pixi run mojo lightbug.mojo
+# Test: curl http://localhost:8080/items
 #       curl http://localhost:8080/items/42
-#       curl "http://localhost:8080/items/42?verbose=true"
 #       curl -X POST http://localhost:8080/items \
 #            -H 'Content-Type: application/json' \
 #            -d '{"name":"Widget","price":9.99}'
-#       curl -X PUT http://localhost:8080/items/42 \
-#            -H 'Content-Type: application/json' \
-#            -d '{"name":"Updated","price":19.99}'
-#       curl -X DELETE http://localhost:8080/items/42
-#       curl http://localhost:8080/v1/status
 #       curl http://localhost:8080/notes
 #       curl http://localhost:8080/notes/1
 # ─────────────────────────────────────────────────────────────────────────────
@@ -23,7 +16,7 @@ from lightbug_api.response import Response
 from lightbug_http.http.json import JsonSerializable, JsonDeserializable
 
 
-# ----------------------------------------------------------------- data types
+# ── Models ────────────────────────────────────────────────────────────────────
 
 @fieldwise_init
 struct Item(JsonSerializable, Movable, Defaultable):
@@ -32,9 +25,7 @@ struct Item(JsonSerializable, Movable, Defaultable):
     var price: Float64
 
     fn __init__(out self):
-        self.id = 0
-        self.name = ""
-        self.price = 0.0
+        self.id = 0; self.name = ""; self.price = 0.0
 
 
 @fieldwise_init
@@ -43,18 +34,7 @@ struct CreateItemRequest(JsonDeserializable, Movable, Defaultable):
     var price: Float64
 
     fn __init__(out self):
-        self.name = ""
-        self.price = 0.0
-
-
-@fieldwise_init
-struct StatusResponse(JsonSerializable, Movable, Defaultable):
-    var status: String
-    var version: String
-
-    fn __init__(out self):
-        self.status = ""
-        self.version = ""
+        self.name = ""; self.price = 0.0
 
 
 @fieldwise_init
@@ -63,60 +43,35 @@ struct Note(JsonSerializable, Movable, Defaultable):
     var text: String
 
     fn __init__(out self):
-        self.id = 0
-        self.text = ""
-
-
-# ------------------------------------------------------------------ handlers
-# Handlers that need full control (non-200 status, redirects, plain text) keep
-# the HandlerResponse return type.  Handlers that just return a model use the
-# model type directly — the framework auto-serialises as JSON 200 OK.
+        self.id = 0; self.text = ""
 
-fn index(ctx: Context) raises -> HandlerResponse:
-    return Response.text("Welcome to lightbug_api 🔥")
 
+# ── Handlers — return your model directly, no Response.json() needed ──────────
 
-# ── Before (old style) ───────────────────────────────────────────────────────
-
-fn list_items(ctx: Context) raises -> Item:          # ← returns Item directly
+fn list_items(ctx: Context) raises -> Item:
     return Item(1, "Widget", 9.99)
 
 
-fn get_item(ctx: Context) raises -> Item:            # ← returns Item directly
-    var id      = ctx.param("id", 0)
-    var verbose = ctx.query("verbose", False)
-    if verbose:
-        print("GET /items/", id, " (verbose mode)")
+fn get_item(ctx: Context) raises -> Item:
+    var id = ctx.param("id", 0)
     return Item(id, String("Item ", id), 9.99)
 
 
 fn create_item(ctx: Context) raises -> HandlerResponse:
-    # Still HandlerResponse — needs 201 Created status code
-    var body    = ctx.json[CreateItemRequest]()
-    var created = Item(100, body.name, body.price)
-    return Response.created(created)
+    var body = ctx.json[CreateItemRequest]()
+    return Response.created(Item(100, body.name, body.price))  # 201 Created
 
 
-fn update_item(ctx: Context) raises -> Item:         # ← returns Item directly
+fn update_item(ctx: Context) raises -> Item:
     var body = ctx.json[CreateItemRequest]()
-    var id   = ctx.param("id", 0)
-    return Item(id, body.name, body.price)
+    return Item(ctx.param("id", 0), body.name, body.price)
 
 
 fn delete_item(ctx: Context) raises -> HandlerResponse:
-    # Still HandlerResponse — needs 204 No Content
-    var id = ctx.param("id", 0)
-    print("Deleting item", id)
-    return Response.no_content()
-
+    return Response.no_content()  # 204 No Content
 
-fn health(ctx: Context) raises -> StatusResponse:    # ← returns StatusResponse directly
-    return StatusResponse("ok", "1.0.0")
 
-
-# ── Resource / controller pattern ────────────────────────────────────────────
-# Group CRUD handlers in a struct; `resource[Notes]("notes")` registers all
-# five standard routes under /notes in one call.
+# ── Resource controller — one struct registers all five CRUD routes ────────────
 
 struct Notes(Resource):
     @staticmethod
@@ -134,40 +89,23 @@ struct Notes(Resource):
 
     @staticmethod
     fn update(ctx: Context) raises -> HandlerResponse:
-        var id = ctx.param("id", 0)
-        return Response.json(Note(id, "updated"))
+        return Response.json(Note(ctx.param("id", 0), "updated"))
 
     @staticmethod
     fn destroy(ctx: Context) raises -> HandlerResponse:
         return Response.no_content()
 
 
-# --------------------------------------------------------------------- main
+# ── App ───────────────────────────────────────────────────────────────────────
 
 fn main() raises:
     var app = App(
-        # Plain-text response — HandlerResponse (unchanged)
-        GET("/",              index),
-
-        # Typed return — framework auto-serialises Item as JSON 200 OK
         GET[Item, list_items]("/items"),
         GET[Item, get_item]("/items/{id}"),
-
-        # 201 Created — still HandlerResponse (needs explicit status)
         POST("/items",        create_item),
-
-        # Typed return
         PUT[Item, update_item]("/items/{id}"),
-
-        # 204 No Content — still HandlerResponse
         DELETE("/items/{id}", delete_item),
 
-        mount("v1",
-            # Typed return inside a mount
-            GET[StatusResponse, health]("status"),
-        ),
-
-        # Resource controller — five routes registered in one line
-        resource[Notes]("notes"),
+        resource[Notes]("notes"),  # GET /notes, GET /notes/{id}, POST, PUT, DELETE
     )
     app.run()