glical_bin.ml

(* -*- coding: utf-8; -*- *)
(* ********************************************************************* *)
(* glical: A library to glance at iCal data using OCaml                  *)
(* (c) 2013/2014, Philippe Wang <philippe.wang@cl.cam.ac.uk>             *)
(* Licence: ISC                                                          *)
(* ********************************************************************* *)

open Printf
open Glical
open Ical

let ical = ref true
let inputs = ref []
let template = ref None
let out = ref stdout
let filters =
  object
    val mutable s = SSet.empty
    method add x = s <- SSet.add x s
    method mem x = SSet.mem x s
    method is_empty = s = SSet.empty
    method get_filter = s
  end

let values_only = ref false

let assert_ a m =
  if not a then
    begin
      eprintf "Error: %s\n%!" m;
      exit 1
    end

let wrap f x =
  try f x
  with
  | Sys_error(s) ->
    assert_ false s;
    assert false
  | e ->
    assert_ false (Printexc.to_string e);
    assert false

let () =
  Arg.(
    parse
    [
      ("-input",
       String(wrap(fun s -> inputs := open_in_bin s :: !inputs)),
       "f  Use/add f as an input");
      ("-ical",
       Set(ical),
       "    Output in iCalendar format (this option has no effect at all for the moment)");
      ("-tpl",
       String(wrap(fun s -> template := Some(open_in_bin s))),
       "t    Output according to the template t (not yet implemented)");
      ("-o",
       String(wrap(fun s -> out := open_out_bin s)),
       "f      Output to file f (default is stdout)");
      ("-f",
       String(filters#add),
       "l      Specify label l that should be kept. \
        Use multiple -F for multiple labels. If no label is given, all are kept.");
      ("-F",
       String(fun s -> values_only := true; filters#add s),
       "l      Same as -f but only values will printed. \
       If at least one -F is used, all other -f will behave like -F.");
    ]
    (wrap(fun s -> inputs := open_in_bin s :: !inputs))
    "glical takes some iCalendar data and allows you to play a little with it"
  )

let _ =
  assert_ (List.filter (fun x -> x)
             [!template <> None; !ical; ]
           = [true])
    "You can only have one of -ical and -tpl";
  if !inputs = [] then
    inputs := [stdin];
  ()

let _ =
  if !template <> None then
    assert_ false "Not yet implemented.";
  let data =
    let b = Buffer.create 42 in
    List.iter (fun i -> Buffer.add_string b (channel_contents i)) !inputs;
    parse_ical(Lexing.lex_ical(Buffer.contents b))
  in
  if !values_only then
    let data =
      Glical.extract_values ~ks:(filters#get_filter) data
    in
    List.iter (fun e ->
        fprintf
          !out
          "%s\n"
          (e.to_string())
      )
      data
  else
    let data =
      if filters#is_empty then
        data
      else
        Glical.extract_assocs ~ks:(filters#get_filter) data
    in
    fprintf
      !out
      "%s"
      (to_string ~f:(fun _ -> None) data)












(* ********************************************************************* *)
(* Permission to use, copy, modify, and/or distribute this software
   for any purpose with or without fee is hereby granted, provided
   that the above copyright notice and this permission notice appear
   in all copies.

   THE SOFTWARE IS PROVIDED “AS IS” AND ISC DISCLAIMS ALL WARRANTIES
   WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
   MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL ISC BE LIABLE FOR
   ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY
   DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS,
   WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS
   ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE
   OF THIS SOFTWARE.                                                     *)
(* ********************************************************************* *)