cmd/xedn: fly region annotation middleware

Signed-off-by: Xe Iaso <me@xeiaso.net>

1 files changed, 13 insertions, 1 deletions

diff --git a/cmd/xedn/main.go b/cmd/xedn/main.go
index 3ad0624..a6762a8 100644
--- a/cmd/xedn/main.go
+++ b/cmd/xedn/main.go
@@ -213,6 +213,18 @@ func main() {
 		http.Error(w, "wait, what, how did you do that?", http.StatusBadRequest)
 	})
 
+	var h http.Handler = topLevel
+	h = xffMW.Handler(h)
+	h = cors.Default().Handler(h)
+	h = FlyRegionAnnotation(h)
+
 	slog.Info("starting up", "addr", *addr)
-	http.ListenAndServe(*addr, cors.Default().Handler(xffMW.Handler(topLevel)))
+	http.ListenAndServe(*addr, h)
+}
+
+func FlyRegionAnnotation(next http.Handler) http.Handler {
+	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		w.Header().Set("Fly-Region", os.Getenv("FLY_REGION"))
+		next.ServeHTTP(w, r)
+	})
 }