58 lines
1.7 KiB
Rust
58 lines
1.7 KiB
Rust
|
use axum::{
|
||
|
|
extract::DefaultBodyLimit,
|
||
|
|
routing::{get, post},
|
||
|
|
Extension, Json, Router,
|
||
|
|
};
|
||
|
|
use std::net::SocketAddr;
|
||
|
|
use tokio::net::TcpListener;
|
||
|
|
use tower_http::limit::RequestBodyLimitLayer;
|
||
|
|
|
||
|
|
use crate::{
|
||
|
|
discovery::http::register_device,
|
||
|
|
transfer::upload::{register_prepare_upload, register_upload},
|
||
|
|
Client,
|
||
|
|
};
|
||
|
|
|
||
|
|
impl Client {
|
||
|
|
pub async fn start_http_server(&self) -> crate::error::Result<()> {
|
||
|
|
let app = self.create_router();
|
||
|
|
let addr = SocketAddr::from(([0, 0, 0, 0], self.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) -> Router {
|
||
|
|
let peers = self.peers.clone();
|
||
|
|
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(self.device.clone()))
|
||
|
|
.layer(Extension(self.sessions.clone()))
|
||
|
|
.layer(Extension(self.download_dir.clone()))
|
||
|
|
.with_state(peers)
|
||
|
|
}
|
||
|
|
}
|