mirsal
cd590b73fe
This is a rewrite of our old feedbot in rust, heavily inspired from rek2's INN matrix bot and making use of some bits from matrix-rust-sdk This is an asynchronous tokio-based matrix client using a stateless feed fetcher implementation based on reqwest, it uses feed_rs for parsing RSS and Atom feeds. State persistence is achieved using a simple file-backed datastore with serde_yaml as a serialization format. Published under the GNU General Public License version 3 or later.
116 lines
3.7 KiB
Rust
116 lines
3.7 KiB
Rust
/**
|
|
* matrix-feedbot v0.1.0
|
|
*
|
|
* Copyright (C) 2024 The 1312 Media Collective
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
use matrix_sdk::{
|
|
config::SyncSettings,
|
|
ruma::events::room::{
|
|
member::StrippedRoomMemberEvent,
|
|
message::RoomMessageEventContent
|
|
},
|
|
ruma::RoomId,
|
|
Client, Room
|
|
};
|
|
|
|
use tokio::{
|
|
time::{sleep, Duration},
|
|
sync::broadcast
|
|
};
|
|
|
|
async fn on_stripped_state_member(
|
|
room_member: StrippedRoomMemberEvent,
|
|
client: Client,
|
|
room: Room,
|
|
) {
|
|
if room_member.state_key != client.user_id().unwrap() {
|
|
return;
|
|
}
|
|
|
|
tokio::spawn(async move {
|
|
println!("Autojoining room {}", room.room_id());
|
|
let mut delay = 2;
|
|
|
|
while let Err(err) = room.join().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
|
|
eprintln!("Failed to join room {} ({err:?}), retrying in {delay}s", room.room_id());
|
|
|
|
sleep(Duration::from_secs(delay)).await;
|
|
delay *= 2;
|
|
|
|
if delay > 3600 {
|
|
eprintln!("Can't join room {} ({err:?})", room.room_id());
|
|
break;
|
|
}
|
|
}
|
|
println!("Successfully joined room {}", room.room_id());
|
|
});
|
|
}
|
|
|
|
pub async fn login_and_sync<T: Clone>(
|
|
homeserver_url: String,
|
|
username: &str,
|
|
password: &str,
|
|
default_room_id: &str,
|
|
mut rx: broadcast::Receiver<(String, Vec<String>)>
|
|
) -> anyhow::Result<()> {
|
|
// We are not reading encrypted messages, so we don't care about session persistence
|
|
let client = Client::builder().homeserver_url(homeserver_url).build().await?;
|
|
|
|
client
|
|
.matrix_auth()
|
|
.login_username(username, password)
|
|
.initial_device_display_name("bender v0.1.0")
|
|
.await?;
|
|
|
|
println!("logged in as {username}");
|
|
|
|
client.add_event_handler(on_stripped_state_member);
|
|
|
|
#[allow(unused)]
|
|
let default_room = client
|
|
.join_room_by_id(<&RoomId>::try_from(default_room_id).expect("Invalid default room ID"))
|
|
.await.expect("Failed to join room");
|
|
|
|
// make sure we already re-joined rooms before sending events
|
|
let sync_token = client.sync_once(SyncSettings::default()).await?.next_batch;
|
|
// since we called `sync_once` before we entered our sync loop
|
|
// we must pass that sync token to `sync`
|
|
let settings = SyncSettings::default().token(sync_token);
|
|
|
|
loop {
|
|
tokio::select! {
|
|
Ok(_) = client.sync(settings.clone()) => return Ok(()),
|
|
Ok((msg, rooms)) = rx.recv() => {
|
|
|
|
let msg = &msg.clone();
|
|
for r in rooms {
|
|
let room = client.join_room_by_id(<&RoomId>::try_from(r.as_str())
|
|
.expect("invalid room ID")
|
|
).await.expect(format!("Failed to join room {}", r).as_str());
|
|
|
|
room.send(RoomMessageEventContent::text_html(msg.clone(), msg)).await
|
|
.expect("Failed to send matrix event");
|
|
}
|
|
|
|
}
|
|
}
|
|
}
|
|
}
|