blob: aa2e2e7f06d40f6ebe192ed84f3e0c6bf6cde2fc (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
|
open Lwt.Syntax
open Markup
exception ClosedStream
exception InvalidStanza of string
type auth_mechanism =
| PLAIN
| Unknown of string [@@deriving show { with_path = false }]
let parse_auth_mechanism = function
| "PLAIN" -> PLAIN
| other -> Unknown other
type stream_features = {
sasl_mechanisms : auth_mechanism list;
starttls : [`Required | `Optional | `None];
unknown_features : Xml.element list;
}
let get (stream : (signal, async) stream) : (signal, sync) stream Lwt.t =
(** [stanza stream] is a promise containing a full stanza of the fragments of
[stream]. *)
let traverse_stanza depth fragment =
let depth = match fragment with
| `Start_element _ -> depth + 1
| `End_element -> depth - 1
| _ -> depth
in ([fragment], if depth = 0 then None else Some depth)
in transform traverse_stanza 0 stream |> Markup_lwt.load
let start domain : Portal.t Lwt.t =
(** [start domain] is a promise containing a Portal (stream * push) connected to the
XMPP server [domain].
Currently, it doesn't handle anything except the initial [<open/>] stanza. *)
let* stream, _push = Portal.connect domain
in let push = function
| None -> _push (Some Portal.stanza_close);
_push None;
| anything -> _push anything
in Some (Portal.stanza_open domain) |> push;
(* TODO: check this is a good stanza *)
let+ _ = get stream
in stream, push
let parse_features (el : Xml.element) : stream_features =
(** [parse_features el] is a [stream_features] record with all the features of the
[<stream:features>] stanza contained in [el]. *)
let open Xml in
let open Either in
let parse_mechanism_stanza = function
| Left {local_name = "mechanism"; children = [Right mechanism]; _} ->
Some (parse_auth_mechanism mechanism)
| _ -> None
in
let parse_feature (acc : stream_features) (feature : Xml.element) : stream_features =
let parse_mechanisms ch =
List.filter_map parse_mechanism_stanza ch
and parse_starttls = function
| [Left {local_name="required"; _}] -> `Required
| [] -> `Optional
| _ -> raise (InvalidStanza (element_to_string el))
in match feature.local_name with
| "mechanisms" -> {acc with sasl_mechanisms=parse_mechanisms feature.children}
| "starttls" -> {acc with starttls=parse_starttls feature.children}
| _ -> {acc with unknown_features = feature :: acc.unknown_features}
in List.fold_left
parse_feature
{sasl_mechanisms=[]; starttls=`None; unknown_features=[]}
(List.filter_map find_left el.children)
|
