1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
|
using System.Net.Http.Headers;
using LibMatrix.Homeservers;
using LibMatrix.Services;
using Microsoft.AspNetCore.Mvc;
using MxApiExtensions.Classes.LibMatrix;
using MxApiExtensions.Services;
namespace MxApiExtensions.Controllers;
[ApiController]
[Route("/")]
public class MediaProxyController(ILogger<GenericController> logger, MxApiExtensionsConfiguration config, AuthenticationService authenticationService,
AuthenticatedHomeserverProviderService authenticatedHomeserverProviderService, HomeserverProviderService hsProvider)
: ControllerBase {
private class MediaCacheEntry {
public DateTime LastRequested { get; set; } = DateTime.Now;
public byte[] Data { get; set; }
public string ContentType { get; set; }
public long Size => Data.LongCount();
}
private static Dictionary<string, MediaCacheEntry> _mediaCache = new();
private static SemaphoreSlim _semaphore = new(1, 1);
[HttpGet("/_matrix/media/{_}/download/{serverName}/{mediaId}")]
public async Task ProxyMedia(string? _, string serverName, string mediaId) {
try {
logger.LogInformation("Proxying media: {}{}", serverName, mediaId);
await _semaphore.WaitAsync();
MediaCacheEntry entry;
if (!_mediaCache.ContainsKey($"{serverName}/{mediaId}")) {
_mediaCache.Add($"{serverName}/{mediaId}", entry = new());
List<RemoteHomeserver> FeasibleHomeservers = new();
{
var a = await authenticatedHomeserverProviderService.TryGetRemoteHomeserver();
if (a is not null)
FeasibleHomeservers.Add(a);
if (a is AuthenticatedHomeserverGeneric ahg) {
var rooms = await ahg.GetJoinedRooms();
foreach (var room in rooms) {
var ahs = (await room.GetMembersByHomeserverAsync()).Keys.Select(x => x.ToString()).ToList();
foreach (var ah in ahs) {
try {
if (!FeasibleHomeservers.Any(x => x.BaseUrl == ah)) {
FeasibleHomeservers.Add(await hsProvider.GetRemoteHomeserver(ah));
}
}
catch { }
}
}
}
}
FeasibleHomeservers.Add(await hsProvider.GetRemoteHomeserver(serverName));
foreach (var homeserver in FeasibleHomeservers) {
var resp = await homeserver.ClientHttpClient.GetAsync($"{Request.Path}");
if (!resp.IsSuccessStatusCode) continue;
entry.ContentType = resp.Content.Headers.ContentType?.ToString() ?? "application/json";
entry.Data = await resp.Content.ReadAsByteArrayAsync();
if (entry.Data is not { Length: > 0 }) throw new NullReferenceException("No data received?");
break;
}
if (entry.Data is not { Length: > 0 }) throw new NullReferenceException("No data received from any homeserver?");
}
else if (_mediaCache[$"{serverName}/{mediaId}"].Data is not { Length: > 0 }) {
_mediaCache.Remove($"{serverName}/{mediaId}");
await ProxyMedia(_, serverName, mediaId);
return;
}
else entry = _mediaCache[$"{serverName}/{mediaId}"];
if (entry.Data is null) throw new NullReferenceException("No data?");
_semaphore.Release();
Response.StatusCode = 200;
Response.ContentType = entry.ContentType;
await Response.StartAsync();
await Response.Body.WriteAsync(entry.Data.ToArray(), 0, entry.Data.Length);
await Response.Body.FlushAsync();
await Response.CompleteAsync();
}
catch (MxApiMatrixException e) {
logger.LogError(e, "Matrix error");
Response.StatusCode = StatusCodes.Status500InternalServerError;
Response.ContentType = "application/json";
await Response.WriteAsync(e.GetAsJson());
await Response.CompleteAsync();
}
catch (Exception e) {
logger.LogError(e, "Unhandled error");
Response.StatusCode = StatusCodes.Status500InternalServerError;
Response.ContentType = "text/plain";
await Response.WriteAsync(e.ToString());
await Response.CompleteAsync();
}
}
[HttpGet("/_matrix/media/{_}/thumbnail/{serverName}/{mediaId}")]
public async Task ProxyThumbnail(string? _, string serverName, string mediaId) => await ProxyMedia(_, serverName, mediaId);
}
|
