reorg code
This commit is contained in:
parent
46cbccca4e
commit
fdfb126c9c
6 changed files with 157 additions and 123 deletions
19
src/commands/help.rs
Normal file
19
src/commands/help.rs
Normal file
|
@ -0,0 +1,19 @@
|
|||
use anyhow::Result;
|
||||
use enum_iterator::IntoEnumIterator;
|
||||
use matrix_sdk::room::Joined;
|
||||
use matrix_sdk::ruma::events::room::message::RoomMessageEventContent;
|
||||
|
||||
use super::BotCommand;
|
||||
|
||||
pub async fn help(room: Joined) -> Result<()> {
|
||||
let mut msg = "List of commands:\n\n".to_owned();
|
||||
for command in BotCommand::into_enum_iter() {
|
||||
msg.push_str(&command.to_string());
|
||||
msg.push_str(" - ");
|
||||
msg.push_str(command.help_text());
|
||||
msg.push('\n');
|
||||
}
|
||||
let content = RoomMessageEventContent::notice_plain(msg);
|
||||
room.send(content, None).await?;
|
||||
Ok(())
|
||||
}
|
59
src/commands/mod.rs
Normal file
59
src/commands/mod.rs
Normal file
|
@ -0,0 +1,59 @@
|
|||
use std::fmt::Display;
|
||||
use std::str::FromStr;
|
||||
|
||||
use enum_iterator::IntoEnumIterator;
|
||||
|
||||
macro_rules! import_command {
|
||||
($( $command:tt ),* $(,)?) => {
|
||||
$(
|
||||
mod $command;
|
||||
pub use $command::$command;
|
||||
)*
|
||||
};
|
||||
}
|
||||
|
||||
import_command!(help, ping, source, uwuify);
|
||||
|
||||
#[derive(IntoEnumIterator)]
|
||||
pub enum BotCommand {
|
||||
Help,
|
||||
Source,
|
||||
Uwu,
|
||||
Ping,
|
||||
}
|
||||
|
||||
impl BotCommand {
|
||||
pub const fn help_text(&self) -> &'static str {
|
||||
match self {
|
||||
Self::Source => "Links to the source code for this bot.",
|
||||
Self::Uwu => "Uwuifies your message.",
|
||||
Self::Ping => "Pong!",
|
||||
Self::Help => "Prints this help.",
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl FromStr for BotCommand {
|
||||
type Err = ();
|
||||
|
||||
fn from_str(s: &str) -> std::result::Result<Self, Self::Err> {
|
||||
match s.to_lowercase().as_str() {
|
||||
"source" => Ok(Self::Source),
|
||||
"uwu" => Ok(Self::Uwu),
|
||||
"ping" => Ok(Self::Ping),
|
||||
"help" => Ok(Self::Help),
|
||||
_ => Err(()),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl Display for BotCommand {
|
||||
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
||||
match self {
|
||||
BotCommand::Source => write!(f, "source"),
|
||||
BotCommand::Uwu => write!(f, "uwu"),
|
||||
BotCommand::Ping => write!(f, "ping"),
|
||||
BotCommand::Help => write!(f, "help"),
|
||||
}
|
||||
}
|
||||
}
|
9
src/commands/ping.rs
Normal file
9
src/commands/ping.rs
Normal file
|
@ -0,0 +1,9 @@
|
|||
use anyhow::Result;
|
||||
use matrix_sdk::room::Joined;
|
||||
use matrix_sdk::ruma::events::room::message::RoomMessageEventContent;
|
||||
|
||||
pub async fn ping(room: Joined) -> Result<()> {
|
||||
let content = RoomMessageEventContent::text_plain("Pong!");
|
||||
room.send(content, None).await?;
|
||||
Ok(())
|
||||
}
|
9
src/commands/source.rs
Normal file
9
src/commands/source.rs
Normal file
|
@ -0,0 +1,9 @@
|
|||
use anyhow::Result;
|
||||
use matrix_sdk::room::Joined;
|
||||
use matrix_sdk::ruma::events::room::message::RoomMessageEventContent;
|
||||
|
||||
pub async fn source(room: Joined) -> Result<()> {
|
||||
let content = RoomMessageEventContent::text_plain("https://git.eddie.sh/edward/matrix-bot");
|
||||
room.send(content, None).await?;
|
||||
Ok(())
|
||||
}
|
55
src/commands/uwuify.rs
Normal file
55
src/commands/uwuify.rs
Normal file
|
@ -0,0 +1,55 @@
|
|||
use anyhow::{Context, Result};
|
||||
use matrix_sdk::room::{Joined, MessagesOptions};
|
||||
use matrix_sdk::ruma::events::room::message::{
|
||||
MessageType, RoomMessageEventContent, TextMessageEventContent,
|
||||
};
|
||||
use matrix_sdk::ruma::events::{
|
||||
AnyMessageLikeEvent, AnyRoomEvent, MessageLikeEvent, OriginalMessageLikeEvent,
|
||||
};
|
||||
use uwuifier::uwuify_str_sse;
|
||||
|
||||
pub async fn uwuify(room: Joined, message: &str) -> Result<()> {
|
||||
let message = if message.is_empty() {
|
||||
if let Ok(msg) = get_previous_message(&room).await {
|
||||
uwuify_str_sse(&msg.body)
|
||||
} else {
|
||||
"uwu".to_owned()
|
||||
}
|
||||
} else {
|
||||
uwuify_str_sse(message)
|
||||
};
|
||||
let content = RoomMessageEventContent::text_plain(message);
|
||||
room.send(content, None).await?;
|
||||
Ok(())
|
||||
}
|
||||
|
||||
async fn get_previous_message(room: &Joined) -> Result<TextMessageEventContent> {
|
||||
let last_prev_batch = (**room)
|
||||
.last_prev_batch()
|
||||
.context("No previous batch key while trying to find previous message")?;
|
||||
let request = MessagesOptions::backward(&last_prev_batch);
|
||||
let messages = room.messages(request).await?;
|
||||
|
||||
messages
|
||||
.chunk
|
||||
.iter()
|
||||
.filter_map(|msg| msg.event.deserialize().ok())
|
||||
.find_map(|msg| {
|
||||
if let AnyRoomEvent::MessageLike(AnyMessageLikeEvent::RoomMessage(
|
||||
MessageLikeEvent::Original(OriginalMessageLikeEvent {
|
||||
content:
|
||||
RoomMessageEventContent {
|
||||
msgtype: MessageType::Text(content),
|
||||
..
|
||||
},
|
||||
..
|
||||
}),
|
||||
)) = msg
|
||||
{
|
||||
Some(content)
|
||||
} else {
|
||||
None
|
||||
}
|
||||
})
|
||||
.context("Failed to find previous message")
|
||||
}
|
129
src/main.rs
129
src/main.rs
|
@ -1,23 +1,18 @@
|
|||
#![warn(clippy::nursery, clippy::pedantic)]
|
||||
|
||||
use std::fmt::Display;
|
||||
use std::path::PathBuf;
|
||||
use std::str::FromStr;
|
||||
use std::time::Duration;
|
||||
|
||||
use anyhow::{Context, Result};
|
||||
use clap::Parser;
|
||||
use enum_iterator::IntoEnumIterator;
|
||||
use matrix_sdk::config::SyncSettings;
|
||||
use matrix_sdk::room::{Joined, MessagesOptions, Room};
|
||||
use matrix_sdk::room::{Joined, Room};
|
||||
use matrix_sdk::ruma::events::room::member::StrippedRoomMemberEvent;
|
||||
use matrix_sdk::ruma::events::room::message::{
|
||||
MessageType, RoomMessageEventContent, SyncRoomMessageEvent, TextMessageEventContent,
|
||||
};
|
||||
use matrix_sdk::ruma::events::{
|
||||
AnyMessageLikeEvent, AnyRoomEvent, MessageLikeEvent, OriginalMessageLikeEvent,
|
||||
OriginalSyncMessageLikeEvent, SyncMessageLikeEvent,
|
||||
};
|
||||
use matrix_sdk::ruma::events::{OriginalSyncMessageLikeEvent, SyncMessageLikeEvent};
|
||||
use matrix_sdk::ruma::OwnedUserId;
|
||||
use matrix_sdk::store::CryptoStore;
|
||||
use matrix_sdk::{Client, Session};
|
||||
|
@ -25,6 +20,10 @@ use serde::{Deserialize, Serialize};
|
|||
use tracing::{error, info, Level};
|
||||
use tracing_subscriber::util::SubscriberInitExt;
|
||||
|
||||
use commands::*;
|
||||
|
||||
mod commands;
|
||||
|
||||
#[derive(Parser)]
|
||||
struct Args {
|
||||
#[clap(subcommand)]
|
||||
|
@ -203,127 +202,11 @@ async fn parse_message(event: SyncRoomMessageEvent, room: Joined) -> Result<()>
|
|||
Ok(())
|
||||
}
|
||||
|
||||
#[derive(IntoEnumIterator)]
|
||||
enum BotCommand {
|
||||
Help,
|
||||
Source,
|
||||
Uwu,
|
||||
Ping,
|
||||
}
|
||||
|
||||
impl BotCommand {
|
||||
const fn help_text(&self) -> &'static str {
|
||||
match self {
|
||||
Self::Source => "Links to the source code for this bot.",
|
||||
Self::Uwu => "Uwuifies your message.",
|
||||
Self::Ping => "Pong!",
|
||||
Self::Help => "Prints this help.",
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl FromStr for BotCommand {
|
||||
type Err = ();
|
||||
|
||||
fn from_str(s: &str) -> std::result::Result<Self, Self::Err> {
|
||||
match s.to_lowercase().as_str() {
|
||||
"source" => Ok(Self::Source),
|
||||
"uwu" => Ok(Self::Uwu),
|
||||
"ping" => Ok(Self::Ping),
|
||||
"help" => Ok(Self::Help),
|
||||
_ => Err(()),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl Display for BotCommand {
|
||||
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
||||
match self {
|
||||
BotCommand::Source => write!(f, "source"),
|
||||
BotCommand::Uwu => write!(f, "uwu"),
|
||||
BotCommand::Ping => write!(f, "ping"),
|
||||
BotCommand::Help => write!(f, "help"),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
async fn source(room: Joined) -> Result<()> {
|
||||
let content = RoomMessageEventContent::text_plain("https://git.eddie.sh/edward/matrix-bot");
|
||||
room.send(content, None).await?;
|
||||
Ok(())
|
||||
}
|
||||
|
||||
async fn uwuify(room: Joined, message: &str) -> Result<()> {
|
||||
let message = if message.is_empty() {
|
||||
if let Ok(msg) = get_previous_message(&room).await {
|
||||
uwuifier::uwuify_str_sse(&msg.body)
|
||||
} else {
|
||||
"uwu".to_owned()
|
||||
}
|
||||
} else {
|
||||
uwuifier::uwuify_str_sse(message)
|
||||
};
|
||||
let content = RoomMessageEventContent::text_plain(message);
|
||||
room.send(content, None).await?;
|
||||
Ok(())
|
||||
}
|
||||
|
||||
async fn get_previous_message(room: &Joined) -> Result<TextMessageEventContent> {
|
||||
let last_prev_batch = (**room)
|
||||
.last_prev_batch()
|
||||
.context("No previous batch key while trying to find previous message")?;
|
||||
let request = MessagesOptions::backward(&last_prev_batch);
|
||||
let messages = room.messages(request).await?;
|
||||
|
||||
messages
|
||||
.chunk
|
||||
.iter()
|
||||
.filter_map(|msg| msg.event.deserialize().ok())
|
||||
.find_map(|msg| {
|
||||
if let AnyRoomEvent::MessageLike(AnyMessageLikeEvent::RoomMessage(
|
||||
MessageLikeEvent::Original(OriginalMessageLikeEvent {
|
||||
content:
|
||||
RoomMessageEventContent {
|
||||
msgtype: MessageType::Text(content),
|
||||
..
|
||||
},
|
||||
..
|
||||
}),
|
||||
)) = msg
|
||||
{
|
||||
Some(content)
|
||||
} else {
|
||||
None
|
||||
}
|
||||
})
|
||||
.context("Failed to find previous message")
|
||||
}
|
||||
|
||||
async fn ping(room: Joined) -> Result<()> {
|
||||
let content = RoomMessageEventContent::text_plain("Pong!");
|
||||
room.send(content, None).await?;
|
||||
Ok(())
|
||||
}
|
||||
|
||||
async fn help(room: Joined) -> Result<()> {
|
||||
let mut msg = "List of commands:\n\n".to_owned();
|
||||
for command in BotCommand::into_enum_iter() {
|
||||
msg.push_str(&command.to_string());
|
||||
msg.push_str(" - ");
|
||||
msg.push_str(command.help_text());
|
||||
msg.push('\n');
|
||||
}
|
||||
let content = RoomMessageEventContent::notice_plain(msg);
|
||||
room.send(content, None).await?;
|
||||
Ok(())
|
||||
}
|
||||
|
||||
async fn unsupported_command(
|
||||
event: SyncRoomMessageEvent,
|
||||
room: Joined,
|
||||
command: &str,
|
||||
) -> Result<()> {
|
||||
// let resp = .into_full_event(room.room_id().to_owned());
|
||||
let event = if let SyncRoomMessageEvent::Original(event) = event {
|
||||
event.into_full_event(room.room_id().to_owned())
|
||||
} else {
|
||||
|
|
Loading…
Reference in a new issue