use std::str; use axum::{RequestExt, RequestPartsExt, extract::Path}; use bytes::Bytes; use conduwuit::{Result, err}; use http::request::Parts; use serde::Deserialize; use service::Services; #[derive(Deserialize)] pub(super) struct QueryParams { pub(super) access_token: Option, pub(super) user_id: Option, } pub(super) struct Request { pub(super) path: Path>, pub(super) query: QueryParams, pub(super) body: Bytes, pub(super) parts: Parts, } pub(super) async fn from( services: &Services, request: hyper::Request, ) -> Result { let limited = request.with_limited_body(); let (mut parts, body) = limited.into_parts(); let path: Path> = parts.extract().await?; let query = parts.uri.query().unwrap_or_default(); let query = serde_html_form::from_str(query) .map_err(|e| err!(Request(Unknown("Failed to read query parameters: {e}"))))?; let max_body_size = services.server.config.max_request_size; // Check if the Content-Length header is present and valid, saves us streaming // the response into memory if let Some(content_length) = parts.headers.get(http::header::CONTENT_LENGTH) { if let Ok(content_length) = content_length .to_str() .map(|s| s.parse::().unwrap_or_default()) { if content_length > max_body_size { return Err(err!(Request(TooLarge("Request body too large")))); } } } let body = axum::body::to_bytes(body, max_body_size) .await .map_err(|e| err!(Request(TooLarge("Request body too large: {e}"))))?; Ok(Request { path, query, body, parts }) }