External/buzz
1
0
Fork 0
mirror of https://github.com/block/buzz.git synced 2026-08-18 06:50:31 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix(desktop): forward cache headers in media proxy (#549)

Browse Source 
Signed-off-by: tlongwell-block <109685178+tlongwell-block@users.noreply.github.com>
This commit is contained in:
tlongwell-block
2026-05-12 14:40:07 -04:00
committed by GitHub
parent afe072fa42
commit effa24f06c
1 changed files with 39 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
desktop/src-tauri/src/media_proxy.rs
+39
View File
@@ -78,6 +78,15 @@ async fn proxy_handler(AxumState(state): AxumState<ProxyState>, req: Request) ->
"content-range",
"accept-ranges",
"content-length",
// Cache-related headers — let WKWebView's HTTP cache do its job so
// images don't re-fetch on every channel switch. Media URLs are
// content-addressed (sha256 in path), so the relay sends
// `Cache-Control: public, max-age=31536000, immutable` and we
// forward it verbatim. `etag`/`last-modified` are forwarded for
// future-proofing if upstream ever adds them.
"cache-control",
"etag",
"last-modified",
] {
if let Some(val) = resp.headers().get(*key) {
if let Ok(v) = HeaderValue::from_bytes(val.as_bytes()) {
@@ -205,6 +214,27 @@ pub async fn handle_sprout_media(
.and_then(|v| v.to_str().ok())
.map(|s| s.to_string());
// Propagate cache-related headers so WKWebView's HTTP cache
// can avoid re-fetching content-addressed media on every
// channel switch. The relay sends
// `Cache-Control: public, max-age=31536000, immutable`;
// `etag`/`last-modified` are forwarded if upstream supplies them.
let cache_control = resp
.headers()
.get("cache-control")
.and_then(|v| v.to_str().ok())
.map(|s| s.to_string());
let etag = resp
.headers()
.get("etag")
.and_then(|v| v.to_str().ok())
.map(|s| s.to_string());
let last_modified = resp
.headers()
.get("last-modified")
.and_then(|v| v.to_str().ok())
.map(|s| s.to_string());
// OOM guard: if this is a non-range GET and the upstream body is
// larger than our cap, bail with 413 instead of buffering into RAM.
// Tauri's protocol handler requires Vec<u8> so we can't truly stream.
@@ -235,6 +265,15 @@ pub async fn handle_sprout_media(
if let Some(ref cl) = content_length {
builder = builder.header("content-length", cl);
}
if let Some(ref cc) = cache_control {
builder = builder.header("cache-control", cc);
}
if let Some(ref e) = etag {
builder = builder.header("etag", e);
}
if let Some(ref lm) = last_modified {
builder = builder.header("last-modified", lm);
}
builder
.body(bytes.to_vec())
.unwrap_or_else(|_| error_response(500, "response build failed"))