ktrs-lint 0.3.0

Port of the ktlint 2.0 rule engine and standard rules over the ktrs mutable AST
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

//! Port of ktlint-ruleset-standard `NoBlankLineBeforeRbraceRule.kt`.

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

use crate::ast_node_edit::AstNodeEdit;
use crate::ast_node_extension::AstNodeExtension;
use crate::rule::{About, Emit, RuleId, RuleV2, TokenSet};
use crate::rules::STANDARD_RULE_ABOUT;

const VISITED_TYPES: TokenSet = TokenSet::create(&[WHITE_SPACE]);

pub struct NoBlankLineBeforeRbraceRule;

impl RuleV2 for NoBlankLineBeforeRbraceRule {
    fn rule_id(&self) -> RuleId {
        RuleId("standard:no-blank-line-before-rbrace")
    }

    fn visited_types(&self) -> Option<TokenSet> {
        Some(VISITED_TYPES)
    }

    fn about(&self) -> About {
        STANDARD_RULE_ABOUT
    }

    fn before_visit_child_nodes(&mut self, ast: &mut Ast, node: NodeId, emit: &mut Emit<'_>) {
        if ast.is_white_space_with_newline(node) && ast.next_leaf(node).map(|it| ast.element_type(it)) == Some(RBRACE) {
            let text = ast.leaf_text(node).to_owned();
            let split: Vec<&str> = text.split('\n').collect();
            if split.len() > 2 {
                emit(ast, ast.start_offset(node) + split[0].len() + split[1].len() + 1, "Unexpected blank line(s) before \"}\"", true)
                    .if_autocorrect_allowed(|| ast.replace_text_with(node, &format!("{}\n{}", split[0], split[split.len() - 1])));
            }
        }
    }
}