diff --git a/src/prompts/fuzzy_select.rs b/src/prompts/fuzzy_select.rs index 706cd5e..9f06798 100644 --- a/src/prompts/fuzzy_select.rs +++ b/src/prompts/fuzzy_select.rs @@ -81,10 +81,14 @@ impl FuzzySelect<'_> { } /// Adds multiple items to the fuzzy selector. - pub fn items(mut self, items: &[T]) -> Self { - for item in items { - self.items.push(item.to_string()); - } + pub fn items(mut self, items: I) -> Self + where + T: ToString, + I: IntoIterator, + { + self.items + .extend(items.into_iter().map(|item| item.to_string())); + self } @@ -404,4 +408,12 @@ mod tests { let _ = fuzzy_select.clone(); } + + #[test] + fn test_iterator() { + let items = ["First", "Second", "Third"]; + let iterator = items.iter().skip(1); + + assert_eq!(FuzzySelect::new().items(iterator).items, &items[1..]); + } } diff --git a/src/prompts/multi_select.rs b/src/prompts/multi_select.rs index 2afa828..42b301d 100644 --- a/src/prompts/multi_select.rs +++ b/src/prompts/multi_select.rs @@ -101,17 +101,21 @@ impl MultiSelect<'_> { } /// Adds multiple items to the selector. - pub fn items(mut self, items: &[T]) -> Self { - for item in items { - self.items.push(item.to_string()); - self.defaults.push(false); - } - self + pub fn items(self, items: I) -> Self + where + T: ToString, + I: IntoIterator, + { + self.items_checked(items.into_iter().map(|item| (item, false))) } /// Adds multiple items to the selector with checked state - pub fn items_checked(mut self, items: &[(T, bool)]) -> Self { - for &(ref item, checked) in items { + pub fn items_checked(mut self, items: I) -> Self + where + T: ToString, + I: IntoIterator, + { + for (item, checked) in items.into_iter() { self.items.push(item.to_string()); self.defaults.push(checked); } @@ -382,4 +386,12 @@ mod tests { let _ = multi_select.clone(); } + + #[test] + fn test_iterator() { + let items = ["First", "Second", "Third"]; + let iterator = items.iter().skip(1); + + assert_eq!(MultiSelect::new().items(iterator).items, &items[1..]); + } } diff --git a/src/prompts/select.rs b/src/prompts/select.rs index b932c6b..872975d 100644 --- a/src/prompts/select.rs +++ b/src/prompts/select.rs @@ -99,14 +99,19 @@ impl Select<'_> { /// ``` pub fn item(mut self, item: T) -> Self { self.items.push(item.to_string()); + self } /// Adds multiple items to the selector. - pub fn items(mut self, items: &[T]) -> Self { - for item in items { - self.items.push(item.to_string()); - } + pub fn items(mut self, items: I) -> Self + where + T: ToString, + I: IntoIterator, + { + self.items + .extend(items.into_iter().map(|item| item.to_string())); + self } @@ -363,7 +368,7 @@ mod tests { let selections = vec!["a".to_string(), "b".to_string()]; assert_eq!( - Select::new().default(0).items(&selections[..]).items, + Select::new().default(0).items(&selections).items, selections ); } @@ -375,9 +380,14 @@ mod tests { let selections = &[a, b]; - assert_eq!( - Select::new().default(0).items(&selections[..]).items, - selections - ); + assert_eq!(Select::new().default(0).items(selections).items, selections); + } + + #[test] + fn test_iterator() { + let items = ["First", "Second", "Third"]; + let iterator = items.iter().skip(1); + + assert_eq!(Select::new().default(0).items(iterator).items, &items[1..]); } } diff --git a/src/prompts/sort.rs b/src/prompts/sort.rs index e385685..4b1b0c8 100644 --- a/src/prompts/sort.rs +++ b/src/prompts/sort.rs @@ -83,10 +83,14 @@ impl Sort<'_> { } /// Adds multiple items to the selector. - pub fn items(mut self, items: &[T]) -> Self { - for item in items { - self.items.push(item.to_string()); - } + pub fn items(mut self, items: I) -> Self + where + T: ToString, + I: IntoIterator, + { + self.items + .extend(items.into_iter().map(|item| item.to_string())); + self } @@ -373,4 +377,12 @@ mod tests { let _ = sort.clone(); } + + #[test] + fn test_iterator() { + let items = ["First", "Second", "Third"]; + let iterator = items.iter().skip(1); + + assert_eq!(Sort::new().items(iterator).items, &items[1..]); + } }