use std::{env, process::exit, time::Duration};

use futures_util::stream::{StreamExt as _, TryStreamExt as _};
use http::Uri;
use ruma::{
    events::{
        room::message::{MessageEventContent, TextMessageEventContent},
        AnySyncMessageEvent, AnySyncRoomEvent, SyncMessageEvent,
    },
    presence::PresenceState,
};
use ruma_client::{self, HttpClient};

async fn log_messages(homeserver_url: Uri, username: &str, password: &str) -> anyhow::Result<()> {
    let client = HttpClient::new(homeserver_url, None);

    client.log_in(username, password, None, None).await?;

    let mut sync_stream = Box::pin(
        client
            .sync(None, None, PresenceState::Online, Some(Duration::from_secs(30)))
            // TODO: This is a horrible way to obtain an initial next_batch token that generates way
            //       too much server load and network traffic. Fix this!
            .skip(1),
    );

    while let Some(res) = sync_stream.try_next().await? {
        // Only look at rooms the user hasn't left yet
        for (room_id, room) in res.rooms.join {
            for event in room.timeline.events.into_iter().flat_map(|r| r.deserialize()) {
                // Filter out the text messages
                if let AnySyncRoomEvent::Message(AnySyncMessageEvent::RoomMessage(
                    SyncMessageEvent {
                        content:
                            MessageEventContent::Text(TextMessageEventContent {
                                body: msg_body, ..
                            }),
                        sender,
                        ..
                    },
                )) = event
                {
                    println!("{:?} in {:?}: {}", sender, room_id, msg_body);
                }
            }
        }
    }

    Ok(())
}

#[tokio::main]
async fn main() -> anyhow::Result<()> {
    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)
            }
        };

    let server = homeserver_url.parse()?;
    log_messages(server, &username, &password).await
}