diff --git a/src/api/controller.rs b/src/api/controller.rs new file mode 100644 index 0000000..368be32 --- /dev/null +++ b/src/api/controller.rs @@ -0,0 +1,89 @@ +//! # Controller +//! +//! this generic trait is implemented by actors managing stream procedures. +//! events can be enqueued for dispatching without blocking ([Controller::send]), and an async blocking +//! api ([Controller::recv]) is provided to wait for server events. Additional sync blocking +//! ([Controller::blocking_recv]) and callback-based ([Controller::callback]) are implemented. + +use crate::Result; +use std::sync::Arc; +use tokio::runtime::Runtime; + +#[tonic::async_trait] +pub(crate) trait ControllerWorker { + type Controller : Controller; + type Tx; + type Rx; + + fn subscribe(&self) -> Self::Controller; + async fn work(self, tx: Self::Tx, rx: Self::Rx); +} + +/// async and threadsafe handle to a generic bidirectional stream +#[tonic::async_trait] +pub trait Controller : Sized + Send + Sync { + /// type of upstream values, used in [Self::send] + type Input; + + /// enqueue a new value to be sent + fn send(&self, x: Self::Input) -> Result<()>; + + /// get next value from stream, blocking until one is available + /// + /// this is just an async trait function wrapped by `async_trait`: + /// + /// `async fn recv(&self) -> codemp::Result;` + async fn recv(&self) -> Result { + if let Some(x) = self.try_recv()? { + return Ok(x); // short circuit if already available + } + + self.poll().await?; + Ok(self.try_recv()?.expect("no message available after polling")) + } + + /// block until next value is added to the stream without removing any element + /// + /// this is just an async trait function wrapped by `async_trait`: + /// + /// `async fn poll(&self) -> codemp::Result<()>;` + async fn poll(&self) -> Result<()>; + + /// attempt to receive a value without blocking, return None if nothing is available + fn try_recv(&self) -> Result>; + + /// sync variant of [Self::recv], blocking invoking thread + fn blocking_recv(&self, rt: &Runtime) -> Result { + rt.block_on(self.recv()) + } + + /// register a callback to be called for each received stream value + /// + /// this will spawn a new task on given runtime invoking [Self::recv] in loop and calling given + /// callback for each received value. a stop channel should be provided, and first value sent + /// into it will stop the worker loop. + /// + /// note: creating a callback handler will hold an Arc reference to the given controller, + /// preventing it from being dropped (and likely disconnecting). using the stop channel is + /// important for proper cleanup + fn callback( + self: &Arc, + rt: &tokio::runtime::Runtime, + mut stop: tokio::sync::mpsc::UnboundedReceiver<()>, + mut cb: F + ) where + Self : 'static, + F : FnMut(T) + Sync + Send + 'static + { + let _self = self.clone(); + rt.spawn(async move { + loop { + tokio::select! { + Ok(data) = _self.recv() => cb(data), + Some(()) = stop.recv() => break, + else => break, + } + } + }); + } +} diff --git a/src/api/factory.rs b/src/api/factory.rs new file mode 100644 index 0000000..51950e8 --- /dev/null +++ b/src/api/factory.rs @@ -0,0 +1,173 @@ +//! ### factory +//! +//! a helper trait to produce Operation Sequences, knowing the current +//! state of the buffer +//! +//! ```rust +//! use codemp::buffer::factory::{OperationFactory, SimpleOperationFactory}; +//! +//! let mut factory = SimpleOperationFactory::from(""); +//! let op = factory.insert("asd", 0); +//! factory.update(op)?; +//! assert_eq!(factory.content(), "asd"); +//! # Ok::<(), codemp::ot::OTError>(()) +//! ``` +//! +//! use [OperationFactory::insert] to add new characters at a specific index +//! +//! ```rust +//! # use codemp::buffer::factory::{OperationFactory, SimpleOperationFactory}; +//! # let mut factory = SimpleOperationFactory::from("asd"); +//! factory.update(factory.insert(" dsa", 3))?; +//! assert_eq!(factory.content(), "asd dsa"); +//! # Ok::<(), codemp::ot::OTError>(()) +//! ``` +//! +//! use [OperationFactory::delta] to arbitrarily change text at any position +//! +//! ```rust +//! # use codemp::buffer::factory::{OperationFactory, SimpleOperationFactory}; +//! # let mut factory = SimpleOperationFactory::from("asd dsa"); +//! let op = factory.delta(2, " xxx ", 5).expect("replaced region is equal to origin"); +//! factory.update(op)?; +//! assert_eq!(factory.content(), "as xxx sa"); +//! # Ok::<(), codemp::ot::OTError>(()) +//! ``` +//! +//! use [OperationFactory::delete] to remove characters from given index +//! +//! ```rust +//! # use codemp::buffer::factory::{OperationFactory, SimpleOperationFactory}; +//! # let mut factory = SimpleOperationFactory::from("as xxx sa"); +//! factory.update(factory.delete(2, 5))?; +//! assert_eq!(factory.content(), "assa"); +//! # Ok::<(), codemp::ot::OTError>(()) +//! ``` +//! +//! use [OperationFactory::replace] to completely replace buffer content +//! +//! ```rust +//! # use codemp::buffer::factory::{OperationFactory, SimpleOperationFactory}; +//! # let mut factory = SimpleOperationFactory::from("assa"); +//! let op = factory.replace("from scratch").expect("replace is equal to origin"); +//! factory.update(op)?; +//! assert_eq!(factory.content(), "from scratch"); +//! # Ok::<(), codemp::ot::OTError>(()) +//! ``` +//! +//! use [OperationFactory::cancel] to remove characters at index, but backwards +//! +//! ```rust +//! # use codemp::buffer::factory::{OperationFactory, SimpleOperationFactory}; +//! # let mut factory = SimpleOperationFactory::from("from scratch"); +//! factory.update(factory.cancel(12, 8))?; +//! assert_eq!(factory.content(), "from"); +//! # Ok::<(), codemp::ot::OTError>(()) +//! ``` +//! +//! +//! This trait provides an implementation for String, but plugin developers +//! should implement their own operation factory interfacing directly with the editor +//! buffer when possible. + +use std::ops::Range; + +use operational_transform::{OperationSeq, Operation}; +use similar::{TextDiff, ChangeTag}; + +/// calculate leading no-ops in given opseq +pub const fn leading_noop(seq: &[Operation]) -> u64 { count_noop(seq.first()) } + +/// calculate tailing no-ops in given opseq +pub const fn tailing_noop(seq: &[Operation]) -> u64 { count_noop(seq.last()) } + +const fn count_noop(op: Option<&Operation>) -> u64 { + match op { + None => 0, + Some(Operation::Retain(n)) => *n, + Some(_) => 0, + } +} + +/// return the range on which the operation seq is actually applying its changes +pub fn op_effective_range(op: &OperationSeq) -> Range { + let first = leading_noop(op.ops()); + let last = op.base_len() as u64 - tailing_noop(op.ops()); + first..last +} + +/// a helper trait that any string container can implement, which generates opseqs +pub trait OperationFactory { + /// the current content of the buffer + fn content(&self) -> String; + + /// completely replace the buffer with given text + fn replace(&self, txt: &str) -> Option { + self.delta(0, txt, self.content().len()) + } + + /// transform buffer in range [start..end] with given text + fn delta(&self, start: usize, txt: &str, end: usize) -> Option { + let mut out = OperationSeq::default(); + let content = self.content(); + let tail_skip = content.len() - end; // TODO len is number of bytes, not chars + let content_slice = &content[start..end]; + + if content_slice == txt { + // if slice equals given text, no operation should be taken + return None; + } + + out.retain(start as u64); + + let diff = TextDiff::from_chars(content_slice, txt); + + for change in diff.iter_all_changes() { + match change.tag() { + ChangeTag::Equal => out.retain(1), + ChangeTag::Delete => out.delete(1), + ChangeTag::Insert => out.insert(change.value()), + } + } + + out.retain(tail_skip as u64); + + Some(out) + } + + /// insert given chars at target position + fn insert(&self, txt: &str, pos: u64) -> OperationSeq { + let mut out = OperationSeq::default(); + let total = self.content().len() as u64; + out.retain(pos); + out.insert(txt); + out.retain(total - pos); + out + } + + /// delete n characters forward at given position + fn delete(&self, pos: u64, count: u64) -> OperationSeq { + let mut out = OperationSeq::default(); + let len = self.content().len() as u64; + out.retain(pos); + out.delete(count); + out.retain(len - (pos+count)); + out + } + + /// delete n characters backwards at given position + fn cancel(&self, pos: u64, count: u64) -> OperationSeq { + let mut out = OperationSeq::default(); + let len = self.content().len() as u64; + out.retain(pos - count); + out.delete(count); + out.retain(len - pos); + out + } +} + +impl OperationFactory for String { + fn content(&self) -> String { + self.clone() + } +} diff --git a/src/api/mod.rs b/src/api/mod.rs new file mode 100644 index 0000000..4a69729 --- /dev/null +++ b/src/api/mod.rs @@ -0,0 +1,14 @@ +//! # api +//! +//! these traits represent the internal api for the codemp library. +//! more methods and structs are provided but these are the core interfaces to +//! interact with the client. + +/// a generic async provider for bidirectional communication +pub mod controller; + +/// a helper trait to generate operation sequences +pub mod factory; + +pub use controller::Controller; +pub use factory::OperationFactory;