Add a 'categories' parameter to 'select_words_except'

This way the -c/--category flag can also work together with the
-i/--inflection one.

Signed-off-by: Miquel Sabaté Solà <mssola@mssola.com>

1 files changed, 5 insertions, 1 deletions

diff --git a/crates/cli/src/run.rs b/crates/cli/src/run.rs
index 31f160e..06d4551 100644
--- a/crates/cli/src/run.rs
+++ b/crates/cli/src/run.rs
@@ -496,7 +496,11 @@ pub fn run(args: Vec<String>) {
                 if !inflection_only && !run_words(&list, &locale) {
                     break;
                 }
-                if let Ok(words_to_inflect) = mihi::select_words_except(&list, &flags) {
+                let cats = match category {
+                    Some(cat) => vec![cat],
+                    None => vec![Category::Noun, Category::Adjective],
+                };
+                if let Ok(words_to_inflect) = mihi::select_words_except(&list, &cats, &flags) {
                     if !run_inflect_words(&words_to_inflect, &locale) {
                         break;
                     }