use std::{convert::TryFrom, env};

use matrix_sdk_appservice::{
    matrix_sdk::{
        room::Room,
        ruma::{
            events::{
                room::member::{MemberEventContent, MembershipState},
                SyncStateEvent,
            },
            UserId,
        },
    },
    AppService, AppServiceRegistration, Result,
};
use tracing::trace;

pub async fn handle_room_member(
    appservice: AppService,
    room: Room,
    event: SyncStateEvent<MemberEventContent>,
) -> Result<()> {
    if !appservice.user_id_is_in_namespace(&event.state_key)? {
        trace!("not an appservice user: {}", event.state_key);
    } else if let MembershipState::Invite = event.content.membership {
        let user_id = UserId::try_from(event.state_key.as_str())?;
        appservice.register_virtual_user(user_id.localpart()).await?;

        let client = appservice.virtual_user_client(user_id.localpart()).await?;
        client.join_room_by_id(room.room_id()).await?;
    }

    Ok(())
}

#[tokio::main]
pub async fn main() -> Result<(), Box<dyn std::error::Error>> {
    env::set_var("RUST_LOG", "matrix_sdk=debug,matrix_sdk_appservice=debug");
    tracing_subscriber::fmt::init();

    let homeserver_url = "http://localhost:8008";
    let server_name = "localhost";
    let registration = AppServiceRegistration::try_from_yaml_file("./tests/registration.yaml")?;

    let mut appservice = AppService::new(homeserver_url, server_name, registration).await?;
    appservice
        .register_event_handler({
            let appservice = appservice.clone();
            move |event: SyncStateEvent<MemberEventContent>, room: Room| {
                handle_room_member(appservice.clone(), room, event)
            }
        })
        .await?;

    let (host, port) = appservice.registration().get_host_and_port()?;
    appservice.run(host, port).await?;

    Ok(())
}