rahix/forgejo-api
1
0
Fork 0
Code Activity

don't implement as_str for enums that don't need it

Browse source
This commit is contained in:
Cyborus 2024-03-15 14:28:28 -04:00
parent 7938d81544
commit 5431303ecb
No known key found for this signature in database
3 changed files with 18 additions and 138 deletions
Download patch file Download diff file Expand all files Collapse all files

2
generator/src/main.rs
View file

@ -261,7 +261,7 @@ fn schema_subtypes(
..
} => {
let name = format!("{parent_name}{}", name.to_pascal_case());
let subtype = structs::create_enum(&name, schema.description.as_deref(), _enum)?;
let subtype = structs::create_enum(&name, schema.description.as_deref(), _enum, false)?;
subtypes.push(subtype);
}
Schema {

27
generator/src/structs.rs
View file

@ -44,7 +44,7 @@ pub fn create_struct_for_definition(
if schema._type == Some(SchemaType::One(Primitive::String)) {
if let Some(_enum) = &schema._enum {
return create_enum(name, schema.description.as_deref(), _enum);
return create_enum(name, schema.description.as_deref(), _enum, false);
}
}
@ -124,6 +124,7 @@ pub fn create_enum(
name: &str,
desc: Option<&str>,
_enum: &[serde_json::Value],
imp_as_str: bool,
) -> eyre::Result<String> {
let mut variants = String::new();
let mut imp = String::new();
@ -143,20 +144,26 @@ pub fn create_enum(
}
imp.push_str("}");
let out = format!(
let strukt = format!(
"
{docs}
#[derive(Debug, Clone, PartialEq, serde::Serialize, serde::Deserialize)]
pub enum {name} {{
{variants}
}}
impl {name} {{
fn as_str(&self) -> &'static str {{
{imp}
}}
}}"
);
let out = if imp_as_str {
let imp = format!(
"\n\nimpl {name} {{
fn as_str(&self) -> &'static str {{
{imp}
}}
}}"
);
format!("{strukt} {imp}")
} else {
strukt
};
Ok(out)
}
@ -241,7 +248,7 @@ fn create_query_struct(spec: &OpenApiV2, op: &Operation) -> eyre::Result<String>
..
} => {
let name = format!("{op_name}{}", param.name.to_pascal_case());
let enum_def = create_enum(&name, None, _enum)?;
let enum_def = create_enum(&name, None, _enum, true)?;
enums.push(enum_def);
name
}
@ -256,7 +263,7 @@ fn create_query_struct(spec: &OpenApiV2, op: &Operation) -> eyre::Result<String>
..
} => {
let name = format!("{op_name}{}", param.name.to_pascal_case());
let enum_def = create_enum(&name, None, _enum)?;
let enum_def = create_enum(&name, None, _enum, true)?;
enums.push(enum_def);
format!("Vec<{name}>")
}