//! ### controller //! //! a controller implementation for buffer actions use operational_transform::OperationSeq; use tokio::sync::{watch, mpsc, broadcast, Mutex}; use tonic::async_trait; use crate::errors::IgnorableError; use crate::{Controller, Error}; use crate::buffer::factory::{leading_noop, tailing_noop, OperationFactory}; use super::TextChange; /// the buffer controller implementation /// /// this contains /// * a watch channel which always contains an updated view of the buffer content /// * a sink to send buffer operations into /// * a mutexed broadcast receiver for buffer operations /// * a channel to stop the associated worker /// /// for each controller a worker exists, managing outgoing and inbound /// queues, transforming outbound delayed ops and applying remote changes /// to the local buffer /// /// this controller implements [crate::buffer::OperationFactory], allowing to produce /// Operation Sequences easily /// /// upon dropping this handle will stop the associated worker pub struct BufferController { content: watch::Receiver, operations: mpsc::UnboundedSender, stream: Mutex>, stop: mpsc::UnboundedSender<()>, } impl BufferController { pub(crate) fn new( content: watch::Receiver, operations: mpsc::UnboundedSender, stream: Mutex>, stop: mpsc::UnboundedSender<()>, ) -> Self { BufferController { content, operations, stream, stop } } } impl Drop for BufferController { fn drop(&mut self) { self.stop.send(()).unwrap_or_warn("could not send stop message to worker"); } } #[async_trait] impl OperationFactory for BufferController { fn content(&self) -> String { self.content.borrow().clone() } } #[async_trait] impl Controller for BufferController { type Input = OperationSeq; /// receive an operation seq and transform it into a TextChange from buffer content async fn recv(&self) -> Result { let op = self.stream.lock().await.recv().await?; let after = self.content.borrow().clone(); let skip = leading_noop(op.ops()) as usize; let before_len = op.base_len(); let tail = tailing_noop(op.ops()) as usize; let span = skip..before_len-tail; let content = after[skip..after.len()-tail].to_string(); Ok(TextChange { span, content }) } /// enqueue an opseq for processing fn send(&self, op: OperationSeq) -> Result<(), Error> { Ok(self.operations.send(op)?) } }