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

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

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

    #[regex("[a-zA-Z_$][a-zA-Z0-9_]*")]
    Identifier,

    #[regex("\"([^\"\\\\]|\\\\[.\n])*\"")]
    #[regex("`([^`]|\\\\`)*`")]
    #[regex("'([^']|\\\\')'")]
    #[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"\.")]
    GlyphCtx,

    #[regex("arguments|async|await|break|case|catch|const|continue")]
    #[regex("debugger|default|delete|do|else|enum|eval|export|extends")]
    #[regex("false|finally|for|if|implements|import|in|instanceof")]
    #[regex("interface|let|long|native|new|null|package|private")]
    #[regex("protected|public|return|static|super|switch|this|throw")]
    #[regex("true|try|typeof|var|void|while|with|yield")]
    Keyword,

    #[regex("function|class")]
    KeywordCtx,

    #[regex("undefined|Object|Array|Number|String|NaN|Infinity|Date|Math")]
    Special,

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

impl Highlight for JavaScript {
    const LANG: &'static str = "js";

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

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