Factoid: Truncate messages to 512 characters

Mavulp · Jun 27, 2020 · 33403d0 · 33403d0
1 parent a92e622
commit 33403d0
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 5 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/src/lib.rs b/src/lib.rs
@@ -230,7 +230,7 @@ where
 
 #[derive(Clone, Debug)]
 struct ThreadedPlugins<C: FrippyClient> {
-    plugins: HashMap<String, Arc<Plugin<Client = C>>>,
+    plugins: HashMap<String, Arc<dyn Plugin<Client = C>>>,
 }
 
 impl<C: FrippyClient + 'static> ThreadedPlugins<C> {

diff --git a/src/plugins/factoid/mod.rs b/src/plugins/factoid/mod.rs
@@ -139,7 +139,8 @@ impl<T: Database, C: Client> Factoid<T, C> {
             .get_factoid(name, idx)
             .context(ErrorKind::NotFound)?;
 
-        let message = factoid.content.replace("\n", "|").replace("\r", "");
+        let mut message = factoid.content.replace("\n", "|").replace("\r", "");
+        message.truncate(512);
 
         Ok(format!("{}: {}", factoid.name, message))
     }
@@ -179,7 +180,7 @@ impl<T: Database, C: Client> Factoid<T, C> {
             let factoid = self.factoids.read().get_factoid(&name, count - 1)?;
 
             let content = factoid.content;
-            let value = if content.starts_with('>') {
+            let mut message = if content.starts_with('>') {
                 let content = String::from(&content[1..]);
 
                 if content.starts_with('>') {
@@ -197,7 +198,8 @@ impl<T: Database, C: Client> Factoid<T, C> {
                 content
             };
 
-            Ok(value.replace("\n", "|").replace("\r", ""))
+            message.truncate(512);
+            Ok(message.replace("\n", "|").replace("\r", ""))
         }
     }