X-Git-Url: https://git.xandkar.net/?p=dups.git;a=blobdiff_plain;f=dups.ml;h=e88d8885240378b84714202793ab850878588232;hp=0bcc83a8a312f52ff3dbd91dab32493e67538dd8;hb=61a05dbbfdda5f3539813fa196d960fb499c2f24;hpb=e13e9ef5acbfd0affadcf6d23223fa61ffd8cd6b

diff --git a/dups.ml b/dups.ml
index 0bcc83a..e88d888 100644
--- a/dups.ml
+++ b/dups.ml
@@ -41,7 +41,6 @@ end = struct
   let find_files root =
     let dirs  = Queue.create () in
     let files = Queue.create () in
-    Queue.add root dirs;
     let explore parent =
       Array.iter (Sys.readdir parent) ~f:(fun child ->
         let path = Filename.concat parent child in
@@ -59,24 +58,16 @@ end = struct
             ()
       )
     in
-    let next_dir () =
-      match Queue.take dirs with
-      | exception Queue.Empty ->
-          ()
-      | dir ->
-          explore dir
+    explore root;
+    let rec next () =
+      match Queue.is_empty files, Queue.is_empty dirs with
+      | false, _     -> Some (Queue.take files)
+      | true , true  -> None
+      | true , false ->
+          explore (Queue.take dirs);
+          next ()
     in
-    let next_file () =
-      match Queue.take files with
-      | exception Queue.Empty ->
-          None
-      | file_path ->
-          Some file_path
-    in
-    Stream.create (fun () ->
-      next_dir ();
-      next_file ()
-    )
+    Stream.create next
 end
 
 type input =
@@ -121,5 +112,14 @@ let main input =
 
 let () =
   let input = ref Paths_on_stdin in
-  Arg.parse [] (fun path -> input := Root_path path) "";
+  Arg.parse
+    []
+    (function
+    | path when Sys.file_exists path ->
+        input := Root_path path
+    | path ->
+        eprintf "File does not exist: %S\n%!" path;
+        exit 1
+    )
+    "";
   main !input