glua_parser_desc 0.1.5

A parser for markup within Lua comments
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
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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118

use glua_parser::{LexerConfig, LexerState, LuaLexer, LuaTokenData, LuaTokenKind, Reader};

use crate::{CodeBlockHighlightKind, DescItemKind, util::ResultContainer};

pub fn process_lua_code_block<'a, C: ResultContainer>(
    c: &mut C,
    line_reader: Reader<'a>,
    state: LexerState,
) -> LexerState {
    let mut lexer = LuaLexer::new_with_state(line_reader, state, LexerConfig::default(), None);
    let tokens = lexer.tokenize();
    for (i, token) in tokens.iter().enumerate() {
        if !matches!(
            token.kind,
            LuaTokenKind::TkEof | LuaTokenKind::TkEndOfLine | LuaTokenKind::TkWhitespace
        ) {
            let highlight_kind = to_highlight_kind(token, i, &tokens);
            if highlight_kind != CodeBlockHighlightKind::None {
                c.emit_range(token.range, DescItemKind::CodeBlockHl(highlight_kind));
            }
        }
    }

    lexer.get_state()
}

fn to_highlight_kind(
    token: &LuaTokenData,
    i: usize,
    tokens: &[LuaTokenData],
) -> CodeBlockHighlightKind {
    match token.kind {
        LuaTokenKind::TkLongString | LuaTokenKind::TkString => CodeBlockHighlightKind::String,
        LuaTokenKind::TkAnd
        | LuaTokenKind::TkBreak
        | LuaTokenKind::TkDo
        | LuaTokenKind::TkElse
        | LuaTokenKind::TkElseIf
        | LuaTokenKind::TkEnd
        | LuaTokenKind::TkFor
        | LuaTokenKind::TkFunction
        | LuaTokenKind::TkGoto
        | LuaTokenKind::TkIf
        | LuaTokenKind::TkIn
        | LuaTokenKind::TkNot
        | LuaTokenKind::TkOr
        | LuaTokenKind::TkRepeat
        | LuaTokenKind::TkReturn
        | LuaTokenKind::TkThen
        | LuaTokenKind::TkUntil
        | LuaTokenKind::TkWhile
        // GMod: dead path — Lua 5.5 `global` keyword disabled
        | LuaTokenKind::TkGlobal
        | LuaTokenKind::TkLocal => CodeBlockHighlightKind::Keyword,
        LuaTokenKind::TkPlus
        | LuaTokenKind::TkMinus
        | LuaTokenKind::TkMul
        | LuaTokenKind::TkDiv
        // GMod: dead paths below — Lua 5.3+ operators disabled
        | LuaTokenKind::TkIDiv
        | LuaTokenKind::TkShl
        | LuaTokenKind::TkShr
        | LuaTokenKind::TkBitAnd
        | LuaTokenKind::TkBitOr
        | LuaTokenKind::TkBitXor
        // end dead paths
        | LuaTokenKind::TkDot
        | LuaTokenKind::TkConcat
        | LuaTokenKind::TkEq
        | LuaTokenKind::TkGe
        | LuaTokenKind::TkLe
        | LuaTokenKind::TkNe
        | LuaTokenKind::TkLt
        | LuaTokenKind::TkGt
        | LuaTokenKind::TkMod
        | LuaTokenKind::TkPow
        | LuaTokenKind::TkLen
        | LuaTokenKind::TkLeftBrace
        | LuaTokenKind::TkRightBrace
        | LuaTokenKind::TkLeftBracket
        | LuaTokenKind::TkRightBracket
        | LuaTokenKind::TkLeftParen
        | LuaTokenKind::TkRightParen
        | LuaTokenKind::TkComma
        | LuaTokenKind::TkSemicolon
        | LuaTokenKind::TkAssign => CodeBlockHighlightKind::Operators,
        LuaTokenKind::TkComplex | LuaTokenKind::TkInt | LuaTokenKind::TkFloat => {
            CodeBlockHighlightKind::Number
        }
        LuaTokenKind::TkShortComment | LuaTokenKind::TkLongComment => {
            CodeBlockHighlightKind::Comment
        }
        LuaTokenKind::TkName => {
            if let Some(next_token) = tokens.get(i + 1) {
                match next_token.kind {
                    LuaTokenKind::TkLeftBrace
                    | LuaTokenKind::TkLeftParen
                    | LuaTokenKind::TkString
                    | LuaTokenKind::TkLongString => return CodeBlockHighlightKind::Function,
                    _ => {}
                }
            }

            if let Some(prev_token) = tokens.get(i.wrapping_sub(1))
                && matches!(
                    prev_token.kind,
                    LuaTokenKind::TkDot | LuaTokenKind::TkDbColon
                )
            {
                return CodeBlockHighlightKind::Property;
            }

            CodeBlockHighlightKind::Variable
        }

        _ => CodeBlockHighlightKind::None,
    }
}