(* * 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 . *) open CalendarLib open Lwt.Infix open Cohttp open Cohttp_lwt_unix let format_date = Printer.Date.sprint "%Y-%m-%d" and a_month = Date.Period.make 0 1 0 and a_year = Date.Period.make 1 0 0 let fetch celcat_url group = let current_date = Date.today () in let lower_date = format_date @@ Date.rem current_date a_month and upper_date = format_date @@ Date.add current_date a_year in let parameters = Uri.encoded_of_query ["start", [lower_date]; "end", [upper_date]; "resType", ["103"]; "calView", ["month"]; "federationIds[]", [group]; "colourScheme", ["3"]] in let body = Cohttp_lwt.Body.of_string parameters and headers = Header.init_with "Content-Type" "application/x-www-form-urlencoded" in Client.post ~body ~headers celcat_url >>= fun (_resp, body) -> Cohttp_lwt.Body.to_string body let log uri = let datetime = Printer.Calendar.sprint "%Y-%m-%d %H:%M:%S" @@ Calendar.now () in Lwt_io.printlf "[%s] %s" datetime (Uri.path_and_query uri) let respond ?(mime="text/html; charset=utf-8") ?(status=`OK) body = let headers = Header.init_with "Content-Type" mime in Server.respond_string ~status ~body ~headers () let get_tz () = Lwt_unix.file_exists "/etc/timezone" >>= fun exists -> if exists then Lwt_io.(open_file ~mode:Input "/etc/timezone") >>= fun file -> Lwt.finalize (fun () -> Lwt_io.read file >|= String.trim) (fun () -> Lwt_io.close file) else Lwt.return "" let serve base_url celcat_url mode stop = get_tz () >>= fun tzname -> let fetch = fetch celcat_url in let callback _conn req _body = let meth = Request.meth req and uri = Request.uri req in let path = Uri.path uri |> Astring.String.cuts ~empty:false ~sep:"/" |> List.map Uri.pct_decode and query = Uri.query uri in log uri >>= fun () -> match meth, path, query with | `GET, ([] | ["index.html"]), [] -> respond Pages.main | `GET, ["lnk"], ["group", [group]] -> let lnk = base_url ^ "/ics/" ^ group ^ ".ics" in respond @@ Pages.link lnk | `GET, ["lnk"], _ -> Server.respond_string ~status:`Bad_request ~body:"Bad request\n" () | `GET, ["ics"; file], [] when Astring.String.is_suffix ~affix:".ics" file -> let group = String.(sub file 0 (length file - 4)) in fetch group >>= fun body -> Course.decode body |> Ics.to_string tzname |> respond ~mime:"text/calendar; charset=utf-8" | `GET, _, _ -> Server.respond_string ~status:`Not_found ~body:"Not found\n" () | _ -> Server.respond_string ~status:`Method_not_allowed ~body:"Method not allowed\n" () in Server.create ~stop ~mode (Server.make ~callback ())