// CodeMirror, copyright (c) by Marijn Haverbeke and others
// Distributed under an MIT license: https://codemirror.net/LICENSE

;(function (mod) {
    if (typeof exports == 'object' && typeof module == 'object')
        // CommonJS
        mod(require('../../lib/codemirror'), require('../../addon/mode/simple'))
    else if (typeof define == 'function' && define.amd)
        // AMD
        define(['../../lib/codemirror', '../../addon/mode/simple'], mod)
    // Plain browser env
    else mod(CodeMirror)
})(function (CodeMirror) {
    'use strict'

    CodeMirror.defineSimpleMode('rust', {
        start: [
            // string and byte string
            { regex: /b?"/, token: 'string', next: 'string' },
            // raw string and raw byte string
            { regex: /b?r"/, token: 'string', next: 'string_raw' },
            { regex: /b?r#+"/, token: 'string', next: 'string_raw_hash' },
            // character
            { regex: /'(?:[^'\\]|\\(?:[nrt0'"]|x[\da-fA-F]{2}|u\{[\da-fA-F]{6}\}))'/, token: 'string-2' },
            // byte
            { regex: /b'(?:[^']|\\(?:['\\nrt0]|x[\da-fA-F]{2}))'/, token: 'string-2' },

            {
                regex: /(?:(?:[0-9][0-9_]*)(?:(?:[Ee][+-]?[0-9_]+)|\.[0-9_]+(?:[Ee][+-]?[0-9_]+)?)(?:f32|f64)?)|(?:0(?:b[01_]+|(?:o[0-7_]+)|(?:x[0-9a-fA-F_]+))|(?:[0-9][0-9_]*))(?:u8|u16|u32|u64|i8|i16|i32|i64|isize|usize)?/,
                token: 'number',
            },
            { regex: /(let(?:\s+mut)?|fn|enum|mod|struct|type|union)(\s+)([a-zA-Z_][a-zA-Z0-9_]*)/, token: ['keyword', null, 'def'] },
            {
                regex: /(?:abstract|alignof|as|async|await|box|break|continue|const|crate|do|dyn|else|enum|extern|fn|for|final|if|impl|in|loop|macro|match|mod|move|offsetof|override|priv|proc|pub|pure|ref|return|self|sizeof|static|struct|super|trait|type|typeof|union|unsafe|unsized|use|virtual|where|while|yield)\b/,
                token: 'keyword',
            },
            { regex: /\b(?:Self|isize|usize|char|bool|u8|u16|u32|u64|f16|f32|f64|i8|i16|i32|i64|str|Option)\b/, token: 'atom' },
            { regex: /\b(?:true|false|Some|None|Ok|Err)\b/, token: 'builtin' },
            { regex: /\b(fn)(\s+)([a-zA-Z_][a-zA-Z0-9_]*)/, token: ['keyword', null, 'def'] },
            { regex: /#!?\[.*\]/, token: 'meta' },
            { regex: /\/\/.*/, token: 'comment' },
            { regex: /\/\*/, token: 'comment', next: 'comment' },
            { regex: /[-+\/*=<>!]+/, token: 'operator' },
            { regex: /[a-zA-Z_]\w*!/, token: 'variable-3' },
            { regex: /[a-zA-Z_]\w*/, token: 'variable' },
            { regex: /[\{\[\(]/, indent: true },
            { regex: /[\}\]\)]/, dedent: true },
        ],
        string: [
            { regex: /"/, token: 'string', next: 'start' },
            { regex: /(?:[^\\"]|\\(?:.|$))*/, token: 'string' },
        ],
        string_raw: [
            { regex: /"/, token: 'string', next: 'start' },
            { regex: /[^"]*/, token: 'string' },
        ],
        string_raw_hash: [
            { regex: /"#+/, token: 'string', next: 'start' },
            { regex: /(?:[^"]|"(?!#))*/, token: 'string' },
        ],
        comment: [
            { regex: /.*?\*\//, token: 'comment', next: 'start' },
            { regex: /.*/, token: 'comment' },
        ],
        meta: {
            dontIndentStates: ['comment'],
            electricInput: /^\s*\}$/,
            blockCommentStart: '/*',
            blockCommentEnd: '*/',
            lineComment: '//',
            fold: 'brace',
        },
    })

    CodeMirror.defineMIME('text/x-rustsrc', 'rust')
    CodeMirror.defineMIME('text/rust', 'rust')
})