rahix/forgejo-api
1
0
Fork 0
Code Activity

add derives for query structs

Browse source
This commit is contained in:
Cyborus 2024-04-20 16:16:32 -04:00
parent a4c081a770
commit f029ba5737
No known key found for this signature in database
2 changed files with 114 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

10
generator/src/structs.rs
View file

@ -241,6 +241,8 @@ fn create_query_struct(spec: &OpenApiV2, op: &Operation) -> eyre::Result<String>
let mut enums = Vec::new();
let mut fields = String::new();
let mut imp = String::new();
// only derive default if every field is optional
let mut can_derive_default = true;
for param in params {
let param = param.deref(spec)?;
if let ParameterIn::Query { param: query_param } = &param._in {
@ -287,6 +289,7 @@ fn create_query_struct(spec: &OpenApiV2, op: &Operation) -> eyre::Result<String>
fields.push_str(&field_name);
fields.push_str(": ");
if query_param.required {
can_derive_default = false;
fields.push_str(&ty);
} else {
fields.push_str("Option<");
@ -425,11 +428,18 @@ fn create_query_struct(spec: &OpenApiV2, op: &Operation) -> eyre::Result<String>
}
}
let derives = if can_derive_default {
"Debug, Clone, PartialEq, Default"
} else {
"Debug, Clone, PartialEq"
};
let result = if fields.is_empty() {
String::new()
} else {
let mut out = format!(
"
#[derive({derives})]
pub struct {op_name} {{
{fields}
}}