Skip to content

Commit 00b610a

Browse files
drahnrdrahnr
committed
refactor; somewhat cleaner approach
1 parent af28d71 commit 00b610a

File tree

2 files changed

+79
-53
lines changed

2 files changed

+79
-53
lines changed

‎progenitor-impl/src/lib.rs‎

Lines changed: 53 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@ use proc_macro2::TokenStream;
77
use quote::quote;
88
use serde::Deserialize;
99
use thiserror::Error;
10+
use typify::{TypeId, TypeDetails};
1011
use typify::{TypeSpace, TypeSpaceSettings};
1112

1213
use crate::to_schema::ToSchema;
@@ -40,6 +41,7 @@ pub type Result<T> = std::result::Result<T, Error>;
4041

4142
pub struct Generator {
4243
type_space: TypeSpace,
44+
forms: HashSet<TypeId>,
4345
settings: GenerationSettings,
4446
uses_futures: bool,
4547
uses_websockets: bool,
@@ -163,6 +165,7 @@ impl Default for Generator {
163165
type_space: TypeSpace::new(
164166
TypeSpaceSettings::default().with_type_mod("types"),
165167
),
168+
forms: Default::default(),
166169
settings: Default::default(),
167170
uses_futures: Default::default(),
168171
uses_websockets: Default::default(),
@@ -204,6 +207,7 @@ impl Generator {
204207
Self {
205208
type_space: TypeSpace::new(&type_settings),
206209
settings: settings.clone(),
210+
forms: Default::default(),
207211
uses_futures: false,
208212
uses_websockets: false,
209213
}
@@ -262,6 +266,44 @@ impl Generator {
262266

263267
let types = self.type_space.to_stream();
264268

269+
let extra_impl = TokenStream::from_iter(
270+
self.forms
271+
.iter()
272+
.map(|type_id| {
273+
let typ = self.get_type_space().get_type(type_id).unwrap();
274+
let form_name = typ.name();
275+
let td = typ.details();
276+
let TypeDetails::Struct(tstru) = td else { unreachable!() };
277+
let properties = indexmap::IndexMap::<&'_ str, _>::from_iter(
278+
tstru
279+
.properties()
280+
.filter_map(|(prop_name, prop_id)| {
281+
self.get_type_space()
282+
.get_type(&prop_id).ok()
283+
.map(|prop_typ| (prop_name, prop_typ))
284+
})
285+
286+
);
287+
let properties = syn::punctuated::Punctuated::<_, syn::Token![,]>::from_iter(
288+
properties
289+
.into_iter()
290+
.map(|(prop_name, prop_ty)| {
291+
let ident = quote::format_ident!("{}", prop_name);
292+
quote!{ (#prop_name, &self. #ident) }
293+
}));
294+
295+
let form_name = quote::format_ident!("{}",typ.name());
296+
297+
quote! {
298+
impl #form_name {
299+
pub fn as_form<'f>(&'f self) -> impl std::iter::Iterator<Item=(&'static str, &'f [u8])> {
300+
[#properties]
301+
.into_iter()
302+
.filter_map(|(name, val)| val.as_ref().map(|val| (name, val.as_slice())))
303+
}
304+
}
305+
}
306+
}));
265307
// Generate an implementation of a `Self::as_inner` method, if an inner
266308
// type is defined.
267309
let maybe_inner = self.settings.inner_type.as_ref().map(|inner| {
@@ -290,20 +332,20 @@ impl Generator {
290332
});
291333

292334
let client_docstring = {
293-
let mut s = format!("Client for {}", spec.info.title);
335+
let mut doc = format!("Client for {}", spec.info.title);
294336

295-
if let Some(ss) = &spec.info.description {
296-
s.push_str("\n\n");
297-
s.push_str(ss);
337+
if let Some(desc) = &spec.info.description {
338+
doc.push_str("\n\n");
339+
doc.push_str(desc);
298340
}
299-
if let Some(ss) = &spec.info.terms_of_service {
300-
s.push_str("\n\n");
301-
s.push_str(ss);
341+
if let Some(tos) = &spec.info.terms_of_service {
342+
doc.push_str("\n\n");
343+
doc.push_str(tos);
302344
}
303345

304-
s.push_str(&format!("\n\nVersion: {}", &spec.info.version));
346+
doc.push_str(&format!("\n\nVersion: {}", &spec.info.version));
305347

306-
s
348+
doc
307349
};
308350

309351
let version_str = &spec.info.version;
@@ -325,6 +367,8 @@ impl Generator {
325367
use std::convert::TryFrom;
326368

327369
#types
370+
371+
#extra_impl
328372
}
329373

330374
#[derive(Clone, Debug)]

‎progenitor-impl/src/method.rs‎

Lines changed: 26 additions & 44 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@ use indexmap::{IndexMap, IndexSet};
1010
use openapiv3::{Components, Parameter, ReferenceOr, Response, StatusCode};
1111
use proc_macro2::TokenStream;
1212
use quote::{format_ident, quote, ToTokens};
13-
use typify::{TypeId, TypeSpace};
13+
use typify::{TypeId, TypeSpace, TypeSpacePatch};
1414

1515
use crate::{
1616
template::PathTemplate,
@@ -105,7 +105,7 @@ pub struct OperationParameter {
105105
#[derive(Debug, Eq, PartialEq)]
106106
pub enum OperationParameterType {
107107
Type(TypeId),
108-
Form(IndexSet<String>),
108+
Form(TypeId),
109109
RawBody,
110110
}
111111

@@ -585,27 +585,15 @@ impl Generator {
585585
.map(|param| {
586586
let name = format_ident!("{}", param.name);
587587
match &param.typ {
588-
OperationParameterType::Type(type_id) => {
588+
OperationParameterType::Type(type_id) |
589+
OperationParameterType::Form(type_id) => {
589590
let typ = self
590591
.type_space
591592
.get_type(type_id)
592593
.expect("TypeIDs are _never_ deleted. qed")
593594
.parameter_ident_with_lifetime("a");
594595
quote! { #name: #typ}
595596
}
596-
OperationParameterType::Form(keys) => {
597-
let ts = TokenStream::from_iter(
598-
itertools::Itertools::intersperse(
599-
keys.iter().map(|form_prop_name| {
600-
let form_prop_name =
601-
format_ident!("{}", form_prop_name);
602-
quote! { #form_prop_name: Vec<u8> }
603-
}),
604-
quote! {, },
605-
),
606-
);
607-
ts
608-
}
609597
OperationParameterType::RawBody => {
610598
quote! { #name: B }
611599
}
@@ -935,15 +923,11 @@ impl Generator {
935923
OperationParameterKind::Body(
936924
BodyContentType::FormData
937925
),
938-
OperationParameterType::Form(map),
926+
OperationParameterType::Form(_),
939927
) => {
940-
let form_prop_names = map.iter().cloned().map(|form_prop_name| {
941-
let ident= format_ident!("{}", form_prop_name);
942-
quote! { (#form_prop_name, #ident) }
943-
});
944928
Some(quote! {
945929
// This uses progenitor_client::RequestBuilderExt which sets up a simple form data based on bytes
946-
.form_from_raw(vec![ #(#form_prop_names),* ])?
930+
.form_from_raw(body.as_form())?
947931
})},
948932
(OperationParameterKind::Body(_), _) => {
949933
unreachable!("invalid body kind/type combination")
@@ -1408,7 +1392,7 @@ impl Generator {
14081392
.params
14091393
.iter()
14101394
.map(|param| match &param.typ {
1411-
OperationParameterType::Type(type_id) => {
1395+
OperationParameterType::Type(type_id) | OperationParameterType::Form(type_id) => {
14121396
let ty = self.type_space.get_type(type_id)?;
14131397

14141398
// For body parameters only, if there's a builder we'll
@@ -1425,10 +1409,6 @@ impl Generator {
14251409
}
14261410
}
14271411

1428-
OperationParameterType::Form(_form) => {
1429-
todo!("Form is nit expected here")
1430-
}
1431-
14321412
OperationParameterType::RawBody => {
14331413
cloneable = false;
14341414
Ok(quote! { Result<reqwest::Body, String> })
@@ -1441,7 +1421,7 @@ impl Generator {
14411421
.params
14421422
.iter()
14431423
.map(|param| match &param.typ {
1444-
OperationParameterType::Type(type_id) => {
1424+
OperationParameterType::Type(type_id) | OperationParameterType::Form(type_id) => {
14451425
let ty = self.type_space.get_type(type_id)?;
14461426
let details = ty.details();
14471427
let optional =
@@ -1460,9 +1440,6 @@ impl Generator {
14601440
Ok(quote! { Err(#err_msg.to_string()) })
14611441
}
14621442
}
1463-
OperationParameterType::Form(_form) => {
1464-
todo!("Form is nit expected here")
1465-
}
14661443
OperationParameterType::RawBody => {
14671444
let err_msg = format!("{} was not initialized", param.name);
14681445
Ok(quote! { Err(#err_msg.to_string()) })
@@ -1474,7 +1451,7 @@ impl Generator {
14741451
.params
14751452
.iter()
14761453
.map(|param| match &param.typ {
1477-
OperationParameterType::Type(type_id) => {
1454+
OperationParameterType::Type(type_id) | OperationParameterType::Form(type_id) => {
14781455
let ty = self.type_space.get_type(type_id)?;
14791456
if ty.builder().is_some() {
14801457
let type_name = ty.ident();
@@ -1488,14 +1465,6 @@ impl Generator {
14881465
}
14891466
}
14901467

1491-
OperationParameterType::Form(_form) => {
1492-
todo!("Form is nit expected here")
1493-
}
1494-
1495-
OperationParameterType::Form(_form) => {
1496-
todo!("Form is nit expected here")
1497-
}
1498-
14991468
OperationParameterType::RawBody => Ok(quote! {}),
15001469
})
15011470
.collect::<Result<Vec<_>>>()?;
@@ -1508,11 +1477,12 @@ impl Generator {
15081477
.map(|param| {
15091478
let param_name = format_ident!("{}", param.name);
15101479
match &param.typ {
1511-
OperationParameterType::Type(type_id) => {
1480+
OperationParameterType::Type(type_id) |
1481+
OperationParameterType::Form(type_id) => {
15121482
let ty = self.type_space.get_type(type_id)?;
15131483
let details = ty.details();
15141484
match (&details, ty.builder()) {
1515-
// TODO right now optional body paramters are not
1485+
// TODO right now optional body parameters are not
15161486
// addressed
15171487
(typify::TypeDetails::Option(_), Some(_)) => {
15181488
unreachable!()
@@ -1596,7 +1566,7 @@ impl Generator {
15961566
}
15971567
}
15981568

1599-
OperationParameterType::Form(form_keys) => {
1569+
OperationParameterType::Form(type_id) => {
16001570
let err_msg = format!(
16011571
"conversion to `reqwest::Body` for {} failed",
16021572
param.name,
@@ -2161,7 +2131,19 @@ impl Generator {
21612131
schema
21622132
))),
21632133
}?;
2164-
OperationParameterType::Form(mapped)
2134+
2135+
let form_name = sanitize(
2136+
&format!(
2137+
"{}-form",
2138+
operation.operation_id.as_ref().unwrap(),
2139+
),
2140+
Case::Pascal,
2141+
);
2142+
let type_id = self
2143+
.type_space
2144+
.add_type_with_name(&schema.to_schema(), Some(form_name))?;
2145+
self.forms.insert(type_id.clone());
2146+
OperationParameterType::Form(type_id)
21652147
}
21662148
BodyContentType::Json | BodyContentType::FormUrlencoded => {
21672149
// TODO it would be legal to have the encoding field set for

0 commit comments

Comments
 (0)