Handle authed media endpoint (#731)

matrix-org · Jul 30, 2024 · 6e4426a · 6e4426a
1 parent 5d64767
commit 6e4426a
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/federation/handle.go b/federation/handle.go
@@ -471,6 +471,7 @@ func HandleKeyRequests() func(*Server) {
 func HandleMediaRequests(mediaIds map[string]func(w http.ResponseWriter)) func(*Server) {
 	return func(srv *Server) {
 		mediamux := srv.mux.PathPrefix("/_matrix/media").Subrouter()
+		mediamuxAuthenticated := srv.mux.PathPrefix("/_matrix/federation/v1/media").Subrouter()
 
 		downloadFn := http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
 			vars := mux.Vars(req)
@@ -497,6 +498,9 @@ func HandleMediaRequests(mediaIds map[string]func(w http.ResponseWriter)) func(*
 		mediamux.Handle("/r0/download/{origin}/{mediaId}", downloadFn).Methods("GET")
 		mediamux.Handle("/v1/download/{origin}/{mediaId}", downloadFn).Methods("GET")
 		mediamux.Handle("/v3/download/{origin}/{mediaId}", downloadFn).Methods("GET")
+
+		// Also handle authenticated media requests
+		mediamuxAuthenticated.Handle("/download/{mediaId}", downloadFn).Methods("GET")
 	}
 }