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

Small change and cargo fmt.

Browse source
This commit is contained in:
Kavin 2023-10-10 02:08:00 +01:00
parent 4472a17049
commit bd9232d6c2
No known key found for this signature in database
GPG key ID: 6E4598CA5C92C41F
1 changed files with 12 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
src/main.rs
View file

@ -29,7 +29,8 @@ async fn main() -> std::io::Result<()> {
// get socket/port from env // get socket/port from env
// backwards compat when only UDS is set // backwards compat when only UDS is set
if env::var("UDS").is_ok() { if env::var("UDS").is_ok() {
let socket_path = env::var("BIND_UNIX").unwrap_or_else(|_| "./socket/actix.sock".to_string()); let socket_path =
env::var("BIND_UNIX").unwrap_or_else(|_| "./socket/actix.sock".to_string());
server.bind_uds(socket_path)? server.bind_uds(socket_path)?
} else { } else {
let bind = env::var("BIND").unwrap_or_else(|_| "0.0.0.0:8080".to_string()); let bind = env::var("BIND").unwrap_or_else(|_| "0.0.0.0:8080".to_string());
@ -321,13 +322,13 @@ async fn index(req: HttpRequest) -> Result<HttpResponse, Box<dyn Error>> {
return Ok(response.body(modified)); return Ok(response.body(modified));
} }
if content_type == "video/vnd.mpeg.dash.mpd" || content_type == "application/dash+xml" { if content_type == "video/vnd.mpeg.dash.mpd" || content_type == "application/dash+xml" {
let mut resp_str = resp.text().await.unwrap(); let resp_str = resp.text().await.unwrap();
let clone_resp = resp_str.clone(); let mut new_resp = resp_str.clone();
let captures = RE_DASH_MANIFEST.captures_iter(&clone_resp); let captures = RE_DASH_MANIFEST.captures_iter(&resp_str);
for capture in captures { for capture in captures {
let url = capture.get(1).unwrap().as_str(); let url = capture.get(1).unwrap().as_str();
let new_url = localize_url(url, host.as_str()); let new_url = localize_url(url, host.as_str());
resp_str = resp_str.replace(url, new_url.as_str()); new_resp = new_resp.replace(url, new_url.as_str());
} }
return Ok(response.body(resp_str)); return Ok(response.body(resp_str));
} }
@ -352,7 +353,12 @@ fn localize_url(url: &str, host: &str) -> String {
return format!("{}?{}", url.path(), url.query().unwrap()); return format!("{}?{}", url.path(), url.query().unwrap());
} else if url.ends_with(".m3u8") || url.ends_with(".ts") { } else if url.ends_with(".m3u8") || url.ends_with(".ts") {
return format!("{}{}host={}", url, if url.contains('?') { "&" } else { "?" }, host); return format!(
"{}{}host={}",
url,
if url.contains('?') { "&" } else { "?" },
host
);
} }
url.to_string() url.to_string()