PluralKit/PluralKit.Bot/CommandSystem/Context/ContextParametersExt.cs

using PluralKit.Core;

namespace PluralKit.Bot;

public static class ContextParametersExt
{
    public static async Task<string?> ParamResolveOpaque(this Context ctx, string param_name)
    {
        return await ctx.Parameters.ResolveParameter(
            ctx, param_name,
            param => (param as Parameter.Opaque)?.value
        );
    }

    public static async Task<PKMember?> ParamResolveMember(this Context ctx, string param_name)
    {
        return await ctx.Parameters.ResolveParameter(
            ctx, param_name,
            param => (param as Parameter.MemberRef)?.member
        );
    }

    public static async Task<PKSystem?> ParamResolveSystem(this Context ctx, string param_name)
    {
        return await ctx.Parameters.ResolveParameter(
            ctx, param_name,
            param => (param as Parameter.SystemRef)?.system
        );
    }

    public static async Task<MemberPrivacySubject?> ParamResolveMemberPrivacyTarget(this Context ctx, string param_name)
    {
        return await ctx.Parameters.ResolveParameter(
            ctx, param_name,
            param => (param as Parameter.MemberPrivacyTarget)?.target
        );
    }

    public static async Task<string?> ParamResolvePrivacyLevel(this Context ctx, string param_name)
    {
        return await ctx.Parameters.ResolveParameter(
            ctx, param_name,
            param => (param as Parameter.PrivacyLevel)?.level
        );
    }

    public static async Task<bool?> ParamResolveToggle(this Context ctx, string param_name)
    {
        return await ctx.Parameters.ResolveParameter(
            ctx, param_name,
            param => (param as Parameter.Toggle)?.value
        );
    }

    // this can never really be false (either it's present and is true or it's not present)
    // but we keep it nullable for consistency with the other methods
    public static async Task<bool?> ParamResolveReset(this Context ctx, string param_name)
    {
        return await ctx.Parameters.ResolveParameter<bool?>(
            ctx, param_name,
            param => param is Parameter.Reset
        );
    }
}