1 files changed, 5 insertions, 1 deletions
diff --git a/src/database.rs b/src/database.rs
index b231675..7905552 100644
--- a/src/database.rs
+++ b/src/database.rs
@@ -103,7 +103,7 @@ impl Database {
         return message_id_vec;
     }
 
-    pub fn send_message(&mut self, msg: &Message) -> Result<UID, &'static str> {
+    pub fn send_message(&mut self, msg: &mut Message) -> Result<UID, &'static str> {
         // Check if user exists
         match self.get_user(msg.sender()) {
             Some(n) => match n.deleted() {
@@ -113,6 +113,10 @@ impl Database {
             None => return Err("User not found"),
         }
 
+        // Truncate whitespace on the ends of the message
+        let trunc_message = msg.message().trim().to_string();
+        msg.set_message(trunc_message);
+
         let query = "INSERT INTO messages (date, sender, message, id, reply_to, deleted) VALUES (:date, :sender, :message, :id, :reply_to, :deleted)";
         let statement = self.db.prepare(query).unwrap().into_iter()
         .bind::<&[(_, sqlite::Value)]>(&[