eightball/PluralKit
1
0
Fork 0
mirror of https://github.com/PluralKit/PluralKit.git synced 2026-02-12 00:30:11 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(bot): fix ffi with flags

Browse source
This commit is contained in:
dusk 2025-01-14 23:30:53 +09:00
parent 50981c1750
commit 6c54551a9e
No known key found for this signature in database
1 changed files with 37 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

44
PluralKit.Bot/CommandSystem/Parameters.cs
View file

@ -19,7 +19,7 @@ public abstract record Parameter()
public class Parameters public class Parameters
{ {
private string _cb { get; init; } private string _cb { get; init; }
private Dictionary<string, string?> _flags { get; init; } private Dictionary<string, uniffi.commands.Parameter?> _flags { get; init; }
private Dictionary<string, uniffi.commands.Parameter> _params { get; init; } private Dictionary<string, uniffi.commands.Parameter> _params { get; init; }
// just used for errors, temporarily // just used for errors, temporarily
@ -52,11 +52,9 @@ public class Parameters
return potentialMatches.Any(_flags.ContainsKey); return potentialMatches.Any(_flags.ContainsKey);
} }
// resolves a single parameter private async Task<Parameter?> ResolveFfiParam(Context ctx, uniffi.commands.Parameter ffi_param)
private async Task<Parameter?> ResolveParameter(Context ctx, string param_name)
{ {
if (!_params.ContainsKey(param_name)) return null; switch (ffi_param)
switch (_params[param_name])
{ {
case uniffi.commands.Parameter.MemberRef memberRef: case uniffi.commands.Parameter.MemberRef memberRef:
var byId = HasFlag("id", "by-id"); var byId = HasFlag("id", "by-id");
@ -85,8 +83,40 @@ public class Parameters
case uniffi.commands.Parameter.Reset _: case uniffi.commands.Parameter.Reset _:
return new Parameter.Reset(); return new Parameter.Reset();
} }
// this should also never happen return null;
throw new PKError($"Unknown parameter type for parameter {param_name}"); }
// resolves a single flag with value
private async Task<Parameter?> ResolveFlag(Context ctx, string flag_name)
{
if (!HasFlag(flag_name)) return null;
var flag_value = _flags[flag_name];
if (flag_value == null) return null;
var resolved = await ResolveFfiParam(ctx, flag_value);
if (resolved != null) return resolved;
// this should never happen, types are handled rust side
return null;
}
// resolves a single parameter
private async Task<Parameter?> ResolveParameter(Context ctx, string param_name)
{
if (!_params.ContainsKey(param_name)) return null;
var resolved = await ResolveFfiParam(ctx, _params[param_name]);
if (resolved != null) return resolved;
// this should never happen, types are handled rust side
return null;
}
public async Task<T?> ResolveFlag<T>(Context ctx, string flag_name, Func<Parameter, T?> extract_func)
{
var param = await ResolveFlag(ctx, flag_name);
// todo: i think this should return null for everything...?
if (param == null) return default;
return extract_func(param)
// this should never really happen (hopefully!), but in case the parameter names dont match up (typos...) between rust <-> c#...
// (it would be very cool to have this statically checked somehow..?)
?? throw new PKError($"Flag {flag_name.AsCode()} was not found or did not have a value defined for command {Callback().AsCode()} -- this is a bug!!");
} }
public async Task<T> ResolveParameter<T>(Context ctx, string param_name, Func<Parameter, T?> extract_func) public async Task<T> ResolveParameter<T>(Context ctx, string param_name, Func<Parameter, T?> extract_func)