1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
//
// Copyright (c) 2025 murilo ijanc' <murilo@ijanc.org>
//
// Permission to use, copy, modify, and distribute this software for any
// purpose with or without fee is hereby granted, provided that the above
// copyright notice and this permission notice appear in all copies.
//
// THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
// WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
// MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
// ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
// WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
// ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
// OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
//
use std::{sync::Arc, time::Duration};
use axum::{
Router,
extract::State,
http::{HeaderName, Request, StatusCode},
middleware,
response::{Html, IntoResponse},
routing::get,
};
use minijinja::context;
use tower_http::{
request_id::{
MakeRequestUuid, PropagateRequestIdLayer, SetRequestIdLayer,
},
services::ServeDir,
timeout::TimeoutLayer,
trace::TraceLayer,
};
use tracing::{error, info_span};
use crate::metric::track_metrics;
use crate::state::AppState;
const REQUEST_ID_HEADER: &str = "x-request-id";
pub(crate) fn route(app_state: Arc<AppState>) -> Router {
let x_request_id = HeaderName::from_static(REQUEST_ID_HEADER);
Router::new()
.route("/", get(handler_home))
.route("/content", get(handler_content))
.route("/about", get(handler_about))
// TODO(msi): from config folder asssets
.nest_service("/assets", ServeDir::new("assets"))
.layer((
SetRequestIdLayer::new(x_request_id.clone(), MakeRequestUuid),
TraceLayer::new_for_http().make_span_with(
|request: &Request<_>| {
// Log the request id as generated.
let request_id = request.headers().get(REQUEST_ID_HEADER);
match request_id {
Some(request_id) => info_span!(
"http_request",
request_id = ?request_id,
),
None => {
error!("could not extract request_id");
info_span!("http_request")
}
}
},
),
// TODO(msi): from config
TimeoutLayer::new(Duration::from_secs(10)),
PropagateRequestIdLayer::new(x_request_id),
))
.route_layer(middleware::from_fn(track_metrics))
.route("/healthz", get(healthz))
.with_state(app_state)
}
async fn healthz() -> impl IntoResponse {
StatusCode::OK
}
async fn handler_home(
State(state): State<Arc<AppState>>,
) -> Result<Html<String>, StatusCode> {
let template = state.env.get_template("home").unwrap();
let rendered = template
.render(context! {
title => "Home",
welcome_text => "Hello World!",
})
.unwrap();
Ok(Html(rendered))
}
async fn handler_content(
State(state): State<Arc<AppState>>,
) -> Result<Html<String>, StatusCode> {
let template = state.env.get_template("content").unwrap();
let some_example_entries = vec!["Data 1", "Data 2", "Data 3"];
let rendered = template
.render(context! {
title => "Content",
entries => some_example_entries,
})
.unwrap();
Ok(Html(rendered))
}
async fn handler_about(
State(state): State<Arc<AppState>>,
) -> Result<Html<String>, StatusCode> {
let template = state.env.get_template("about").unwrap();
let rendered = template.render(context!{
title => "About",
about_text => "Simple demonstration layout for an axum project with minijinja as templating engine.",
}).unwrap();
Ok(Html(rendered))
}
|