// 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'))
    else if (typeof define == 'function' && define.amd)
        // AMD
        define(['../../lib/codemirror'], mod)
    // Plain browser env
    else mod(CodeMirror)
})(function (CodeMirror) {
    'use strict'

    CodeMirror.defineMode('groovy', function (config) {
        function words(str) {
            var obj = {},
                words = str.split(' ')
            for (var i = 0; i < words.length; ++i) obj[words[i]] = true
            return obj
        }
        var keywords = words(
            'abstract as assert boolean break byte case catch char class const continue def default ' +
                'do double else enum extends final finally float for goto if implements import in ' +
                'instanceof int interface long native new package private protected public return ' +
                'short static strictfp super switch synchronized threadsafe throw throws trait transient ' +
                'try void volatile while'
        )
        var blockKeywords = words('catch class def do else enum finally for if interface switch trait try while')
        var standaloneKeywords = words('return break continue')
        var atoms = words('null true false this')

        var curPunc
        function tokenBase(stream, state) {
            var ch = stream.next()
            if (ch == '"' || ch == "'") {
                return startString(ch, stream, state)
            }
            if (/[\[\]{}\(\),;\:\.]/.test(ch)) {
                curPunc = ch
                return null
            }
            if (/\d/.test(ch)) {
                stream.eatWhile(/[\w\.]/)
                if (stream.eat(/eE/)) {
                    stream.eat(/\+\-/)
                    stream.eatWhile(/\d/)
                }
                return 'number'
            }
            if (ch == '/') {
                if (stream.eat('*')) {
                    state.tokenize.push(tokenComment)
                    return tokenComment(stream, state)
                }
                if (stream.eat('/')) {
                    stream.skipToEnd()
                    return 'comment'
                }
                if (expectExpression(state.lastToken, false)) {
                    return startString(ch, stream, state)
                }
            }
            if (ch == '-' && stream.eat('>')) {
                curPunc = '->'
                return null
            }
            if (/[+\-*&%=<>!?|\/~]/.test(ch)) {
                stream.eatWhile(/[+\-*&%=<>|~]/)
                return 'operator'
            }
            stream.eatWhile(/[\w\$_]/)
            if (ch == '@') {
                stream.eatWhile(/[\w\$_\.]/)
                return 'meta'
            }
            if (state.lastToken == '.') return 'property'
            if (stream.eat(':')) {
                curPunc = 'proplabel'
                return 'property'
            }
            var cur = stream.current()
            if (atoms.propertyIsEnumerable(cur)) {
                return 'atom'
            }
            if (keywords.propertyIsEnumerable(cur)) {
                if (blockKeywords.propertyIsEnumerable(cur)) curPunc = 'newstatement'
                else if (standaloneKeywords.propertyIsEnumerable(cur)) curPunc = 'standalone'
                return 'keyword'
            }
            return 'variable'
        }
        tokenBase.isBase = true

        function startString(quote, stream, state) {
            var tripleQuoted = false
            if (quote != '/' && stream.eat(quote)) {
                if (stream.eat(quote)) tripleQuoted = true
                else return 'string'
            }
            function t(stream, state) {
                var escaped = false,
                    next,
                    end = !tripleQuoted
                while ((next = stream.next()) != null) {
                    if (next == quote && !escaped) {
                        if (!tripleQuoted) {
                            break
                        }
                        if (stream.match(quote + quote)) {
                            end = true
                            break
                        }
                    }
                    if (quote == '"' && next == '$' && !escaped && stream.eat('{')) {
                        state.tokenize.push(tokenBaseUntilBrace())
                        return 'string'
                    }
                    escaped = !escaped && next == '\\'
                }
                if (end) state.tokenize.pop()
                return 'string'
            }
            state.tokenize.push(t)
            return t(stream, state)
        }

        function tokenBaseUntilBrace() {
            var depth = 1
            function t(stream, state) {
                if (stream.peek() == '}') {
                    depth--
                    if (depth == 0) {
                        state.tokenize.pop()
                        return state.tokenize[state.tokenize.length - 1](stream, state)
                    }
                } else if (stream.peek() == '{') {
                    depth++
                }
                return tokenBase(stream, state)
            }
            t.isBase = true
            return t
        }

        function tokenComment(stream, state) {
            var maybeEnd = false,
                ch
            while ((ch = stream.next())) {
                if (ch == '/' && maybeEnd) {
                    state.tokenize.pop()
                    break
                }
                maybeEnd = ch == '*'
            }
            return 'comment'
        }

        function expectExpression(last, newline) {
            return !last || last == 'operator' || last == '->' || /[\.\[\{\(,;:]/.test(last) || last == 'newstatement' || last == 'keyword' || last == 'proplabel' || (last == 'standalone' && !newline)
        }

        function Context(indented, column, type, align, prev) {
            this.indented = indented
            this.column = column
            this.type = type
            this.align = align
            this.prev = prev
        }
        function pushContext(state, col, type) {
            return (state.context = new Context(state.indented, col, type, null, state.context))
        }
        function popContext(state) {
            var t = state.context.type
            if (t == ')' || t == ']' || t == '}') state.indented = state.context.indented
            return (state.context = state.context.prev)
        }

        // Interface

        return {
            startState: function (basecolumn) {
                return {
                    tokenize: [tokenBase],
                    context: new Context((basecolumn || 0) - config.indentUnit, 0, 'top', false),
                    indented: 0,
                    startOfLine: true,
                    lastToken: null,
                }
            },

            token: function (stream, state) {
                var ctx = state.context
                if (stream.sol()) {
                    if (ctx.align == null) ctx.align = false
                    state.indented = stream.indentation()
                    state.startOfLine = true
                    // Automatic semicolon insertion
                    if (ctx.type == 'statement' && !expectExpression(state.lastToken, true)) {
                        popContext(state)
                        ctx = state.context
                    }
                }
                if (stream.eatSpace()) return null
                curPunc = null
                var style = state.tokenize[state.tokenize.length - 1](stream, state)
                if (style == 'comment') return style
                if (ctx.align == null) ctx.align = true

                if ((curPunc == ';' || curPunc == ':') && ctx.type == 'statement') popContext(state)
                // Handle indentation for {x -> \n ... }
                else if (curPunc == '->' && ctx.type == 'statement' && ctx.prev.type == '}') {
                    popContext(state)
                    state.context.align = false
                } else if (curPunc == '{') pushContext(state, stream.column(), '}')
                else if (curPunc == '[') pushContext(state, stream.column(), ']')
                else if (curPunc == '(') pushContext(state, stream.column(), ')')
                else if (curPunc == '}') {
                    while (ctx.type == 'statement') ctx = popContext(state)
                    if (ctx.type == '}') ctx = popContext(state)
                    while (ctx.type == 'statement') ctx = popContext(state)
                } else if (curPunc == ctx.type) popContext(state)
                else if (ctx.type == '}' || ctx.type == 'top' || (ctx.type == 'statement' && curPunc == 'newstatement')) pushContext(state, stream.column(), 'statement')
                state.startOfLine = false
                state.lastToken = curPunc || style
                return style
            },

            indent: function (state, textAfter) {
                if (!state.tokenize[state.tokenize.length - 1].isBase) return CodeMirror.Pass
                var firstChar = textAfter && textAfter.charAt(0),
                    ctx = state.context
                if (ctx.type == 'statement' && !expectExpression(state.lastToken, true)) ctx = ctx.prev
                var closing = firstChar == ctx.type
                if (ctx.type == 'statement') return ctx.indented + (firstChar == '{' ? 0 : config.indentUnit)
                else if (ctx.align) return ctx.column + (closing ? 0 : 1)
                else return ctx.indented + (closing ? 0 : config.indentUnit)
            },

            electricChars: '{}',
            closeBrackets: { triples: '\'"' },
            fold: 'brace',
            blockCommentStart: '/*',
            blockCommentEnd: '*/',
            lineComment: '//',
        }
    })

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