feat(session): move connect from flesh to session

1 files changed, 30 insertions, 0 deletions

diff --git a/lib/session.ml b/lib/session.ml
index c640080..1b591b5 100644
--- a/lib/session.ml
+++ b/lib/session.ml
@@ -1,2 +1,32 @@
+open Lwt.Syntax
+open Lwt.Infix
+
 type state =
   | Disconnected
+
+(** [connect domain config] is a promise containing the portal connected to the XMPP
+    server located at [domain], and all its supported features.
+
+    Basically, it conforms to {{:
+    https://datatracker.ietf.org/doc/html/rfc6120#section-4.3 }}, and gets the provided
+    Portal in a "ready" state. *)
+let connect (domain : string) (config : Stream.config) : (Portal.t * Stream.features) Lwt.t =
+  let open Portal in
+  let open Stream in
+  let* portal = connect domain
+  in
+  let needs_restart = function
+    | Feature.Mechanisms _ | STARTTLS -> true
+    | _ -> false
+  in
+  let+ features =
+    let rec handle_features (features : Stream.features) : Stream.features Lwt.t =
+      match features with
+      | feature :: rest ->
+         let* () = negotiate feature portal config
+         in if needs_restart (Feature.unwrap feature)
+            then start domain portal >>= handle_features
+            else handle_features rest
+      | features -> Lwt.return features
+    in start domain portal >>= handle_features
+  in (portal, features)