Add ip extracter

6 files changed, 24 insertions, 8 deletions

diff --git a/web/README.md b/web/README.md
index 556952c..9b0bebc 100644
--- a/web/README.md
+++ b/web/README.md
@@ -4,7 +4,7 @@
 * [x] Graceful Shutdown
 * [x] Minijinja
 * [x] Prometheus
-* [ ] Middleware Ip
+* [x] Middleware Ip
 * [x] Request Id Header
 * [x] Static files
 * [x] Config
diff --git a/web/template/Cargo.toml b/web/template/Cargo.toml
index 207eafb..d266dac 100644
--- a/web/template/Cargo.toml
+++ b/web/template/Cargo.toml
@@ -9,6 +9,7 @@ edition = "2024"
 [dependencies]
 anyhow = "=1.0.100"
 axum = { version = "=0.8.6", features = ["macros"] }
+axum-client-ip = "=1.1.3"
 axum-messages = "=0.8.0"
 axum_csrf = { version = "=0.11.0", features = ["layer"] }
 config = { version = "=0.15.19", default-features = false, features = ["toml"] }
diff --git a/web/template/src/main.rs b/web/template/src/main.rs
index f9c6340..d1f56d9 100644
--- a/web/template/src/main.rs
+++ b/web/template/src/main.rs
@@ -14,6 +14,7 @@
 // OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 //
 
+use std::net::SocketAddr;
 use std::sync::Arc;
 
 use minijinja::Environment;
@@ -23,8 +24,8 @@ use tracing::info;
 mod helpers;
 mod metric;
 mod router;
-mod state;
 mod settings;
+mod state;
 
 #[tokio::main]
 async fn main() -> anyhow::Result<()> {
@@ -52,8 +53,11 @@ async fn start_main_server() -> anyhow::Result<()> {
     // TODO(msi): from config
     let listener = TcpListener::bind("0.0.0.0:3000").await?;
     info!("listening on http://{}", listener.local_addr().unwrap());
-    axum::serve(listener, app)
-        .with_graceful_shutdown(helpers::shutdown_signal())
-        .await?;
+    axum::serve(
+        listener,
+        app.into_make_service_with_connect_info::<SocketAddr>(),
+    )
+    .with_graceful_shutdown(helpers::shutdown_signal())
+    .await?;
     Ok(())
 }
diff --git a/web/template/src/router.rs b/web/template/src/router.rs
index 170890a..ca2f2a8 100644
--- a/web/template/src/router.rs
+++ b/web/template/src/router.rs
@@ -23,6 +23,7 @@ use axum::{
     response::{Html, IntoResponse, Redirect},
     routing::get,
 };
+use axum_client_ip::{ClientIp, ClientIpSource};
 use axum_csrf::{CsrfConfig, CsrfLayer, CsrfToken, Key};
 use axum_messages::{Messages, MessagesManagerLayer};
 use minijinja::context;
@@ -62,6 +63,9 @@ pub(crate) fn route(app_state: Arc<AppState>) -> Router {
         .with_key(Some(cookie_key))
         .with_cookie_domain(Some("127.0.0.1"));
 
+    // TODO(msi): from config, if debug mode
+    let ip_source = ClientIpSource::ConnectInfo;
+
     Router::new()
         .route("/", get(handler_home))
         .route("/content", get(handler_content))
@@ -70,6 +74,7 @@ pub(crate) fn route(app_state: Arc<AppState>) -> Router {
         .route("/message", get(set_messages_handler))
         .route("/read-messages", get(read_messages_handler))
         .route("/csrf", get(csrf_root).post(csrf_check_key))
+        .route("/ip", get(ip_handler))
         .layer(MessagesManagerLayer)
         // TODO(msi): from config folder asssets
         .nest_service("/assets", ServeDir::new("assets"))
@@ -97,6 +102,7 @@ pub(crate) fn route(app_state: Arc<AppState>) -> Router {
                 .with_expiry(Expiry::OnInactivity(Duration::seconds(10))),
             MessagesManagerLayer,
             CsrfLayer::new(config),
+            ip_source.into_extension(),
             // TODO(msi): from config
             TimeoutLayer::new(std::time::Duration::from_secs(10)),
             PropagateRequestIdLayer::new(x_request_id),
@@ -106,6 +112,10 @@ pub(crate) fn route(app_state: Arc<AppState>) -> Router {
         .with_state(app_state)
 }
 
+async fn ip_handler(ClientIp(ip): ClientIp) -> String {
+    ip.to_string()
+}
+
 async fn csrf_root(
     token: CsrfToken,
     State(state): State<Arc<AppState>>,
diff --git a/web/template/src/settings.rs b/web/template/src/settings.rs
index 897cbf4..9814764 100644
--- a/web/template/src/settings.rs
+++ b/web/template/src/settings.rs
@@ -48,7 +48,8 @@ pub(crate) struct Settings {
 impl Settings {
     pub(crate) fn new() -> Result<Self, ConfigError> {
         info!("loading settings");
-        let run_mode = env::var("RUN_MODE").unwrap_or_else(|_| "development".into());
+        let run_mode =
+            env::var("RUN_MODE").unwrap_or_else(|_| "development".into());
 
         let s = Config::builder()
             // Start off by merging in the "default" configuration file
@@ -57,8 +58,7 @@ impl Settings {
             // Default to 'development' env
             // Note that this file is _optional_
             .add_source(
-                File::with_name(&format!("config/{run_mode}"))
-                    .required(false),
+                File::with_name(&format!("config/{run_mode}")).required(false),
             )
             // Add in a local configuration file
             // This file shouldn't be checked in to git
diff --git a/web/template/templates/layout.jinja b/web/template/templates/layout.jinja
index f162c30..6abc2fd 100644
--- a/web/template/templates/layout.jinja
+++ b/web/template/templates/layout.jinja
@@ -12,6 +12,7 @@
             <li><a href="/message">Set Message</a></li>
             <li><a href="/read-messages">Read Messages</a></li>
             <li><a href="/csrf">Csrf</a></li>
+            <li><a href="/ip">Ip</a></li>
         </ul>
     </nav>
     <h1><h1>Hello, World web =]</h1>