chore: fix media type option

Author: appflowy

collab-database/src/fields/type_option/media_type_option.rs

@@ -5,6 +5,7 @@ use crate::fields::{
 };
 use crate::rows::{new_cell_builder, Cell};
 
+use crate::error::DatabaseError;
 use crate::template::entity::CELL_DATA;
 use crate::template::util::TypeOptionCellData;
 use collab::util::AnyMapExt;
@@ -13,6 +14,7 @@ use serde_json::{json, Value};
 use serde_repr::Serialize_repr;
 use std::fmt::{Display, Formatter};
 use std::path::Path;
+use std::str::FromStr;
 use std::sync::Arc;
 use yrs::Any;
 
@@ -144,6 +146,22 @@ impl ToString for MediaCellData {
   }
 }
 
+impl FromStr for MediaCellData {
+  type Err = DatabaseError;
+
+  fn from_str(s: &str) -> Result<Self, Self::Err> {
+    if s.is_empty() {
+      return Ok(MediaCellData { files: vec![] });
+    }
+    let files = s
+      .split(", ")
+      .map(|file: &str| serde_json::from_str::<MediaFile>(file).unwrap_or_default())
+      .collect::<Vec<_>>();
+
+    Ok(MediaCellData { files })
+  }
+}
+
 #[derive(Clone, Debug, Default, PartialEq, Eq, Serialize, Deserialize)]
 pub struct MediaFile {
   pub id: String,
@@ -348,6 +366,44 @@ impl<'de> Deserialize<'de> for MediaUploadType {
 mod tests {
   use super::*;
   use serde_json::json;
+  #[test]
+  fn test_is_cell_empty() {
+    let empty_media_cell_data = MediaCellData { files: vec![] };
+    assert!(empty_media_cell_data.is_cell_empty());
+
+    let non_empty_media_cell_data = MediaCellData {
+      files: vec![MediaFile::new(
+        "file1.jpg".to_string(),
+        "http://example.com/file1.jpg".to_string(),
+        MediaUploadType::Local,
+        MediaFileType::Image,
+      )],
+    };
+    assert!(!non_empty_media_cell_data.is_cell_empty());
+  }
+
+  #[test]
+  fn test_media_file_rename() {
+    let original = MediaFile::new(
+      "original_name.jpg".to_string(),
+      "http://example.com/file.jpg".to_string(),
+      MediaUploadType::Local,
+      MediaFileType::Image,
+    );
+
+    let renamed = original.rename("new_name.jpg".to_string());
+    assert_eq!(renamed.name, "new_name.jpg");
+    assert_eq!(renamed.url, original.url);
+    assert_eq!(renamed.upload_type, original.upload_type);
+    assert_eq!(renamed.file_type, original.file_type);
+  }
+
+  #[test]
+  fn test_invalid_json_deserialization() {
+    let invalid_json = json!("InvalidType");
+    assert!(serde_json::from_value::<MediaUploadType>(invalid_json.clone()).is_err());
+    assert!(serde_json::from_value::<MediaFileType>(invalid_json).is_err());
+  }
 
   #[test]
   fn test_media_cell_data_to_string() {
@@ -467,4 +523,56 @@ mod tests {
       MediaFileType::Text
     );
   }
+
+  #[test]
+  fn test_convert_raw_cell_data() {
+    let media_type_option = MediaTypeOption::default();
+
+    // Test with valid JSON data
+    let valid_data =
+      r#"{"files":[{"id":"1","name":"file1","url":"url1","upload_type":0,"file_type":1}]}"#;
+    assert_eq!(
+      media_type_option.convert_raw_cell_data(valid_data),
+      "file1".to_string()
+    );
+
+    // Test with invalid JSON data
+    let invalid_data = "invalid_json";
+    assert_eq!(
+      media_type_option.convert_raw_cell_data(invalid_data),
+      "".to_string()
+    );
+
+    // Test with empty string
+    let empty_data = "";
+    assert_eq!(
+      media_type_option.convert_raw_cell_data(empty_data),
+      "".to_string()
+    );
+
+    // Test with valid JSON but missing "files" field
+    let missing_field_data = r#"{"other_field":[{"id":"1"}]}"#;
+    assert_eq!(
+      media_type_option.convert_raw_cell_data(missing_field_data),
+      "".to_string()
+    );
+
+    // Test with valid JSON but incorrect structure
+    let incorrect_structure_data = r#"{"files":"not_an_array"}"#;
+    assert_eq!(
+      media_type_option.convert_raw_cell_data(incorrect_structure_data),
+      "".to_string()
+    );
+  }
+
+  #[test]
+  fn test_numeric_cell_conversion() {
+    let mut cell = new_cell_builder(FieldType::Media);
+    cell.insert(CELL_DATA.into(), "123.45".to_string().into());
+
+    let media_type_option = MediaTypeOption::default();
+    let numeric_value = media_type_option.numeric_cell(&cell);
+
+    assert_eq!(numeric_value, Some(123.45));
+  }
 }

collab-database/src/fields/type_option/relation_type_option.rs

@@ -5,6 +5,7 @@ use crate::template::relation_parse::RelationCellData;
 use collab::util::AnyMapExt;
 use serde::{Deserialize, Serialize};
 use serde_json::{json, Value};
+use std::str::FromStr;
 
 #[derive(Debug, Clone, Default, Serialize, Deserialize)]
 pub struct RelationTypeOption {
@@ -35,13 +36,8 @@ impl TypeOptionCellReader for RelationTypeOption {
   }
 
   fn convert_raw_cell_data(&self, cell_data: &str) -> String {
-    let cell_data = RelationCellData::from(cell_data);
-    cell_data
-      .row_ids
-      .into_iter()
-      .map(|id| id.to_string())
-      .collect::<Vec<_>>()
-      .join(", ")
+    let cell_data = RelationCellData::from_str(cell_data).unwrap_or_default();
+    cell_data.to_string()
   }
 }
 

collab-database/src/template/relation_parse.rs

@@ -1,5 +1,7 @@
 use crate::entity::FieldType;
+use std::str::FromStr;
 
+use crate::error::DatabaseError;
 use crate::rows::{new_cell_builder, Cell, RowId};
 use crate::template::entity::CELL_DATA;
 use crate::template::util::TypeOptionCellData;
@@ -12,6 +14,23 @@ pub struct RelationCellData {
   pub row_ids: Vec<RowId>,
 }
 
+impl FromStr for RelationCellData {
+  type Err = DatabaseError;
+
+  fn from_str(s: &str) -> Result<Self, Self::Err> {
+    if s.is_empty() {
+      return Ok(RelationCellData { row_ids: vec![] });
+    }
+
+    let ids = s
+      .split(", ")
+      .map(|id| id.to_string().into())
+      .collect::<Vec<_>>();
+
+    Ok(RelationCellData { row_ids: ids })
+  }
+}
+
 impl TypeOptionCellData for RelationCellData {
   fn is_cell_empty(&self) -> bool {
     self.row_ids.is_empty()
@@ -52,27 +71,6 @@ impl From<RelationCellData> for Cell {
   }
 }
 
-impl From<&str> for RelationCellData {
-  fn from(s: &str) -> Self {
-    if s.is_empty() {
-      return RelationCellData { row_ids: vec![] };
-    }
-
-    let ids = s
-      .split(", ")
-      .map(|id| id.to_string().into())
-      .collect::<Vec<_>>();
-
-    RelationCellData { row_ids: ids }
-  }
-}
-
-impl From<String> for RelationCellData {
-  fn from(s: String) -> Self {
-    RelationCellData::from(s.as_str())
-  }
-}
-
 impl ToString for RelationCellData {
   fn to_string(&self) -> String {
     self