Skip to content

Commit 2e326dd

Browse files
committed
chore: add log middleware
1 parent 6accd3a commit 2e326dd

File tree

3 files changed

+22
-4
lines changed

3 files changed

+22
-4
lines changed

api/src/interactions/mod.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ static PUB_KEY: Lazy<VerifyingKey> = Lazy::new(|| {
3131
pub fn router() -> axum::Router<AppState> {
3232
axum::Router::new()
3333
.route("/", post(post_interactions))
34-
.layer(middleware::from_fn(pubkey_middleware))
34+
.route_layer(middleware::from_fn(pubkey_middleware))
3535
.route("/register", post(register_commands))
3636
}
3737

api/src/main.rs

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
use axum::{http::StatusCode, middleware, routing::get, Json, Router};
1+
use axum::{http::StatusCode, routing::get, Json, Router};
22
use lambda_http::{run, tracing, Error};
33
use serde::{Deserialize, Serialize};
44
use serde_json::{json, Value};
@@ -9,7 +9,7 @@ use axum::body::Body;
99
use http::Response;
1010
use tower_http::cors::CorsLayer;
1111

12-
mod auth;
12+
mod middleware;
1313
mod dynamo;
1414
mod users;
1515
mod guilds;
@@ -72,8 +72,9 @@ async fn main() -> Result<(), Error> {
7272
.nest("/users", users::router())
7373
.nest("/guilds", guilds::router())
7474
.layer(CorsLayer::permissive())
75-
.route_layer(middleware::from_fn(auth::auth_middleware))
75+
.route_layer(axum::middleware::from_fn(middleware::auth_middleware))
7676
.nest("/interactions", interactions::router())
77+
.route_layer(axum::middleware::from_fn(middleware::log_middleware))
7778
.with_state(app_state)
7879
.route("/health", get(health_check))
7980
.route("/bot", get(get_bot_redirect));

api/src/auth.rs renamed to api/src/middleware.rs

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@ use axum::extract::Request;
22
use axum::http::{HeaderMap, StatusCode};
33
use axum::middleware::Next;
44
use axum::response::Response;
5+
use lambda_http::tracing::info;
56
use twilight_http::Client;
67

78
pub async fn auth_middleware(
@@ -27,4 +28,20 @@ pub async fn auth_middleware(
2728
} else {
2829
Err(StatusCode::UNAUTHORIZED)
2930
}
31+
}
32+
33+
pub async fn log_middleware(
34+
request: Request,
35+
next: Next,
36+
) -> Result<Response, StatusCode> {
37+
// Log the request
38+
info!("Received request: {:?}", request);
39+
40+
// Call the next middleware or handler
41+
let response = next.run(request).await;
42+
43+
// Log the response
44+
info!("Response: {:?}", response);
45+
46+
Ok(response)
3047
}

0 commit comments

Comments
 (0)