60 lines
1.8 KiB
Rust
60 lines
1.8 KiB
Rust
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(())
|
|
}
|