From 652cabb51294064f51b6459f000c75941f412a27 Mon Sep 17 00:00:00 2001
From: Rory& <root@rory.gay>
Date: Sat, 16 Nov 2024 03:33:00 +0100
Subject: HSE updates

---
 .../Controllers/Media/MediaController.cs                             | 5 +++++
 1 file changed, 5 insertions(+)

(limited to 'Utilities/LibMatrix.HomeserverEmulator/Controllers/Media')

diff --git a/Utilities/LibMatrix.HomeserverEmulator/Controllers/Media/MediaController.cs b/Utilities/LibMatrix.HomeserverEmulator/Controllers/Media/MediaController.cs
index 7899ada..59d37ff 100644
--- a/Utilities/LibMatrix.HomeserverEmulator/Controllers/Media/MediaController.cs
+++ b/Utilities/LibMatrix.HomeserverEmulator/Controllers/Media/MediaController.cs
@@ -42,6 +42,9 @@ public class MediaController(
 
     [HttpGet("download/{serverName}/{mediaId}")]
     public async Task DownloadMedia(string serverName, string mediaId) {
+        Response.Headers["Access-Control-Allow-Origin"] = "*";
+        Response.Headers["Access-Control-Allow-Methods"] = "GET";
+        
         var stream = await DownloadRemoteMedia(serverName, mediaId);
         await stream.CopyToAsync(Response.Body);
     }
@@ -56,6 +59,7 @@ public class MediaController(
         JsonObject data = new();
 
         using var hc = new HttpClient();
+        logger.LogInformation("Getting URL preview for {}", url);
         using var response = await hc.GetAsync(url);
         var doc = await response.Content.ReadAsStringAsync();
         var match = Regex.Match(doc, "<meta property=\"(.*?)\" content=\"(.*?)\"");
@@ -80,6 +84,7 @@ public class MediaController(
                     };
                 using var client = new HttpClient();
                 var stream = await client.GetStreamAsync(mediaUrl);
+                Directory.CreateDirectory(Path.GetDirectoryName(path)!);
                 await using var fs = System.IO.File.Create(path);
                 await stream.CopyToAsync(fs);
             }
-- 
cgit 1.5.1