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

feature parity with legacy ytproxy

Browse source
This commit is contained in:
Andrea Spacca 2023-03-01 11:38:21 +09:00
parent 8bbda01e9c
commit b1658c94af
1 changed files with 13 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

14
src/main.rs
View file

@ -1,7 +1,9 @@
use std::env; use std::env;
use std::error::Error; use std::error::Error;
use std::string::ToString;
use actix_web::{App, HttpRequest, HttpResponse, HttpResponseBuilder, HttpServer, web}; use actix_web::{App, HttpRequest, HttpResponse, HttpResponseBuilder, HttpServer, web};
use actix_web::dev::ResourcePath;
use actix_web::http::Method; use actix_web::http::Method;
use image::EncodableLayout; use image::EncodableLayout;
use lazy_static::lazy_static; use lazy_static::lazy_static;
@ -49,6 +51,10 @@ lazy_static!(
}; };
); );
lazy_static!(
static ref PREFIX_PATH: String = env::var("PREFIX_PATH").unwrap_or_else(|_| "".to_string());
);
const ALLOWED_DOMAINS: [&str; 7] = [ const ALLOWED_DOMAINS: [&str; 7] = [
"youtube.com", "youtube.com",
"googlevideo.com", "googlevideo.com",
@ -59,6 +65,7 @@ const ALLOWED_DOMAINS: [&str; 7] = [
"odycdn.com", "odycdn.com",
]; ];
fn add_headers(response: &mut HttpResponseBuilder) { fn add_headers(response: &mut HttpResponseBuilder) {
response response
.append_header(("Access-Control-Allow-Origin", "*")) .append_header(("Access-Control-Allow-Origin", "*"))
@ -73,6 +80,11 @@ fn is_header_allowed(header: &str) -> bool {
} }
!matches!(header, "host" | !matches!(header, "host" |
"authorization" |
"origin" |
"referer" |
"cookie" |
"etag" |
"content-length" | "content-length" |
"set-cookie" | "set-cookie" |
"alt-svc" | "alt-svc" |
@ -242,7 +254,7 @@ fn localize_url(url: &str, host: &str) -> String {
url.query_pairs_mut() url.query_pairs_mut()
.append_pair("host", &host); .append_pair("host", &host);
return format!("{}?{}", url.path(), url.query().unwrap()); return format!("{}{}?{}", PREFIX_PATH.as_str(), 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 if url.contains('?') { return if url.contains('?') {
format!("{}&host={}", url, host) format!("{}&host={}", url, host)