ca26755581c45e363a0eea475f033a80c81ec29e
[whenjobs.git] / 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")