diff options
Diffstat (limited to 'test/js')
| -rw-r--r-- | test/js/websockets_hello.ml | 26 |
1 files changed, 11 insertions, 15 deletions
diff --git a/test/js/websockets_hello.ml b/test/js/websockets_hello.ml index 316ae5e..877e543 100644 --- a/test/js/websockets_hello.ml +++ b/test/js/websockets_hello.ml @@ -17,11 +17,9 @@ let rec run t = then next_tick (fun () -> run t) else () -let main (stream, push) localpart password = +let main (stream, push) config = let* _stream = Lwt_stream.get stream - in let+ _auth = Sasl.send_auth_stanza (stream, push) - localpart password - Sasl.PLAIN + in let+ _auth = Sasl.authenticate (stream, push) config in match _auth with | Error (NotAuthorized, Some (_, text)) -> print_endline ("Not authorized: " ^ text) | Error (MalformedRequest, Some (_, text)) -> print_endline ("Malformed request: " ^ text) @@ -30,15 +28,13 @@ let main (stream, push) localpart password = let () = run @@ - let jid = (Sys.getenv "EXAMPLE_JID") - and password = (Sys.getenv "EXAMPLE_PASSWORD") - in - (* Probably not exactly compliant with https://xmpp.org/extensions/xep-0029.html, - but it's just for simplicity's sake in the testing. *) - let domain = (List.nth (String.split_on_char '@' jid) 1) - and localpart = (List.nth (String.split_on_char '@' jid) 0) - in - let* stream, push = Stream.start domain - in Lwt.catch - (fun () -> main (stream, push) localpart password >|= (fun () -> push None)) + let config : Sasl.auth_config = { + jid = (Sys.getenv "EXAMPLE_JID"); + password = (Sys.getenv "EXAMPLE_PASSWORD"); + preferred_mechanisms = [] + } + in let domain = (List.nth (String.split_on_char '@' config.jid) 1) in + let* stream, push = Stream.start domain in + Lwt.catch + (fun () -> main (stream, push) config >|= (fun () -> push None)) (fun exn -> push None; Lwt.fail exn) |