wip: add base for server

This commit is contained in:
2025-04-30 23:39:21 +02:00
parent ce49dd6e3d
commit 06b389b28f

View File

@@ -1,16 +1,41 @@
use std::sync::{atomic::AtomicU32, Arc};
use http_body_util::{combinators::BoxBody, BodyExt, Full};
use hyper::{body::{Bytes, Incoming}, Error, Request, Response};
use hyper::{
body::{Bytes, Incoming}, server::conn::http1, service::service_fn, Error, Request, Response
};
use hyper_util::rt::TokioIo;
use tokio::net::TcpListener;
mod args;
mod config;
async fn router(
_: Request<Incoming>,
_: Arc<SharedState>,
) -> Result<Response<BoxBody<Bytes, Error>>, Error> {
Ok(Response::new(Full::new("test".into()).map_err(|n| match n {}).boxed()))
}
fn load_config(path: &str) -> Result<config::Config, &'static str> {
let Ok(file_contents) = std::fs::read_to_string(path) else {
return Err("could not read the config file");
};
toml::from_str(&file_contents).map_err(|_| "invalid config file structure or fields")
}
struct SharedState {
counter: AtomicU32,
}
impl Default for SharedState {
fn default() -> Self {
Self { counter: 0.into() }
}
}
#[tokio::main]
async fn main() {
// load config
let args = match args::Args::try_from(&mut std::env::args()) {
Ok(v) => v,
Err(e) => {
@@ -18,5 +43,36 @@ async fn main() {
return;
}
};
println!("{}", args.config_path());
let config = match load_config(args.config_path()) {
Ok(v) => v,
Err(e) => {
eprintln!("{}", e);
return;
}
};
// shared state
let state = Arc::new(SharedState::default());
// server
let Ok(listener) = TcpListener::bind(config.server.bind_to).await else {
eprintln!("unable to bind to: {:?}", config.server.bind_to);
return;
};
loop {
let Ok((stream, _)) = listener.accept().await else {
eprintln!("unable to accept new connections");
return;
};
let io = TokioIo::new(stream);
let state_clone = state.clone();
tokio::task::spawn(async move {
if let Err(_) = http1::Builder::new().serve_connection(io, service_fn(move |req| {
router(req, state_clone.clone())
})).await {
println!("closed connection");
}
});
}
}