From 9214bb483289cbcba8268654129e08d1369c4b24 Mon Sep 17 00:00:00 2001
From: Mia Herkt <mia@0x0.st>
Date: Wed, 30 Nov 2022 02:28:19 +0100
Subject: [PATCH] Add X-Expires to file response headers

Tells clients when files will expire, in milliseconds since Unix epoch.

Closes #50.
---
 fhost.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/fhost.py b/fhost.py
index 2e2f8af..026e3bd 100755
--- a/fhost.py
+++ b/fhost.py
@@ -344,6 +344,7 @@ def store_file(f, requested_expiration:  typing.Optional[int], addr):
     sf, isnew = File.store(f, requested_expiration, addr)
 
     response = make_response(sf.geturl())
+    response.headers["X-Expires"] = sf.expiration
 
     if isnew:
         response.headers["X-Token"] = sf.mgmt_token
@@ -427,9 +428,11 @@ def get(path):
                 response.headers["Content-Type"] = f.mime
                 response.headers["Content-Length"] = fpath.stat().st_size
                 response.headers["X-Accel-Redirect"] = "/" + str(fpath)
-                return response
             else:
-                return send_from_directory(app.config["FHOST_STORAGE_PATH"], f.sha256, mimetype = f.mime)
+                response = send_from_directory(app.config["FHOST_STORAGE_PATH"], f.sha256, mimetype = f.mime)
+
+            response.headers["X-Expires"] = f.expiration
+            return response
     else:
         if request.method == "POST":
             abort(405)