mirror of
https://github.com/bitcoinresearchkit/brk.git
synced 2026-04-28 16:49:58 -07:00
git: reset
This commit is contained in:
78
server/src/main.rs
Normal file
78
server/src/main.rs
Normal file
@@ -0,0 +1,78 @@
|
||||
use std::sync::Arc;
|
||||
|
||||
use axum::{extract::State, http::HeaderMap, response::Response, routing::get, serve, Router};
|
||||
use parser::log;
|
||||
use reqwest::header::HOST;
|
||||
use response::generic_to_reponse;
|
||||
use routes::Routes;
|
||||
use serde::Serialize;
|
||||
use tokio::net::TcpListener;
|
||||
use tower_http::compression::CompressionLayer;
|
||||
|
||||
mod chunk;
|
||||
mod handler;
|
||||
mod headers;
|
||||
mod imports;
|
||||
mod kind;
|
||||
mod paths;
|
||||
mod response;
|
||||
mod routes;
|
||||
|
||||
use handler::file_handler;
|
||||
|
||||
#[derive(Clone, Debug, Default, Serialize)]
|
||||
pub struct Grouped<T> {
|
||||
pub date: T,
|
||||
pub height: T,
|
||||
pub last: T,
|
||||
}
|
||||
|
||||
#[derive(Clone)]
|
||||
pub struct AppState {
|
||||
routes: Arc<Routes>,
|
||||
}
|
||||
|
||||
#[tokio::main]
|
||||
async fn main() -> color_eyre::Result<()> {
|
||||
color_eyre::install()?;
|
||||
|
||||
let routes = Routes::build();
|
||||
|
||||
routes.generate_grouped_keys_to_url_path_file();
|
||||
|
||||
let state = AppState {
|
||||
routes: Arc::new(routes),
|
||||
};
|
||||
|
||||
let compression_layer = CompressionLayer::new()
|
||||
.br(true)
|
||||
.deflate(true)
|
||||
.gzip(true)
|
||||
.zstd(true);
|
||||
|
||||
let router = Router::new()
|
||||
.route("/*path", get(file_handler))
|
||||
.route("/", get(fallback))
|
||||
.with_state(state)
|
||||
.layer(compression_layer);
|
||||
|
||||
let port = 3110;
|
||||
|
||||
log(&format!("Starting server on port {port}..."));
|
||||
|
||||
let listener = TcpListener::bind(format!("0.0.0.0:{port}")).await?;
|
||||
|
||||
serve(listener, router).await?;
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
pub async fn fallback(headers: HeaderMap, State(app_state): State<AppState>) -> Response {
|
||||
generic_to_reponse(
|
||||
app_state
|
||||
.routes
|
||||
.to_full_paths(headers[HOST].to_str().unwrap().to_string()),
|
||||
None,
|
||||
60,
|
||||
)
|
||||
}
|
||||
Reference in New Issue
Block a user