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

use bytes::Bytes;
use flowy_error::{FlowyError, FlowyResult};
use flowy_revision::{RevisionObjectDeserializer, RevisionObjectSerializer};
use flowy_sync::entities::revision::Revision;
use lib_ot::core::{Body, Extension, Interval, NodeDataBuilder, NodeOperation, NodeTree, NodeTreeContext, Transaction};
use lib_ot::text_delta::TextOperationBuilder;

#[derive(Debug)]
pub struct Document {
    tree: NodeTree,
}

impl Document {
    pub fn new(tree: NodeTree) -> Self {
        Self { tree }
    }

    pub fn from_transaction(transaction: Transaction) -> FlowyResult<Self> {
        let tree = NodeTree::from_operations(transaction.operations, make_tree_context())?;
        Ok(Self { tree })
    }

    pub fn get_content(&self, pretty: bool) -> FlowyResult<String> {
        if pretty {
            serde_json::to_string_pretty(self).map_err(|err| FlowyError::serde().context(err))
        } else {
            serde_json::to_string(self).map_err(|err| FlowyError::serde().context(err))
        }
    }

    pub fn get_tree(&self) -> &NodeTree {
        &self.tree
    }
}

pub(crate) fn make_tree_context() -> NodeTreeContext {
    NodeTreeContext {}
}

pub fn initial_document_content() -> String {
    let delta = TextOperationBuilder::new().insert("").build();
    let node_data = NodeDataBuilder::new("text").insert_body(Body::Delta(delta)).build();
    let editor_node = NodeDataBuilder::new("editor").add_node_data(node_data).build();
    let node_operation = NodeOperation::Insert {
        path: vec![0].into(),
        nodes: vec![editor_node],
    };
    let extension = Extension::TextSelection {
        before_selection: Interval::default(),
        after_selection: Interval::default(),
    };
    let transaction = Transaction {
        operations: vec![node_operation].into(),
        extension,
    };
    transaction.to_json().unwrap()
}

impl std::ops::Deref for Document {
    type Target = NodeTree;

    fn deref(&self) -> &Self::Target {
        &self.tree
    }
}

impl std::ops::DerefMut for Document {
    fn deref_mut(&mut self) -> &mut Self::Target {
        &mut self.tree
    }
}

pub struct DocumentRevisionSerde();
impl RevisionObjectDeserializer for DocumentRevisionSerde {
    type Output = Document;

    fn deserialize_revisions(_object_id: &str, revisions: Vec<Revision>) -> FlowyResult<Self::Output> {
        let mut tree = NodeTree::new(make_tree_context());
        let transaction = make_transaction_from_revisions(revisions)?;
        let _ = tree.apply_transaction(transaction)?;
        let document = Document::new(tree);
        Result::<Document, FlowyError>::Ok(document)
    }
}

impl RevisionObjectSerializer for DocumentRevisionSerde {
    fn combine_revisions(revisions: Vec<Revision>) -> FlowyResult<Bytes> {
        let transaction = make_transaction_from_revisions(revisions)?;
        Ok(Bytes::from(transaction.to_bytes()?))
    }
}

fn make_transaction_from_revisions(revisions: Vec<Revision>) -> FlowyResult<Transaction> {
    let mut transaction = Transaction::new();
    for revision in revisions {
        let _ = transaction.compose(Transaction::from_bytes(&revision.bytes)?)?;
    }
    Ok(transaction)
}