open Pusk.Net open Pusk.Utils type credentials = { username : string ; password : string } let inject_username session_id creds = (* Find username input *) let strat = CSS "input[name='text']" in let input_username = match find session_id strat with | [] -> raise (Any (fmt "Username input not found")) | _ as l -> if List.length l > 1 then raise (Any "Too many elements found as the username input") else List.nth l 0 in (* Insert the username *) send_keys session_id input_username creds.username; Unix.sleep 1; send_keys session_id input_username Keys.return; Unix.sleep 3 ;; let rec _inject_password session_id creds try_count = if try_count == 0 then raise (Any "Password input not found"); let input_password = match find session_id (CSS "input[name='password']") with | [] -> (* Retry to inject username with the second page *) (* NOTE: I think this is only when a email is first sent, so the second * injection NEED to be a username (or phone number) * * Maybe allow user to provides us with username and email and use email * by default and fallback to username in the second attempt *) inject_username session_id creds; _inject_password session_id creds (try_count - 1); None | _ as l -> if List.length l > 1 then raise (Any "Too many elements found as the password input") else Some (List.nth l 0) in match input_password with | Some input -> (* Insert password *) send_keys session_id input creds.password; Unix.sleep 1; send_keys session_id input Keys.return; Unix.sleep 3 | None -> () ;; let inject_password session_id creds = _inject_password session_id creds 1 let inject_2fa session_id secret input = let code = match secret with | Some seed -> Twostep.TOTP.code ~secret:seed () | None -> raise (Any "No TOTP code given, but 2FA code required") in (* Insert 2FA code *) send_keys session_id input code; Unix.sleep 1; send_keys session_id input Keys.return; Unix.sleep 5 ;; let login_twitter ctx username password secret = if ctx.debug then print_endline "Login to twitter..."; (* Navigate to login page and wait for page loaded*) ignore (navigate ctx.session_id "https://twitter.com/i/flow/login"); Unix.sleep 5; let creds = { username; password } in (* Insert the username *) if ctx.debug then print_endline "Type username..."; inject_username ctx.session_id creds; (* Find password input *) if ctx.debug then print_endline "Type password..."; inject_password ctx.session_id creds; (* Detection and injection of 2FA code if needed *) match find ctx.session_id (CSS "input[name='text']") with | [] -> print_endline "Doesn't use 2FA as no input found" | _ as l -> if List.length l > 1 then raise (Any "Too many elements found as 2FA input") else ( if ctx.debug then print_endline "Type 2FA code..."; inject_2fa ctx.session_id secret (List.nth l 0)) ;; let go_to_profile ctx = if ctx.debug then print_endline "Locate profile button..."; let profile_button = match find ctx.session_id (XPath "//a[@data-testid='AppTabBar_Profile_Link']") with | [] -> raise (Any (fmt "Profile button not found")) | _ as l -> if List.length l > 1 then raise (Any "Too many profile button found") else List.nth l 0 in if ctx.debug then print_endline "Navigate to user replies..."; ignore (navigate ctx.session_id (fmt "https://twitter.com%s/with_replies" (get_attribute ctx.session_id profile_button "href"))); Unix.sleep 4 ;; let find_latest_tweet ctx = match find ctx.session_id (XPath "//article[@data-testid='tweet']") with | [] -> None | _ as tweets -> (* Get dates attached to each tweets *) let dates = (* When a tweet is a RT, two dates are attached *) List.flatten (List.map (fun tweet -> match find_in_element ctx.session_id (CSS "time[datetime]") tweet with | [] -> raise (Any (fmt "No dates found for tweet '%s'" tweet)) | _ as l -> l) tweets) in (* Turn datetime from ISO 8601 format to epoch int *) let datetimes = List.map (fun date -> let time = Core.Time.of_string_with_utc_offset (get_attribute ctx.session_id date "datetime") in Float.to_int (Core.Time.Span.to_sec (Core.Time.to_span_since_epoch time))) dates in (* Returns the most recent date *) Some (List.fold_left max min_int datetimes) ;; let tweet ctx msg = let tweet_button = match find ctx.session_id (XPath "//a[@data-testid='SideNav_NewTweet_Button']") with | [] -> raise (Any (fmt "Tweet button not found")) | _ as l -> if List.length l > 1 then raise (Any "Too many tweet button found") else List.nth l 0 in click ctx.session_id tweet_button; let tweet_area = match find ctx.session_id (XPath "//div[@data-testid='tweetTextarea_0']") with | [] -> raise (Any (fmt "Tweet button not found")) | _ as l -> if List.length l > 1 then raise (Any "Too many tweet button found") else List.nth l 0 in send_keys ctx.session_id tweet_area msg; Unix.sleep 1; let send_tweet_button = match find ctx.session_id (XPath "//div[@data-testid='tweetButton']") with | [] -> raise (Any (fmt "Send-tweet button not found")) | _ as l -> if List.length l > 1 then raise (Any "Too many send-tweet button found") else List.nth l 0 in click ctx.session_id send_tweet_button; Unix.sleep 4 ;;