fineui/src/base/sql/codemirrior/sql-hint.js

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

(function (mod) {
    mod(CodeMirror);
})(function (CodeMirror) {

    var tables;
    var defaultTable;
    var keywords;
    var identifierQuote;
    var CONS = {
        QUERY_DIV: ";",
        ALIAS_KEYWORD: "AS"
    };
    var Pos = CodeMirror.Pos, cmpPos = CodeMirror.cmpPos;

    function isArray (val) { return Object.prototype.toString.call(val) == "[object Array]"; }

    function getKeywords (editor) {
        var mode = editor.doc.modeOption;
        if (mode === "sql") mode = "text/x-sql";
        return CodeMirror.resolveMode(mode).keywords;
    }

    function getIdentifierQuote (editor) {
        var mode = editor.doc.modeOption;
        if (mode === "sql") mode = "text/x-sql";
        return CodeMirror.resolveMode(mode).identifierQuote || "`";
    }

    function getText (item) {
        return typeof item === "string" ? item : item.text;
    }

    function wrapTable (name, value) {
        if (isArray(value)) value = {columns: value};
        if (!value.text) value.text = name;
        return value;
    }

    function parseTables (input) {
        var result = {};
        if (isArray(input)) {
            for (var i = input.length - 1; i >= 0; i--) {
                var item = input[i];
                result[getText(item).toUpperCase()] = wrapTable(getText(item), item);
            }
        } else if (input) {
            for (var name in input) {result[name.toUpperCase()] = wrapTable(name, input[name]);}
        }
        return result;
    }

    function getTable (name) {
        return tables[name.toUpperCase()];
    }

    function shallowClone (object) {
        var result = {};
        for (var key in object) {
            if (object.hasOwnProperty(key)) {result[key] = object[key];}
        }
        return result;
    }

    function match (string, word) {
        if (BI.isNotEmptyString(string) && word.length !== string.length) {
            var len = string.length;
            var sub = getText(word).substr(0, len);
            return string.toUpperCase() === sub.toUpperCase();
        }
    }

    function addMatches (result, search, wordlist, formatter) {
        if (isArray(wordlist)) {
            for (var i = 0; i < wordlist.length; i++) {if (match(search, wordlist[i])) result.push(formatter(wordlist[i], i));}
        } else {
            for (var word in wordlist) {
                if (wordlist.hasOwnProperty(word)) {
                    var val = wordlist[word];
                    if (!val || val === true) {val = word;} else {val = val.displayText ? {text: val.text, displayText: val.displayText} : val.text;}
                    if (match(search, val)) result.push(formatter(val));
                }
            }
        }
    }

    function cleanName (name) {
    // Get rid name from identifierQuote and preceding dot(.)
        if (name.charAt(0) == ".") {
            name = name.substr(1);
        }
        // replace doublicated identifierQuotes with single identifierQuotes
        // and remove single identifierQuotes
        var nameParts = name.split(identifierQuote + identifierQuote);
        for (var i = 0; i < nameParts.length; i++) {nameParts[i] = nameParts[i].replace(new RegExp(identifierQuote, "g"), "");}
        return nameParts.join(identifierQuote);
    }

    function insertIdentifierQuotes (name) {
        var nameParts = getText(name).split(".");
        for (var i = 0; i < nameParts.length; i++) {
            nameParts[i] = identifierQuote +
        // doublicate identifierQuotes
        nameParts[i].replace(new RegExp(identifierQuote, "g"), identifierQuote + identifierQuote) +
        identifierQuote;
        }
        var escaped = nameParts.join(".");
        if (typeof name === "string") return escaped;
        name = shallowClone(name);
        name.text = escaped;
        return name;
    }

    function nameCompletion (cur, token, result, editor) {
    // Try to complete table, column names and return start position of completion
        var useIdentifierQuotes = false;
        var nameParts = [];
        var start = token.start;
        var cont = true;
        while (cont) {
            cont = (token.string.charAt(0) == ".");
            useIdentifierQuotes = useIdentifierQuotes || (token.string.charAt(0) == identifierQuote);

            start = token.start;
            nameParts.unshift(cleanName(token.string));

            token = editor.getTokenAt(Pos(cur.line, token.start));
            if (token.string == ".") {
                cont = true;
                token = editor.getTokenAt(Pos(cur.line, token.start));
            }
        }

        // Try to complete table names
        var string = nameParts.join(".");
        addMatches(result, string, tables, function (w) {
            return useIdentifierQuotes ? insertIdentifierQuotes(w) : w;
        });

        // Try to complete columns from defaultTable
        addMatches(result, string, defaultTable, function (w) {
            return useIdentifierQuotes ? insertIdentifierQuotes(w) : w;
        });

        // Try to complete columns
        string = nameParts.pop();
        var table = nameParts.join(".");

        var alias = false;
        var aliasTable = table;
        // Check if table is available. If not, find table by Alias
        if (!getTable(table)) {
            var oldTable = table;
            table = findTableByAlias(table, editor);
            if (table !== oldTable) alias = true;
        }

        var columns = getTable(table);
        if (columns && columns.columns) {columns = columns.columns;}

        if (columns) {
            addMatches(result, string, columns, function (w) {
                var tableInsert = table;
                if (alias == true) tableInsert = aliasTable;
                if (typeof w === "string") {
                    w = tableInsert + "." + w;
                } else {
                    w = shallowClone(w);
                    w.text = tableInsert + "." + w.text;
                }
                return useIdentifierQuotes ? insertIdentifierQuotes(w) : w;
            });
        }

        return start;
    }

    function eachWord (lineText, f) {
        var words = lineText.split(/\s+/);
        for (var i = 0; i < words.length; i++) {if (words[i]) f(words[i].replace(/[,;]/g, ""));}
    }

    function findTableByAlias (alias, editor) {
        var doc = editor.doc;
        var fullQuery = doc.getValue();
        var aliasUpperCase = alias.toUpperCase();
        var previousWord = "";
        var table = "";
        var separator = [];
        var validRange = {
            start: Pos(0, 0),
            end: Pos(editor.lastLine(), editor.getLineHandle(editor.lastLine()).length)
        };

        // add separator
        var indexOfSeparator = fullQuery.indexOf(CONS.QUERY_DIV);
        while(indexOfSeparator != -1) {
            separator.push(doc.posFromIndex(indexOfSeparator));
            indexOfSeparator = fullQuery.indexOf(CONS.QUERY_DIV, indexOfSeparator + 1);
        }
        separator.unshift(Pos(0, 0));
        separator.push(Pos(editor.lastLine(), editor.getLineHandle(editor.lastLine()).text.length));

        // find valid range
        var prevItem = null;
        var current = editor.getCursor();
        for (var i = 0; i < separator.length; i++) {
            if ((prevItem == null || cmpPos(current, prevItem) > 0) && cmpPos(current, separator[i]) <= 0) {
                validRange = {start: prevItem, end: separator[i]};
                break;
            }
            prevItem = separator[i];
        }

        var query = doc.getRange(validRange.start, validRange.end, false);

        for (var i = 0; i < query.length; i++) {
            var lineText = query[i];
            eachWord(lineText, function (word) {
                var wordUpperCase = word.toUpperCase();
                if (wordUpperCase === aliasUpperCase && getTable(previousWord)) {table = previousWord;}
                if (wordUpperCase !== CONS.ALIAS_KEYWORD) {previousWord = word;}
            });
            if (table) break;
        }
        return table;
    }

    CodeMirror.registerHelper("hint", "sql", function (editor, options) {
        tables = parseTables(options && options.tables);
        var defaultTableName = options && options.defaultTable;
        var disableKeywords = options && options.disableKeywords;
        defaultTable = defaultTableName && getTable(defaultTableName);
        keywords = getKeywords(editor);
        var keywordsCount = BI.size(keywords);
        var functions = [];
        var desc = {};
        var cur = editor.getCursor();
        var token = editor.getTokenAt(cur);
        BI.each(BI.FormulaCollections, function (idx, formula) {
            if(formula.lastIndexOf(token.string, 0) == 0 && !BI.contains(functions, formula)) {
                functions.push(formula);
            }
        });
        BI.each(BI.FormulaJson, function (idx, formula) {
            desc[formula.name.toLowerCase()] = formula.def;
        });
        keywords = BI.concat(BI.keys(keywords), functions);
        identifierQuote = getIdentifierQuote(editor);

        if (defaultTableName && !defaultTable) {defaultTable = findTableByAlias(defaultTableName, editor);}

        defaultTable = defaultTable || [];

        if (defaultTable.columns) {defaultTable = defaultTable.columns;}

        var result = [];
        var start, end, search;
        if (token.end > cur.ch) {
            token.end = cur.ch;
            token.string = token.string.slice(0, cur.ch - token.start);
        }

        if (token.string.match(/^[.`"\w@]\w*$/)) {
            search = token.string;
            start = token.start;
            end = token.end;
        } else {
            start = end = cur.ch;
            search = "";
        }
        if (search.charAt(0) == "." || search.charAt(0) == identifierQuote) {
            start = nameCompletion(cur, token, result, editor);
        } else {
            addMatches(result, search, tables, function (w) {return w;});
            addMatches(result, search, defaultTable, function (w) {return w;});
            if (!disableKeywords) {
                addMatches(result, search, keywords, function (w, i) {
                    var isKeyword = i < keywordsCount;
                    return {
                        isKeyword: isKeyword,
                        text: w
                    // description: desc[w] || "SQL关键字",
                    // className: isKeyword ? "sql-keyword" : "sql-fr-function",
                    // render: function (Element, self, data) {
                    //   var label = BI.createWidget({
                    //     type: "bi.label",
                    //     element: Element,
                    //     text: data.displayText || getText(data)
                    //   });
                    //   label.setTitle(data.description, {
                    //     container: "body"
            
                    //   });
                    // }
                    };
                });
            }
        }

        return {list: result, from: Pos(cur.line, start), to: Pos(cur.line, end)};
    });
});