open! Lwt.Syntax
open! Lwt.Infix
open! Flesh
open! Session
open! Lwt_react

let main =
  let config : Stream.config = {
      starttls = {prefer = true};
      sasl = {
          jid = Jid.of_string (Sys.getenv "FLESH_JID");
          password = (Sys.getenv "FLESH_PASSWORD");
          preferred_mechanisms = [Sasl.PLAIN]
        };
      other = [];
    }
  in
  let waiter, wakener = Lwt.wait () in
  try%lwt let* { state; update } = create config in
             S.map (function
                 | Connected (portal, Logged_in _) -> portal.push None;
                                                      Lwt.wakeup wakener ()
                 | _ -> ())
               state |> S.keep;
             connect { state; update } config.sasl.jid.domainpart;
             waiter
  with exn ->
    begin
      (* I suspect JavaScript's [wrap_callback] swallows the Exceptions thrown by
         OCaml, so... The next best thing is probably printing something. *)
      print_endline
        (match exn with
         | Xml.InvalidStanza stanza -> "Invalid stanza: " ^ stanza
         | Portal.MalformedStanza err -> "Server sent malformed stanza: " ^ (Markup.Error.to_string err)
         | _ -> "... and so I stumble back to bed.");
      Lwt.fail exn
    end