emmylua_parser/syntax/
comment_trait.rs

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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
use crate::{
    kind::{LuaSyntaxKind, LuaTokenKind},
    LuaKind,
};

use super::{node::LuaComment, traits::LuaAstNode, LuaSyntaxNode};

#[allow(unused)]
pub trait LuaCommentOwner: LuaAstNode {
    fn get_comments(&self) -> Vec<LuaComment> {
        let mut comments = vec![];
        if let Some(attached_comment) = find_attached_comment(self.syntax()) {
            comments.push(LuaComment::cast(attached_comment).unwrap());
        }

        if let Some(inline_comment) = find_inline_comment(self.syntax()) {
            comments.push(LuaComment::cast(inline_comment).unwrap());
        }

        comments
    }

    fn get_left_comment(&self) -> Option<LuaComment> {
        if let Some(attached_comment) = find_attached_comment(self.syntax()) {
            return Some(LuaComment::cast(attached_comment)?);
        }

        None
    }
}

fn find_attached_comment(node: &LuaSyntaxNode) -> Option<LuaSyntaxNode> {
    let mut prev_sibling = node.prev_sibling_or_token();
    let mut meet_end_of_line = false;
    for _ in 0..=2 {
        if prev_sibling.is_none() {
            return None;
        }

        if let Some(sibling) = &prev_sibling {
            match sibling.kind() {
                LuaKind::Token(LuaTokenKind::TkWhitespace) => {}
                LuaKind::Token(LuaTokenKind::TkEndOfLine) => {
                    if meet_end_of_line {
                        return None;
                    }
                    meet_end_of_line = true;
                }
                LuaKind::Syntax(LuaSyntaxKind::Comment) => return sibling.clone().into_node(),
                _ => {
                    return None;
                }
            }
        }
        prev_sibling = prev_sibling.unwrap().prev_sibling_or_token();
    }

    None
}

fn find_inline_comment(node: &LuaSyntaxNode) -> Option<LuaSyntaxNode> {
    let mut next_sibling = node.next_sibling_or_token();
    for _ in 0..=3 {
        if next_sibling.is_none() {
            return None;
        }

        if let Some(sibling) = &next_sibling {
            match sibling.kind() {
                LuaKind::Token(
                    LuaTokenKind::TkWhitespace | LuaTokenKind::TkComma | LuaTokenKind::TkSemicolon,
                ) => {}
                LuaKind::Syntax(LuaSyntaxKind::Comment) => return sibling.clone().into_node(),
                _ => {
                    return None;
                }
            }
        }
        next_sibling = next_sibling.unwrap().next_sibling_or_token();
    }

    None
}