T7321: add utility needed for validate_tree in vyconfd load config

2 files changed, 24 insertions, 0 deletions

diff --git a/src/config_tree.ml b/src/config_tree.ml
index 712db2b..4d0f5df 100644
--- a/src/config_tree.ml
+++ b/src/config_tree.ml
@@ -135,6 +135,28 @@ let get_subtree ?(with_node=false) node path =
             Vytree.make_full default_data "" (Vytree.children_of_node n)
     with Vytree.Nonexistent_path -> make ""
 
+let value_paths_of_tree node =
+    let func ct (p, a) _t =
+        match p with
+        | [] -> (p, a)
+        | _ ->
+            let q = List.rev p in
+            if not (Vytree.is_terminal_path ct q) then
+                (p, a)
+            else
+                let vals = get_values ct q in
+                match vals with
+                | [] -> (p, q::a)
+                | _ as vs ->
+                    let a' =
+                        let f acc v =
+                            let q' = q @ [v] in
+                            q'::acc
+                        in List.fold_left f a vs
+                    in (p, a')
+    in List.rev (snd (Vytree.fold_tree_with_path (func node) ([], []) node))
+
+
 module Renderer =
 struct
     (* Rendering configs as set commands *)
diff --git a/src/config_tree.mli b/src/config_tree.mli
index f24a906..be746a0 100644
--- a/src/config_tree.mli
+++ b/src/config_tree.mli
@@ -47,6 +47,8 @@ val is_leaf : t -> string list -> bool
 
 val get_subtree : ?with_node:bool -> t -> string list -> t
 
+val value_paths_of_tree : t -> string list list
+
 val render_commands : ?op:command -> t -> string list -> string
 
 val render_config : ?ord_val:bool -> t -> string