about summary refs log tree commit diff
path: root/MatrixRoomUtils.Web/Classes/MRUStorageWrapper.cs
blob: 41f604de84b878c91307bd44329a6c7cd61a5ac6 (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 MatrixRoomUtils.Core;
using MatrixRoomUtils.Core.Responses;
using MatrixRoomUtils.Core.Services;
using Microsoft.AspNetCore.Components;

namespace MatrixRoomUtils.Web.Classes;

public class MRUStorageWrapper {
    private readonly TieredStorageService _storageService;
    private readonly HomeserverProviderService _homeserverProviderService;
    private readonly NavigationManager _navigationManager;

    public MRUStorageWrapper(
        TieredStorageService storageService,
        HomeserverProviderService homeserverProviderService,
        NavigationManager navigationManager
    ) {
        _storageService = storageService;
        _homeserverProviderService = homeserverProviderService;
        _navigationManager = navigationManager;
    }

    public async Task<List<LoginResponse>?> GetAllTokens() {
        return await _storageService.DataStorageProvider.LoadObjectAsync<List<LoginResponse>>("mru.tokens") ?? new List<LoginResponse>();
    }
    public async Task<LoginResponse?> GetCurrentToken() {
        var currentToken = await _storageService.DataStorageProvider.LoadObjectAsync<LoginResponse>("token");
        var allTokens = await GetAllTokens();
        if (allTokens is null or { Count: 0 }) {
            await SetCurrentToken(null);
            return null;
        }
        if (currentToken is null) {
            await SetCurrentToken(currentToken = allTokens[0]);
        }
        if(!allTokens.Any(x=>x.AccessToken == currentToken.AccessToken)) {
            await SetCurrentToken(currentToken = allTokens[0]);
        }
        return currentToken;
    }

    public async Task AddToken(LoginResponse loginResponse) {
        var tokens = await GetAllTokens();
        if (tokens == null) {
            tokens = new List<LoginResponse>();
        }

        tokens.Add(loginResponse);
        await _storageService.DataStorageProvider.SaveObjectAsync("mru.tokens", tokens);
    }

    public async Task<AuthenticatedHomeServer?> GetCurrentSession() {
        var token = await GetCurrentToken();
        if (token == null) {
            return null;
        }
        
        return await _homeserverProviderService.GetAuthenticatedWithToken(token.Homeserver, token.AccessToken); 
    }

    public async Task<AuthenticatedHomeServer?> GetCurrentSessionOrNavigate() {
        var session = await GetCurrentSession();
        if (session == null) {
            _navigationManager.NavigateTo("/Login");
        }
        return session;
    }
    public class Settings {
        public DeveloperSettings DeveloperSettings { get; set; } = new();
    }

    public class DeveloperSettings {
        public bool EnableLogViewers { get; set; } = false;
        public bool EnableConsoleLogging { get; set; } = true;
        public bool EnablePortableDevtools { get; set; } = false;
    }

    public async Task RemoveToken(LoginResponse auth) {
        var tokens = await GetAllTokens();
        if (tokens == null) {
            return;
        }

        tokens.RemoveAll(x=>x.AccessToken == auth.AccessToken);
        await _storageService.DataStorageProvider.SaveObjectAsync("mru.tokens", tokens);
    }

    public async Task SetCurrentToken(LoginResponse? auth) {
        _storageService.DataStorageProvider.SaveObjectAsync("token", auth);
    }
}