pusk/bin/main.ml

48 lines
1.2 KiB
OCaml
Raw Normal View History

open Pusk.Net
open Pusk.Drivers
open Pusk.Utils
2023-05-13 19:58:48 +02:00
open Twitter
2023-05-13 13:06:15 +02:00
let start driver =
let name_driver = prepare driver in
let data_driver = run_process name_driver [] in
2023-05-14 01:44:11 +02:00
let session_id = get_session ~headless:false () in
2023-05-13 13:06:15 +02:00
data_driver, session_id
;;
2023-05-12 20:30:02 +02:00
2023-05-13 13:06:15 +02:00
let stop (driver_process, session_id) =
2023-05-13 13:33:46 +02:00
if not (close_session session_id) then print_endline "Can't close the session";
2023-05-13 11:59:09 +02:00
stop_process driver_process
2023-05-12 19:36:27 +02:00
;;
2023-05-13 13:06:15 +02:00
2023-05-13 19:58:48 +02:00
let main ctx =
(* Load credentials *)
load_dotenv;
let username, password =
match Sys.getenv_opt "TWITTER_USERNAME", Sys.getenv_opt "TWITTER_PASSWORD" with
| Some u, Some p -> u, p
| None, None -> raise (Any "Username and password not set")
| None, Some _ -> raise (Any "Username not set")
| Some _, None -> raise (Any "Password not set")
in
2023-05-15 11:44:21 +02:00
login_twitter ctx username password (Sys.getenv_opt "TWITTER_TOTP");
go_to_profile ctx
2023-05-13 16:17:18 +02:00
;;
2023-05-13 13:06:15 +02:00
2023-05-15 11:02:56 +02:00
let handler data (signal : int) =
stop_process data;
exit
(match signal with
| v when v = Sys.sigint -> 130
| _ -> 1)
;;
2023-05-13 13:06:15 +02:00
let () =
let data = start (Gecko "0.33.0") in
2023-05-15 11:02:56 +02:00
Sys.set_signal Sys.sigint (Sys.Signal_handle (handler (fst data)));
2023-05-13 19:58:48 +02:00
let ctx = { session_id = snd data } in
(try main ctx with
| Any why -> print_endline why);
2023-05-13 13:06:15 +02:00
stop data
;;