feat: support ai meeting block (#432)

Author: LucasXu0

collab/src/document/block_parser/document_parser.rs

@@ -1,4 +1,5 @@
 use super::{
+  AiMeetingNotesParser, AiMeetingParser, AiMeetingSummaryParser, AiMeetingTranscriptionParser,
   BlockParserRegistry, BulletedListParser, CalloutParser, CodeBlockParser, DividerParser,
   DocumentParserDelegate, FileBlockParser, HeadingParser, ImageParser, LinkPreviewParser,
   MathEquationParser, NumberedListParser, OutputFormat, PageParser, ParagraphParser, ParseContext,
@@ -82,7 +83,11 @@ impl DocumentParser {
       .register(Arc::new(SimpleTableParser))
       .register(Arc::new(SimpleTableRowParser))
       .register(Arc::new(SimpleTableCellParser))
-      .register(Arc::new(SubpageParser));
+      .register(Arc::new(SubpageParser))
+      .register(Arc::new(AiMeetingParser))
+      .register(Arc::new(AiMeetingSummaryParser))
+      .register(Arc::new(AiMeetingNotesParser))
+      .register(Arc::new(AiMeetingTranscriptionParser));
 
     parser
   }

collab/src/document/block_parser/parsers/ai_meeting.rs

@@ -0,0 +1,15 @@
+use super::super::{BlockParser, ParseContext, ParseResult};
+use crate::document::blocks::{Block, BlockType};
+use crate::error::CollabError;
+
+pub struct AiMeetingParser;
+
+impl BlockParser for AiMeetingParser {
+  fn parse(&self, _block: &Block, _context: &ParseContext) -> Result<ParseResult, CollabError> {
+    Ok(ParseResult::container("".to_string()))
+  }
+
+  fn block_type(&self) -> &'static str {
+    BlockType::AiMeeting.as_str()
+  }
+}

collab/src/document/block_parser/parsers/ai_meeting_notes.rs

@@ -0,0 +1,15 @@
+use super::super::{BlockParser, ParseContext, ParseResult};
+use crate::document::blocks::{Block, BlockType};
+use crate::error::CollabError;
+
+pub struct AiMeetingNotesParser;
+
+impl BlockParser for AiMeetingNotesParser {
+  fn parse(&self, _block: &Block, _context: &ParseContext) -> Result<ParseResult, CollabError> {
+    Ok(ParseResult::container("".to_string()))
+  }
+
+  fn block_type(&self) -> &'static str {
+    BlockType::AiMeetingNotes.as_str()
+  }
+}

collab/src/document/block_parser/parsers/ai_meeting_summary.rs

@@ -0,0 +1,15 @@
+use super::super::{BlockParser, ParseContext, ParseResult};
+use crate::document::blocks::{Block, BlockType};
+use crate::error::CollabError;
+
+pub struct AiMeetingSummaryParser;
+
+impl BlockParser for AiMeetingSummaryParser {
+  fn parse(&self, _block: &Block, _context: &ParseContext) -> Result<ParseResult, CollabError> {
+    Ok(ParseResult::container("".to_string()))
+  }
+
+  fn block_type(&self) -> &'static str {
+    BlockType::AiMeetingSummary.as_str()
+  }
+}

collab/src/document/block_parser/parsers/ai_meeting_transcription.rs

@@ -0,0 +1,15 @@
+use super::super::{BlockParser, ParseContext, ParseResult};
+use crate::document::blocks::{Block, BlockType};
+use crate::error::CollabError;
+
+pub struct AiMeetingTranscriptionParser;
+
+impl BlockParser for AiMeetingTranscriptionParser {
+  fn parse(&self, _block: &Block, _context: &ParseContext) -> Result<ParseResult, CollabError> {
+    Ok(ParseResult::container("".to_string()))
+  }
+
+  fn block_type(&self) -> &'static str {
+    BlockType::AiMeetingTranscription.as_str()
+  }
+}

collab/src/document/block_parser/parsers/mod.rs

@@ -1,3 +1,7 @@
+pub mod ai_meeting;
+pub mod ai_meeting_notes;
+pub mod ai_meeting_summary;
+pub mod ai_meeting_transcription;
 pub mod bulleted_list;
 pub mod callout;
 pub mod code_block;
@@ -20,6 +24,10 @@ pub mod subpage;
 pub mod todo_list;
 pub mod toggle_list;
 
+pub use ai_meeting::*;
+pub use ai_meeting_notes::*;
+pub use ai_meeting_summary::*;
+pub use ai_meeting_transcription::*;
 pub use bulleted_list::*;
 pub use callout::*;
 pub use code_block::*;

collab/src/document/blocks/block_types.rs

@@ -35,6 +35,10 @@ pub enum BlockType {
   SimpleTableCell,
   SimpleColumns,
   SimpleColumn,
+  AiMeeting,
+  AiMeetingSummary,
+  AiMeetingNotes,
+  AiMeetingTranscription,
   Custom(String),
 
   // Legacy types
@@ -74,6 +78,10 @@ impl BlockType {
       BlockType::SimpleTableCell => "simple_table_cell",
       BlockType::SimpleColumns => "simple_columns",
       BlockType::SimpleColumn => "simple_column",
+      BlockType::AiMeeting => "ai_meeting",
+      BlockType::AiMeetingSummary => "ai_meeting_summary",
+      BlockType::AiMeetingNotes => "ai_meeting_notes",
+      BlockType::AiMeetingTranscription => "ai_meeting_transcription",
       BlockType::Table => "table",
       BlockType::TableCell => "table/cell",
       BlockType::Custom(s) => s,
@@ -111,6 +119,10 @@ impl BlockType {
       "simple_table_cell" => BlockType::SimpleTableCell,
       "simple_columns" => BlockType::SimpleColumns,
       "simple_column" => BlockType::SimpleColumn,
+      "ai_meeting" => BlockType::AiMeeting,
+      "ai_meeting_summary" => BlockType::AiMeetingSummary,
+      "ai_meeting_notes" => BlockType::AiMeetingNotes,
+      "ai_meeting_transcription" => BlockType::AiMeetingTranscription,
       "table" => BlockType::Table,
       "table/cell" => BlockType::TableCell,
       _ => BlockType::Custom(s.to_string()),

collab/tests/document/block_parser/ai_meeting_test.rs

@@ -0,0 +1,143 @@
+use crate::blocks::block_test_core::{BlockTestCore, generate_id};
+use collab::document::block_parser::parsers::ai_meeting::AiMeetingParser;
+use collab::document::block_parser::{BlockParser, DocumentParser, OutputFormat, ParseContext};
+use collab::document::blocks::{Block, BlockType};
+use serde_json::json;
+use std::collections::HashMap;
+
+fn create_ai_meeting_block(test: &mut BlockTestCore, parent_id: &str) -> Block {
+  create_block(test, BlockType::AiMeeting, parent_id)
+}
+
+fn create_ai_meeting_summary_block(test: &mut BlockTestCore, parent_id: &str) -> Block {
+  create_block(test, BlockType::AiMeetingSummary, parent_id)
+}
+
+fn create_ai_meeting_notes_block(test: &mut BlockTestCore, parent_id: &str) -> Block {
+  create_block(test, BlockType::AiMeetingNotes, parent_id)
+}
+
+fn create_ai_meeting_transcription_block(test: &mut BlockTestCore, parent_id: &str) -> Block {
+  create_block(test, BlockType::AiMeetingTranscription, parent_id)
+}
+
+fn create_block(test: &mut BlockTestCore, ty: BlockType, parent_id: &str) -> Block {
+  let data = HashMap::new();
+  let actual_parent_id = if parent_id.is_empty() {
+    test.get_page().id
+  } else {
+    parent_id.to_string()
+  };
+
+  let block = Block {
+    id: generate_id(),
+    ty: ty.as_str().to_string(),
+    parent: actual_parent_id,
+    children: generate_id(),
+    external_id: None,
+    external_type: None,
+    data,
+  };
+
+  test.document.insert_block(block, None).unwrap()
+}
+
+fn create_text_block(test: &mut BlockTestCore, text: &str, parent_id: &str) -> Block {
+  let delta = json!([{ "insert": text }]).to_string();
+  let external_id = test.create_text(delta);
+  let data = HashMap::new();
+
+  let block = Block {
+    id: generate_id(),
+    ty: BlockType::Paragraph.as_str().to_string(),
+    parent: parent_id.to_string(),
+    children: generate_id(),
+    external_id: Some(external_id),
+    external_type: Some("text".to_string()),
+    data,
+  };
+  test.document.insert_block(block, None).unwrap()
+}
+
+fn create_heading_block(
+  test: &mut BlockTestCore,
+  text: &str,
+  level: u32,
+  parent_id: &str,
+) -> Block {
+  let delta = json!([{ "insert": text }]).to_string();
+  let external_id = test.create_text(delta);
+  let mut data = HashMap::new();
+  data.insert("level".to_string(), json!(level));
+
+  let block = Block {
+    id: generate_id(),
+    ty: BlockType::Heading.as_str().to_string(),
+    parent: parent_id.to_string(),
+    children: generate_id(),
+    external_id: Some(external_id),
+    external_type: Some("text".to_string()),
+    data,
+  };
+  test.document.insert_block(block, None).unwrap()
+}
+
+#[test]
+fn test_ai_meeting_parser() {
+  let mut test = BlockTestCore::new();
+  let document_parser = DocumentParser::with_default_parsers();
+
+  let ai_meeting = create_ai_meeting_block(&mut test, "");
+
+  let transcription = create_ai_meeting_transcription_block(&mut test, &ai_meeting.id);
+  create_text_block(&mut test, "Transcription content.", &transcription.id);
+
+  let notes = create_ai_meeting_notes_block(&mut test, &ai_meeting.id);
+  create_text_block(&mut test, "Notes content.", &notes.id);
+
+  let summary = create_ai_meeting_summary_block(&mut test, &ai_meeting.id);
+  create_text_block(&mut test, "Summary content.", &summary.id);
+  create_heading_block(&mut test, "Overview", 4, &summary.id);
+
+  let document_data = test.get_document_data();
+  let context = ParseContext::new(&document_data, &document_parser, OutputFormat::Markdown);
+
+  let result = document_parser.parse_block(&ai_meeting, &context).unwrap();
+
+  let expected = r#"#### Overview
+Summary content.
+Notes content.
+Transcription content."#;
+
+  assert_eq!(result, expected);
+}
+
+#[test]
+fn test_ai_meeting_parser_plain_text() {
+  let mut test = BlockTestCore::new();
+  let document_parser = DocumentParser::with_default_parsers();
+
+  let ai_meeting = create_ai_meeting_block(&mut test, "");
+
+  let transcription = create_ai_meeting_transcription_block(&mut test, &ai_meeting.id);
+  create_text_block(&mut test, "Transcription content.", &transcription.id);
+
+  let notes = create_ai_meeting_notes_block(&mut test, &ai_meeting.id);
+  create_text_block(&mut test, "Notes content.", &notes.id);
+
+  let summary = create_ai_meeting_summary_block(&mut test, &ai_meeting.id);
+  create_text_block(&mut test, "Summary content.", &summary.id);
+  create_heading_block(&mut test, "Overview", 4, &summary.id);
+
+  let document_data = test.get_document_data();
+  let context = ParseContext::new(&document_data, &document_parser, OutputFormat::PlainText);
+
+  let result = document_parser.parse_block(&ai_meeting, &context).unwrap();
+
+  let expected = r#"Overview
+Summary content.
+Notes content.
+Transcription content."#;
+
+  assert_eq!(result, expected);
+}

collab/tests/document/block_parser/mod.rs

@@ -1,3 +1,4 @@
+mod ai_meeting_test;
 mod bulleted_list_test;
 mod callout_test;
 mod code_block_test;