123 lines
3.8 KiB
Rust
123 lines
3.8 KiB
Rust
mod frontend_api;
|
|
mod frontend_render;
|
|
mod manifest;
|
|
mod static_serve;
|
|
mod summary_proxy;
|
|
|
|
use crate::frontend_api::create_frontend_api;
|
|
use crate::frontend_render::{new_frontend_render_router, FrontendConfig};
|
|
use crate::manifest::handle_manifest;
|
|
use crate::static_serve::{static_serve, static_serve_svg, static_serve_sw};
|
|
use crate::summary_proxy::generate_summary;
|
|
use axum::headers::CacheControl;
|
|
use axum::routing::{any, get};
|
|
use axum::{Router, TypedHeader};
|
|
use dotenvy::dotenv;
|
|
use hyper::StatusCode;
|
|
use magnetar_common::config::MagnetarConfig;
|
|
use miette::{miette, IntoDiagnostic};
|
|
use std::net::SocketAddr;
|
|
use std::sync::Arc;
|
|
use std::time::Duration;
|
|
use tera::Tera;
|
|
use thiserror::Error;
|
|
use tower_http::services::ServeFile;
|
|
use tower_http::trace::TraceLayer;
|
|
use tracing::log::info;
|
|
use tracing_subscriber::EnvFilter;
|
|
|
|
#[derive(Debug, Error)]
|
|
pub enum RouterSetupError {
|
|
#[error("Failed to load template: {0}")]
|
|
TemplateLoadError(#[from] tera::Error),
|
|
}
|
|
|
|
fn new_calckey_fe_router(config: &'static MagnetarConfig) -> Result<Router, RouterSetupError> {
|
|
let tera = Tera::new("fe_calckey/frontend/assets-be/template/**/*.html")?;
|
|
let frontend_renderer_config = FrontendConfig {
|
|
magnetar_config: config,
|
|
templater: Arc::new(tera),
|
|
};
|
|
|
|
Ok(Router::new()
|
|
.nest_service(
|
|
"/favicon.ico",
|
|
ServeFile::new("fe_calckey/frontend/assets/favicon.ico"),
|
|
)
|
|
.nest_service(
|
|
"/favicon.png",
|
|
ServeFile::new("fe_calckey/frontend/assets/favicon.png"),
|
|
)
|
|
.nest_service(
|
|
"/favicon.svg",
|
|
ServeFile::new("fe_calckey/frontend/assets/favicon.svg"),
|
|
)
|
|
.route("/manifest.json", get(handle_manifest).with_state(config))
|
|
.nest(
|
|
"/sw.js",
|
|
static_serve_sw("fe_calckey/frontend/built/_sw_dist_/sw.js"),
|
|
)
|
|
.nest("/static-assets", static_serve("fe_calckey/frontend/assets"))
|
|
.nest(
|
|
"/client-assets",
|
|
static_serve("fe_calckey/frontend/client/assets"),
|
|
)
|
|
.nest(
|
|
"/assets",
|
|
static_serve("fe_calckey/frontend/built/_client_dist_"),
|
|
)
|
|
.nest(
|
|
"/twemoji",
|
|
static_serve_svg("fe_calckey/frontend/assets-be/twemoji"),
|
|
)
|
|
.nest("/fe-api", create_frontend_api("fe_calckey/frontend/assets"))
|
|
.route("/url", get(generate_summary))
|
|
.route(
|
|
"/streaming",
|
|
any(|| async {
|
|
(
|
|
StatusCode::SERVICE_UNAVAILABLE,
|
|
TypedHeader(
|
|
CacheControl::new()
|
|
.with_max_age(Duration::from_secs(0))
|
|
.with_private(),
|
|
),
|
|
)
|
|
}),
|
|
)
|
|
.nest("", new_frontend_render_router(frontend_renderer_config)))
|
|
}
|
|
|
|
#[tokio::main]
|
|
async fn main() -> miette::Result<()> {
|
|
dotenv().ok();
|
|
|
|
let filter_layer = EnvFilter::try_from_default_env()
|
|
.or_else(|_| EnvFilter::try_new("info"))
|
|
.unwrap();
|
|
|
|
tracing_subscriber::fmt()
|
|
.with_env_filter(filter_layer)
|
|
.with_test_writer()
|
|
.init();
|
|
|
|
let config = &*Box::leak::<'static>(Box::new(
|
|
magnetar_common::config::load_config().map_err(|e| miette!(e))?,
|
|
));
|
|
info!("Loaded configuration: {config:#?}");
|
|
|
|
let app = Router::new()
|
|
.layer(TraceLayer::new_for_http())
|
|
.nest("", new_calckey_fe_router(config).into_diagnostic()?);
|
|
|
|
let addr = SocketAddr::from((
|
|
config.calckey_frontend.bind_addr,
|
|
config.calckey_frontend.port,
|
|
));
|
|
info!("Serving on: {addr}");
|
|
axum::Server::bind(&addr)
|
|
.serve(app.into_make_service())
|
|
.await
|
|
.map_err(|e| miette!("Error running server: {}", e))
|
|
}
|