Skip to main content

example_image_bot/
main.rs

1#![recursion_limit = "256"]
2
3use std::{env, fs, process::exit};
4
5use matrix_sdk::{
6    Client, Room, RoomState,
7    attachment::AttachmentConfig,
8    config::SyncSettings,
9    ruma::events::room::message::{
10        MessageType, OriginalSyncRoomMessageEvent, TextMessageEventContent,
11    },
12};
13use url::Url;
14
15async fn on_room_message(event: OriginalSyncRoomMessageEvent, room: Room, image: Vec<u8>) {
16    if room.state() != RoomState::Joined {
17        return;
18    }
19    let MessageType::Text(text_content) = event.content.msgtype else { return };
20
21    if text_content.body.contains("!image") {
22        println!("sending image");
23        room.send_attachment(
24            "cat.jpg",
25            &mime::IMAGE_JPEG,
26            image,
27            AttachmentConfig::new().caption(Some(TextMessageEventContent::plain("my pretty cat"))),
28        )
29        .await
30        .unwrap();
31
32        println!("message sent");
33    }
34}
35
36async fn login_and_sync(
37    homeserver_url: String,
38    username: String,
39    password: String,
40    image: Vec<u8>,
41) -> matrix_sdk::Result<()> {
42    let homeserver_url = Url::parse(&homeserver_url).expect("Couldn't parse the homeserver URL");
43    let client = Client::new(homeserver_url).await.unwrap();
44
45    client
46        .matrix_auth()
47        .login_username(&username, &password)
48        .initial_device_display_name("command bot")
49        .await?;
50
51    let response = client.sync_once(SyncSettings::default()).await.unwrap();
52
53    client.add_event_handler(move |ev, room| on_room_message(ev, room, image));
54
55    let settings = SyncSettings::default().token(response.next_batch);
56    client.sync(settings).await?;
57
58    Ok(())
59}
60
61#[tokio::main]
62async fn main() -> anyhow::Result<()> {
63    tracing_subscriber::fmt::init();
64    let (Some(homeserver_url), Some(username), Some(password), Some(image_path)) =
65        (env::args().nth(1), env::args().nth(2), env::args().nth(3), env::args().nth(4))
66    else {
67        eprintln!(
68            "Usage: {} <homeserver_url> <username> <password> <image>",
69            env::args().next().unwrap()
70        );
71        exit(1)
72    };
73
74    println!("helloooo {homeserver_url} {username} {password} {image_path:#?}");
75    let image = fs::read(&image_path).expect("Can't open image file.");
76
77    login_and_sync(homeserver_url, username, password, image).await?;
78    Ok(())
79}