lib/whenutils.mli

   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 (** Types and utility functions. *)
  20
  21 module StringMap : sig
  22   type key = String.t
  23   type 'a t
  24   val empty : 'a t
  25   val is_empty : 'a t -> bool
  26   val mem : key -> 'a t -> bool
  27   val add : key -> 'a -> 'a t -> 'a t
  28   (*val singleton : key -> 'a -> 'a t*)
  29   val remove : key -> 'a t -> 'a t
  30   (*val merge :
  31     (key -> 'a option -> 'b option -> 'c option) -> 'a t -> 'b t -> 'c t*)
  32   val compare : ('a -> 'a -> int) -> 'a t -> 'a t -> int
  33   val equal : ('a -> 'a -> bool) -> 'a t -> 'a t -> bool
  34   val iter : (key -> 'a -> unit) -> 'a t -> unit
  35   val fold : (key -> 'a -> 'b -> 'b) -> 'a t -> 'b -> 'b
  36   (*val for_all : (key -> 'a -> bool) -> 'a t -> bool
  37   val exists : (key -> 'a -> bool) -> 'a t -> bool
  38   val filter : (key -> 'a -> bool) -> 'a t -> 'a t
  39   val partition : (key -> 'a -> bool) -> 'a t -> 'a t * 'a t
  40   val cardinal : 'a t -> int
  41   val bindings : 'a t -> (key * 'a) list
  42   val min_binding : 'a t -> key * 'a
  43   val max_binding : 'a t -> key * 'a
  44   val choose : 'a t -> key * 'a
  45   val split : key -> 'a t -> 'a t * 'a option * 'a t*)
  46   val find : key -> 'a t -> 'a
  47   val map : ('a -> 'b) -> 'a t -> 'b t
  48   val mapi : (key -> 'a -> 'b) -> 'a t -> 'b t
  49   val keys : 'a t -> key list
  50   val values : 'a t -> 'a list
  51 end
  52 (** A map from string to any type. *)
  53
  54 module IntMap : sig
  55   type key = int
  56   type 'a t
  57   val empty : 'a t
  58   val is_empty : 'a t -> bool
  59   val mem : key -> 'a t -> bool
  60   val add : key -> 'a -> 'a t -> 'a t
  61   (*val singleton : key -> 'a -> 'a t*)
  62   val remove : key -> 'a t -> 'a t
  63   (*val merge :
  64     (key -> 'a option -> 'b option -> 'c option) -> 'a t -> 'b t -> 'c t*)
  65   val compare : ('a -> 'a -> int) -> 'a t -> 'a t -> int
  66   val equal : ('a -> 'a -> bool) -> 'a t -> 'a t -> bool
  67   val iter : (key -> 'a -> unit) -> 'a t -> unit
  68   val fold : (key -> 'a -> 'b -> 'b) -> 'a t -> 'b -> 'b
  69   (*val for_all : (key -> 'a -> bool) -> 'a t -> bool
  70   val exists : (key -> 'a -> bool) -> 'a t -> bool
  71   val filter : (key -> 'a -> bool) -> 'a t -> 'a t
  72   val partition : (key -> 'a -> bool) -> 'a t -> 'a t * 'a t
  73   val cardinal : 'a t -> int
  74   val bindings : 'a t -> (key * 'a) list
  75   val min_binding : 'a t -> key * 'a
  76   val max_binding : 'a t -> key * 'a
  77   val choose : 'a t -> key * 'a
  78   val split : key -> 'a t -> 'a t * 'a option * 'a t*)
  79   val find : key -> 'a t -> 'a
  80   val map : ('a -> 'b) -> 'a t -> 'b t
  81   val mapi : (key -> 'a -> 'b) -> 'a t -> 'b t
  82   val keys : 'a t -> key list
  83   val values : 'a t -> 'a list
  84 end
  85 (** A map from int to any type. *)
  86
  87 module StringSet : sig
  88   type elt = String.t
  89   type t = Set.Make(String).t
  90   val empty : t
  91   val is_empty : t -> bool
  92   val mem : elt -> t -> bool
  93   val add : elt -> t -> t
  94   val singleton : elt -> t
  95   val remove : elt -> t -> t
  96   val union : t -> t -> t
  97   val inter : t -> t -> t
  98   val diff : t -> t -> t
  99   val compare : t -> t -> int
 100   val equal : t -> t -> bool
 101   val subset : t -> t -> bool
 102   val iter : (elt -> unit) -> t -> unit
 103   val fold : (elt -> 'a -> 'a) -> t -> 'a -> 'a
 104   val for_all : (elt -> bool) -> t -> bool
 105   val exists : (elt -> bool) -> t -> bool
 106   val filter : (elt -> bool) -> t -> t
 107   val partition : (elt -> bool) -> t -> t * t
 108   val cardinal : t -> int
 109   val elements : t -> elt list
 110   val min_elt : t -> elt
 111   val max_elt : t -> elt
 112   val choose : t -> elt
 113   val split : elt -> t -> t * bool * t
 114 end
 115 (** A set of strings. *)
 116
 117 val (//) : string -> string -> string
 118 (** [dir // file] concatenates directory and file. *)
 119
 120 val filter_map : ('a -> 'b option) -> 'a list -> 'b list
 121 (** Filter + map. *)
 122
 123 type whenexpr =
 124   | Expr_bool of bool                   (** A boolean constant. *)
 125   | Expr_str of string                  (** A string constant. *)
 126   | Expr_int of Big_int.big_int         (** An integer constant. *)
 127   | Expr_float of float                 (** A float constant. *)
 128   | Expr_var of string                  (** A variable name. *)
 129   | Expr_and of whenexpr * whenexpr     (** && *)
 130   | Expr_or of whenexpr * whenexpr      (** || *)
 131   | Expr_eq of whenexpr * whenexpr      (** == *)
 132   | Expr_not of whenexpr                (** ! *)
 133   | Expr_changes of string              (** changes var *)
 134 (** Internal type used to represent 'when' expressions. *)
 135
 136 type periodexpr =
 137   | Every_seconds of int
 138   | Every_days of int
 139   | Every_months of int
 140   | Every_years of int
 141 (** Internal type used to represent 'every' expressions. *)
 142
 143 type shell_script = {
 144   sh_loc : Camlp4.PreCast.Loc.t;
 145   sh_script : string;
 146 }
 147 (** A shell script. *)
 148
 149 type variable =
 150   | T_bool of bool
 151   | T_string of string
 152   | T_int of Big_int.big_int
 153   | T_float of float
 154 (** Typed variable (see also [whenproto.x]) *)
 155
 156 val string_of_variable : variable -> string
 157
 158 val variable_of_rpc : Whenproto_aux.variable -> variable
 159 val rpc_of_variable : variable -> Whenproto_aux.variable
 160
 161 type variables = variable StringMap.t
 162 (** A set of variables. *)
 163
 164 type job_private
 165 (** Private state associated with a job, used for evaluation. *)
 166
 167 val no_job_private : job_private
 168 (* XXX any use of no_job_private is wrong XXX *)
 169
 170 type job_cond =
 171   | When_job of whenexpr                (** when ... : << >> *)
 172   | Every_job of periodexpr             (** every ... : << >> *)
 173
 174 type job = {
 175   job_loc : Camlp4.PreCast.Loc.t;
 176   job_name : string;
 177   job_cond : job_cond;
 178   job_script : shell_script;
 179   job_private : job_private;
 180 }
 181 (** A job. *)
 182
 183 val expr_of_ast : Camlp4.PreCast.Ast.Loc.t -> Camlp4.PreCast.Ast.expr -> whenexpr
 184 (** Convert OCaml AST to an expression.  Since OCaml ASTs are much
 185     more general than the expressions we can use, this can raise
 186     [Invalid_argument] in many different situations. *)
 187
 188 val string_of_whenexpr : whenexpr -> string
 189 (** Pretty-print an expression to a string. *)
 190
 191 val string_of_periodexpr : periodexpr -> string
 192 (** Pretty-print a period expression to a string. *)
 193
 194 val dependencies_of_whenexpr : whenexpr -> string list
 195 (** Return list of variables that an expression depends on.  This is
 196     used to work out when an expression needs to be reevaluated. *)
 197
 198 val dependencies_of_job : job -> string list
 199 (** Which variables does this job depend on? *)
 200
 201 val job_evaluate : job -> variables -> bool * job
 202 (** Evaluate [job]'s condition in the context of the [variables], and
 203     return [true] iff it should be run now.  Note that this returns a
 204     possibly-updated [job] structure.
 205
 206     This is a no-op for 'every' jobs. *)
 207
 208 val next_periodexpr : float -> periodexpr -> float
 209 (** [next_periodexpr t period] returns the earliest event of [period]
 210     strictly after time [t].
 211
 212     Visualising periods as repeated events on a timeline, this
 213     returns [t']:
 214
 215     {v
 216     events:  ---+---------+---------+---------+---------+---------+-----
 217     times:          t     t'
 218     }
 219
 220     Note that [periodexpr] events are not necessarily regular.
 221     eg. The start of a month is not a fixed number of seconds
 222     after the start of the previous month.  'Epoch' refers
 223     to the Unix Epoch (ie. 1970-01-01 00:00:00 UTC).
 224
 225     If [period = Every_seconds i] then events are when
 226     [t' mod i == 0] when t' is the number of seconds since
 227     the Epoch.  This returns the next t' > t.
 228
 229     If [period = Every_days i] then events happen at
 230     midnight UTC every [i] days since the Epoch.
 231     This returns the next midnight > t.
 232
 233     If [period = Every_months i] then events happen at
 234     midnight UTC on the 1st day of the month every [i] months
 235     since the Epoch.  This returns midnight on the
 236     1st day of the next month > t.
 237
 238     If [period = Every_years i] then events happen at
 239     midnight UTC on the 1st day of the year when
 240     [(y - 1970) mod i == 0].  This returns midnight on the
 241     1st day of the next year > t. *)