give the option to target only a given access in a base when changing accesses

Author: Elie Canonici Merle

gwpublic/gwaccess.ml

@@ -33,6 +33,7 @@ let input_file list file =
   close_in ic
 
 let () =
+  let target_access = ref None in
   let access = ref None in
   let list = ref [] in
   let bname = ref "" in
@@ -41,10 +42,23 @@ let () =
   let access_opt opt doc x =
     (opt, Arg.Unit (fun () -> set_access x), " Set access flag to " ^ doc ^ ".")
   in
+  let access_of_string s = match s with
+    | "public" -> Some Def.Public
+    | "private" -> Some Def.Private
+    | "default" -> Some Def.IfTitles
+    | _ -> None
+  in
+  let set_target_access_of_string s =
+    assert (!target_access = None);
+    let access_opt = access_of_string s in
+    assert (Option.is_some access_opt);
+    target_access := access_opt
+  in
   let speclist =
     [ access_opt "--private" "PRIVATE" Def.Private
     ; access_opt "--public" "PUBLIC" Def.Public
     ; access_opt "--default" "IFTITLE" Def.IfTitles
+    ; "--target", Arg.String (fun s -> set_target_access_of_string s), "[public|default|private] Proccess only persons with given access"
     ; "--everybody", Arg.Set everybody, " process the wole base."
     ; "--key", Arg.String (fun s -> list := s :: !list), "<KEY>"
     ; "--keys", Arg.String (fun s -> input_file list s), "<FILE> file containing one key per line."
@@ -54,12 +68,17 @@ let () =
   let anonfun i = bname := i in
   let usage = "Usage: " ^ Sys.argv.(0) ^ " [-everybody] [-ind key] [-list-ind file] base" in
   Arg.parse speclist anonfun usage ;
-  match !bname, !access, !everybody, !list with
-  | ("", _, _, _) | (_, None, _, _) | (_, _, false, []) ->
+  match !bname, !access, !everybody, !target_access, !list with
+  | ("", _, _, _, _) | (_, None, _, _, _) | (_, _, false, None, []) ->
     Arg.usage speclist usage ;
     exit 2 ;
-  | (bname, Some access, everybody, list) ->
+  | (bname, Some access, everybody, target_access, list) ->
     Secure.set_base_dir (Filename.dirname bname);
     Lock.control_retry (Files.lock_file bname) ~onerror:Lock.print_error_and_exit @@ fun () ->
-    if everybody then Gwaccess_util.access_everybody access bname
+    if Option.is_some target_access then
+      Gwaccess_util.change_only_old_access
+        ~old_access:(Option.get target_access)
+        ~new_access:access
+        bname
+    else if everybody then Gwaccess_util.access_everybody access bname
     else access_some access bname list

gwpublic/gwaccess_util.ml

@@ -73,3 +73,12 @@ let access_everybody access bname =
       patch_person base p.Def.key_index p
   end (Gwdb.persons base) ;
   commit_patches base
+
+let change_only_old_access ~old_access ~new_access bname =
+  let base = Gwdb.open_base bname in
+  Gwdb.Collection.iter begin fun p ->
+    if Gwdb.get_access p = old_access then
+      let p = {(gen_person_of_person p) with Def.access = new_access} in
+      patch_person base p.Def.key_index p
+  end (Gwdb.persons base) ;
+  commit_patches base