From 995757c05510737509ad634540644a21d7d167d1 Mon Sep 17 00:00:00 2001
From: Fernando Herrera <fernando.j.herrera@gmail.com>
Date: Sun, 27 Feb 2022 11:17:13 +0000
Subject: [PATCH] flags for find (#4663)

---
 crates/nu-command/src/filters/find.rs | 44 ++++++++++++++++++++++++---
 1 file changed, 40 insertions(+), 4 deletions(-)

diff --git a/crates/nu-command/src/filters/find.rs b/crates/nu-command/src/filters/find.rs
index b0aa28514..6fb40879d 100644
--- a/crates/nu-command/src/filters/find.rs
+++ b/crates/nu-command/src/filters/find.rs
@@ -29,6 +29,22 @@ impl Command for Find {
                 "regex to match with",
                 Some('r'),
             )
+            .switch(
+                "insensitive",
+                "case-insensitive search for regex (?i)",
+                Some('i'),
+            )
+            .switch(
+                "multiline",
+                "multi-line mode: ^ and $ match begin/end of line for regex (?m)",
+                Some('m'),
+            )
+            .switch(
+                "dotall",
+                "dotall mode: allow a dot . to match newline character \\n for regex (?s)",
+                Some('s'),
+            )
+            .switch("invert", "invert the match", Some('v'))
             .rest("rest", SyntaxShape::Any, "terms to search")
             .category(Category::Filters)
     }
@@ -94,7 +110,7 @@ impl Command for Find {
             },
             Example {
                 description: "Find using regex case insensitive",
-                example: r#"[aBc bde Arc abf] | find --regex "(?i)ab""#,
+                example: r#"[aBc bde Arc abf] | find --regex "ab" -i"#,
                 result: Some(Value::List {
                     vals: vec![Value::test_string("aBc".to_string()), Value::test_string("abf".to_string())],
                     span: Span::test_data()
@@ -149,13 +165,31 @@ fn find_with_regex(
     let ctrlc = engine_state.ctrlc.clone();
     let config = stack.get_config()?;
 
+    let insensitive = call.has_flag("insensitive");
+    let multiline = call.has_flag("multiline");
+    let dotall = call.has_flag("dotall");
+    let invert = call.has_flag("invert");
+
+    let flags = match (insensitive, multiline, dotall) {
+        (false, false, false) => "",
+        (true, false, false) => "(?i)",
+        (false, true, false) => "(?m)",
+        (false, false, true) => "(?s)",
+        (true, true, false) => "(?im)",
+        (true, false, true) => "(?is)",
+        (false, true, true) => "(?ms)",
+        (true, true, true) => "(?ims)",
+    };
+
+    let regex = flags.to_string() + &regex;
+
     let re = Regex::new(regex.as_str())
         .map_err(|e| ShellError::UnsupportedInput(format!("incorrect regex: {}", e), span))?;
 
     input.filter(
         move |value| {
             let string = value.into_string(" ", &config);
-            re.is_match(string.as_str())
+            re.is_match(string.as_str()) != invert
         },
         ctrlc,
     )
@@ -174,6 +208,7 @@ fn find_with_predicate(
     let redirect_stdout = call.redirect_stdout;
     let redirect_stderr = call.redirect_stderr;
     let engine_state = engine_state.clone();
+    let invert = call.has_flag("invert");
 
     let capture_block = predicate;
     let block_id = capture_block.block_id;
@@ -205,7 +240,7 @@ fn find_with_predicate(
                 redirect_stderr,
             )
             .map_or(false, |pipeline_data| {
-                pipeline_data.into_value(span).is_true()
+                pipeline_data.into_value(span).is_true() != invert
             })
         },
         ctrlc,
@@ -223,6 +258,7 @@ fn find_with_rest(
     let metadata = input.metadata();
     let engine_state = engine_state.clone();
     let config = stack.get_config()?;
+    let invert = call.has_flag("invert");
 
     let terms = call.rest::<Value>(&engine_state, stack, 0)?;
     let lower_terms = terms
@@ -277,7 +313,7 @@ fn find_with_rest(
                     }
                 }),
                 Value::Binary { .. } => false,
-            })
+            }) != invert
         },
         ctrlc,
     )?;