tilo_app/framework/

text.rs

//! Text controls over collaborative plain text: every edit a splice action.

use tilo_resource::text::splice_between;
use tilo_resource::{Doc, Model, Path, PlainText, Resource, Segment, TextEdit};
use ui_core::automerge::{ObjId, ReadDoc as _};
use ui_core::widget::{
    AppModel, Autocomplete, CodeEditor, Decorator, EventCtx, ModelPath, TextInput, Widget,
};
use ui_core::winit::event::WindowEvent;

use super::dispatch::Dispatcher;

/// A text control a [`TextControl`] drives: it shows the text object it is
/// bound to and keeps the user's edits local, for its owner to submit.
pub trait EditsText: Widget {
    /// Show the text object `obj`, reading it on every sync, and keep edits
    /// out of the document.
    fn show_text(&mut self, obj: ObjId);
    /// Stop showing a text object.
    fn hide_text(&mut self);
    /// The text after the user's edits since the last call, if any.
    fn take_edited(&mut self) -> Option<String>;
    /// Show `text`, which is not a user edit.
    fn reset_text(&mut self, text: &str);
    fn shown_text(&self) -> &str;
}

impl EditsText for TextInput {
    fn show_text(&mut self, obj: ObjId) {
        self.bind_text(obj);
    }
    fn hide_text(&mut self) {
        self.clear_text_binding();
    }
    fn take_edited(&mut self) -> Option<String> {
        self.take_dirty()
    }
    fn reset_text(&mut self, text: &str) {
        self.set_text(text);
    }
    fn shown_text(&self) -> &str {
        self.value()
    }
}

impl EditsText for CodeEditor {
    fn show_text(&mut self, obj: ObjId) {
        self.bind_text(obj);
    }
    fn hide_text(&mut self) {
        self.clear_text_binding();
    }
    fn take_edited(&mut self) -> Option<String> {
        self.take_dirty()
    }
    fn reset_text(&mut self, text: &str) {
        self.set_text(text);
    }
    fn shown_text(&self) -> &str {
        self.value()
    }
}

impl EditsText for Autocomplete {
    fn show_text(&mut self, obj: ObjId) {
        self.input_mut().show_text(obj);
    }
    fn hide_text(&mut self) {
        self.input_mut().hide_text();
    }
    fn take_edited(&mut self) -> Option<String> {
        Autocomplete::take_edited(self)
    }
    fn reset_text(&mut self, text: &str) {
        self.set_text(text);
    }
    fn shown_text(&self) -> &str {
        self.text()
    }
}

/// Finds the text object a control edits.
type Locate = Box<dyn Fn(&Doc) -> Option<ObjId>>;

/// Submits one splice of the text a control edits.
type Splice<R> = Box<
    dyn Fn(
        &Dispatcher<R>,
        &mut ui_core::widget::Edits<'_>,
        TextEdit,
    ) -> Result<(), tilo_actions::ActionError>,
>;

/// A text field or code editor over collaborative text — a [`PlainText`]
/// resource or child (a SQL buffer, a report note), or a text field of a
/// record at its [`Path`] ([`Self::at`]) — whose every edit is a splice
/// through `dispatcher`.
///
/// So typing is permission-checked (a viewer's keystrokes change nothing and
/// the control shows the stored text again), attributed, undone a word at a
/// time, and merges with a collaborator typing in the same text character
/// by character. The control reads the text object `locate` finds on every
/// sync; a collaborator's edit, an undo or a restore shows as it lands.
/// Without a dispatcher ([`view_only`](Self::view_only)) it shows the text
/// and every edit is taken back.
pub struct TextControl<W: EditsText, R: Resource = PlainText> {
    control: W,
    dispatch: Option<Dispatcher<R>>,
    locate: Locate,
    splice: Splice<R>,
    /// The text object shown.
    bound: Option<ObjId>,
    /// The text as the document holds it, which the next edit is a splice of.
    stored: String,
}

impl<W: EditsText> TextControl<W> {
    /// `control` over the text `locate` finds, edited through `dispatcher`.
    pub fn new(
        control: W,
        dispatcher: Dispatcher<PlainText>,
        locate: impl Fn(&Doc) -> Option<ObjId> + 'static,
    ) -> Self {
        let mut text = Self::view_only(control, locate);
        text.attach(dispatcher);
        text
    }

    /// `control` showing the text `locate` finds, never editing it.
    pub fn view_only(control: W, locate: impl Fn(&Doc) -> Option<ObjId> + 'static) -> Self {
        Self {
            control,
            dispatch: None,
            locate: Box::new(locate),
            splice: Box::new(|dispatch, edits, edit| dispatch.apply_now(edits, &edit)),
            bound: None,
            stored: String::new(),
        }
    }
}

impl<W: EditsText, R: Resource<State = R> + Model> TextControl<W, R> {
    /// `control` over the text field of the record at `path`, each edit a
    /// field edit there (`path.splice(..)`) through `dispatcher`: for a
    /// hand-written surface's own editor.
    pub fn at(control: W, dispatcher: Dispatcher<R>, path: Path<R, String>) -> Self {
        let mut text = Self::view_only_at(control, path);
        text.attach(dispatcher);
        text
    }

    /// `control` showing the text field at `path`, every edit taken back
    /// until [`attach`](Self::attach) hands it a dispatcher: for a surface
    /// that builds its editors before the framework attaches it.
    pub fn view_only_at(control: W, path: Path<R, String>) -> Self {
        let segments = path.segments().to_vec();
        Self {
            control,
            dispatch: None,
            locate: Box::new(move |doc| text_at(doc, &segments)),
            splice: Box::new(move |dispatch, edits, edit| {
                let edit = match edit {
                    TextEdit::Splice { at, delete, insert } => path.splice(at, delete, insert),
                    TextEdit::ReplaceText { text } => path.replace(text),
                };
                dispatch.submit_now(edits, edit)
            }),
            bound: None,
            stored: String::new(),
        }
    }
}

/// The text object at `path` in `doc`: fields and map entries by key, items
/// of a collection through their entry's value.
fn text_at(doc: &Doc, path: &[Segment]) -> Option<ObjId> {
    let mut obj = ui_core::automerge::ROOT;
    for segment in path {
        let child = |obj: &ObjId, key: &str| match doc.get(obj, key) {
            Ok(Some((ui_core::automerge::Value::Object(_), id))) => Some(id),
            _ => None,
        };
        obj = match segment {
            Segment::Field(key) | Segment::Key(key) => child(&obj, key)?,
            Segment::Item(id) => {
                let entry = child(&obj, id.as_str())?;
                child(&entry, tilo_resource::shape::ENTRY_VALUE_KEY)?
            }
            Segment::Cell { .. } => return None,
        };
    }
    Some(obj)
}

impl<W: EditsText, R: Resource> TextControl<W, R> {
    /// Edit through `dispatcher` from now on.
    pub fn attach(&mut self, dispatcher: Dispatcher<R>) {
        self.dispatch = Some(dispatcher);
    }

    pub fn control(&self) -> &W {
        &self.control
    }

    /// The control, to configure it (a caret, a selection, a highlighter).
    pub fn control_mut(&mut self) -> &mut W {
        &mut self.control
    }

    /// Bind the text `locate` finds now in `doc`: for a host outside the
    /// widget tree, before it reads the control each frame.
    pub fn pull(&mut self, doc: &Doc) {
        self.follow(doc);
        self.control.sync(doc, &ui_core::automerge::ROOT);
        self.stored = self.control.shown_text().to_owned();
    }

    /// Submit the control's edit since the last call as one splice: for a
    /// host outside the widget tree, after it handed the control an event.
    pub fn submit(&mut self, edits: &mut ui_core::widget::Edits<'_>) {
        let Some(edited) = self.control.take_edited() else {
            return;
        };
        let Some(edit) = splice_between(&self.stored, &edited) else {
            return;
        };
        let applied = match &self.dispatch {
            Some(dispatch) => (self.splice)(dispatch, edits, edit),
            None => Err(tilo_actions::ActionError::Refused(
                "this text is open to view only".into(),
            )),
        };
        match applied {
            Ok(()) => self.stored = edited,
            Err(error) => {
                tracing::warn!(%error, "text edit refused");
                let stored = self.stored.clone();
                self.control.reset_text(&stored);
            }
        }
    }

    fn follow(&mut self, doc: &Doc) {
        let located = (self.locate)(doc);
        if located == self.bound {
            return;
        }
        if let Some(obj) = &located {
            self.control.show_text(obj.clone());
        } else {
            self.control.hide_text();
            self.control.reset_text("");
        }
        self.bound = located;
    }
}

// The control's layout, drawing and input are its own; the wrapper binds it
// to the text before each sync and turns its edits into splices after.
impl<W: EditsText + 'static, R: Resource> Decorator for TextControl<W, R> {
    type Inner = W;

    fn inner(&self) -> &W {
        &self.control
    }

    fn inner_mut(&mut self) -> &mut W {
        &mut self.control
    }

    fn sync(&mut self, state: &AppModel, path: &ModelPath) {
        {
            let (doc, _) = ui_core::doc::document_ref(state, path);
            self.follow(doc);
        }
        self.control.sync(state, path);
        self.stored = self.control.shown_text().to_owned();
    }

    fn handle_event(&mut self, event: &WindowEvent, ctx: &mut EventCtx<'_>) {
        self.control.handle_event(event, ctx);
        self.submit(&mut ctx.edits());
    }

    fn perform_action(
        &mut self,
        action: &ui_core::a11y::Action,
        key_path: &[u64],
        state: &mut ui_core::widget::Edits<'_>,
        path: &ModelPath,
    ) -> bool {
        let handled = self.control.perform_action(action, key_path, state, path);
        self.submit(state);
        handled
    }

    fn perform_semantic_action(
        &mut self,
        target: &str,
        action: &ui_core::a11y::Action,
        state: &mut ui_core::widget::Edits<'_>,
        path: &ModelPath,
    ) -> bool {
        let handled = self
            .control
            .perform_semantic_action(target, action, state, path);
        self.submit(state);
        handled
    }
}