stdlib/fedora: Use grep -F when matching %fedora-rebuild-name

[goals.git] / src / utils.ml

diff --git a/src/utils.ml b/src/utils.ml
index d1b0efa..c4722a8 100644 (file)
--- a/src/utils.ml
+++ b/src/utils.ml
@@ -19,11 +19,15 @@
 
 open Printf
 
+external nprocs : unit -> int = "nprocs"
+
 let failwithf fs = ksprintf failwith fs
 
 let (//) = Filename.concat
 let is_directory d = try Sys.is_directory d with Sys_error _ -> false
 
+let unique = let i = ref 0 in fun () -> incr i; !i
+
 (* From OCaml 4.08 sources.  We can remove this when we can
  * depend on min OCaml 4.08.
  *)
@@ -58,6 +62,30 @@ let rec string_find s sub =
   in
   loop 0
 
+let rec split sep str =
+  let len = String.length sep in
+  let seplen = String.length str in
+  let i = string_find str sep in
+  if i = -1 then str, ""
+  else (
+    String.sub str 0 i, String.sub str (i + len) (seplen - i - len)
+  )
+
+and nsplit ?(max = 0) sep str =
+  if max < 0 then
+    invalid_arg "String.nsplit: max parameter should not be negative";
+
+  (* If we reached the limit, OR if the pattern does not match the string
+   * at all, return the rest of the string as a single element list.
+   *)
+  if max = 1 || string_find str sep = -1 then
+    [str]
+  else (
+    let s1, s2 = split sep str in
+    let max = if max = 0 then 0 else max - 1 in
+    s1 :: nsplit ~max sep s2
+  )
+
 let isspace c =
   c = ' '
   (* || c = '\f' *) || c = '\n' || c = '\r' || c = '\t' (* || c = '\v' *)