43 lines
1.3 KiB
OCaml
43 lines
1.3 KiB
OCaml
open Cohttp_lwt_unix
|
|
open Utils
|
|
|
|
let driver_url = "http://127.0.0.1:4444/session"
|
|
|
|
let send_post_request url json =
|
|
let headers = Cohttp.Header.init_with "Content-Type" "application/json" in
|
|
let body = Cohttp_lwt.Body.of_string json in
|
|
Lwt.bind
|
|
(Client.post ~headers ~body (Uri.of_string url))
|
|
(fun (_response, body) -> Cohttp_lwt.Body.to_string body)
|
|
;;
|
|
|
|
let send_delete_request url =
|
|
Lwt.bind
|
|
(Client.delete (Uri.of_string url))
|
|
(fun (_response, body) -> Cohttp_lwt.Body.to_string body)
|
|
;;
|
|
|
|
let execute_request url json = Lwt_main.run (send_post_request url json)
|
|
|
|
(* Server MUST be started already *)
|
|
let get_session () =
|
|
let response = execute_request (fmt "%s" driver_url) Json.connection_payload in
|
|
match Yojson.Safe.from_string response with
|
|
| `Assoc fields ->
|
|
let value = List.assoc "value" fields in
|
|
let rec find_session_id = function
|
|
| ("sessionId", `String session_id) :: _ -> session_id
|
|
| _ :: rest -> find_session_id rest
|
|
| [] -> failwith "Session ID not found"
|
|
in
|
|
find_session_id (Yojson.Safe.Util.to_assoc value)
|
|
| _ -> failwith "Invalid JSON"
|
|
;;
|
|
|
|
let close_session id =
|
|
Lwt_main.run (send_delete_request (fmt "%s/%s" driver_url id)) = "{\"value\":null}"
|
|
;;
|
|
|
|
let navigate url session_id =
|
|
execute_request (fmt "%s/%s/url" driver_url session_id) (Json.navigate_payload url)
|
|
;;
|