#![warn(missing_docs)]

//! In case you and your friends ever felt like you don't have enough
//! Slavoj Žižek in your life, this matrix bot can help you satisfying that
//! need.

use matrix_sdk::room::Room;
use matrix_sdk::events::room::member::MemberEventContent;
use matrix_sdk::events::StrippedStateEvent;
use matrix_sdk::{async_trait, Client, ClientConfig, EventHandler, SyncSettings};
use tokio::time::{sleep, Duration};
use url::Url;

pub mod bot_config;
pub mod zizek_db;

use bot_config::BotConfig;
use zizek_db::ZizekDb;

struct AutoJoinBot
{
    client: Client,
}

impl AutoJoinBot
{
    pub fn new(client: Client) -> Self { Self { client } }
}

#[async_trait]
impl EventHandler for AutoJoinBot {
    async fn on_stripped_state_member(&self, room: Room, room_member: &StrippedStateEvent<MemberEventContent>, _: Option<MemberEventContent>){
        /* If the event does not concern the bot user, ignore it */
        if room_member.state_key != self.client.user_id().await.unwrap() {
            return;
        }

        if let Room::Invited(room) = room {
          println!("Autojoining room {}", room.room_id());
            let mut delay = 2;

            while let Err(err) = room.accept_invitation().await {
                // retry autojoin due to synapse sending invites, before the
                // invited user can join for more information see
                // https://github.com/matrix-org/synapse/issues/4345
                eprintln!(
                    "Failed to join room {} ({:?}), retrying in {}s",
                    room.room_id(),
                    err,
                    delay
                );

                sleep(Duration::from_secs(delay)).await;
                delay *= 2;

                if delay > 3600 {
                    eprintln!("Can't join room {} ({:?})", room.room_id(), err);
                    break;
                }
            }
            println!("Successfully joined room {}", room.room_id());
        }
    }
}

async fn login_and_sync(homeserver_url: &str, username: &str, password: &str) -> Result<(), matrix_sdk::Error> {
    let mut config_dir = dirs::home_dir().expect("No home directory found");
    config_dir.push("zizek_bot");
    let config = ClientConfig::new().store_path(config_dir);


        let homeserver_url = Url::parse(&homeserver_url).expect("Couldn't parse the homeserver URL");
    let client = Client::new_with_config(homeserver_url, config).unwrap();

    client.login(username, password, None, Some("Slavoj Žižek")).await?;
    println!("logged in as {}", username);

    client.set_event_handler(Box::new(AutoJoinBot::new(client.clone()))).await;

    client.sync(SyncSettings::default()).await;

    Ok(())
}

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

    let bot_config = BotConfig::from_config_file().expect("Unable to read bot configuration file");
    let zizek_db =
        ZizekDb::from_file("zizek_quotes.db").expect("Unable to load zizek quote database");

    login_and_sync(bot_config.homeserver_url(), bot_config.username(), bot_config.password()).await?;

    Ok(())
}