PluralKit/PluralKit.Bot/Commands/SystemLink.cs

using Myriad.Extensions;

using PluralKit.Core;

namespace PluralKit.Bot;

public class SystemLink
{
    private readonly ModelRepository _repo;

    public SystemLink(ModelRepository repo)
    {
        _repo = repo;
    }

    public async Task LinkSystem(Context ctx)
    {
        ctx.CheckSystem();

        var account = await ctx.MatchUser() ??
                      throw new PKSyntaxError("You must pass an account to link with (either ID or @mention).");
        var accountIds = await _repo.GetSystemAccounts(ctx.System.Id);
        if (accountIds.Contains(account.Id))
            throw Errors.AccountAlreadyLinked;

        var existingAccount = await _repo.GetSystemByAccount(account.Id);
        if (existingAccount != null)
            throw Errors.AccountInOtherSystem(existingAccount);

        var msg = $"{account.Mention()}, please confirm the link.";
        if (!await ctx.PromptYesNo(msg, "Confirm", account, false)) throw Errors.MemberLinkCancelled;
        await _repo.AddAccount(ctx.System.Id, account.Id);
        await ctx.Reply($"{Emojis.Success} Account linked to system.");
    }

    public async Task UnlinkAccount(Context ctx)
    {
        ctx.CheckSystem();

        ulong id;
        if (!ctx.MatchUserRaw(out id))
            throw new PKSyntaxError("You must pass an account to link with (either ID or @mention).");

        var accountIds = (await _repo.GetSystemAccounts(ctx.System.Id)).ToList();
        if (!accountIds.Contains(id)) throw Errors.AccountNotLinked;
        if (accountIds.Count == 1) throw Errors.UnlinkingLastAccount;

        var msg = $"Are you sure you want to unlink <@{id}> from your system?";
        if (!await ctx.PromptYesNo(msg, "Unlink")) throw Errors.MemberUnlinkCancelled;

        await _repo.RemoveAccount(ctx.System.Id, id);
        await ctx.Reply($"{Emojis.Success} Account unlinked.");
    }
}