Don't set response headers if they're not present

matrix-org · Sep 5, 2024 · b2ce852 · b2ce852
1 parent afdfd30
commit b2ce852
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/src/components/media-proxy.ts b/src/components/media-proxy.ts
@@ -159,9 +159,15 @@ export class MediaProxy {
             },
         }, (getRes) => {
             const { statusCode } = res;
-            res.setHeader('content-disposition', getRes.headers['content-disposition'] as string);
-            res.setHeader('content-type', getRes.headers['content-type'] as string);
-            res.setHeader('content-length', getRes.headers['content-length'] as string);
+            if (getRes.headers['content-disposition']) {
+                res.setHeader('content-disposition', getRes.headers['content-disposition']);
+            }
+            if (getRes.headers['content-type']) {
+                res.setHeader('content-type', getRes.headers['content-type']);
+            }
+            if (getRes.headers['content-length']) {
+                res.setHeader('content-length', getRes.headers['content-length']);
+            }
             res.status(statusCode);
             getRes.pipe(res);
         });