download gecko

bin/main.ml

@@ -41,4 +41,47 @@ let download_selenium version =
   download (Uri.of_string (fmt url version version)) (fmt "./selenium-%s.jar" version)
 ;;
 
-let () = Lwt_main.run (download_selenium "4.9.0")
+let download_gecko_driver version output =
+  let url =
+    format_of_string
+      "https://github.com/mozilla/geckodriver/releases/download/v%s/geckodriver-v%s-linux64.tar.gz"
+  in
+  download (Uri.of_string (fmt url version version)) output
+;;
+
+let run_program_in_background program args =
+  let pid = Unix.fork () in
+  match pid with
+  | 0 ->
+    (* Child process *)
+    let dev_null = Unix.openfile "/dev/null" [ O_WRONLY ] 0o666 in
+    Unix.dup2 dev_null Unix.stdout;
+    Unix.dup2 dev_null Unix.stderr;
+    Unix.close dev_null;
+    Unix.execvp program (Array.of_list (program :: args))
+  | _ -> pid (* Parent process *)
+;;
+
+let () =
+  (* Selenium *)
+  let version_selenium = "4.9.0" in
+  let selenium = fmt "selenium-%s.jar" version_selenium in
+  if not (Sys.file_exists selenium) then Lwt_main.run (download_selenium version_selenium);
+  (* Gecko Driver *)
+  let driver = fmt "geckodriver" in
+  if not (Sys.file_exists driver)
+  then (
+    let version_driver = "0.33.0" in
+    let archive = fmt "./gecko-%s.tar.gz" version_driver in
+    Lwt_main.run (download_gecko_driver version_driver archive);
+    let _ = Sys.command (fmt "tar xvzf %s" archive) in
+    ());
+  let selenium_pid =
+    run_program_in_background "java" [ fmt "-jar %s" selenium; "standalone" ]
+  in
+  print_endline (fmt "Java running in %d" selenium_pid);
+  let closed_pid, _ = Unix.waitpid [] selenium_pid in
+  if not (closed_pid = selenium_pid)
+  then print_endline "WTF???"
+  else print_endline (fmt "Program %d closed!" closed_pid)
+;;