lib/whenutils.ml

   1 (* whenjobs
   2  * Copyright (C) 2012 Red Hat Inc.
   3  *
   4  * This program is free software; you can redistribute it and/or modify
   5  * it under the terms of the GNU General Public License as published by
   6  * the Free Software Foundation; either version 2 of the License, or
   7  * (at your option) any later version.
   8  *
   9  * This program is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  * GNU General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License along
  15  * with this program; if not, write to the Free Software Foundation, Inc.,
  16  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  17  *)
  18
  19 open Camlp4.PreCast
  20 open Ast
  21
  22 open CalendarLib
  23
  24 open Big_int
  25 open Unix
  26 open Printf
  27
  28 module StringMap = struct
  29   include Map.Make (String)
  30   let keys m = fold (fun k _ ks -> k :: ks) m []
  31   let values m = fold (fun _ v vs -> v :: vs) m []
  32 end
  33
  34 module IntMap = struct
  35   include Map.Make (struct type t = int let compare = compare end)
  36   let keys m = fold (fun k _ ks -> k :: ks) m []
  37   let values m = fold (fun _ v vs -> v :: vs) m []
  38 end
  39
  40 module BigIntMap = struct
  41   include Map.Make (struct type t = big_int let compare = compare_big_int end)
  42   let keys m = fold (fun k _ ks -> k :: ks) m []
  43   let values m = fold (fun _ v vs -> v :: vs) m []
  44 end
  45
  46 module StringSet = Set.Make (String)
  47
  48 let (//) = Filename.concat
  49
  50 let isalpha = function 'a'..'z' | 'A'..'Z' -> true | _ -> false
  51 let isalnum = function 'a'..'z' | 'A'..'Z' | '0'..'9' -> true | _ -> false
  52
  53 let rec filter_map f = function
  54   | [] -> []
  55   | x :: xs ->
  56     match f x with
  57     | Some y -> y :: filter_map f xs
  58     | None -> filter_map f xs
  59
  60 let string_of_time_t ?(localtime = false) t =
  61   let tm = (if localtime then Unix.localtime else gmtime) t in
  62   sprintf "%04d-%02d-%02d %02d:%02d:%02d%s"
  63     (1900+tm.tm_year) (1+tm.tm_mon) tm.tm_mday
  64     tm.tm_hour tm.tm_min tm.tm_sec
  65     (if localtime then "" else " UTC")
  66
  67 let string_startswith str prefix =
  68   let len = String.length str in
  69   let plen = String.length prefix in
  70   len >= plen && String.sub str 0 plen = prefix
  71
  72 let string_endswith str suffix =
  73   let len = String.length str in
  74   let slen = String.length suffix in
  75   len >= slen && String.sub str (len-slen) slen = suffix