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

use crate::{Highlight, Kind};
use logos::Logos;

#[derive(Logos, PartialEq, Eq, Clone, Copy, Debug)]
pub enum Rust {
    #[error]
    Error,

    #[regex("[a-z_$][a-zA-Z0-9_]*!?")]
    #[regex("[A-Z_]*!?", priority = 3)]
    Identifier,

    #[regex("[A-Z][a-zA-Z0-9_]*!?")]
    StrongIdentifier,

    #[regex("\"([^\"\\n]|\\\\[\"\\n])*\"")]
    #[regex("'[^'\\n]*'")]
    #[regex("r\"[^\"]*\"")]
    #[regex("r#\"#?([^#]|[^\"]#)*\"#")]
    #[regex("b\"([^\"\\n]|\\\\[\"\\n])*\"")]
    #[regex("b'[^'\\n]*'")]
    #[regex("br\"[^\"]*\"")]
    #[regex("br#\"#?([^#]|[^\"]#)*\"#")]
    #[regex("[0-9][0-9_]*")]
    #[regex("0[xX][0-9a-fA-F_]+")]
    #[regex("0[oO][0-7_]+")]
    #[regex("0[bB][01_]+")]
    Literal,

    #[regex(r#"\?|!|\^|-|\+|\*|&|/|\\|<|>|=|=>|->|_|#\[[^\]]*\]"#)]
    Glyph,

    #[regex(
        r"\.|:|(&|'[a-zA-Z_][a-zA-Z0-9_]*)([ \t\n\r]*mut[ \t\n\r]+)?",
        priority = 3
    )]
    GlyphCtx,

    #[regex("as|break|const|continue|crate|dyn|else|extern")]
    #[regex("false|for|if|impl|in|let|loop|match|mod|move|mut")]
    #[regex("pub|ref|return|self|Self|static|super")]
    #[regex("true|unsafe|use|where|while")]
    #[regex("abstract|async|await|become|box|do|final|macro")]
    #[regex("override|priv|try|typeof|unsized|virtual|yield")]
    Keyword,

    #[regex("fn|enum|struct|type|trait")]
    KeywordCtx,

    #[regex("Some|None|Ok|Err|str|bool|[ui](8|16|32|64|size)|f32|f64")]
    Special,

    #[regex("//[^\n]*")]
    Comment,
}

impl Highlight for Rust {
    const LANG: &'static str = "rust";

    fn kind(tokens: &[Self; 2]) -> Kind {
        use Rust::*;

        match tokens {
            [KeywordCtx, StrongIdentifier]
            | [GlyphCtx, StrongIdentifier]
            | [KeywordCtx, Identifier]
            | [GlyphCtx, Identifier]
            | [_, Special] => Kind::SpecialIdentifier,
            [_, Identifier] => Kind::Identifier,
            [_, StrongIdentifier] => Kind::StrongIdentifier,
            [_, Literal] => Kind::Literal,
            [_, Glyph] | [_, GlyphCtx] => Kind::Glyph,
            [_, Keyword] | [_, KeywordCtx] => Kind::Keyword,
            [_, Comment] => Kind::Comment,
            _ => Kind::None,
        }
    }
}