AppFlowy/frontend/rust-lib/flowy-document/src/entities.rs

use crate::errors::ErrorCode;
use flowy_derive::{ProtoBuf, ProtoBuf_Enum};
use std::convert::TryInto;

#[derive(PartialEq, Eq, Debug, ProtoBuf_Enum, Clone)]
pub enum ExportType {
  Text = 0,
  Markdown = 1,
  Link = 2,
}

impl Default for ExportType {
  fn default() -> Self {
    ExportType::Text
  }
}

impl From<i32> for ExportType {
  fn from(val: i32) -> Self {
    match val {
      0 => ExportType::Text,
      1 => ExportType::Markdown,
      2 => ExportType::Link,
      _ => {
        tracing::error!("Invalid export type: {}", val);
        ExportType::Text
      },
    }
  }
}

#[derive(Default, ProtoBuf)]
pub struct EditPayloadPB {
  #[pb(index = 1)]
  pub doc_id: String,

  // Encode in JSON format
  #[pb(index = 2)]
  pub operations: String,
}

#[derive(Default)]
pub struct EditParams {
  pub doc_id: String,

  // Encode in JSON format
  pub operations: String,
}

impl TryInto<EditParams> for EditPayloadPB {
  type Error = ErrorCode;
  fn try_into(self) -> Result<EditParams, Self::Error> {
    Ok(EditParams {
      doc_id: self.doc_id,
      operations: self.operations,
    })
  }
}

#[derive(Default, ProtoBuf)]
pub struct DocumentDataPB {
  #[pb(index = 1)]
  pub doc_id: String,

  /// Encode in JSON format
  #[pb(index = 2)]
  pub content: String,
}

#[derive(Default, ProtoBuf)]
pub struct ExportPayloadPB {
  #[pb(index = 1)]
  pub view_id: String,

  #[pb(index = 2)]
  pub export_type: ExportType,

  #[pb(index = 3)]
  pub document_version: DocumentVersionPB,
}

#[derive(PartialEq, Eq, Debug, ProtoBuf_Enum, Clone)]
pub enum DocumentVersionPB {
  /// this version's content of the document is build from `Delta`. It uses
  /// `DeltaDocumentEditor`.
  V0 = 0,
  /// this version's content of the document is build from `NodeTree`. It uses
  /// `AppFlowyDocumentEditor`
  V1 = 1,
}

impl std::default::Default for DocumentVersionPB {
  fn default() -> Self {
    Self::V0
  }
}

#[derive(Default, ProtoBuf)]
pub struct OpenDocumentPayloadPB {
  #[pb(index = 1)]
  pub document_id: String,

  #[pb(index = 2)]
  pub version: DocumentVersionPB,
}

#[derive(Default, Debug)]
pub struct ExportParams {
  pub view_id: String,
  pub export_type: ExportType,
  pub document_version: DocumentVersionPB,
}

impl TryInto<ExportParams> for ExportPayloadPB {
  type Error = ErrorCode;
  fn try_into(self) -> Result<ExportParams, Self::Error> {
    Ok(ExportParams {
      view_id: self.view_id,
      export_type: self.export_type,
      document_version: self.document_version,
    })
  }
}

#[derive(Default, ProtoBuf)]
pub struct ExportDataPB {
  #[pb(index = 1)]
  pub data: String,

  #[pb(index = 2)]
  pub export_type: ExportType,
}