rico 0.1.7

use logos::{Lexer, Logos, Skip};

/// Update the line count and the char index.
fn newline_callback(lex: &mut Lexer<Token>) -> Skip {
    lex.extras.0 += 1;
    lex.extras.1 = lex.span().end;
    Skip
}

#[derive(Logos, Debug, PartialEq, Clone)]
#[logos(skip r"[ \t]+")]
#[logos(extras = (usize, usize))]
pub enum Token {
    #[regex(r"[\r\n\f]", newline_callback)]
    Newline,

    // Keywords
    #[token("namespace")]
    Namespace,
    #[token("include")]
    Include,
    #[token("typedef")]
    Typedef,
    #[token("const")]
    Const,
    #[token("enum")]
    Enum,
    #[token("struct")]
    Struct,
    #[token("union")]
    Union,
    #[token("exception")]
    Exception,
    #[token("service")]
    Service,
    #[token("extends")]
    Extends,
    #[token("throws")]
    Throws,
    #[token("required")]
    Required,
    #[token("optional")]
    Optional,

    // Base types
    #[token("void")]
    Void,
    #[token("bool")]
    Bool,
    #[token("byte")]
    Byte,
    #[token("i16")]
    I16,
    #[token("i32")]
    I32,
    #[token("i64")]
    I64,
    #[token("double")]
    Double,
    #[token("string")]
    String,
    #[token("binary")]
    Binary,
    #[token("map")]
    Map,
    #[token("list")]
    List,
    #[token("set")]
    Set,

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

    #[regex(r#"(?:"([^"\\]|\\.)*"|'([^'\\]|\\.)*')"#)]
    StringLiteral,

    #[regex(r"[+-]?[0-9]+")]
    IntegerLiteral,

    #[regex(r"[+-]?[0-9]*\.[0-9]+([eE][-+]?[0-9]+)?")]
    DoubleLiteral,

    #[token("true")]
    #[token("false")]
    BooleanLiteral,

    #[regex(r"0[xX][0-9a-fA-F]+")]
    HexLiteral,

    // Punctuation
    #[token("{")]
    LeftBrace,
    #[token("}")]
    RightBrace,
    #[token("[")]
    LeftBracket,
    #[token("]")]
    RightBracket,
    #[token("<")]
    LeftAngle,
    #[token(">")]
    RightAngle,
    #[token("(")]
    LeftParen,
    #[token(")")]
    RightParen,
    #[token(";")]
    Semicolon,
    #[token(":")]
    Colon,
    #[token(",")]
    Comma,
    #[token("=")]
    Equals,
    #[token(".")]
    Dot,

    // Comments
    #[regex(r"(//|#).*")]
    LineComment,
    #[regex(r"/\*[^*]*\*+(?:[^*/][^*]*\*+)*/")]
    BlockComment,

    #[token("oneway")]
    Oneway,
}