aboutsummaryrefslogtreecommitdiff
path: root/src/transform.rs
diff options
context:
space:
mode:
authorMartin Fischer <martin@push-f.com>2021-11-22 08:55:14 +0100
committerMartin Fischer <martin@push-f.com>2021-11-22 09:21:02 +0100
commitef698909223b2f29f7a6cae17f0aba3461c8803d (patch)
tree5f91a28cf98b6742c899bf515e99fcb8f89261ff /src/transform.rs
parent19823e09b7fe5e6a5b3965c7096c9daba18ce028 (diff)
support HashMap and BTreeMap
Diffstat (limited to 'src/transform.rs')
-rw-r--r--src/transform.rs26
1 files changed, 19 insertions, 7 deletions
diff --git a/src/transform.rs b/src/transform.rs
index 4dc88f0..4625f4c 100644
--- a/src/transform.rs
+++ b/src/transform.rs
@@ -33,6 +33,12 @@ fn is_supported_collection(ident: &Ident) -> bool {
|| ident == "BTreeSet"
}
+fn is_supported_collection_map(ident: &Ident) -> bool {
+ // collections added here must implement IntoIterator & FromIterator
+ // when adding a type here don't forget to document it in the README
+ ident == "HashMap" || ident == "BTreeMap"
+}
+
impl AssocTypeConversions<'_> {
pub fn parse_type_path(&self, type_: &mut Type) -> Result<TypeTransform, TransformError> {
if !iter_type(type_).any(match_assoc_type) {
@@ -123,14 +129,20 @@ impl AssocTypeConversions<'_> {
if let (GenericArgument::Type(arg1), GenericArgument::Type(arg2)) =
(args_iter.next().unwrap(), args_iter.next().unwrap())
{
- if (iter_type(arg1).any(match_assoc_type)
- || iter_type(arg2).any(match_assoc_type))
- && last_seg.ident == "Result"
+ if iter_type(arg1).any(match_assoc_type)
+ || iter_type(arg2).any(match_assoc_type)
{
- return Ok(TypeTransform::Result(
- self.parse_type_path(arg1)?.into(),
- self.parse_type_path(arg2)?.into(),
- ));
+ if last_seg.ident == "Result" {
+ return Ok(TypeTransform::Result(
+ self.parse_type_path(arg1)?.into(),
+ self.parse_type_path(arg2)?.into(),
+ ));
+ } else if is_supported_collection_map(&last_seg.ident) {
+ return Ok(TypeTransform::CollectionMap(
+ self.parse_type_path(arg1)?.into(),
+ self.parse_type_path(arg2)?.into(),
+ ));
+ }
}
}
}