diff --git a/BugMine.Web/Classes/BugMineStorage.cs b/BugMine.Web/Classes/BugMineStorage.cs
new file mode 100644
index 0000000..b45e253
--- /dev/null
+++ b/BugMine.Web/Classes/BugMineStorage.cs
@@ -0,0 +1,100 @@
+using LibMatrix;
+using LibMatrix.Homeservers;
+using LibMatrix.Services;
+using Microsoft.AspNetCore.Components;
+
+namespace BugMine.Web.Classes;
+
+public class BugMineStorage(ILogger<BugMineStorage> logger, TieredStorageService storageService, HomeserverProviderService homeserverProviderService, NavigationManager navigationManager) {
+ public async Task<List<UserAuth>?> GetAllTokens() {
+ logger.LogTrace("Getting all tokens.");
+ return await storageService.DataStorageProvider!.LoadObjectAsync<List<UserAuth>>("bugmine.tokens") ??
+ new List<UserAuth>();
+ }
+
+ public async Task<UserAuth?> GetCurrentToken() {
+ logger.LogTrace("Getting current token.");
+ var currentToken = await storageService.DataStorageProvider!.LoadObjectAsync<UserAuth>("bugmine.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(UserAuth UserAuth) {
+ logger.LogTrace("Adding token.");
+ var tokens = await GetAllTokens() ?? new List<UserAuth>();
+
+ tokens.Add(UserAuth);
+ await storageService.DataStorageProvider!.SaveObjectAsync("bugmine.tokens", tokens);
+ }
+
+ private async Task<AuthenticatedHomeserverGeneric?> GetCurrentSession() {
+ logger.LogTrace("Getting current session.");
+ var token = await GetCurrentToken();
+ if (token == null) {
+ return null;
+ }
+
+ return await homeserverProviderService.GetAuthenticatedWithToken(token.Homeserver, token.AccessToken, token.Proxy);
+ }
+
+ public async Task<AuthenticatedHomeserverGeneric?> GetSession(UserAuth userAuth) {
+ logger.LogTrace("Getting session.");
+ return await homeserverProviderService.GetAuthenticatedWithToken(userAuth.Homeserver, userAuth.AccessToken, userAuth.Proxy);
+ }
+
+ public async Task<AuthenticatedHomeserverGeneric?> GetCurrentSessionOrNavigate() {
+ logger.LogTrace("Getting current session or navigating.");
+ AuthenticatedHomeserverGeneric? session = null;
+
+ try {
+ //catch if the token is invalid
+ session = await GetCurrentSession();
+ }
+ catch (MatrixException e) {
+ if (e.ErrorCode == "M_UNKNOWN_TOKEN") {
+ var token = await GetCurrentToken();
+ logger.LogWarning("Encountered invalid token for {user} on {homeserver}", token.UserId, token.Homeserver);
+ navigationManager.NavigateTo("/InvalidSession?ctx=" + token.AccessToken);
+ return null;
+ }
+
+ throw;
+ }
+
+ if (session is null) {
+ logger.LogInformation("No session found. Navigating to login.");
+ navigationManager.NavigateTo("/Login");
+ }
+
+ return session;
+ }
+
+ public async Task RemoveToken(UserAuth auth) {
+ logger.LogTrace("Removing token.");
+ var tokens = await GetAllTokens();
+ if (tokens == null) {
+ return;
+ }
+
+ tokens.RemoveAll(x => x.AccessToken == auth.AccessToken);
+ await storageService.DataStorageProvider.SaveObjectAsync("bugmine.tokens", tokens);
+ }
+
+ public async Task SetCurrentToken(UserAuth? auth) {
+ logger.LogTrace("Setting current token.");
+ await storageService.DataStorageProvider.SaveObjectAsync("bugmine.token", auth);
+ }
+}
diff --git a/BugMine.Web/Classes/Constants.cs b/BugMine.Web/Classes/Constants.cs
new file mode 100644
index 0000000..f222879
--- /dev/null
+++ b/BugMine.Web/Classes/Constants.cs
@@ -0,0 +1,9 @@
+namespace BugMine.Web.Classes;
+
+public static class Constants {
+#if DEBUG
+ public const bool Debug = true;
+#else
+ public const bool Debug = false;
+#endif
+}
\ No newline at end of file
diff --git a/BugMine.Web/Classes/LocalStorageProviderService.cs b/BugMine.Web/Classes/LocalStorageProviderService.cs
new file mode 100644
index 0000000..247d789
--- /dev/null
+++ b/BugMine.Web/Classes/LocalStorageProviderService.cs
@@ -0,0 +1,28 @@
+using Blazored.LocalStorage;
+using LibMatrix.Interfaces.Services;
+
+namespace BugMine.Web.Classes;
+
+public class LocalStorageProviderService : IStorageProvider {
+ private readonly ILocalStorageService _localStorageService;
+
+ public LocalStorageProviderService(ILocalStorageService localStorageService) {
+ _localStorageService = localStorageService;
+ }
+
+ Task IStorageProvider.SaveAllChildrenAsync<T>(string key, T value) {
+ throw new NotImplementedException();
+ }
+
+ Task<T?> IStorageProvider.LoadAllChildrenAsync<T>(string key) where T : default => throw new NotImplementedException();
+
+ async Task IStorageProvider.SaveObjectAsync<T>(string key, T value) => await _localStorageService.SetItemAsync(key, value);
+
+ async Task<T?> IStorageProvider.LoadObjectAsync<T>(string key) where T : default => await _localStorageService.GetItemAsync<T>(key);
+
+ async Task<bool> IStorageProvider.ObjectExistsAsync(string key) => await _localStorageService.ContainKeyAsync(key);
+
+ async Task<List<string>> IStorageProvider.GetAllKeysAsync() => (await _localStorageService.KeysAsync()).ToList();
+
+ async Task IStorageProvider.DeleteObjectAsync(string key) => await _localStorageService.RemoveItemAsync(key);
+}
diff --git a/BugMine.Web/Classes/SessionStorageProviderService.cs b/BugMine.Web/Classes/SessionStorageProviderService.cs
new file mode 100644
index 0000000..c9f1c8e
--- /dev/null
+++ b/BugMine.Web/Classes/SessionStorageProviderService.cs
@@ -0,0 +1,28 @@
+using Blazored.SessionStorage;
+using LibMatrix.Interfaces.Services;
+
+namespace BugMine.Web.Classes;
+
+public class SessionStorageProviderService : IStorageProvider {
+ private readonly ISessionStorageService _sessionStorageService;
+
+ public SessionStorageProviderService(ISessionStorageService sessionStorage) {
+ _sessionStorageService = sessionStorage;
+ }
+
+ Task IStorageProvider.SaveAllChildrenAsync<T>(string key, T value) {
+ throw new NotImplementedException();
+ }
+
+ Task<T?> IStorageProvider.LoadAllChildrenAsync<T>(string key) where T : default => throw new NotImplementedException();
+
+ async Task IStorageProvider.SaveObjectAsync<T>(string key, T value) => await _sessionStorageService.SetItemAsync(key, value);
+
+ async Task<T?> IStorageProvider.LoadObjectAsync<T>(string key) where T : default => await _sessionStorageService.GetItemAsync<T>(key);
+
+ async Task<bool> IStorageProvider.ObjectExistsAsync(string key) => await _sessionStorageService.ContainKeyAsync(key);
+
+ async Task<List<string>> IStorageProvider.GetAllKeysAsync() => (await _sessionStorageService.KeysAsync()).ToList();
+
+ async Task IStorageProvider.DeleteObjectAsync(string key) => await _sessionStorageService.RemoveItemAsync(key);
+}
diff --git a/BugMine.Web/Classes/UserAuth.cs b/BugMine.Web/Classes/UserAuth.cs
new file mode 100644
index 0000000..3085df7
--- /dev/null
+++ b/BugMine.Web/Classes/UserAuth.cs
@@ -0,0 +1,15 @@
+using LibMatrix.Responses;
+
+namespace BugMine.Web.Classes;
+
+public class UserAuth : LoginResponse {
+ public UserAuth() { }
+ public UserAuth(LoginResponse login) {
+ Homeserver = login.Homeserver;
+ UserId = login.UserId;
+ AccessToken = login.AccessToken;
+ DeviceId = login.DeviceId;
+ }
+
+ public string? Proxy { get; set; }
+}
|
