/*
Language: Rust
Author: Andrey Vlasovskikh <andrey.vlasovskikh@gmail.com>
Contributors: Roman Shmatov <romanshmatov@gmail.com>, Kasper Andersen <kma_untrusted@protonmail.com>
Category: system
*/

import { KeywordsDef, LanguageDef } from '../types';
import {
    IDENT_RE,
    C_LINE_COMMENT_MODE,
    COMMENT,
    QUOTE_STRING_MODE,
    UNDERSCORE_TITLE_MODE,
} from '../common';

const NUM_SUFFIX = '([ui](8|16|32|64|128|size)|f(32|64))\?';

const KEYWORDS: KeywordsDef =
    'alignof as be box break const continue crate do else enum extern ' +
    'false fn for if impl in let loop match mod mut offsetof once priv ' +
    'proc pub pure ref return self Self sizeof static struct super trait true ' +
    'type typeof unsafe unsized use virtual while where yield move default';

const BUILTINS =
    // functions
    'drop ' +
    // types
    'i8 i16 i32 i64 i128 isize ' +
    'u8 u16 u32 u64 u128 usize ' +
    'f32 f64 ' +
    'str char bool ' +
    'Box Option Result String Vec ' +
    // traits
    'Copy Send Sized Sync Drop Fn FnMut FnOnce ToOwned Clone Debug ' +
    'PartialEq PartialOrd Eq Ord AsRef AsMut Into From Default Iterator ' +
    'Extend IntoIterator DoubleEndedIterator ExactSizeIterator ' +
    'SliceConcatExt ToString ' +
    // macros
    'assert! assert_eq! bitflags! bytes! cfg! col! concat! concat_idents! ' +
    'debug_assert! debug_assert_eq! env! panic! file! format! format_args! ' +
    'include_bin! include_str! line! local_data_key! module_path! ' +
    'option_env! print! println! select! stringify! try! unimplemented! ' +
    'unreachable! vec! write! writeln! macro_rules! assert_ne! debug_assert_ne!';

export const Rust: LanguageDef = {
    name: 'rust',
    aliases: ['rs'],
    keywords: {
        keyword:
            KEYWORDS,
        literal:
            'true false Some None Ok Err',
        built_in:
            BUILTINS
    },
    lexemes: IDENT_RE + '!?',
    illegal: '</',
    contains: [
        C_LINE_COMMENT_MODE,
        COMMENT('/\\*', '\\*/', { contains: ['self'] }),
        { ...QUOTE_STRING_MODE, begin: /b?"/, illegal: undefined },
        {
            className: 'string',
            variants: [
                { begin: /r(#*)"(.|\n)*?"\1(?!#)/ },
                { begin: /b?'\\?(x\w{2}|u\w{4}|U\w{8}|.)'/ }
            ]
        },
        {
            className: 'symbol',
            begin: /'[a-zA-Z_][a-zA-Z0-9_]*/
        },
        {
            className: 'number',
            variants: [
                { begin: '\\b0b([01_]+)' + NUM_SUFFIX },
                { begin: '\\b0o([0-7_]+)' + NUM_SUFFIX },
                { begin: '\\b0x([A-Fa-f0-9_]+)' + NUM_SUFFIX },
                {
                    begin: '\\b(\\d[\\d_]*(\\.[0-9_]+)?([eE][+-]?[0-9_]+)?)' +
                        NUM_SUFFIX
                }
            ],
            relevance: 0
        },
        {
            className: 'function',
            beginKeywords: 'fn', end: '(\\(|<)', excludeEnd: true,
            contains: [UNDERSCORE_TITLE_MODE]
        },
        {
            className: 'meta',
            begin: '#\\!?\\[', end: '\\]',
            contains: [
                {
                    className: 'meta-string',
                    begin: /"/, end: /"/
                }
            ]
        },
        {
            className: 'class',
            beginKeywords: 'type', end: ';',
            contains: [
                { ...UNDERSCORE_TITLE_MODE, endsParent: true }
            ],
            illegal: '\\S'
        },
        {
            className: 'class',
            beginKeywords: 'trait enum struct union', end: '{',
            contains: [
                { ...UNDERSCORE_TITLE_MODE, endsParent: true }
            ],
            illegal: '[\\w\\d]'
        },
        {
            begin: IDENT_RE + '::',
            keywords: { built_in: BUILTINS }
        },
        {
            begin: '->'
        }
    ]
};