Add support for g flag

vinc · Sep 17, 2024 · 1bfafbc · 1bfafbc
1 parent bc3c0db
commit 1bfafbc
Showing 1 changed file with 21 additions and 13 deletions.
diff --git a/src/usr/edit.rs b/src/usr/edit.rs
@@ -610,25 +610,33 @@ impl Editor {
 
     pub fn exec(&mut self) {
         if let Some(query) = prompt(&mut self.command_prompt, ":") {
-            if !query.is_empty() {
-                let params: Vec<&str> = query.split('/').collect();
-                match params[0] {
-                    "s" if params.len() == 4 => { // Replace current line
-                        let re = Regex::new(params[1]);
-                        let s = params[2];
-                        let y = self.offset.y + self.cursor.y;
+            let params: Vec<&str> = query.split('/').collect();
+            match params[0] {
+                "s" if params.len() == 4 => { // Replace current line
+                    let re = Regex::new(params[1]);
+                    let s = params[2];
+                    let y = self.offset.y + self.cursor.y;
+                    if params[3] == "g" { // Replace all occurrences
+                        self.lines[y] = re.replace_all(&self.lines[y], s);
+                    } else {
                         self.lines[y] = re.replace(&self.lines[y], s);
                     }
-                    "%s" if params.len() == 4 => { // Replace all lines
-                        let re = Regex::new(params[1]);
-                        let s = params[2];
-                        let n = self.lines.len();
-                        for y in 0..n {
+                }
+                "%s" if params.len() == 4 => { // Replace all lines
+                    let re = Regex::new(params[1]);
+                    let s = params[2];
+                    let n = self.lines.len();
+                    for y in 0..n {
+                        if params[3] == "g" { // Replace all occurrences
+                            self.lines[y] = re.replace_all(&self.lines[y], s);
+                        } else {
                             self.lines[y] = re.replace(&self.lines[y], s);
                         }
                     }
-                    _ => {}
                 }
+                _ => {}
+            }
+            if !query.is_empty() {
                 self.command_prompt.history.add(&query);
                 self.command_prompt.history.save(&self.command_history);
             }