From 933bf2efbd540642446a512415849cc0e7bcdd94 Mon Sep 17 00:00:00 2001 From: atagen Date: Sun, 24 Nov 2024 11:34:33 +1100 Subject: [PATCH] tighten acceptance of fresh cmd flags --- lib/meat.ml | 15 +++++++++++++-- 1 file changed, 13 insertions(+), 2 deletions(-) diff --git a/lib/meat.ml b/lib/meat.ml index 759e413..1e6c13d 100644 --- a/lib/meat.ml +++ b/lib/meat.ml @@ -62,8 +62,19 @@ let look () = do_cmd "nix flake show $NH_FLAKE"; print_string footer -let all_flag () = Array.mem "-a" argv || Array.mem "--all" argv -let sub_flag () = Array.mem "-s" argv || Array.mem "--subflake" argv +(* let all_flag () = Array.mem "-a" argv || Array.mem "--all" argv +let sub_flag () = Array.mem "-s" argv || Array.mem "--subflake" argv *) +let all_flag () = if Array.length argv >= 3 then match Array.get argv 2 with + | "-a" | "--all" -> true + | _ -> false +else + false +let sub_flag () = if Array.length argv >= 3 then match Array.get argv 2 with + | "-s" | "--sub" -> true + | _ -> false +else + false + let derelativise base = List.map (fun a -> base ^ "/" ^ a) let filter_dirs fullpath dirs =