mirror of
https://github.com/PluralKit/PluralKit.git
synced 2026-02-16 10:40:12 +00:00
fix(commands): add csharp glue codegen binary, it was gitignored
This commit is contained in:
parent
35f7bdbaf5
commit
f804e7629f
1 changed files with 173 additions and 0 deletions
173
crates/commands/src/bin/write_cs_glue.rs
Normal file
173
crates/commands/src/bin/write_cs_glue.rs
Normal 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
|
||||||
|
}
|
||||||
Loading…
Add table
Add a link
Reference in a new issue