WIP

cot-rs · Sep 25, 2024 · 6612554 · 6612554
1 parent c04e7e3
commit 6612554
Show file tree

Hide file tree

Showing 7 changed files with 98 additions and 3 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -9,6 +9,7 @@ members = [
     # Examples
     "examples/hello-world",
     "examples/todo-list",
+    "examples/sessions",
 ]
 resolver = "2"
 

diff --git a/examples/sessions/Cargo.toml b/examples/sessions/Cargo.toml
@@ -0,0 +1,10 @@
+[package]
+name = "example-sessions"
+version = "0.1.0"
+publish = false
+description = "Sessions - Flareon example."
+edition = "2021"
+
+[dependencies]
+flareon = { path = "../../flareon" }
+tokio = { version = "1.40.0", features = ["macros", "rt-multi-thread"] }
diff --git a/examples/sessions/src/main.rs b/examples/sessions/src/main.rs
@@ -0,0 +1,35 @@
+use flareon::middleware::SessionMiddleware;
+use flareon::request::{Request, RequestExt};
+use flareon::response::{Response, ResponseExt};
+use flareon::router::Route;
+use flareon::{Body, Error, FlareonApp, FlareonProject, StatusCode};
+
+async fn return_hello(mut request: Request) -> Result<Response, Error> {
+    request
+        .session_mut()
+        .insert("user_name", "XD")
+        .await
+        .unwrap();
+
+    Ok(Response::new_html(
+        StatusCode::OK,
+        Body::fixed("<h1>Hello Flareon!</h1>".as_bytes().to_vec()),
+    ))
+}
+
+#[tokio::main]
+async fn main() {
+    let hello_app = FlareonApp::builder()
+        .urls([Route::with_handler("/", return_hello)])
+        .build()
+        .unwrap();
+
+    let flareon_project = FlareonProject::builder()
+        .register_app_with_views(hello_app, "")
+        .middleware(SessionMiddleware::new())
+        .build();
+
+    flareon::run(flareon_project, "127.0.0.1:8000")
+        .await
+        .unwrap();
+}
diff --git a/examples/sessions/templates/index.html b/examples/sessions/templates/index.html
@@ -0,0 +1,17 @@
+{% let request = request %}
+
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Sessions example</title>
+</head>
+<body>
+<h1>TODO List</h1>
+<form id="todo-form" action="{{ flareon::reverse_str!(request, "add-todo") }}" method="post">
+    <input type="text" id="title" name="title" placeholder="Enter a new TODO" required>
+    <button type="submit">Add TODO</button>
+</form>
+</body>
+</html>
diff --git a/flareon/src/lib.rs b/flareon/src/lib.rs
@@ -20,6 +20,7 @@ mod headers;
 #[doc(hidden)]
 #[path = "private.rs"]
 pub mod __private;
+pub mod middleware;
 pub mod request;
 pub mod response;
 pub mod router;
@@ -279,7 +280,7 @@ pub struct FlareonProject<S> {
     handler: S,
 }
 
-#[derive(Debug)]
+#[derive(Debug, Clone)]
 pub struct FlareonProjectBuilder {
     apps: Vec<FlareonApp>,
     urls: Vec<Route>,
@@ -304,10 +305,12 @@ impl FlareonProjectBuilder {
 
     #[must_use]
     pub fn middleware<M: tower::Layer<RouterService>>(
-        self,
+        &mut self,
         middleware: M,
     ) -> FlareonProjectBuilderWithMiddleware<M::Service> {
-        self.to_builder_with_middleware().middleware(middleware)
+        self.clone()
+            .to_builder_with_middleware()
+            .middleware(middleware)
     }
 
     /// Builds the Flareon project instance.

diff --git a/flareon/src/middleware.rs b/flareon/src/middleware.rs
@@ -0,0 +1,21 @@
+use tower_sessions::{MemoryStore, SessionManagerLayer};
+
+#[derive(Debug, Copy, Clone)]
+pub struct SessionMiddleware {}
+
+impl SessionMiddleware {
+    #[must_use]
+    pub fn new() -> Self {
+        Self {}
+    }
+}
+
+impl<S> tower::Layer<S> for SessionMiddleware {
+    type Service = <SessionManagerLayer<MemoryStore> as tower::Layer<S>>::Service;
+
+    fn layer(&self, inner: S) -> Self::Service {
+        let session_store = MemoryStore::default();
+        let session_layer = SessionManagerLayer::new(session_store);
+        session_layer.layer(inner)
+    }
+}