From e0c91d22f26ea1a5cebbb2c8a725265a2859739c Mon Sep 17 00:00:00 2001
From: Cory Forsstrom <cforsstrom18@gmail.com>
Date: Sun, 21 Jul 2024 10:28:21 -0700
Subject: [PATCH] Serialize text field for downgrade compatability

---
 data/src/message.rs | 52 ++++++++++++++++++++++++++++++++++++++++-----
 1 file changed, 47 insertions(+), 5 deletions(-)

diff --git a/data/src/message.rs b/data/src/message.rs
index 9cc3461bf..a22736927 100644
--- a/data/src/message.rs
+++ b/data/src/message.rs
@@ -1,6 +1,9 @@
+use std::borrow::Cow;
+
 use chrono::{DateTime, Utc};
 use irc::proto;
 use irc::proto::Command;
+use itertools::Itertools;
 use once_cell::sync::Lazy;
 use regex::Regex;
 use serde::{Deserialize, Serialize};
@@ -86,7 +89,7 @@ pub enum Direction {
     Received,
 }
 
-#[derive(Debug, Clone, Serialize)]
+#[derive(Debug, Clone)]
 pub struct Message {
     pub received_at: Posix,
     pub server_time: DateTime<Utc>,
@@ -158,6 +161,35 @@ impl Message {
     }
 }
 
+impl Serialize for Message {
+    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
+    where
+        S: serde::Serializer,
+    {
+        #[derive(Serialize)]
+        struct Data<'a> {
+            received_at: &'a Posix,
+            server_time: &'a DateTime<Utc>,
+            direction: &'a Direction,
+            target: &'a Target,
+            content: &'a Content,
+            // Old field before we had fragments,
+            // added for downgrade compatability
+            text: Cow<'a, str>,
+        }
+
+        Data {
+            received_at: &self.received_at,
+            server_time: &self.server_time,
+            direction: &self.direction,
+            target: &self.target,
+            content: &self.content,
+            text: self.content.text(),
+        }
+        .serialize(serializer)
+    }
+}
+
 impl<'de> Deserialize<'de> for Message {
     fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
     where
@@ -169,7 +201,7 @@ impl<'de> Deserialize<'de> for Message {
             server_time: DateTime<Utc>,
             direction: Direction,
             target: Target,
-            // New field
+            // New field, optional for upgrade compatability
             content: Option<Content>,
             // Old field before we had fragments
             text: Option<String>,
@@ -184,10 +216,11 @@ impl<'de> Deserialize<'de> for Message {
             text,
         } = Data::deserialize(deserializer)?;
 
-        let content = if let Some(text) = text {
-            parse_fragments(text)
-        } else if let Some(content) = content {
+        let content = if let Some(content) = content {
             content
+        } else if let Some(text) = text {
+            // First time upgrading, convert text into content
+            parse_fragments(text)
         } else {
             // Unreachable
             Content::Plain("".to_string())
@@ -239,6 +272,15 @@ pub enum Content {
     Fragments(Vec<Fragment>),
 }
 
+impl Content {
+    fn text(&self) -> Cow<str> {
+        match self {
+            Content::Plain(s) => s.into(),
+            Content::Fragments(fragments) => fragments.iter().map(Fragment::as_str).join("").into(),
+        }
+    }
+}
+
 #[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
 pub enum Fragment {
     Text(String),