1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22

use tree_sitter::{Node, Tree};

// TODO(lb): Perhaps make an associated error type so that these operations can
// fail?

/// Modify a tree-sitter parse tree when printing.
pub trait Editor {
    /// Does this editor have an edit for this node?
    fn has_edit(&self, tree: &Tree, node: &Node) -> bool;

    /// Does this editor have an edit for some descendant of this node?
    fn contains_edit(&self, tree: &Tree, node: &Node) -> bool {
        if self.has_edit(tree, node) {
            return true;
        }
        node.children(&mut tree.walk())
            .any(|c| self.contains_edit(tree, &c))
    }

    /// Edit this node (precondition: [Editor::has_edit]).
    fn edit(&self, source: &[u8], tree: &Tree, node: &Node) -> Vec<u8>;
}