refactor: support IntoIterMapCollect with N types

1 files changed, 7 insertions, 7 deletions

diff --git a/src/transform.rs b/src/transform.rs
index fa7877e..f2744cf 100644
--- a/src/transform.rs
+++ b/src/transform.rs
@@ -120,9 +120,9 @@ impl TypeConverter<'_> {
                                 } else if self.known_as_collection_with_one_type(&last_seg.ident)
                                     && path_len == 1
                                 {
-                                    return Ok(TypeTransform::CollectionOneType(
-                                        self.convert_type(arg)?.into(),
-                                    ));
+                                    return Ok(TypeTransform::IntoIterMapCollect(vec![
+                                        self.convert_type(arg)?
+                                    ]));
                                 }
                             }
                         }
@@ -140,10 +140,10 @@ impl TypeConverter<'_> {
                                         self.convert_type(arg2)?.into(),
                                     ));
                                 } else if self.known_as_collection_with_two_types(&last_seg.ident) {
-                                    return Ok(TypeTransform::CollectionTwoTypes(
-                                        self.convert_type(arg1)?.into(),
-                                        self.convert_type(arg2)?.into(),
-                                    ));
+                                    return Ok(TypeTransform::IntoIterMapCollect(vec![
+                                        self.convert_type(arg1)?,
+                                        self.convert_type(arg2)?,
+                                    ]));
                                 }
                             }
                         }