about summary refs log tree commit diff
path: root/LibMatrix/Services/WellKnownResolver/WellKnownResolverService.cs
blob: 4c78347b7665571d764dd81d272ce8fae6c9769d (plain) (blame)
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
using System.Diagnostics;
using System.Text.Json.Serialization;
using LibMatrix.Extensions;
using LibMatrix.Services.WellKnownResolver.WellKnownResolvers;
using Microsoft.Extensions.Logging;
using Microsoft.Extensions.Logging.Abstractions;

namespace LibMatrix.Services.WellKnownResolver;

public class WellKnownResolverService {
    private readonly MatrixHttpClient _httpClient = new();

    private readonly ILogger<WellKnownResolverService> _logger;
    private readonly ClientWellKnownResolver _clientWellKnownResolver;
    private readonly SupportWellKnownResolver _supportWellKnownResolver;
    private readonly ServerWellKnownResolver _serverWellKnownResolver;
    private readonly WellKnownResolverConfiguration _configuration;

    public WellKnownResolverService(ILogger<WellKnownResolverService> logger, ClientWellKnownResolver clientWellKnownResolver, SupportWellKnownResolver supportWellKnownResolver,
        WellKnownResolverConfiguration configuration, ServerWellKnownResolver serverWellKnownResolver) {
        _logger = logger;
        _clientWellKnownResolver = clientWellKnownResolver;
        _supportWellKnownResolver = supportWellKnownResolver;
        _configuration = configuration;
        _serverWellKnownResolver = serverWellKnownResolver;
        if (logger is NullLogger<WellKnownResolverService>) {
            var stackFrame = new StackTrace(true).GetFrame(1);
            Console.WriteLine(
                $"WARN | Null logger provided to WellKnownResolverService!\n{stackFrame?.GetMethod()?.DeclaringType?.ToString() ?? "null"} at {stackFrame?.GetFileName() ?? "null"}:{stackFrame?.GetFileLineNumber().ToString() ?? "null"}");
        }
    }

    public async Task<WellKnownRecords> TryResolveWellKnownRecords(string homeserver, bool includeClient = true, bool includeServer = true, bool includeSupport = true,
        WellKnownResolverConfiguration? config = null) {
        WellKnownRecords records = new();
        _logger.LogDebug($"Resolving well-knowns for {homeserver}");
        if (includeClient && await _clientWellKnownResolver.TryResolveWellKnown(homeserver, config ?? _configuration) is { } clientResult) {
            records.ClientWellKnown = clientResult;
        }
        
        if (includeServer && await _serverWellKnownResolver.TryResolveWellKnown(homeserver, config ?? _configuration) is { } serverResult) {
            records.ServerWellKnown = serverResult;
        }
        
        if (includeSupport && await _supportWellKnownResolver.TryResolveWellKnown(homeserver, config ?? _configuration) is { } supportResult) {
            records.SupportWellKnown = supportResult;
        }

        return records;
    }

    public class WellKnownRecords {
        public WellKnownResolutionResult<ClientWellKnown?>? ClientWellKnown { get; set; }
        public WellKnownResolutionResult<ServerWellKnown?>? ServerWellKnown { get; set; }
        public WellKnownResolutionResult<SupportWellKnown?>? SupportWellKnown { get; set; }
    }

    public class WellKnownResolutionResult<T> {
        public WellKnownSource Source { get; set; }
        public string? SourceUri { get; set; }
        public T? Content { get; set; }
        public List<WellKnownResolutionWarning> Warnings { get; set; } = [];
    }

    [JsonConverter(typeof(JsonStringEnumConverter))]
    public enum WellKnownSource {
        None,
        Https,
        Dns,
        Http,
        ManualCheck,
        Search
    }

    public struct WellKnownResolutionWarning {
        public WellKnownResolutionWarningType Type { get; set; }
        public string Message { get; set; }
        [JsonIgnore]
        public Exception? Exception { get; set; }
        public string? ExceptionMessage => Exception?.Message;

        [JsonConverter(typeof(JsonStringEnumConverter))]
        public enum WellKnownResolutionWarningType {
            None,
            Exception,
            InvalidResponse,
            Timeout,
            SlowResponse
        }
    }
}