fix(commands): add csharp glue codegen binary, it was gitignored

This commit is contained in:
dusk 2025-01-24 01:57:13 +09:00
parent 35f7bdbaf5
commit f804e7629f
No known key found for this signature in database

View file

@ -0,0 +1,173 @@
use std::{env, fmt::Write, fs, path::PathBuf, str::FromStr};
use command_parser::{
parameter::{Parameter, ParameterKind},
token::Token,
};
fn main() -> Result<(), Box<dyn std::error::Error>> {
let write_location = env::args()
.nth(1)
.expect("file location should be provided");
let write_location = PathBuf::from_str(&write_location).unwrap();
let commands = command_definitions::all().collect::<Vec<_>>();
let mut glue = String::new();
writeln!(&mut glue, "#nullable enable\n")?;
writeln!(&mut glue, "using PluralKit.Core;\n")?;
writeln!(&mut glue, "namespace PluralKit.Bot;\n")?;
let mut record_fields = String::new();
for command in &commands {
writeln!(
&mut record_fields,
r#"public record {command_name}({command_name}Params parameters, {command_name}Flags flags): Commands;"#,
command_name = command_callback_to_name(&command.cb),
)?;
}
let mut match_branches = String::new();
for command in &commands {
let mut command_params_init = String::new();
let command_params = find_parameters(&command.tokens);
for param in &command_params {
writeln!(
&mut command_params_init,
r#"{name} = await ctx.ParamResolve{extract_fn_name}("{name}") ?? throw new PKError("this is a bug"),"#,
name = param.name(),
extract_fn_name = get_param_param_ty(param.kind()),
)?;
}
let mut command_flags_init = String::new();
for flag in &command.flags {
if let Some(kind) = flag.get_value() {
writeln!(
&mut command_flags_init,
r#"{name} = await ctx.FlagResolve{extract_fn_name}("{name}"),"#,
name = flag.get_name(),
extract_fn_name = get_param_param_ty(kind),
)?;
} else {
writeln!(
&mut command_flags_init,
r#"{name} = ctx.Parameters.HasFlag("{name}"),"#,
name = flag.get_name(),
)?;
}
}
write!(
&mut match_branches,
r#"
"{command_callback}" => new {command_name}(
new {command_name}Params {{ {command_params_init} }},
new {command_name}Flags {{ {command_flags_init} }}
),
"#,
command_name = command_callback_to_name(&command.cb),
command_callback = command.cb,
)?;
}
write!(
&mut glue,
r#"
public abstract record Commands()
{{
{record_fields}
public static async Task<Commands?> FromContext(Context ctx)
{{
return ctx.Parameters.Callback() switch
{{
{match_branches}
_ => null,
}};
}}
}}
"#,
)?;
for command in &commands {
let mut command_params_fields = String::new();
let command_params = find_parameters(&command.tokens);
for param in &command_params {
writeln!(
&mut command_params_fields,
r#"public required {ty} {name};"#,
name = param.name(),
ty = get_param_ty(param.kind()),
)?;
}
let mut command_flags_fields = String::new();
for flag in &command.flags {
if let Some(kind) = flag.get_value() {
writeln!(
&mut command_flags_fields,
r#"public {ty}? {name};"#,
name = flag.get_name(),
ty = get_param_ty(kind),
)?;
} else {
writeln!(
&mut command_flags_fields,
r#"public required bool {name};"#,
name = flag.get_name(),
)?;
}
}
write!(
&mut glue,
r#"
public class {command_name}Params
{{
{command_params_fields}
}}
public class {command_name}Flags
{{
{command_flags_fields}
}}
"#,
command_name = command_callback_to_name(&command.cb),
)?;
}
fs::write(write_location, glue)?;
Ok(())
}
fn command_callback_to_name(cb: &str) -> String {
cb.split("_")
.map(|w| w.chars().nth(0).unwrap().to_uppercase().collect::<String>() + &w[1..])
.collect()
}
fn get_param_ty(kind: ParameterKind) -> &'static str {
match kind {
ParameterKind::OpaqueString | ParameterKind::OpaqueStringRemainder => "string",
ParameterKind::MemberRef => "PKMember",
ParameterKind::SystemRef => "PKSystem",
ParameterKind::MemberPrivacyTarget => "MemberPrivacySubject",
ParameterKind::PrivacyLevel => "string",
ParameterKind::Toggle => "bool",
}
}
fn get_param_param_ty(kind: ParameterKind) -> &'static str {
match kind {
ParameterKind::OpaqueString | ParameterKind::OpaqueStringRemainder => "Opaque",
ParameterKind::MemberRef => "Member",
ParameterKind::SystemRef => "System",
ParameterKind::MemberPrivacyTarget => "MemberPrivacyTarget",
ParameterKind::PrivacyLevel => "PrivacyLevel",
ParameterKind::Toggle => "Toggle",
}
}
fn find_parameters(tokens: &[Token]) -> Vec<&Parameter> {
let mut result = Vec::new();
for token in tokens {
match token {
Token::Parameter(param) => result.push(param),
_ => {}
}
}
result
}