1
0
Fork 0

feat(definitions): impl Display

Instead of adding a `to_string` method on each definition struct, this
commit implements `Display`. This shouldn't result in any breaking
changes since the trait also exposes a method with the same name.

I also tried to format the generated text here and there to make it more
readable in the generate code base as well as in the generate text.

Authored-by: Aviac <aviac@mailbox.org>
This commit is contained in:
aviac 2024-01-19 11:26:14 +01:00
parent 331bb80acf
commit 1b6cd3467a
No known key found for this signature in database
GPG key ID: 644781002BDEA982

View file

@ -925,20 +925,23 @@ fn create_query_struct(spec: &OpenApiV2, path: &str, op: &Operation) -> eyre::Re
match ty { match ty {
ParameterType::String => match param.format.as_deref() { ParameterType::String => match param.format.as_deref() {
Some("date-time" | "date") => { Some("date-time" | "date") => {
writeln!(&mut handler, "s.push_str(\"{}=\");", param.name)?; writeln!(
writeln!(&mut handler, "s.push_str(&{field_name}.format(&time::format_description::well_known::Rfc3339).unwrap());")?; &mut handler,
writeln!(&mut handler, "s.push('&');")?; "write!(f, \"{}={{field_name}}&\", field_name = {field_name}.format(&time::format_description::well_known::Rfc3339).unwrap())?;",
param.name)?;
} }
_ => { _ => {
writeln!(&mut handler, "s.push_str(\"{}=\");", param.name)?; writeln!(
writeln!(&mut handler, "s.push_str(&{field_name});")?; &mut handler,
writeln!(&mut handler, "s.push('&');")?; "write!(f, \"{}={{{}}}&\")?;",
param.name, field_name
)?;
} }
}, },
ParameterType::Number | ParameterType::Integer | ParameterType::Boolean => { ParameterType::Number | ParameterType::Integer | ParameterType::Boolean => {
writeln!( writeln!(
&mut handler, &mut handler,
"write!(&mut s, \"{}={{}}&\", {field_name}).unwrap();", "write!(f, \"{}={{{field_name}}}&\")?;",
param.name param.name
)?; )?;
} }
@ -952,17 +955,17 @@ fn create_query_struct(spec: &OpenApiV2, path: &str, op: &Operation) -> eyre::Re
ParameterType::String => { ParameterType::String => {
match param.format.as_deref() { match param.format.as_deref() {
Some("date-time" | "date") => { Some("date-time" | "date") => {
"s.push_str(&item.format(&time::format_description::well_known::Rfc3339).unwrap());" "write!(f, \"{{date}}\", item.format(&time::format_description::well_known::Rfc3339).unwrap())?;"
}, },
_ => { _ => {
"s.push_str(&item);" "write!(f, \"{item}\")?;"
} }
} }
}, },
ParameterType::Number | ParameterType::Number |
ParameterType::Integer | ParameterType::Integer |
ParameterType::Boolean => { ParameterType::Boolean => {
"write!(&mut s, \"{item}\").unwrap();" "write!(f, \"{item}\")?;"
}, },
ParameterType::Array => { ParameterType::Array => {
eyre::bail!("nested arrays not supported in query"); eyre::bail!("nested arrays not supported in query");
@ -1003,13 +1006,13 @@ fn create_query_struct(spec: &OpenApiV2, path: &str, op: &Operation) -> eyre::Re
)?); )?);
} }
CollectionFormat::Multi => { CollectionFormat::Multi => {
writeln!(&mut handler, "")?; writeln!(&mut handler)?;
writeln!(&mut handler, "if !{field_name}.is_empty() {{")?; writeln!(&mut handler, "if !{field_name}.is_empty() {{")?;
writeln!(&mut handler, "for item in {field_name} {{")?; writeln!(&mut handler, "for item in {field_name} {{")?;
writeln!(&mut handler, "s.push_str(\"{}=\");", param.name)?; writeln!(&mut handler, "write!(f, \"{}=\")?;", param.name)?;
handler.push_str(item_pusher); handler.push_str(item_pusher);
handler.push('\n'); handler.push('\n');
writeln!(&mut handler, "s.push('&')")?; writeln!(&mut handler, "write!(f, '&')?;")?;
writeln!(&mut handler, "}}")?; writeln!(&mut handler, "}}")?;
writeln!(&mut handler, "}}")?; writeln!(&mut handler, "}}")?;
} }
@ -1023,16 +1026,36 @@ fn create_query_struct(spec: &OpenApiV2, path: &str, op: &Operation) -> eyre::Re
imp.push_str(&handler); imp.push_str(&handler);
} }
} }
if fields.is_empty() {
return Ok(String::new()); let result = if fields.is_empty() {
String::new()
} else { } else {
let op_name = op let op_name = op
.operation_id .operation_id
.as_ref() .as_ref()
.ok_or_else(|| eyre::eyre!("no op id found"))? .ok_or_else(|| eyre::eyre!("no op id found"))?
.to_pascal_case(); .to_pascal_case();
return Ok(format!("pub struct {op_name}Query {{\n{fields}\n}}\n\nimpl {op_name}Query {{\npub(crate) fn to_string(self) -> String {{\n{imp}\n}}\n}}")); // human readable format string + added whitespace so that the generate text is human
} // readable as well
format!(
"
pub struct {op_name}Query {{
{fields}
}}
impl std::fmt::Display for {op_name}Query {{
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {{
{imp}
Ok(())
}}
}}
",
fields = fields.replace('\n', "\n ").trim(),
imp = imp.replace('\n', "\n ").trim()
)
};
Ok(result)
} }
fn simple_query_array( fn simple_query_array(
@ -1042,17 +1065,19 @@ fn simple_query_array(
sep: &str, sep: &str,
) -> eyre::Result<String> { ) -> eyre::Result<String> {
let mut out = String::new(); let mut out = String::new();
writeln!(&mut out, "s.push_str(\"{}=\");", param.name)?;
writeln!(&mut out, "")?; writeln!(&mut out)?;
writeln!(&mut out, "if !{name}.is_empty() {{")?; writeln!(&mut out, "if !{name}.is_empty() {{")?;
writeln!(&mut out, "for (i, item) in {name}.iter().enumerate() {{")?; writeln!(&mut out, "for item in {name} {{")?;
writeln!(&mut out, "write!(f, \"{}=\")?;", param.name)?;
out.push_str(item_pusher); out.push_str(item_pusher);
out.push('\n'); out.push('\n');
writeln!(&mut out, "if i < {name}.len() - 1 {{")?; writeln!(&mut out, "if i < {name}.len() - 1 {{")?;
writeln!(&mut out, "s.push('{sep}')")?; writeln!(&mut out, "s.push('{sep}')")?;
writeln!(&mut out, "}}")?; writeln!(&mut out, "}}")?;
writeln!(&mut out, "}}")?; writeln!(&mut out, "}}")?;
writeln!(&mut out, "s.push('&')")?; writeln!(&mut out, "write!(f, '&')?;")?;
writeln!(&mut out, "}}")?; writeln!(&mut out, "}}")?;
Ok(out) Ok(out)
} }