joecalsend/src/http_server.rs

use std::net::SocketAddr;

use axum::{
    Extension, Json, Router,
    extract::DefaultBodyLimit,
    routing::{get, post},
};
use tokio::net::TcpListener;
use tower_http::limit::RequestBodyLimitLayer;

use crate::{
    Config, JoecalState,
    discovery::register_device,
    transfer::{register_prepare_upload, register_upload},
};

impl JoecalState {
    pub async fn start_http_server(&self, config: &Config) -> crate::error::Result<()> {
        let app = self.create_router(&config);
        let addr = SocketAddr::from(([0, 0, 0, 0], config.port));

        let listener = TcpListener::bind(&addr).await?;
        println!("HTTP server listening on {addr}");

        axum::serve(
            listener,
            app.into_make_service_with_connect_info::<SocketAddr>(),
        )
        .await?;
        Ok(())
    }

    fn create_router(&self, config: &Config) -> Router {
        let device = self.device.clone();
        Router::new()
            .route("/api/localsend/v2/register", post(register_device))
            .route(
                "/api/localsend/v2/info",
                get(move || {
                    let device = device.clone();
                    async move { Json(device) }
                }),
            )
            .route(
                "/api/localsend/v2/prepare-upload",
                post(register_prepare_upload),
            )
            .route("/api/localsend/v2/upload", post(register_upload))
            .layer(DefaultBodyLimit::disable())
            .layer(RequestBodyLimitLayer::new(1024 * 1024 * 1024))
            .layer(Extension(config.download_dir.clone()))
            .with_state(self.clone())
    }
}