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
|
use std::sync::Arc;
use tokio::sync::Mutex;
use axum::extract::DefaultBodyLimit;
use axum::middleware::{self};
use axum::routing::{get, post};
use axum::Router;
use sqlx::SqlitePool;
use tower_http::add_extension::AddExtensionLayer;
use tower_http::services::ServeDir;
use admin::{admin_healthcheck, admin_middleware, generate_token, upload_story};
use frontend_builder::home;
use crate::db::story::Story;
use self::frontend_builder::build_page;
mod admin;
pub mod frontend_builder;
#[derive(Debug, Clone)]
pub struct ApiContext {
pool: SqlitePool,
}
#[derive(Clone)]
struct AppState {
admin_token: Arc<Mutex<String>>,
}
pub async fn router(pool: SqlitePool) -> Router {
let state = AppState {
admin_token: Arc::new(Mutex::new(generate_token())),
};
Router::new()
.route("/admin/story", post(upload_story))
.route("/admin", get(admin_healthcheck))
.layer(middleware::from_fn_with_state(
state.clone(),
admin_middleware,
))
.with_state(state)
.route("/", get(home))
.route("/stories", get(build_page::<Story>))
.nest_service("/static", ServeDir::new("static"))
.layer(DefaultBodyLimit::max(1074000000))
.layer(AddExtensionLayer::new(ApiContext { pool }))
}
|
