rahix/forgejo-api
1
0
Fork 0
Code Activity

generate subtypes and enums

Browse source
This commit is contained in:
Cyborus 2024-02-09 17:39:06 -05:00
parent 4d63cd09c1
commit 945647baf1
No known key found for this signature in database
3 changed files with 194 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

3
generator/src/methods.rs
View file

@ -103,8 +103,7 @@ fn method_docs(spec: &OpenApiV2, op: &Operation) -> eyre::Result<String> {
for param in params {
let param = param.deref(spec)?;
match &param._in {
ParameterIn::Path { param: _ }
| ParameterIn::FormData { param: _ } => {
ParameterIn::Path { param: _ } | ParameterIn::FormData { param: _ } => {
write!(&mut out, "/// - `{}`", param.name)?;
if let Some(description) = &param.description {
write!(&mut out, ": {}", description)?;

47
generator/src/structs.rs
View file

@ -39,6 +39,14 @@ pub fn create_struct_for_definition(
return Ok(String::new());
}
if schema._type == Some(SchemaType::One(Primitive::String)) {
if let Some(_enum) = &schema._enum {
return create_enum(name, schema);
}
}
let mut subtypes = Vec::new();
let docs = create_struct_docs(schema)?;
let mut fields = String::new();
let required = schema.required.as_deref().unwrap_or_default();
@ -84,6 +92,17 @@ pub fn create_struct_for_definition(
fields.push_str(": ");
fields.push_str(&field_ty);
fields.push_str(",\n");
if let MaybeRef::Value { value } = &prop_schema {
if value._type == Some(SchemaType::One(Primitive::Object))
|| (value._type == Some(SchemaType::One(Primitive::String))
&& value._enum.is_some())
{
let name = format!("{name}{}", prop_name.to_pascal_case());
let subtype = create_struct_for_definition(spec, &name, value)?;
subtypes.push(subtype);
}
}
}
}
@ -95,7 +114,33 @@ pub fn create_struct_for_definition(
fields.push_str(">,\n");
}
let out = format!("{docs}#[derive(Debug, Clone, PartialEq, serde::Serialize, serde::Deserialize)]\npub struct {name} {{\n{fields}}}\n\n");
let mut out = format!("{docs}#[derive(Debug, Clone, PartialEq, serde::Serialize, serde::Deserialize)]\npub struct {name} {{\n{fields}}}\n\n");
for subtype in subtypes {
out.push_str(&subtype);
}
Ok(out)
}
fn create_enum(name: &str, schema: &Schema) -> eyre::Result<String> {
let _enum = schema
._enum
.as_deref()
.ok_or_else(|| eyre::eyre!("cannot create enum from non-enum schema"))?;
let mut variants = String::new();
let docs = create_struct_docs(schema)?;
for variant in _enum {
match variant {
serde_json::Value::String(s) => {
let variant_name = s.to_pascal_case();
variants.push_str(&variant_name);
variants.push_str(",\n");
}
x => eyre::bail!("cannot create enum variant. expected string, got {x:?}"),
}
}
let out = format!("{docs}#[derive(Debug, Clone, PartialEq, serde::Serialize, serde::Deserialize)]\npub enum {name} {{\n{variants}}}\n\n");
Ok(out)
}