ktrs-compose 0.5.0

Port of mrmans0n/compose-rules (the ktlint rule set `compose`) over the ktrs-lint engine
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45

//! Port of `core/util/ASTNodes.kt`.

use ktrs_ast::psi::PsiComment;
use ktrs_ast::{Ast, NodeId};
use ktrs_syntax::SyntaxKind::WHITE_SPACE;

pub fn last_child_leaf_or_self(ast: &Ast, node: NodeId) -> NodeId {
    let mut node = node;
    while let Some(last) = ast.last_child_node(node) {
        node = last;
    }
    node
}

pub fn first_child_leaf_or_self(ast: &Ast, node: NodeId) -> NodeId {
    let mut node = node;
    while let Some(first) = ast.first_child_node(node) {
        node = first;
    }
    node
}

/// `ASTNode.parent(p, strict)`.
pub fn parent(ast: &Ast, node: NodeId, p: impl Fn(&Ast, NodeId) -> bool, strict: bool) -> Option<NodeId> {
    let mut n = if strict { ast.tree_parent(node) } else { Some(node) };
    while let Some(current) = n {
        if p(ast, current) {
            return Some(current);
        }
        n = ast.tree_parent(current);
    }
    None
}

pub fn is_part_of_comment(ast: &Ast, node: NodeId) -> bool {
    parent(ast, node, PsiComment::is, false).is_some()
}

pub fn next_code_sibling(ast: &Ast, node: NodeId) -> Option<NodeId> {
    next_sibling(ast, node, |ast, it| ast.element_type(it) != WHITE_SPACE && !is_part_of_comment(ast, it))
}

pub fn next_sibling(ast: &Ast, node: NodeId, p: impl Fn(&Ast, NodeId) -> bool) -> Option<NodeId> {
    ast.siblings(node, true).find(|&it| p(ast, it))
}