277 lines
15 KiB
C#
277 lines
15 KiB
C#
using Discord.Interactions;
|
|
using Discord;
|
|
using MareSynchronosShared.Data;
|
|
using MareSynchronosShared.Utils;
|
|
using MareSynchronosShared.Models;
|
|
using Microsoft.EntityFrameworkCore;
|
|
using System.Security.Policy;
|
|
|
|
namespace MareSynchronosServices.Discord;
|
|
|
|
public partial class MareWizardModule
|
|
{
|
|
[ComponentInteraction("wizard-relink")]
|
|
public async Task ComponentRelink()
|
|
{
|
|
if (!(await ValidateInteraction().ConfigureAwait(false))) return;
|
|
|
|
_logger.LogInformation("{method}:{userId}", nameof(ComponentRelink), Context.Interaction.User.Id);
|
|
|
|
EmbedBuilder eb = new();
|
|
eb.WithTitle("Relink");
|
|
eb.WithColor(Color.Blue);
|
|
eb.WithDescription("Use this in case you already have a registered Mare account, but lost access to your previous Discord account." + Environment.NewLine + Environment.NewLine
|
|
+ "- Have your original registered Lodestone URL ready (i.e. https://eu.finalfantasyxiv.com/lodestone/character/XXXXXXXXX)" + Environment.NewLine
|
|
+ " - The relink process requires you to modify your Lodestone profile with a generated code for verification" + Environment.NewLine
|
|
+ "- Do not use this on mobile because you will need to be able to copy the generated secret key");
|
|
ComponentBuilder cb = new();
|
|
AddHome(cb);
|
|
cb.WithButton("Start Relink", "wizard-relink-start", ButtonStyle.Primary, emote: new Emoji("🔗"));
|
|
await ModifyInteraction(eb, cb).ConfigureAwait(false);
|
|
}
|
|
|
|
[ComponentInteraction("wizard-relink-start")]
|
|
public async Task ComponentRelinkStart()
|
|
{
|
|
if (!(await ValidateInteraction().ConfigureAwait(false))) return;
|
|
|
|
_logger.LogInformation("{method}:{userId}", nameof(ComponentRelinkStart), Context.Interaction.User.Id);
|
|
|
|
using var db = GetDbContext();
|
|
db.LodeStoneAuth.RemoveRange(db.LodeStoneAuth.Where(u => u.DiscordId == Context.User.Id));
|
|
_botServices.DiscordVerifiedUsers.TryRemove(Context.User.Id, out _);
|
|
_botServices.DiscordRelinkLodestoneMapping.TryRemove(Context.User.Id, out _);
|
|
await db.SaveChangesAsync().ConfigureAwait(false);
|
|
|
|
await RespondWithModalAsync<LodestoneModal>("wizard-relink-lodestone-modal").ConfigureAwait(false);
|
|
}
|
|
|
|
[ModalInteraction("wizard-relink-lodestone-modal")]
|
|
public async Task ModalRelink(LodestoneModal lodestoneModal)
|
|
{
|
|
if (!(await ValidateInteraction().ConfigureAwait(false))) return;
|
|
|
|
_logger.LogInformation("{method}:{userId}:{url}", nameof(ModalRelink), Context.Interaction.User.Id, lodestoneModal.LodestoneUrl);
|
|
|
|
EmbedBuilder eb = new();
|
|
eb.WithColor(Color.Purple);
|
|
var result = await HandleRelinkModalAsync(eb, lodestoneModal).ConfigureAwait(false);
|
|
ComponentBuilder cb = new();
|
|
cb.WithButton("Cancel", "wizard-relink", ButtonStyle.Secondary, emote: new Emoji("❌"));
|
|
if (result.Success) cb.WithButton("Verify", "wizard-relink-verify:" + result.LodestoneAuth + "," + result.UID, ButtonStyle.Primary, emote: new Emoji("✅"));
|
|
else cb.WithButton("Try again", "wizard-relink-start", ButtonStyle.Primary, emote: new Emoji("🔁"));
|
|
await ModifyModalInteraction(eb, cb).ConfigureAwait(false);
|
|
}
|
|
|
|
[ComponentInteraction("wizard-relink-verify:*,*")]
|
|
public async Task ComponentRelinkVerify(string verificationCode, string uid)
|
|
{
|
|
if (!(await ValidateInteraction().ConfigureAwait(false))) return;
|
|
|
|
_logger.LogInformation("{method}:{userId}:{uid}:{verificationCode}", nameof(ComponentRelinkVerify), Context.Interaction.User.Id, uid, verificationCode);
|
|
|
|
|
|
_botServices.VerificationQueue.Enqueue(new KeyValuePair<ulong, Func<DiscordBotServices, Task>>(Context.User.Id,
|
|
(services) => HandleVerifyRelinkAsync(Context.User.Id, verificationCode, services)));
|
|
EmbedBuilder eb = new();
|
|
ComponentBuilder cb = new();
|
|
eb.WithColor(Color.Purple);
|
|
cb.WithButton("Cancel", "wizard-relink", ButtonStyle.Secondary, emote: new Emoji("❌"));
|
|
cb.WithButton("Check", "wizard-relink-verify-check:" + verificationCode + "," + uid, ButtonStyle.Primary, emote: new Emoji("❓"));
|
|
eb.WithTitle("Relink Verification Pending");
|
|
eb.WithDescription("Please wait until the bot verifies your registration." + Environment.NewLine
|
|
+ "Press \"Check\" to check if the verification has been already processed" + Environment.NewLine + Environment.NewLine
|
|
+ "__This will not advance automatically, you need to press \"Check\".__");
|
|
await ModifyInteraction(eb, cb).ConfigureAwait(false);
|
|
}
|
|
|
|
[ComponentInteraction("wizard-relink-verify-check:*,*")]
|
|
public async Task ComponentRelinkVerifyCheck(string verificationCode, string uid)
|
|
{
|
|
if (!(await ValidateInteraction().ConfigureAwait(false))) return;
|
|
|
|
_logger.LogInformation("{method}:{userId}:{uid}:{verificationCode}", nameof(ComponentRelinkVerifyCheck), Context.Interaction.User.Id, uid, verificationCode);
|
|
|
|
EmbedBuilder eb = new();
|
|
ComponentBuilder cb = new();
|
|
bool stillEnqueued = _botServices.VerificationQueue.Any(k => k.Key == Context.User.Id);
|
|
bool verificationRan = _botServices.DiscordVerifiedUsers.TryGetValue(Context.User.Id, out bool verified);
|
|
if (!verificationRan)
|
|
{
|
|
if (stillEnqueued)
|
|
{
|
|
eb.WithColor(Color.Gold);
|
|
eb.WithTitle("Your relink verification is still pending");
|
|
eb.WithDescription("Please try again and click Check in a few seconds");
|
|
cb.WithButton("Cancel", "wizard-relink", ButtonStyle.Secondary, emote: new Emoji("❌"));
|
|
cb.WithButton("Check", "wizard-relink-verify-check:" + verificationCode + "," + uid, ButtonStyle.Primary, emote: new Emoji("❓"));
|
|
}
|
|
else
|
|
{
|
|
eb.WithColor(Color.Red);
|
|
eb.WithTitle("Something went wrong");
|
|
eb.WithDescription("Your relink verification was processed but did not arrive properly. Please try to start the relink process from the start.");
|
|
cb.WithButton("Restart", "wizard-relink", ButtonStyle.Primary, emote: new Emoji("🔁"));
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if (verified)
|
|
{
|
|
eb.WithColor(Color.Green);
|
|
using var db = _services.CreateScope().ServiceProvider.GetRequiredService<MareDbContext>();
|
|
var (_, key) = await HandleRelinkUser(db, uid).ConfigureAwait(false);
|
|
eb.WithTitle($"Relink successful, your UID is again: {uid}");
|
|
eb.WithDescription("This is your private secret key. Do not share this private secret key with anyone. **If you lose it, it is irrevocably lost.**"
|
|
+ Environment.NewLine + Environment.NewLine
|
|
+ $"**{key}**"
|
|
+ Environment.NewLine + Environment.NewLine
|
|
+ "Enter this key in Mare Synchronos and hit save to connect to the service."
|
|
+ Environment.NewLine
|
|
+ "Have fun.");
|
|
AddHome(cb);
|
|
}
|
|
else
|
|
{
|
|
eb.WithColor(Color.Gold);
|
|
eb.WithTitle("Failed to verify relink");
|
|
eb.WithDescription("The bot was not able to find the required verification code on your Lodestone profile." + Environment.NewLine + Environment.NewLine
|
|
+ "Please restart your relink process, make sure to save your profile _twice_ for it to be properly saved." + Environment.NewLine + Environment.NewLine
|
|
+ "The code the bot is looking for is**" + Environment.NewLine + Environment.NewLine
|
|
+ "**" + verificationCode + "**");
|
|
cb.WithButton("Cancel", "wizard-relink", emote: new Emoji("❌"));
|
|
cb.WithButton("Retry", "wizard-relink-verify:" + verificationCode + "," + uid, ButtonStyle.Primary, emote: new Emoji("🔁"));
|
|
}
|
|
}
|
|
|
|
await ModifyInteraction(eb, cb).ConfigureAwait(false);
|
|
}
|
|
|
|
private async Task<(bool Success, string LodestoneAuth, string UID)> HandleRelinkModalAsync(EmbedBuilder embed, LodestoneModal arg)
|
|
{
|
|
ulong userId = Context.User.Id;
|
|
|
|
var lodestoneId = ParseCharacterIdFromLodestoneUrl(arg.LodestoneUrl);
|
|
if (lodestoneId == null)
|
|
{
|
|
embed.WithTitle("Invalid Lodestone URL");
|
|
embed.WithDescription("The lodestone URL was not valid. It should have following format:" + Environment.NewLine
|
|
+ "https://eu.finalfantasyxiv.com/lodestone/character/YOUR_LODESTONE_ID/");
|
|
return (false, string.Empty, string.Empty);
|
|
}
|
|
// check if userid is already in db
|
|
using var scope = _services.CreateScope();
|
|
|
|
var hashedLodestoneId = StringUtils.Sha256String(lodestoneId.ToString());
|
|
|
|
using var db = scope.ServiceProvider.GetService<MareDbContext>();
|
|
|
|
// check if discord id or lodestone id is banned
|
|
if (db.BannedRegistrations.Any(a => a.DiscordIdOrLodestoneAuth == hashedLodestoneId))
|
|
{
|
|
embed.WithTitle("Illegal operation");
|
|
embed.WithDescription("Your account is banned");
|
|
return (false, string.Empty, string.Empty);
|
|
}
|
|
|
|
if (!db.LodeStoneAuth.Any(a => a.HashedLodestoneId == hashedLodestoneId))
|
|
{
|
|
// character already in db
|
|
embed.WithTitle("Impossible operation");
|
|
embed.WithDescription("This lodestone character does not exist in the database.");
|
|
return (false, string.Empty, string.Empty);
|
|
}
|
|
|
|
var expectedUser = await db.LodeStoneAuth.Include(u => u.User).SingleAsync(u => u.HashedLodestoneId == hashedLodestoneId).ConfigureAwait(false);
|
|
|
|
string lodestoneAuth = await GenerateLodestoneAuth(Context.User.Id, hashedLodestoneId, db).ConfigureAwait(false);
|
|
// check if lodestone id is already in db
|
|
embed.WithTitle("Authorize your character for relinking");
|
|
embed.WithDescription("Add following key to your character profile at https://na.finalfantasyxiv.com/lodestone/my/setting/profile/"
|
|
+ Environment.NewLine + Environment.NewLine
|
|
+ $"**{lodestoneAuth}**"
|
|
+ Environment.NewLine + Environment.NewLine
|
|
+ $"**! THIS IS NOT THE KEY YOU HAVE TO ENTER IN MARE !**"
|
|
+ Environment.NewLine
|
|
+ "__You can delete the entry from your profile after verification.__"
|
|
+ Environment.NewLine + Environment.NewLine
|
|
+ "The verification will expire in approximately 15 minutes. If you fail to verify the relink will be invalidated and you have to relink again.");
|
|
_botServices.DiscordRelinkLodestoneMapping[Context.User.Id] = lodestoneId.ToString();
|
|
|
|
return (true, lodestoneAuth, expectedUser.User.UID);
|
|
}
|
|
|
|
private async Task HandleVerifyRelinkAsync(ulong userid, string authString, DiscordBotServices services)
|
|
{
|
|
using var req = new HttpClient();
|
|
|
|
services.DiscordVerifiedUsers.Remove(userid, out _);
|
|
if (services.DiscordRelinkLodestoneMapping.ContainsKey(userid))
|
|
{
|
|
var randomServer = services.LodestoneServers[random.Next(services.LodestoneServers.Length)];
|
|
var url = $"https://{randomServer}.finalfantasyxiv.com/lodestone/character/{services.DiscordRelinkLodestoneMapping[userid]}";
|
|
_logger.LogInformation("Verifying {userid} with URL {url}", userid, url);
|
|
using var response = await req.GetAsync(url).ConfigureAwait(false);
|
|
if (response.IsSuccessStatusCode || response.StatusCode == System.Net.HttpStatusCode.Forbidden)
|
|
{
|
|
var content = await response.Content.ReadAsStringAsync().ConfigureAwait(false);
|
|
if (content.Contains(authString))
|
|
{
|
|
services.DiscordVerifiedUsers[userid] = true;
|
|
_logger.LogInformation("Relink: Verified {userid} from lodestone {lodestone}", userid, services.DiscordRelinkLodestoneMapping[userid]);
|
|
await _botServices.LogToChannel($"<@{userid}> RELINK VERIFY: Success. ({url})").ConfigureAwait(false);
|
|
services.DiscordRelinkLodestoneMapping.TryRemove(userid, out _);
|
|
}
|
|
else
|
|
{
|
|
services.DiscordVerifiedUsers[userid] = false;
|
|
_logger.LogInformation("Relink: Could not verify {userid} from lodestone {lodestone}, did not find authString: {authString}, status code was: {code}",
|
|
userid, services.DiscordRelinkLodestoneMapping[userid], authString, response.StatusCode);
|
|
await _botServices.LogToChannel($"<@{userid}> RELINK VERIFY: Failed: No Authstring. ({url})").ConfigureAwait(false);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
_logger.LogWarning("Could not verify {userid}, HttpStatusCode: {code}", userid, response.StatusCode);
|
|
await _botServices.LogToChannel($"<@{userid}> RELINK VERIFY: Failed: HttpStatusCode {response.StatusCode}. ({url})").ConfigureAwait(false);
|
|
}
|
|
}
|
|
}
|
|
|
|
private async Task<(string, string)> HandleRelinkUser(MareDbContext db, string uid)
|
|
{
|
|
var oldLodestoneAuth = await db.LodeStoneAuth.Include(u => u.User).SingleOrDefaultAsync(u => u.User.UID == uid && u.DiscordId != Context.User.Id).ConfigureAwait(false);
|
|
var newLodestoneAuth = await db.LodeStoneAuth.Include(u => u.User).SingleOrDefaultAsync(u => u.DiscordId == Context.User.Id).ConfigureAwait(false);
|
|
|
|
var user = oldLodestoneAuth.User;
|
|
|
|
var computedHash = StringUtils.Sha256String(StringUtils.GenerateRandomString(64) + DateTime.UtcNow.ToString());
|
|
var auth = new Auth()
|
|
{
|
|
HashedKey = StringUtils.Sha256String(computedHash),
|
|
User = user,
|
|
};
|
|
|
|
var previousAuth = await db.Auth.SingleOrDefaultAsync(u => u.UserUID == user.UID).ConfigureAwait(false);
|
|
if (previousAuth != null)
|
|
{
|
|
db.Remove(previousAuth);
|
|
}
|
|
|
|
newLodestoneAuth.LodestoneAuthString = null;
|
|
newLodestoneAuth.StartedAt = null;
|
|
newLodestoneAuth.User = user;
|
|
db.Update(newLodestoneAuth);
|
|
db.Remove(oldLodestoneAuth);
|
|
await db.Auth.AddAsync(auth).ConfigureAwait(false);
|
|
|
|
_botServices.Logger.LogInformation("User relinked: {userUID}", user.UID);
|
|
|
|
await db.SaveChangesAsync().ConfigureAwait(false);
|
|
|
|
await _botServices.LogToChannel($"{Context.User.Mention} RELINK COMPLETE: => {user.UID}").ConfigureAwait(false);
|
|
|
|
return (user.UID, computedHash);
|
|
}
|
|
}
|