Implemented --interactive for rm

rm/rm.rs

@@ -12,7 +12,7 @@
 extern mod extra;
 
 use std::os;
-use std::io::{stderr,fs,io_error};
+use std::io::{stdin,stderr,stdio,fs,buffered,io_error};
 use extra::getopts::groups;
 
 enum InteractiveMode {
@@ -21,6 +21,21 @@ enum InteractiveMode {
     InteractiveAlways
 }
 
+impl Eq for InteractiveMode {
+    fn eq(&self, other: &InteractiveMode) -> bool {
+        match (*self, *other) {
+            (InteractiveNone, InteractiveNone) |
+            (InteractiveOnce, InteractiveOnce) |
+            (InteractiveAlways, InteractiveAlways) => true,
+            _ => false
+        }
+    }
+
+    fn ne(&self, other: &InteractiveMode) -> bool {
+        !self.eq(other)
+    }
+}
+
 fn main() {
     let args = os::args();
     let program = args[0].clone();
@@ -71,6 +86,11 @@ fn main() {
         println("assurance that the contents are truly unrecoverable, consider using shred.");
     } else if matches.opt_present("version") {
         println("rm 1.0.0");
+    } else if matches.free.is_empty() {
+        writeln!(&mut stderr() as &mut Writer, "Missing an argument");
+        writeln!(&mut stderr() as &mut Writer,
+                 "For help, try '{0:s} --help'", program);
+        os::set_exit_status(1);
     } else {
         let force = matches.opt_present("force");
         let interactive =
@@ -98,36 +118,33 @@ fn main() {
         let recursive = matches.opt_present("recursive");
         let dir = matches.opt_present("dir");
         let verbose = matches.opt_present("verbose");
-        remove_files(matches.free, force, interactive, one_fs, preserve_root,
+        if interactive == InteractiveOnce && (recursive || matches.free.len() > 3) {
+            let msg =
+                if recursive {
+                    "Remove all arguments recursively? "
+                } else {
+                    "Remove all arguments? "
+                };
+            if !prompt(msg) {
+                return;
+            }
+        }
+        remove(matches.free, force, interactive, one_fs, preserve_root,
                recursive, dir, verbose);
     }
 }
 
-// TODO: implement one-file-system and interactive
+// TODO: implement one-file-system
-fn remove_files(files: &[~str], force: bool, interactive: InteractiveMode, one_fs: bool, preserve_root: bool, recursive: bool, dir: bool, verbose: bool) {
+fn remove(files: &[~str], force: bool, interactive: InteractiveMode, one_fs: bool, preserve_root: bool, recursive: bool, dir: bool, verbose: bool) {
     for filename in files.iter() {
         let file = Path::new(filename.to_owned());
         if file.exists() {
             if file.is_dir() {
                 if recursive && (*filename != ~"/" || !preserve_root) {
-                    remove_files(fs::walk_dir(&file).map(|x| x.as_str().unwrap().to_owned()).to_owned_vec(), force, interactive, one_fs, preserve_root, recursive, dir, verbose);
+                    remove(fs::walk_dir(&file).map(|x| x.as_str().unwrap().to_owned()).to_owned_vec(), force, interactive, one_fs, preserve_root, recursive, dir, verbose);
-                    io_error::cond.trap(|_| {
+                    remove_dir(&file, *filename, interactive, verbose);
-                        writeln!(&mut stderr() as &mut Writer,
-                                 "Could not remove directory: '{}'", *filename);
-                        os::set_exit_status(1);
-                    }).inside(|| {
-                        fs::rmdir(&file);
-                        println!("Removed '{}'", *filename);
-                    });
                 } else if dir && (*filename != ~"/" || !preserve_root) {
-                    io_error::cond.trap(|_| {
+                    remove_dir(&file, *filename, interactive, verbose);
-                        writeln!(&mut stderr() as &mut Writer,
-                                 "Could not remove directory '{}'", *filename);
-                        os::set_exit_status(1);
-                    }).inside(|| {
-                        fs::rmdir(&file);
-                        println!("Removed '{}'", *filename);
-                    });
                 } else {
                     if recursive {
                         writeln!(&mut stderr() as &mut Writer,
@@ -137,18 +154,11 @@ fn remove_files(files: &[~str], force: bool, interactive: InteractiveMode, one_f
                         writeln!(&mut stderr() as &mut Writer,
                                  "Could not remove directory '{}' (did you mean to pass '-r'?)",
                                  *filename);
-                        os::set_exit_status(1);
                     }
+                    os::set_exit_status(1);
                 }
             } else {
-                io_error::cond.trap(|_| {
+                remove_file(&file, *filename, interactive, verbose);
-                    writeln!(&mut stderr() as &mut Writer,
-                             "Could not remove file: '{}'", *filename);
-                    os::set_exit_status(1);
-                }).inside(|| {
-                    fs::unlink(&file);
-                    println!("Removed '{}'", *filename);
-                });
             }
         } else if !force {
             writeln!(&mut stderr() as &mut Writer,
@@ -158,3 +168,75 @@ fn remove_files(files: &[~str], force: bool, interactive: InteractiveMode, one_f
     }
 }
 
+fn remove_dir(path: &Path, name: &str, interactive: InteractiveMode, verbose: bool) {
+    let response =
+        if interactive == InteractiveAlways {
+            prompt_file(path, name)
+        } else {
+            true
+        };
+    if response {
+        io_error::cond.trap(|_| {
+            writeln!(&mut stderr() as &mut Writer,
+                     "Could not remove directory '{}'", name);
+            os::set_exit_status(1);
+        }).inside(|| {
+            fs::rmdir(path);
+            if verbose {
+                println!("Removed '{}'", name);
+            }
+        });
+    }
+}
+
+fn remove_file(path: &Path, name: &str, interactive: InteractiveMode, verbose: bool) {
+    let response =
+        if interactive == InteractiveAlways {
+            prompt_file(path, name)
+        } else {
+            true
+        };
+    if response {
+        io_error::cond.trap(|_| {
+            writeln!(&mut stderr() as &mut Writer,
+                     "Could not remove file '{}'", name);
+            os::set_exit_status(1);
+        }).inside(|| {
+            fs::unlink(path);
+            if verbose {
+                println!("Removed '{}'", name);
+            }
+        });
+    }
+}
+
+fn prompt_file(path: &Path, name: &str) -> bool {
+    if path.is_dir() {
+        prompt(format!("Remove directory '{}'? ", name))
+    } else {
+        prompt(format!("Remove file '{}'? ", name))
+    }
+}
+
+fn prompt(msg: &str) -> bool {
+    print(msg);
+    read_prompt()
+}
+
+fn read_prompt() -> bool {
+    stdio::flush();
+    match buffered::BufferedReader::new(stdin()).read_line() {
+        Some(line) => {
+            match line.char_at(0) {
+                'y' | 'Y' => true,
+                'n' | 'N' => false,
+                _ => {
+                    print("Please enter either Y or N: ");
+                    read_prompt()
+                }
+            }
+        }
+        None => true
+    }
+}
+