summary refs log tree commit diff
path: root/src/system/plugin
diff options
context:
space:
mode:
Diffstat (limited to 'src/system/plugin')
-rw-r--r--src/system/plugin/edit.rs11
-rw-r--r--src/system/plugin/reproxy.rs12
2 files changed, 3 insertions, 20 deletions
diff --git a/src/system/plugin/edit.rs b/src/system/plugin/edit.rs
index 2edc504..1fd4e23 100644
--- a/src/system/plugin/edit.rs
+++ b/src/system/plugin/edit.rs
@@ -24,16 +24,7 @@ impl<'system> SeancePlugin<'system> for Edit {
     }
 
     async fn handle_command<'message>(&self, logger: &'system Logger, system: &'system System, message: &'message Message, command: PluginCommand, args: Vec<&'message str>) {
-        let most_recent_message = system.get_most_recent_message(message.channel_id).await;
-
-        if let Some((edit_target, update_most_recent)) = match message.kind {
-            twilight_model::channel::message::MessageType::Regular => most_recent_message.map(|(message, _)| (message, true)),
-            twilight_model::channel::message::MessageType::Reply => async {Some((
-                message.referenced_message.as_ref()?.as_ref().clone(),
-                message.referenced_message.as_ref()?.id == most_recent_message?.0.id
-            ))}.await,
-            _ => todo!(),
-        } {
+        if let Some((edit_target, update_most_recent)) = system.resolve_message_target(message).await {
             if let Some(authoring_member) = system.get_member_by_id(edit_target.author.id).await {
                 if let Some(edit_contents) = match command {
                     PluginCommand::Word("edit") => Some(args.join(" ")),
diff --git a/src/system/plugin/reproxy.rs b/src/system/plugin/reproxy.rs
index 145bed8..30b0c93 100644
--- a/src/system/plugin/reproxy.rs
+++ b/src/system/plugin/reproxy.rs
@@ -27,19 +27,11 @@ impl<'system> SeancePlugin<'system> for Reproxy {
     async fn handle_message<'message>(&self, logger: &'system Logger, system: &'system System, message: &'message Message, response: &'message mut Response) {
         if let Response::Proxy { member, content } = response {
             if content.as_str().trim() == "*" {
-                let most_recent_message = system.get_most_recent_message(message.channel_id).await;
-                if let Some((reproxy_target, update_most_recent)) = match message.kind {
-                    twilight_model::channel::message::MessageType::Regular => most_recent_message.map(|(message, _)| (message, true)),
-                    twilight_model::channel::message::MessageType::Reply => async {Some((
-                        message.referenced_message.as_ref()?.as_ref().clone(),
-                        message.referenced_message.as_ref()?.id == most_recent_message?.0.id
-                    ))}.await,
-                    _ => todo!(),
-                } {
+                if let Some((reproxy_target, is_most_recent)) = system.resolve_message_target(message).await {
                     if let Ok(new_message) = util::duplicate_message(&member.client, &reproxy_target, reproxy_target.content.as_str()).await {
                         let _ = {member.client.lock().await.delete_message(reproxy_target.channel_id, reproxy_target.id).await};
 
-                        if update_most_recent {
+                        if is_most_recent {
                             system.cache_most_recent_message(new_message.channel_id, new_message, member.clone()).await;
                         }