use std::{env, process::exit}; use matrix_sdk::{ self, events::room::message::{MessageEvent, MessageEventContent, TextMessageEventContent}, Client, ClientConfig, EventEmitter, JsonStore, SyncRoom, SyncSettings, }; use url::Url; struct CommandBot { /// This clone of the `Client` will send requests to the server, /// while the other keeps us in sync with the server using `sync_forever`. client: Client, } impl CommandBot { pub fn new(client: Client) -> Self { Self { client } } } #[async_trait::async_trait] impl EventEmitter for CommandBot { async fn on_room_message(&self, room: SyncRoom, event: &MessageEvent) { if let SyncRoom::Joined(room) = room { let msg_body = if let MessageEvent { content: MessageEventContent::Text(TextMessageEventContent { body: msg_body, .. }), .. } = event { msg_body.clone() } else { String::new() }; if msg_body.contains("!party") { let content = MessageEventContent::Text(TextMessageEventContent { body: "🎉🎊🥳 let's PARTY!! 🥳🎊🎉".to_string(), format: None, formatted_body: None, relates_to: None, }); // we clone here to hold the lock for as little time as possible. let room_id = room.read().await.room_id.clone(); println!("sending"); self.client // send our message to the room we found the "!party" command in // the last parameter is an optional Uuid which we don't care about. .room_send(&room_id, content, None) .await .unwrap(); println!("message sent"); } } } } async fn login_and_sync( homeserver_url: String, username: String, password: String, ) -> Result<(), matrix_sdk::Error> { // the location for `JsonStore` to save files to let mut home = dirs::home_dir().expect("no home directory found"); home.push("party_bot"); let store = JsonStore::open(&home)?; let client_config = ClientConfig::new() .proxy("http://localhost:8080")? .disable_ssl_verification() .state_store(Box::new(store)); let homeserver_url = Url::parse(&homeserver_url).expect("Couldn't parse the homeserver URL"); // create a new Client with the given homeserver url and config let mut client = Client::new_with_config(homeserver_url, None, client_config).unwrap(); client .login( username.clone(), password, None, Some("command bot".to_string()), ) .await?; println!("logged in as {}", username); // An initial sync to set up state and so our bot doesn't respond to old messages. // If the `StateStore` finds saved state in the location given the initial sync will // be skipped in favor of loading state from the store client.sync(SyncSettings::default()).await.unwrap(); // add our CommandBot to be notified of incoming messages, we do this after the initial // sync to avoid responding to messages before the bot was running. client .add_event_emitter(Box::new(CommandBot::new(client.clone()))) .await; // since we called sync before we `sync_forever` we must pass that sync token to // `sync_forever` let settings = SyncSettings::default().token(client.sync_token().await.unwrap()); // this keeps state from the server streaming in to CommandBot via the EventEmitter trait client.sync_forever(settings, |_| async {}).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: {} ", env::args().next().unwrap() ); exit(1) } }; login_and_sync(homeserver_url, username, password).await?; Ok(()) }