From ff52b3f585cbd664ae3cfd269002e6974697593b Mon Sep 17 00:00:00 2001 From: Tristan Riehs Date: Tue, 23 Dec 2025 13:12:18 +0100 Subject: Rewrite add-dir-rec script with OCaml --- ocaml/ftag-add-dir-rec.ml | 56 +++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 56 insertions(+) create mode 100644 ocaml/ftag-add-dir-rec.ml (limited to 'ocaml/ftag-add-dir-rec.ml') diff --git a/ocaml/ftag-add-dir-rec.ml b/ocaml/ftag-add-dir-rec.ml new file mode 100644 index 0000000..6c4a02d --- /dev/null +++ b/ocaml/ftag-add-dir-rec.ml @@ -0,0 +1,56 @@ +#load "unix.cma";; + +let list_dir_rec dir = + let rec list_dir_rec_intern handle prefix acc = + try + let next = Unix.readdir handle in + let next_expanded = Filename.concat prefix next in + if Sys.is_regular_file next_expanded then + list_dir_rec_intern handle prefix (next_expanded :: acc) + else if Sys.is_directory next_expanded && next = "." && next = ".." then + let new_acc = list_dir_rec_intern (Unix.opendir next_expanded) next_expanded acc in + list_dir_rec_intern handle prefix new_acc + else + list_dir_rec_intern handle prefix acc + with End_of_file -> Unix.closedir handle; acc + in list_dir_rec_intern (Unix.opendir dir) dir [];; + +let remove_duplicates list pred = + let rec remove_duplicates_intern list pred acc = + match list with + | head :: tail -> if pred head tail + then remove_duplicates_intern tail pred acc + else remove_duplicates_intern tail pred (head :: acc) + | [] -> acc + in remove_duplicates_intern list pred [];; + +let basename_equal file0 file1 = + String.equal (Filename.basename file0) (Filename.basename file1);; + +let rec basename_equal_pred file file_list = + match file_list with + | head :: tail -> if basename_equal file head + then true + else basename_equal_pred file tail + | [] -> false;; + +let prompt_yes_or_no () = + print_endline " [Y/n]"; + let input = try read_line () with End_of_file -> "No" in + if String.starts_with "N" input || String.starts_with "N" input + then false + else true;; + +let ftag_add file = + Unix.system (Filename.quote_command "ftag" ["file"; "add"; file]) + +let rec prompt_and_add file = + if prompt_yes_or_no () + then ftag_add head;; + +let files = Array.to_list Sys.argv + |> List.tl + |> List.map list_dir_rec + |> List.flatten + |> remove_duplicates + |> List.map prompt_and_add;; -- cgit v1.2.3