PluralKit/PluralKit.Bot/Commands/Privacy/ContextPrivacyExt.cs

using PluralKit.Core;

namespace PluralKit.Bot;

public static class ContextPrivacyExt
{
    public static PrivacyLevel PopPrivacyLevel(this Context ctx)
    {
        if (ctx.Match("public", "show", "shown", "visible"))
            return PrivacyLevel.Public;

        if (ctx.Match("private", "hide", "hidden"))
            return PrivacyLevel.Private;

        if (!ctx.HasNext())
            throw new PKSyntaxError("You must pass a privacy level (`public` or `private`)");

        throw new PKSyntaxError(
            $"Invalid privacy level {ctx.PopArgument().AsCode()} (must be `public` or `private`).");
    }

    public static SystemPrivacySubject PopSystemPrivacySubject(this Context ctx)
    {
        if (!SystemPrivacyUtils.TryParseSystemPrivacy(ctx.PeekArgument(), out var subject))
            throw new PKSyntaxError(
                $"Invalid privacy subject {ctx.PopArgument().AsCode()} (must be `description`, `members`, `front`, `fronthistory`, `groups`, or `all`).");

        ctx.PopArgument();
        return subject;
    }

    public static MemberPrivacySubject PopMemberPrivacySubject(this Context ctx)
    {
        if (!MemberPrivacyUtils.TryParseMemberPrivacy(ctx.PeekArgument(), out var subject))
            throw new PKSyntaxError(
                $"Invalid privacy subject {ctx.PopArgument().AsCode()} (must be `name`, `description`, `avatar`, `birthday`, `pronouns`, `metadata`, `visibility`, or `all`).");

        ctx.PopArgument();
        return subject;
    }

    public static GroupPrivacySubject PopGroupPrivacySubject(this Context ctx)
    {
        if (!GroupPrivacyUtils.TryParseGroupPrivacy(ctx.PeekArgument(), out var subject))
            throw new PKSyntaxError(
                $"Invalid privacy subject {ctx.PopArgument().AsCode()} (must be `description`, `icon`, `visibility`, or `all`).");

        ctx.PopArgument();
        return subject;
    }

    public static bool MatchPrivateFlag(this Context ctx, LookupContext pctx)
    {
        var privacy = true;
        if (ctx.MatchFlag("a", "all")) privacy = false;
        if (pctx == LookupContext.ByNonOwner && !privacy) throw Errors.LookupNotAllowed;

        return privacy;
    }
}