TeamPiped-mirror/piped-proxy
1
0
Fork 0
mirror of https://github.com/TeamPiped/piped-proxy.git synced 2024-08-14 23:50:45 +00:00
Code Issues Projects Releases Packages Wiki Activity

Send 206 response in UMP response content.

Browse source
This commit is contained in:
Kavin 2023-12-06 06:39:18 +00:00
parent b9c8dc8f58
commit a9bc07e98d
No known key found for this signature in database
GPG key ID: 6E4598CA5C92C41F
1 changed files with 7 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
src/main.rs
View file

@ -1,4 +1,4 @@
use actix_web::http::Method; use actix_web::http::{Method, StatusCode};
use actix_web::{web, App, HttpRequest, HttpResponse, HttpResponseBuilder, HttpServer}; use actix_web::{web, App, HttpRequest, HttpResponse, HttpResponseBuilder, HttpServer};
use once_cell::sync::Lazy; use once_cell::sync::Lazy;
use qstring::QString; use qstring::QString;
@ -120,6 +120,7 @@ fn is_header_allowed(header: &str) -> bool {
| "report-to" | "report-to"
| "strict-transport-security" | "strict-transport-security"
| "user-agent" | "user-agent"
| "range"
) )
} }
@ -244,6 +245,8 @@ async fn index(req: HttpRequest) -> Result<HttpResponse, Box<dyn Error>> {
} }
} }
let has_range = query.has("range");
let qs = { let qs = {
let collected = query let collected = query
.into_pairs() .into_pairs()
@ -427,6 +430,9 @@ async fn index(req: HttpRequest) -> Result<HttpResponse, Box<dyn Error>> {
if let Some(mime_type) = mime_type { if let Some(mime_type) = mime_type {
response.content_type(mime_type); response.content_type(mime_type);
} }
if has_range {
response.status(StatusCode::PARTIAL_CONTENT);
}
let transformed_stream = UmpTransformStream::new(resp); let transformed_stream = UmpTransformStream::new(resp);
return Ok(response.streaming(transformed_stream)); return Ok(response.streaming(transformed_stream));
} }