heimoshuiyu/imageproxy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

cors and LISTEN env

Browse Source
This commit is contained in:
黑墨水鱼
2025-01-05 22:35:06 +08:00
parent f5ef691324
commit db83d0adb0
1 changed files with 11 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/main.rs
View File

@@ -58,7 +58,11 @@ async fn main() -> anyhow::Result<()> {
let app = axum::Router::new() let app = axum::Router::new()
.fallback(get(api_proxy_image)) .fallback(get(api_proxy_image))
.with_state(downloader); .with_state(downloader);
let listener = tokio::net::TcpListener::bind("0.0.0.0:2999").await?;
let listen = std::env::var("LISTEN").unwrap_or("0.0.0.0:2999".to_string());
println!("listening on: {}", listen);
let listener = tokio::net::TcpListener::bind(&listen).await?;
axum::serve(listener, app).await?; axum::serve(listener, app).await?;
Ok(()) Ok(())
@@ -140,6 +144,12 @@ async fn api_proxy_image(
Ok(Response::builder() Ok(Response::builder()
.header("Content-Type", "image/webp") .header("Content-Type", "image/webp")
.header("Cache-Control", "public, max-age=31536000")
.header("Access-Control-Allow-Origin", "*")
.header("Access-Control-Allow-Methods", "GET")
.header("Access-Control-Allow-Headers", "Content-Type")
.header("Access-Control-Max-Age", "86400")
.header("X-Original-Url", target_url)
.header("X-Original-Content-Type", ori_content_type) .header("X-Original-Content-Type", ori_content_type)
.header("X-Original-Size", original_size.to_string()) .header("X-Original-Size", original_size.to_string())
.header("X-Original-Width", ori_width.to_string()) .header("X-Original-Width", ori_width.to_string())