use std::{env, io, process::exit};
use url::Url;

use matrix_sdk::{self, events::AnyToDeviceEvent, Client, ClientConfig, SyncSettings};

async fn login(
    homeserver_url: String,
    username: String,
    password: String,
) -> Result<(), matrix_sdk::Error> {
    let client_config = ClientConfig::new()
        .proxy("http://localhost:8080")?
        .disable_ssl_verification();
    let homeserver_url = Url::parse(&homeserver_url).expect("Couldn't parse the homeserver URL");
    let client = Client::new_with_config(homeserver_url, client_config).unwrap();

    client
        .login(username, password, None, Some("rust-sdk".to_string()))
        .await?;

    let client_ref = &client;

    client
        .sync_forever(SyncSettings::new(), |response| async move {
            let client = &client_ref;

            for event in &response.to_device.events {
                let e = event
                    .deserialize()
                    .expect("Can't deserialize to-device event");

                println!("Received to-device event {:?}", e);
                match e {
                    AnyToDeviceEvent::KeyVerificationStart(e) => {
                        let sas = client
                            .get_verification(&e.content.transaction_id)
                            .await
                            .expect("Sas object wasn't created");
                        sas.accept().await.unwrap();
                    }
                    AnyToDeviceEvent::KeyVerificationKey(e) => {
                        let sas = client
                            .get_verification(&e.content.transaction_id)
                            .await
                            .expect("Sas object wasn't created");
                        println!("Emoji: {:?}", sas.emoji());

                        let mut input = String::new();
                        io::stdin()
                            .read_line(&mut input)
                            .expect("error: unable to read user input");

                        match input.trim().to_lowercase().as_ref() {
                            "yes" | "true" | "ok" => sas.confirm().await.unwrap(),
                            _ => sas.cancel().await.unwrap(),
                        }
                    }
                    _ => (),
                }
            }
        })
        .await;

    Ok(())
}

#[tokio::main]
async fn main() -> Result<(), matrix_sdk::Error> {
    tracing_subscriber::fmt::init();

    let (homeserver_url, username, password) =
        match (env::args().nth(1), env::args().nth(2), env::args().nth(3)) {
            (Some(a), Some(b), Some(c)) => (a, b, c),
            _ => {
                eprintln!(
                    "Usage: {} <homeserver_url> <username> <password>",
                    env::args().next().unwrap()
                );
                exit(1)
            }
        };

    login(homeserver_url, username, password).await
}