|
|
@ -2,19 +2,21 @@ use std::collections::HashSet; |
|
|
|
use std::sync::Arc; |
|
|
|
use std::sync::Arc; |
|
|
|
use std::time::{Duration, Instant}; |
|
|
|
use std::time::{Duration, Instant}; |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
use elefren::{FediClient, StatusBuilder}; |
|
|
|
use elefren::debug::EventDisplay; |
|
|
|
use elefren::debug::EventDisplay; |
|
|
|
use elefren::debug::NotificationDisplay; |
|
|
|
use elefren::debug::NotificationDisplay; |
|
|
|
|
|
|
|
use elefren::debug::StatusDisplay; |
|
|
|
use elefren::entities::event::Event; |
|
|
|
use elefren::entities::event::Event; |
|
|
|
use elefren::entities::notification::{Notification, NotificationType}; |
|
|
|
use elefren::entities::notification::{Notification, NotificationType}; |
|
|
|
|
|
|
|
use elefren::entities::status::Status; |
|
|
|
use elefren::status_builder::Visibility; |
|
|
|
use elefren::status_builder::Visibility; |
|
|
|
use elefren::{FediClient, StatusBuilder}; |
|
|
|
|
|
|
|
use futures::StreamExt; |
|
|
|
use futures::StreamExt; |
|
|
|
|
|
|
|
|
|
|
|
use crate::command::StatusCommand; |
|
|
|
use crate::command::StatusCommand; |
|
|
|
use crate::error::GroupError; |
|
|
|
use crate::error::GroupError; |
|
|
|
use crate::store::data::GroupConfig; |
|
|
|
|
|
|
|
use crate::store::ConfigStore; |
|
|
|
use crate::store::ConfigStore; |
|
|
|
use crate::utils::{normalize_acct, LogError}; |
|
|
|
use crate::store::data::GroupConfig; |
|
|
|
|
|
|
|
use crate::utils::{LogError, normalize_acct}; |
|
|
|
|
|
|
|
|
|
|
|
/// This is one group's config store capable of persistence
|
|
|
|
/// This is one group's config store capable of persistence
|
|
|
|
#[derive(Debug)] |
|
|
|
#[derive(Debug)] |
|
|
@ -27,6 +29,9 @@ pub struct GroupHandle { |
|
|
|
const DELAY_BEFORE_ACTION: Duration = Duration::from_millis(250); |
|
|
|
const DELAY_BEFORE_ACTION: Duration = Duration::from_millis(250); |
|
|
|
const DELAY_REOPEN_STREAM: Duration = Duration::from_millis(1000); |
|
|
|
const DELAY_REOPEN_STREAM: Duration = Duration::from_millis(1000); |
|
|
|
const MAX_CATCHUP_NOTIFS: usize = 25; |
|
|
|
const MAX_CATCHUP_NOTIFS: usize = 25; |
|
|
|
|
|
|
|
// also statuses
|
|
|
|
|
|
|
|
const MAX_CATCHUP_STATUSES: usize = 100; |
|
|
|
|
|
|
|
// higher because we can expect a lot of non-hashtag statuses here
|
|
|
|
const PERIODIC_SAVE: Duration = Duration::from_secs(60); |
|
|
|
const PERIODIC_SAVE: Duration = Duration::from_secs(60); |
|
|
|
const PING_INTERVAL: Duration = Duration::from_secs(15); // must be < periodic save!
|
|
|
|
const PING_INTERVAL: Duration = Duration::from_secs(15); // must be < periodic save!
|
|
|
|
|
|
|
|
|
|
|
@ -45,6 +50,7 @@ impl GroupHandle { |
|
|
|
Ok(()) |
|
|
|
Ok(()) |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/* |
|
|
|
pub async fn reload(&mut self) -> Result<(), GroupError> { |
|
|
|
pub async fn reload(&mut self) -> Result<(), GroupError> { |
|
|
|
if let Some(g) = self.store.get_group_config(self.config.get_acct()).await { |
|
|
|
if let Some(g) = self.store.get_group_config(self.config.get_acct()).await { |
|
|
|
self.config = g; |
|
|
|
self.config = g; |
|
|
@ -53,6 +59,7 @@ impl GroupHandle { |
|
|
|
Err(GroupError::GroupNotExist) |
|
|
|
Err(GroupError::GroupNotExist) |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
*/ |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
trait NotifTimestamp { |
|
|
|
trait NotifTimestamp { |
|
|
@ -65,6 +72,14 @@ impl NotifTimestamp for Notification { |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
impl NotifTimestamp for Status { |
|
|
|
|
|
|
|
fn timestamp_millis(&self) -> u64 { |
|
|
|
|
|
|
|
// this may not work well for unseen status tracking,
|
|
|
|
|
|
|
|
// if ancient statuses were to appear in the timeline :(
|
|
|
|
|
|
|
|
self.created_at.timestamp_millis().max(0) as u64 |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
impl GroupHandle { |
|
|
|
impl GroupHandle { |
|
|
|
pub async fn run(&mut self) -> Result<(), GroupError> { |
|
|
|
pub async fn run(&mut self) -> Result<(), GroupError> { |
|
|
|
assert!(PERIODIC_SAVE >= PING_INTERVAL); |
|
|
|
assert!(PERIODIC_SAVE >= PING_INTERVAL); |
|
|
@ -89,6 +104,20 @@ impl GroupHandle { |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
match self.catch_up_with_missed_statuses().await { |
|
|
|
|
|
|
|
Ok(true) => { |
|
|
|
|
|
|
|
debug!("Some missed statuses handled"); |
|
|
|
|
|
|
|
// Save asap!
|
|
|
|
|
|
|
|
next_save = Instant::now() - PERIODIC_SAVE |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
Ok(false) => { |
|
|
|
|
|
|
|
debug!("No statuses missed"); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
Err(e) => { |
|
|
|
|
|
|
|
error!("Failed to handle missed statuses: {}", e); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
loop { |
|
|
|
loop { |
|
|
|
if next_save < Instant::now() { |
|
|
|
if next_save < Instant::now() { |
|
|
|
self.save_if_needed().await.log_error("Failed to save group"); |
|
|
|
self.save_if_needed().await.log_error("Failed to save group"); |
|
|
@ -104,7 +133,9 @@ impl GroupHandle { |
|
|
|
Ok(Some(event)) => { |
|
|
|
Ok(Some(event)) => { |
|
|
|
debug!("(@{}) Event: {}", self.config.get_acct(), EventDisplay(&event)); |
|
|
|
debug!("(@{}) Event: {}", self.config.get_acct(), EventDisplay(&event)); |
|
|
|
match event { |
|
|
|
match event { |
|
|
|
Event::Update(_status) => {} |
|
|
|
Event::Update(status) => { |
|
|
|
|
|
|
|
self.handle_status(status).await.log_error("Error handling a status"); |
|
|
|
|
|
|
|
} |
|
|
|
Event::Notification(n) => { |
|
|
|
Event::Notification(n) => { |
|
|
|
self.handle_notification(n).await.log_error("Error handling a notification"); |
|
|
|
self.handle_notification(n).await.log_error("Error handling a notification"); |
|
|
|
} |
|
|
|
} |
|
|
@ -156,7 +187,7 @@ impl GroupHandle { |
|
|
|
return Ok(()); |
|
|
|
return Ok(()); |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
let commands = crate::command::parse_status(&status.content); |
|
|
|
let commands = crate::command::parse_slash_commands(&status.content); |
|
|
|
|
|
|
|
|
|
|
|
let mut replies = vec![]; |
|
|
|
let mut replies = vec![]; |
|
|
|
let mut announcements = vec![]; |
|
|
|
let mut announcements = vec![]; |
|
|
@ -205,6 +236,9 @@ impl GroupHandle { |
|
|
|
any_admin_cmd = true; |
|
|
|
any_admin_cmd = true; |
|
|
|
replies.push(format!("User {} banned from group!", u)); |
|
|
|
replies.push(format!("User {} banned from group!", u)); |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
self.unfollow_user(&u).await |
|
|
|
|
|
|
|
.log_error("Failed to unfollow"); |
|
|
|
|
|
|
|
|
|
|
|
// no announcement here
|
|
|
|
// no announcement here
|
|
|
|
} |
|
|
|
} |
|
|
|
Err(e) => { |
|
|
|
Err(e) => { |
|
|
@ -280,11 +314,7 @@ impl GroupHandle { |
|
|
|
Ok(_) => { |
|
|
|
Ok(_) => { |
|
|
|
any_admin_cmd = true; |
|
|
|
any_admin_cmd = true; |
|
|
|
replies.push(format!("User {} added to the group!", u)); |
|
|
|
replies.push(format!("User {} added to the group!", u)); |
|
|
|
|
|
|
|
self.follow_user(&u).await.log_error("Failed to follow"); |
|
|
|
if self.config.is_member_only() { |
|
|
|
|
|
|
|
announcements |
|
|
|
|
|
|
|
.push(format!("Welcome new member @{} to the group!", u)); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
} |
|
|
|
} |
|
|
|
Err(e) => { |
|
|
|
Err(e) => { |
|
|
|
replies.push(format!("Failed to add user {} to group: {}", u, e)); |
|
|
|
replies.push(format!("Failed to add user {} to group: {}", u, e)); |
|
|
@ -303,6 +333,9 @@ impl GroupHandle { |
|
|
|
Ok(_) => { |
|
|
|
Ok(_) => { |
|
|
|
any_admin_cmd = true; |
|
|
|
any_admin_cmd = true; |
|
|
|
replies.push(format!("User {} removed from the group.", u)); |
|
|
|
replies.push(format!("User {} removed from the group.", u)); |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
self.unfollow_user(&u).await |
|
|
|
|
|
|
|
.log_error("Failed to unfollow"); |
|
|
|
} |
|
|
|
} |
|
|
|
Err(_) => { |
|
|
|
Err(_) => { |
|
|
|
unreachable!() |
|
|
|
unreachable!() |
|
|
@ -313,12 +346,33 @@ impl GroupHandle { |
|
|
|
replies.push("Only admins can manage members".to_string()); |
|
|
|
replies.push("Only admins can manage members".to_string()); |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
StatusCommand::AddTag(tag) => { |
|
|
|
|
|
|
|
if is_admin { |
|
|
|
|
|
|
|
any_admin_cmd = true; |
|
|
|
|
|
|
|
self.config.add_tag(&tag); |
|
|
|
|
|
|
|
replies.push(format!("Tag #{} added to the group!", tag)); |
|
|
|
|
|
|
|
} else { |
|
|
|
|
|
|
|
replies.push("Only admins can manage group tags".to_string()); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
StatusCommand::RemoveTag(tag) => { |
|
|
|
|
|
|
|
if is_admin { |
|
|
|
|
|
|
|
any_admin_cmd = true; |
|
|
|
|
|
|
|
self.config.remove_tag(&tag); |
|
|
|
|
|
|
|
replies.push(format!("Tag #{} removed from the group!", tag)); |
|
|
|
|
|
|
|
} else { |
|
|
|
|
|
|
|
replies.push("Only admins can manage group tags".to_string()); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
} |
|
|
|
StatusCommand::GrantAdmin(u) => { |
|
|
|
StatusCommand::GrantAdmin(u) => { |
|
|
|
let u = normalize_acct(&u, &group_acct)?; |
|
|
|
let u = normalize_acct(&u, &group_acct)?; |
|
|
|
if is_admin { |
|
|
|
if is_admin { |
|
|
|
if !self.config.is_admin(&u) { |
|
|
|
if !self.config.is_admin(&u) { |
|
|
|
match self.config.set_admin(&u, true) { |
|
|
|
match self.config.set_admin(&u, true) { |
|
|
|
Ok(_) => { |
|
|
|
Ok(_) => { |
|
|
|
|
|
|
|
// try to make the config a little more sane, admins should be members
|
|
|
|
|
|
|
|
let _ = self.config.set_member(&u, true); |
|
|
|
|
|
|
|
|
|
|
|
any_admin_cmd = true; |
|
|
|
any_admin_cmd = true; |
|
|
|
replies.push(format!("User {} is now a group admin!", u)); |
|
|
|
replies.push(format!("User {} is now a group admin!", u)); |
|
|
|
announcements |
|
|
|
announcements |
|
|
@ -409,12 +463,13 @@ impl GroupHandle { |
|
|
|
**Supported commands:**\n\ |
|
|
|
**Supported commands:**\n\ |
|
|
|
`/boost, /b` - boost the replied-to post into the group\n\ |
|
|
|
`/boost, /b` - boost the replied-to post into the group\n\ |
|
|
|
`/ignore, /i` - make the group completely ignore the post\n\ |
|
|
|
`/ignore, /i` - make the group completely ignore the post\n\ |
|
|
|
`/ping` - check that the service is alive".to_string(), |
|
|
|
`/ping` - check that the service is alive\n\ |
|
|
|
|
|
|
|
`/join` - join the group\n\ |
|
|
|
|
|
|
|
`/leave` - leave the group".to_string(), |
|
|
|
); |
|
|
|
); |
|
|
|
|
|
|
|
|
|
|
|
if self.config.is_member_only() { |
|
|
|
if self.config.is_member_only() { |
|
|
|
replies.push("`/members, /who` - show group members / admins".to_string()); |
|
|
|
replies.push("`/members, /who` - show group members / admins".to_string()); |
|
|
|
replies.push("`/leave` - leave the group".to_string()); |
|
|
|
|
|
|
|
} else { |
|
|
|
} else { |
|
|
|
replies.push("`/members, /who` - show group admins".to_string()); |
|
|
|
replies.push("`/members, /who` - show group admins".to_string()); |
|
|
|
} |
|
|
|
} |
|
|
@ -441,18 +496,7 @@ impl GroupHandle { |
|
|
|
if is_admin { |
|
|
|
if is_admin { |
|
|
|
if self.config.is_member_only() { |
|
|
|
if self.config.is_member_only() { |
|
|
|
replies.push("Group members:".to_string()); |
|
|
|
replies.push("Group members:".to_string()); |
|
|
|
let admins = self.config.get_admins().collect::<HashSet<_>>(); |
|
|
|
self.list_members(&mut replies); |
|
|
|
let mut members = self.config.get_members().collect::<Vec<_>>(); |
|
|
|
|
|
|
|
members.extend(admins.iter()); |
|
|
|
|
|
|
|
members.sort(); |
|
|
|
|
|
|
|
members.dedup(); |
|
|
|
|
|
|
|
for m in members { |
|
|
|
|
|
|
|
if admins.contains(&m) { |
|
|
|
|
|
|
|
replies.push(format!("{} [admin]", m)); |
|
|
|
|
|
|
|
} else { |
|
|
|
|
|
|
|
replies.push(m.to_string()); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
} else { |
|
|
|
} else { |
|
|
|
show_admins = true; |
|
|
|
show_admins = true; |
|
|
|
} |
|
|
|
} |
|
|
@ -462,18 +506,55 @@ impl GroupHandle { |
|
|
|
|
|
|
|
|
|
|
|
if show_admins { |
|
|
|
if show_admins { |
|
|
|
replies.push("Group admins:".to_string()); |
|
|
|
replies.push("Group admins:".to_string()); |
|
|
|
let mut admins = self.config.get_admins().collect::<Vec<_>>(); |
|
|
|
self.list_admins(&mut replies); |
|
|
|
admins.sort(); |
|
|
|
} |
|
|
|
for a in admins { |
|
|
|
} |
|
|
|
replies.push(a.to_string()); |
|
|
|
StatusCommand::ListTags => { |
|
|
|
} |
|
|
|
replies.push("Group tags:".to_string()); |
|
|
|
|
|
|
|
let mut tags = self.config.get_tags().collect::<Vec<_>>(); |
|
|
|
|
|
|
|
tags.sort(); |
|
|
|
|
|
|
|
for t in tags { |
|
|
|
|
|
|
|
replies.push(format!("#{}", t)); |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
StatusCommand::Leave => { |
|
|
|
StatusCommand::Leave => { |
|
|
|
if self.config.is_member(¬if_acct) { |
|
|
|
if self.config.is_member_or_admin(¬if_acct) { |
|
|
|
|
|
|
|
// admin can leave but that's a bad idea
|
|
|
|
|
|
|
|
|
|
|
|
any_admin_cmd = true; |
|
|
|
any_admin_cmd = true; |
|
|
|
let _ = self.config.set_member(¬if_acct, false); |
|
|
|
let _ = self.config.set_member(¬if_acct, false); |
|
|
|
replies.push("You left the group.".to_string()); |
|
|
|
replies.push("You're no longer a group member. Unfollow the group user to stop receiving group messages.".to_string()); |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
self.unfollow_user(¬if_acct).await |
|
|
|
|
|
|
|
.log_error("Failed to unfollow"); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
StatusCommand::Join => { |
|
|
|
|
|
|
|
if self.config.is_member_or_admin(¬if_acct) { |
|
|
|
|
|
|
|
// Already a member, so let's try to follow the user
|
|
|
|
|
|
|
|
// again, maybe first time it failed
|
|
|
|
|
|
|
|
self.follow_user(¬if_acct).await |
|
|
|
|
|
|
|
.log_error("Failed to follow"); |
|
|
|
|
|
|
|
} else { |
|
|
|
|
|
|
|
// Not a member yet
|
|
|
|
|
|
|
|
if self.config.is_member_only() { |
|
|
|
|
|
|
|
// No you can't
|
|
|
|
|
|
|
|
replies.push(format!( |
|
|
|
|
|
|
|
"Hi, this group is closed to new sign-ups.\n\ |
|
|
|
|
|
|
|
Please ask one of the group admins to add you:")); |
|
|
|
|
|
|
|
self.list_admins(&mut replies); |
|
|
|
|
|
|
|
} else { |
|
|
|
|
|
|
|
// Open access
|
|
|
|
|
|
|
|
self.follow_user(¬if_acct).await |
|
|
|
|
|
|
|
.log_error("Failed to follow"); |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
// This only fails if the user is banned, but that is filtered above
|
|
|
|
|
|
|
|
let _ = self.config.set_member(¬if_acct, true); |
|
|
|
|
|
|
|
replies.push(format!("\ |
|
|
|
|
|
|
|
Thanks for joining, you are now a member and the group user will \ |
|
|
|
|
|
|
|
follow you so you can use group hashtags. Make sure you follow the \ |
|
|
|
|
|
|
|
group user to receive group messages.")); |
|
|
|
|
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
StatusCommand::Ping => { |
|
|
|
StatusCommand::Ping => { |
|
|
@ -524,43 +605,106 @@ impl GroupHandle { |
|
|
|
} |
|
|
|
} |
|
|
|
NotificationType::Follow => { |
|
|
|
NotificationType::Follow => { |
|
|
|
info!("New follower!"); |
|
|
|
info!("New follower!"); |
|
|
|
tokio::time::sleep(Duration::from_millis(500)).await; |
|
|
|
|
|
|
|
|
|
|
|
if self.config.is_member_or_admin(¬if_acct) { |
|
|
|
let text = if self.config.is_member_only() { |
|
|
|
// Already joined, just doing something silly, ignore this
|
|
|
|
// Admins are listed without @, so they won't become handles here.
|
|
|
|
debug!("User already a member, ignoring"); |
|
|
|
// Tagging all admins would be annoying.
|
|
|
|
|
|
|
|
let mut admins = self.config.get_admins().cloned().collect::<Vec<_>>(); |
|
|
|
|
|
|
|
admins.sort(); |
|
|
|
|
|
|
|
format!( |
|
|
|
|
|
|
|
"@{user} welcome to the group! This is a member-only group, you won't be \ |
|
|
|
|
|
|
|
able to post. Ask the group admins if you wish to join!\n\n\ |
|
|
|
|
|
|
|
Admins: {admins}", |
|
|
|
|
|
|
|
user = notif_acct, |
|
|
|
|
|
|
|
admins = admins.join(", ") |
|
|
|
|
|
|
|
) |
|
|
|
|
|
|
|
} else { |
|
|
|
} else { |
|
|
|
format!( |
|
|
|
let text = if self.config.is_member_only() { |
|
|
|
"@{user} welcome to the group! \ |
|
|
|
// Admins are listed without @, so they won't become handles here.
|
|
|
|
To share a post, tag the group user. Use /help for more info.", |
|
|
|
// Tagging all admins would be annoying.
|
|
|
|
user = notif_acct |
|
|
|
let mut admins = self.config.get_admins().cloned().collect::<Vec<_>>(); |
|
|
|
) |
|
|
|
admins.sort(); |
|
|
|
}; |
|
|
|
format!( |
|
|
|
|
|
|
|
"@{user} Hi, this is a member-only group, you won't be \ |
|
|
|
let post = StatusBuilder::new() |
|
|
|
able to post. You can still receive group posts though. If you'd like to join, \ |
|
|
|
.status(text) |
|
|
|
please ask one of the group admins to add you:\n\n\ |
|
|
|
.content_type("text/markdown") |
|
|
|
{admins}", |
|
|
|
.visibility(Visibility::Direct) |
|
|
|
user = notif_acct, |
|
|
|
.build() |
|
|
|
admins = admins.join(", ") |
|
|
|
.expect("error build status"); |
|
|
|
) |
|
|
|
|
|
|
|
} else { |
|
|
|
let _ = self.client.new_status(post).await.log_error("Failed to post"); |
|
|
|
self.follow_user(¬if_acct).await |
|
|
|
|
|
|
|
.log_error("Failed to follow"); |
|
|
|
|
|
|
|
make_welcome_text(¬if_acct) |
|
|
|
|
|
|
|
}; |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
let post = StatusBuilder::new() |
|
|
|
|
|
|
|
.status(text) |
|
|
|
|
|
|
|
.content_type("text/markdown") |
|
|
|
|
|
|
|
.visibility(Visibility::Direct) |
|
|
|
|
|
|
|
.build() |
|
|
|
|
|
|
|
.expect("error build status"); |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
tokio::time::sleep(Duration::from_millis(500)).await; |
|
|
|
|
|
|
|
let _ = self.client.new_status(post).await.log_error("Failed to post"); |
|
|
|
|
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
_ => {} |
|
|
|
NotificationType::Favourite => {} |
|
|
|
|
|
|
|
NotificationType::Reblog => {} |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
Ok(()) |
|
|
|
Ok(()) |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/// Handle a non-mention status
|
|
|
|
|
|
|
|
async fn handle_status(&mut self, s: Status) -> Result<(), GroupError> { |
|
|
|
|
|
|
|
debug!("Handling status #{}", s.id); |
|
|
|
|
|
|
|
let ts = s.timestamp_millis(); |
|
|
|
|
|
|
|
self.config.set_last_status(ts); |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if !s.content.contains('#') { |
|
|
|
|
|
|
|
debug!("No tags in status"); |
|
|
|
|
|
|
|
return Ok(()); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if s.visibility.is_private() { |
|
|
|
|
|
|
|
debug!("Status is direct/private, not boosting"); |
|
|
|
|
|
|
|
return Ok(()); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if s.content.contains("/add ") || s.content.contains("/remove ") { |
|
|
|
|
|
|
|
debug!("Discard, looks like a hashtag manipulation command"); |
|
|
|
|
|
|
|
return Ok(()); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
let gu = self.config.get_acct(); |
|
|
|
|
|
|
|
let su = normalize_acct(&s.account.acct, gu)?; |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if self.config.is_banned(&su) { |
|
|
|
|
|
|
|
debug!("Status author @{} is banned.", su); |
|
|
|
|
|
|
|
return Ok(()); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if !self.config.is_member_or_admin(&su) { |
|
|
|
|
|
|
|
debug!("Status author @{} is not a member.", su); |
|
|
|
|
|
|
|
return Ok(()); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
let tags = crate::command::parse_status_tags(&s.content); |
|
|
|
|
|
|
|
debug!("Tags in status: {:?}", tags); |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
for t in tags { |
|
|
|
|
|
|
|
if self.config.is_tag_followed(&t) { |
|
|
|
|
|
|
|
self.client.reblog(&s.id).await |
|
|
|
|
|
|
|
.log_error("Failed to reblog"); |
|
|
|
|
|
|
|
break; |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
Ok(()) |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
async fn follow_user(&mut self, acct: &str) -> Result<(), GroupError> { |
|
|
|
|
|
|
|
self.client.follow(acct).await?; |
|
|
|
|
|
|
|
Ok(()) |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
async fn unfollow_user(&mut self, acct: &str) -> Result<(), GroupError> { |
|
|
|
|
|
|
|
self.client.unfollow(acct).await?; |
|
|
|
|
|
|
|
Ok(()) |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
/// Catch up with missed notifications, returns true if any were handled
|
|
|
|
/// Catch up with missed notifications, returns true if any were handled
|
|
|
|
async fn catch_up_with_missed_notifications(&mut self) -> Result<bool, GroupError> { |
|
|
|
async fn catch_up_with_missed_notifications(&mut self) -> Result<bool, GroupError> { |
|
|
|
let last_notif = self.config.get_last_notif(); |
|
|
|
let last_notif = self.config.get_last_notif(); |
|
|
@ -601,4 +745,92 @@ impl GroupHandle { |
|
|
|
|
|
|
|
|
|
|
|
Ok(true) |
|
|
|
Ok(true) |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/// Catch up with missed statuses, returns true if any were handled
|
|
|
|
|
|
|
|
async fn catch_up_with_missed_statuses(&mut self) -> Result<bool, GroupError> { |
|
|
|
|
|
|
|
let last_status = self.config.get_last_status(); |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
let notifications = self.client.get_home_timeline().await?; |
|
|
|
|
|
|
|
let mut iter = notifications.items_iter(); |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
let mut statuses_to_handle = vec![]; |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
// They are retrieved newest first, but we want oldest first for chronological handling
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
let mut num = 0; |
|
|
|
|
|
|
|
while let Some(s) = iter.next_item().await { |
|
|
|
|
|
|
|
let ts = s.timestamp_millis(); |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if ts <= last_status { |
|
|
|
|
|
|
|
break; // reached our last seen status (hopefully there arent any retro-bumped)
|
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
if s.content.contains('#') && !s.visibility.is_private() { |
|
|
|
|
|
|
|
statuses_to_handle.push(s); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
num += 1; |
|
|
|
|
|
|
|
if num > MAX_CATCHUP_STATUSES { |
|
|
|
|
|
|
|
warn!("Too many statuses missed to catch up!"); |
|
|
|
|
|
|
|
break; |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if statuses_to_handle.is_empty() { |
|
|
|
|
|
|
|
return Ok(false); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
statuses_to_handle.reverse(); |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
debug!("{} statuses to catch up!", statuses_to_handle.len()); |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
for s in statuses_to_handle { |
|
|
|
|
|
|
|
debug!("Handling missed status: {}", StatusDisplay(&s)); |
|
|
|
|
|
|
|
self.handle_status(s).await |
|
|
|
|
|
|
|
.log_error("Error handling a status"); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
Ok(true) |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
fn list_admins(&self, replies: &mut Vec<String>) { |
|
|
|
|
|
|
|
let mut admins = self.config.get_admins().collect::<Vec<_>>(); |
|
|
|
|
|
|
|
admins.sort(); |
|
|
|
|
|
|
|
for a in admins { |
|
|
|
|
|
|
|
replies.push(a.to_string()); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
fn list_members(&self, replies: &mut Vec<String>) { |
|
|
|
|
|
|
|
let admins = self.config.get_admins().collect::<HashSet<_>>(); |
|
|
|
|
|
|
|
let mut members = self.config.get_members().collect::<Vec<_>>(); |
|
|
|
|
|
|
|
members.extend(admins.iter()); |
|
|
|
|
|
|
|
members.sort(); |
|
|
|
|
|
|
|
members.dedup(); |
|
|
|
|
|
|
|
for m in members { |
|
|
|
|
|
|
|
if admins.contains(&m) { |
|
|
|
|
|
|
|
replies.push(format!("{} [admin]", m)); |
|
|
|
|
|
|
|
} else { |
|
|
|
|
|
|
|
replies.push(m.to_string()); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
fn make_welcome_text(user: &str) -> String { |
|
|
|
|
|
|
|
format!( |
|
|
|
|
|
|
|
"@{user} Welcome to the group! To share a post, tag the group user \ |
|
|
|
|
|
|
|
or use one of the group hashtags. Use /help for more info.", |
|
|
|
|
|
|
|
user = user |
|
|
|
|
|
|
|
) |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
trait VisExt: Copy { |
|
|
|
|
|
|
|
/// Check if is private or direct
|
|
|
|
|
|
|
|
fn is_private(self) -> bool; |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
impl VisExt for Visibility { |
|
|
|
|
|
|
|
fn is_private(self) -> bool { |
|
|
|
|
|
|
|
self == Visibility::Direct || self == Visibility::Private |
|
|
|
|
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|