matrix-rust-sdk/matrix_sdk/examples/autojoin.rs

108 lines
3.0 KiB
Rust
Raw Normal View History

2020-07-08 18:22:50 +00:00
use std::{env, process::exit};
use tokio::time::{sleep, Duration};
2020-07-08 18:22:50 +00:00
use matrix_sdk::{
self, async_trait,
events::{room::member::MemberEventContent, StrippedStateEvent},
Client, ClientConfig, EventHandler, Room, RoomType, SyncSettings,
2020-07-08 18:22:50 +00:00
};
use url::Url;
struct AutoJoinBot {
client: Client,
}
impl AutoJoinBot {
pub fn new(client: Client) -> Self {
Self { client }
}
}
#[async_trait]
impl EventHandler for AutoJoinBot {
2020-07-08 18:22:50 +00:00
async fn on_stripped_state_member(
&self,
room: Room,
room_member: &StrippedStateEvent<MemberEventContent>,
2020-07-08 18:22:50 +00:00
_: Option<MemberEventContent>,
) {
if room_member.state_key != self.client.user_id().await.unwrap() {
return;
}
if room.room_type() == RoomType::Invited {
2020-12-19 19:20:39 +00:00
println!("Autojoining room {}", room.room_id());
2020-10-10 13:57:23 +00:00
let mut delay = 2;
2020-12-19 19:20:39 +00:00
while let Err(err) = self.client.join_room_by_id(room.room_id()).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
2020-10-10 13:57:23 +00:00
eprintln!(
"Failed to join room {} ({:?}), retrying in {}s",
2020-12-19 19:20:39 +00:00
room.room_id(),
err,
delay
2020-10-10 13:57:23 +00:00
);
sleep(Duration::from_secs(delay)).await;
2020-10-10 13:57:23 +00:00
delay *= 2;
2020-10-10 13:57:23 +00:00
if delay > 3600 {
2020-12-19 19:20:39 +00:00
eprintln!("Can't join room {} ({:?})", room.room_id(), err);
2020-10-10 13:57:23 +00:00
break;
}
}
2020-12-19 19:20:39 +00:00
println!("Successfully joined room {}", room.room_id());
2020-07-08 18:22:50 +00:00
}
}
}
async fn login_and_sync(
homeserver_url: String,
2020-08-15 01:09:13 +00:00
username: &str,
password: &str,
2020-07-08 18:22:50 +00:00
) -> Result<(), matrix_sdk::Error> {
let mut home = dirs::home_dir().expect("no home directory found");
home.push("autojoin_bot");
let client_config = ClientConfig::new().store_path(home);
let homeserver_url = Url::parse(&homeserver_url).expect("Couldn't parse the homeserver URL");
2020-12-19 19:20:39 +00:00
let client = Client::new_with_config(homeserver_url, client_config).unwrap();
2020-07-08 18:22:50 +00:00
client
2020-08-15 01:09:13 +00:00
.login(username, password, None, Some("autojoin bot"))
2020-07-08 18:22:50 +00:00
.await?;
println!("logged in as {}", username);
client
.set_event_handler(Box::new(AutoJoinBot::new(client.clone())))
2020-07-08 18:22:50 +00:00
.await;
client.sync(SyncSettings::default()).await;
2020-07-08 18:22:50 +00:00
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: {} <homeserver_url> <username> <password>",
env::args().next().unwrap()
);
exit(1)
}
};
2020-08-15 01:09:13 +00:00
login_and_sync(homeserver_url, &username, &password).await?;
2020-07-08 18:22:50 +00:00
Ok(())
}