aly / main.rs

#!/usr/bin/env python3 from pyotp import TOTP import requests from traceback import print_exc while True: try: code = TOTP("aVeryLongSecretCode").now() requests.get("http://localhost:4646/ffxivlauncher/" + code, timeout=10) break except requests.exceptions.ConnectionError: print_exc()

use std::time::Duration; use log::{error, info, warn, debug}; use reqwest::Client; use totp_rs::{Algorithm, TOTP, Secret}; use clap::Parser; const TOTP_DIGITS: usize = 6; const TOTP_STEP: u64 = 30; const TOTP_ALGORITHM: Algorithm = Algorithm::SHA1; #[derive(Parser, Debug)] #[command(version, about, long_about = None)] struct Args { secret: String, } #[tokio::main] async fn main() -> Result<(), Box<dyn std::error::Error>> { // Initialize logging env_logger::builder() .filter_level(log::LevelFilter::Info) .format_target(false) .format_timestamp(None) .init(); let args: Args = Args::parse(); // Create the TOTP instance let totp: TOTP = TOTP::new( TOTP_ALGORITHM, TOTP_DIGITS, 1, TOTP_STEP, Secret::Encoded(args.secret).to_bytes()?, )?; // Generate the current TOTP code let mut token: String = totp.generate_current()?; info!("Generated TOTP token: {}", token); // Send the TOTP code to the local service let client: Client = Client::new(); loop { match totp.check_current(&token) { Ok(true) => { debug!("TOTP token is valid."); } Ok(false) => { info!("TOTP token is invalid. Regenerating..."); token = totp.generate_current()?; } Err(e) => { error!("Error checking TOTP token: {}", e); } } match client .get(&format!("http://localhost:4646/ffxivlauncher/{}", token)) .timeout(Duration::from_secs(5)) .send() .await { Ok(response ) => { if response.status().is_success() { info!("TOTP token sent successfully."); break; } else { warn!("Failed to send TOTP token. Status: {}", response.status()); } } Err(e) => { if e.is_connect() { info!("Connection refused. Retrying..."); } else { error!("Unexpected error: {}", e); } } } } Ok(()) }