Fix forcelink of existing users.

GianniKoch · Mar 8, 2024 · 4bdefc0 · 4bdefc0
1 parent 74dfd9f
commit 4bdefc0
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 3 deletions.
diff --git a/source/POI.DiscordDotNet/Commands/ChatCommands/BeatSaber/BaseLinkCommand.cs b/source/POI.DiscordDotNet/Commands/ChatCommands/BeatSaber/BaseLinkCommand.cs
@@ -118,9 +118,16 @@ public virtual async Task Handle(CommandContext ctx, string _)
 			return scoreSaberId;
 		}
 
-		protected Task CreateScoreLink(ulong discordId, string scoreSaberId)
+		protected async Task CreateScoreLink(ulong discordId, string scoreSaberId)
 		{
-			return GlobalUserSettingsRepository.CreateOrUpdateScoreSaberLink(discordId, scoreSaberId);
+			var existingUserSetting = await GlobalUserSettingsRepository.GetByScoreSaberId(scoreSaberId);
+
+			if(existingUserSetting != null)
+			{
+				await GlobalUserSettingsRepository.DeleteAsync(existingUserSetting);
+			}
+
+			await GlobalUserSettingsRepository.CreateOrUpdateScoreSaberLink(discordId, scoreSaberId);
 
 			// TODO: Role assignment logic (Preferably call into service)
 		}

diff --git a/source/POI.Persistence.EFCore.Npgsql/Repositories/GlobalUserSettingsRepository.cs b/source/POI.Persistence.EFCore.Npgsql/Repositories/GlobalUserSettingsRepository.cs
@@ -97,4 +97,17 @@ public async Task UpdateBirthday(ulong discordUserId, LocalDate? birthday, Cance
 
 		await context.SaveChangesAsync(cts).ConfigureAwait(false);
 	}
+
+	public async Task<GlobalUserSettings?> GetByScoreSaberId(string scoreSaberId, CancellationToken cts)
+	{
+		await using var context = await _appDbContextFactory.CreateDbContextAsync(cts);
+		return await context.GlobalUserSettings.SingleOrDefaultAsync(g => g.ScoreSaberId == scoreSaberId, cts);
+	}
+
+	public async Task DeleteAsync(GlobalUserSettings existingUserSetting, CancellationToken cts = default)
+	{
+		await using var context = await _appDbContextFactory.CreateDbContextAsync(cts);
+		context.GlobalUserSettings.Remove(existingUserSetting);
+		await context.SaveChangesAsync(cts);
+	}
 }
diff --git a/source/POI.Persistence/Repositories/IGlobalUserSettingsRepository.cs b/source/POI.Persistence/Repositories/IGlobalUserSettingsRepository.cs
@@ -10,7 +10,8 @@ public interface IGlobalUserSettingsRepository
 	Task<GlobalUserSettings?> LookupSettingsByScoreSaberId(string scoreSaberId, CancellationToken cts = default);
 	Task<List<ScoreSaberAccountLink>> GetAllScoreSaberAccountLinks(CancellationToken cts = default);
 	Task CreateOrUpdateScoreSaberLink(ulong discordUserId, string scoreSaberId, CancellationToken cts = default);
-
 	Task<List<GlobalUserSettings>> GetAllBirthdayGirls(int dayOfMonth, int month, CancellationToken cts = default);
 	Task UpdateBirthday(ulong discordUserId, LocalDate? birthday, CancellationToken cts = default);
+	Task<GlobalUserSettings?> GetByScoreSaberId(string scoreSaberId, CancellationToken cts = default);
+	Task DeleteAsync(GlobalUserSettings existingUserSetting, CancellationToken cts = default);
 }