(*
 *    Copyright (C) 2020  Alban Gruin
 *
 *    ucs is free software: you can redistribute it and/or modify
 *    it under the terms of the GNU Affero General Public License as published
 *    by the Free Software Foundation, either version 3 of the License, or
 *    (at your option) any later version.
 *
 *    ucs is distributed in the hope that it will be useful,
 *    but WITHOUT ANY WARRANTY; without even the implied warranty of
 *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *    GNU Affero General Public License for more details.
 *
 *    You should have received a copy of the GNU Affero General Public License
 *    along with ucs.  If not, see <http://www.gnu.org/licenses/>.
 *)

open CalendarLib

module J = Json_encoding

let get_unicode v =
  let b = Buffer.create 1 in
  Buffer.add_utf_8_uchar b (Uchar.of_int v);
  Buffer.contents b

let html_entities_regex = Re.Perl.compile_pat "&#(\\d+);"

let replace_entities str =
  Re.Pcre.full_split ~rex:html_entities_regex str
  |> List.filter_map (function
         | Re.Pcre.Group (_, v) ->
            Some ("&#" ^ v ^ ";", get_unicode @@ int_of_string v)
         | _ -> None)
  |> Stringext.replace_all_assoc str

let check_groups str =
  let group_affixes = ["MAT-Agreg Interne "; "3EME ANNEE "; "2EME ANNEE ";
                       "1ERE ANNEE "; "MAG1 "; "DEUST "; "M2 "; "M1 "; "L3P ";
                       "L3 "; "L2 "; "L1 "] in
  List.fold_left
    (fun res affix ->
      res || Astring.String.is_prefix ~affix str) false group_affixes

let location_and_summary str category =
  let sep = "\r\n\r\n<br />\r\n\r\n" in
  let parts = Astring.String.cuts ~empty:false ~sep str in
  let _, location, summary =
    List.fold_right
      (fun str (has_groups, location, summary) ->
        if not has_groups then
          check_groups str, "", ""
        else if location = "" then
          true, replace_entities str, ""
        else if summary = "" then
          true, replace_entities str, location
        else
          true, location, summary) parts (false, "", "") in
  if summary = "" then
    location, category
  else
    let summary = match Astring.String.cut ~sep:" - " summary with
      | None -> str
      | Some (_, str) -> str in
    location, summary ^ " (" ^ category ^ ")"

let date =
  let date_format = "%FT%T" in
  Printer.Calendar.(
    J.(conv (sprint date_format) (from_fstring date_format) string))

let encoding =
  J.(conv
       (fun _ -> (None, None, "", "", (), (), ()),
                 ((), (), (), (), (), (), (), (), (), ()))
       (fun ((start, stop, description, category, (), (), ()),
             ((), (), (), (), (), (), (), (), (), ())) ->
         let location, summary = location_and_summary description category in
         match start, stop with
         | Some start, Some stop ->
            Some (Ics.Event.make start stop summary location)
         | _, _ -> None)
       (merge_objs
          (obj7
             (req "start" @@ option date)
             (req "end" @@ option date)
             (req "description" string)
             (req "eventCategory" string)
             (req "id" unit)
             (req "allDay" unit)
             (req "backgroundColor" unit))
          (obj10
             (req "textColor" unit)
             (req "department" unit)
             (req "faculty" unit)
             (req "sites" unit)
             (req "modules" unit)
             (req "registerStatus" unit)
             (req "studentMark" unit)
             (req "custom1" unit)
             (req "custom2" unit)
             (req "custom3" unit))))

let decode s =
  let toks =
    match s with
    | "" -> `O []
    | s -> Ezjsonm.from_string s in
  J.(destruct (list encoding) toks)
  |> List.filter_map (fun event -> event)
  |> Ics.make