use crate::commands::execute_command; use crate::config::*; use crate::context::Context; use crate::db::Database; use crate::error::BotError; use crate::state::DiceBotState; use dirs; use log::{error, info}; use matrix_sdk::Error as MatrixError; use matrix_sdk::{ self, events::{ room::message::{MessageEventContent, NoticeMessageEventContent}, AnyMessageEventContent, }, Client, ClientConfig, JsonStore, Room, SyncSettings, }; //use matrix_sdk_common_macros::async_trait; use std::clone::Clone; use std::path::PathBuf; use std::sync::{Arc, RwLock}; use url::Url; pub mod event_handlers; /// The DiceBot struct represents an active dice bot. The bot is not /// connected to Matrix until its run() function is called. pub struct DiceBot { /// A reference to the configuration read in on application start. config: Arc, /// The matrix client. client: Client, /// State of the dicebot state: Arc>, /// Active database layer db: Database, } fn cache_dir() -> Result { let mut dir = dirs::cache_dir().ok_or(BotError::NoCacheDirectoryError)?; dir.push("matrix-dicebot"); Ok(dir) } /// Creates the matrix client. fn create_client(config: &Config) -> Result { let cache_dir = cache_dir()?; let store = JsonStore::open(&cache_dir)?; let client_config = ClientConfig::new().state_store(Box::new(store)); let homeserver_url = Url::parse(&config.matrix_homeserver())?; Ok(Client::new_with_config(homeserver_url, client_config)?) } /// Extracts more detailed error messages out of a matrix SDK error. fn extract_error_message(error: MatrixError) -> String { use matrix_sdk::Error::RumaResponse; match error { RumaResponse(ruma_error) => ruma_error.to_string(), _ => error.to_string(), } } impl DiceBot { /// Create a new dicebot with the given configuration and state /// actor. This function returns a Result because it is possible /// for client creation to fail for some reason (e.g. invalid /// homeserver URL). pub fn new( config: &Arc, state: &Arc>, db: &Database, ) -> Result { Ok(DiceBot { client: create_client(&config)?, config: config.clone(), state: state.clone(), db: db.clone(), }) } /// Logs the bot into Matrix and listens for events until program /// terminated, or a panic occurs. Originally adapted from the /// matrix-rust-sdk command bot example. pub async fn run(self) -> Result<(), BotError> { let username = &self.config.matrix_username(); let password = &self.config.matrix_password(); //TODO provide a device id from config. let mut client = self.client.clone(); client .login(username, password, None, Some("matrix dice bot")) .await?; info!("Logged in as {}", username); // Initial sync without event handler prevents responding to // messages received while bot was offline. TODO: selectively // respond to old messages? e.g. comands missed while offline. self.client.sync_once(SyncSettings::default()).await?; client.add_event_emitter(Box::new(self)).await; info!("Listening for commands"); let token = client .sync_token() .await .ok_or(BotError::SyncTokenRequired)?; let settings = SyncSettings::default().token(token); // TODO replace with sync_with_callback for cleaner shutdown // process. client.sync(settings).await; Ok(()) } async fn execute_commands(&self, room: &Room, sender_username: &str, msg_body: &str) { let room_name = room.display_name().clone(); let room_id = room.room_id.clone(); let mut results = Vec::with_capacity(msg_body.lines().count()); let commands = msg_body.trim().lines().filter(|line| line.starts_with("!")); for command in commands { let ctx = Context::new(&self.db, &room_id.as_str(), &sender_username, &command); if let Some(cmd_result) = execute_command(&ctx).await { results.push(cmd_result); } } if results.len() >= 1 { if results.len() == 1 { let cmd_result = &results[0]; let response = AnyMessageEventContent::RoomMessage(MessageEventContent::Notice( NoticeMessageEventContent::html( cmd_result.plain.clone(), cmd_result.html.clone(), ), )); let result = self.client.room_send(&room_id, response, None).await; if let Err(e) = result { let message = extract_error_message(e); error!("Error sending message: {}", message); }; } else if results.len() > 1 { let message = format!("{}: Executed {} commands", sender_username, results.len()); let response = AnyMessageEventContent::RoomMessage(MessageEventContent::Notice( NoticeMessageEventContent::html(&message, &message), )); let result = self.client.room_send(&room_id, response, None).await; if let Err(e) = result { let message = extract_error_message(e); error!("Error sending message: {}", message); }; } info!("[{}] {} executed: {}", room_name, sender_username, msg_body); } } }