sql-hint.js 8.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271
  1. // CodeMirror, copyright (c) by Marijn Haverbeke and others
  2. // Distributed under an MIT license: http://codemirror.net/LICENSE
  3. (function(mod) {
  4. if (typeof exports == "object" && typeof module == "object") // CommonJS
  5. mod(require("../../lib/codemirror"), require("../../mode/sql/sql"));
  6. else if (typeof define == "function" && define.amd) // AMD
  7. define(["../../lib/codemirror", "../../mode/sql/sql"], mod);
  8. else // Plain browser env
  9. mod(CodeMirror);
  10. })(function(CodeMirror) {
  11. "use strict";
  12. var tables;
  13. var defaultTable;
  14. var keywords;
  15. var CONS = {
  16. QUERY_DIV: ";",
  17. ALIAS_KEYWORD: "AS"
  18. };
  19. var Pos = CodeMirror.Pos, cmpPos = CodeMirror.cmpPos;
  20. function isArray(val) { return Object.prototype.toString.call(val) == "[object Array]" }
  21. function getKeywords(editor) {
  22. var mode = editor.doc.modeOption;
  23. if (mode === "sql") mode = "text/x-sql";
  24. return CodeMirror.resolveMode(mode).keywords;
  25. }
  26. function getText(item) {
  27. return typeof item == "string" ? item : item.text;
  28. }
  29. function wrapTable(name, value) {
  30. if (isArray(value)) value = {columns: value}
  31. if (!value.text) value.text = name
  32. return value
  33. }
  34. function parseTables(input) {
  35. var result = {}
  36. if (isArray(input)) {
  37. for (var i = input.length - 1; i >= 0; i--) {
  38. var item = input[i]
  39. result[getText(item).toUpperCase()] = wrapTable(getText(item), item)
  40. }
  41. } else if (input) {
  42. for (var name in input)
  43. result[name.toUpperCase()] = wrapTable(name, input[name])
  44. }
  45. return result
  46. }
  47. function getTable(name) {
  48. return tables[name.toUpperCase()]
  49. }
  50. function shallowClone(object) {
  51. var result = {};
  52. for (var key in object) if (object.hasOwnProperty(key))
  53. result[key] = object[key];
  54. return result;
  55. }
  56. function match(string, word) {
  57. var len = string.length;
  58. var sub = getText(word).substr(0, len);
  59. return string.toUpperCase() === sub.toUpperCase();
  60. }
  61. function addMatches(result, search, wordlist, formatter) {
  62. if (isArray(wordlist)) {
  63. for (var i = 0; i < wordlist.length; i++)
  64. if (match(search, wordlist[i])) result.push(formatter(wordlist[i]))
  65. } else {
  66. for (var word in wordlist) if (wordlist.hasOwnProperty(word)) {
  67. var val = wordlist[word]
  68. if (!val || val === true)
  69. val = word
  70. else
  71. val = val.displayText ? {text: val.text, displayText: val.displayText} : val.text
  72. if (match(search, val)) result.push(formatter(val))
  73. }
  74. }
  75. }
  76. function cleanName(name) {
  77. // Get rid name from backticks(`) and preceding dot(.)
  78. if (name.charAt(0) == ".") {
  79. name = name.substr(1);
  80. }
  81. return name.replace(/`/g, "");
  82. }
  83. function insertBackticks(name) {
  84. var nameParts = getText(name).split(".");
  85. for (var i = 0; i < nameParts.length; i++)
  86. nameParts[i] = "`" + nameParts[i] + "`";
  87. var escaped = nameParts.join(".");
  88. if (typeof name == "string") return escaped;
  89. name = shallowClone(name);
  90. name.text = escaped;
  91. return name;
  92. }
  93. function nameCompletion(cur, token, result, editor) {
  94. // Try to complete table, column names and return start position of completion
  95. var useBacktick = false;
  96. var nameParts = [];
  97. var start = token.start;
  98. var cont = true;
  99. while (cont) {
  100. cont = (token.string.charAt(0) == ".");
  101. useBacktick = useBacktick || (token.string.charAt(0) == "`");
  102. start = token.start;
  103. nameParts.unshift(cleanName(token.string));
  104. token = editor.getTokenAt(Pos(cur.line, token.start));
  105. if (token.string == ".") {
  106. cont = true;
  107. token = editor.getTokenAt(Pos(cur.line, token.start));
  108. }
  109. }
  110. // Try to complete table names
  111. var string = nameParts.join(".");
  112. addMatches(result, string, tables, function(w) {
  113. return useBacktick ? insertBackticks(w) : w;
  114. });
  115. // Try to complete columns from defaultTable
  116. addMatches(result, string, defaultTable, function(w) {
  117. return useBacktick ? insertBackticks(w) : w;
  118. });
  119. // Try to complete columns
  120. string = nameParts.pop();
  121. var table = nameParts.join(".");
  122. var alias = false;
  123. var aliasTable = table;
  124. // Check if table is available. If not, find table by Alias
  125. if (!getTable(table)) {
  126. var oldTable = table;
  127. table = findTableByAlias(table, editor);
  128. if (table !== oldTable) alias = true;
  129. }
  130. var columns = getTable(table);
  131. if (columns && columns.columns)
  132. columns = columns.columns;
  133. if (columns) {
  134. addMatches(result, string, columns, function(w) {
  135. var tableInsert = table;
  136. if (alias == true) tableInsert = aliasTable;
  137. if (typeof w == "string") {
  138. w = tableInsert + "." + w;
  139. } else {
  140. w = shallowClone(w);
  141. w.text = tableInsert + "." + w.text;
  142. }
  143. return useBacktick ? insertBackticks(w) : w;
  144. });
  145. }
  146. return start;
  147. }
  148. function eachWord(lineText, f) {
  149. if (!lineText) return;
  150. var excepted = /[,;]/g;
  151. var words = lineText.split(" ");
  152. for (var i = 0; i < words.length; i++) {
  153. f(words[i]?words[i].replace(excepted, '') : '');
  154. }
  155. }
  156. function findTableByAlias(alias, editor) {
  157. var doc = editor.doc;
  158. var fullQuery = doc.getValue();
  159. var aliasUpperCase = alias.toUpperCase();
  160. var previousWord = "";
  161. var table = "";
  162. var separator = [];
  163. var validRange = {
  164. start: Pos(0, 0),
  165. end: Pos(editor.lastLine(), editor.getLineHandle(editor.lastLine()).length)
  166. };
  167. //add separator
  168. var indexOfSeparator = fullQuery.indexOf(CONS.QUERY_DIV);
  169. while(indexOfSeparator != -1) {
  170. separator.push(doc.posFromIndex(indexOfSeparator));
  171. indexOfSeparator = fullQuery.indexOf(CONS.QUERY_DIV, indexOfSeparator+1);
  172. }
  173. separator.unshift(Pos(0, 0));
  174. separator.push(Pos(editor.lastLine(), editor.getLineHandle(editor.lastLine()).text.length));
  175. //find valid range
  176. var prevItem = null;
  177. var current = editor.getCursor()
  178. for (var i = 0; i < separator.length; i++) {
  179. if ((prevItem == null || cmpPos(current, prevItem) > 0) && cmpPos(current, separator[i]) <= 0) {
  180. validRange = {start: prevItem, end: separator[i]};
  181. break;
  182. }
  183. prevItem = separator[i];
  184. }
  185. var query = doc.getRange(validRange.start, validRange.end, false);
  186. for (var i = 0; i < query.length; i++) {
  187. var lineText = query[i];
  188. eachWord(lineText, function(word) {
  189. var wordUpperCase = word.toUpperCase();
  190. if (wordUpperCase === aliasUpperCase && getTable(previousWord))
  191. table = previousWord;
  192. if (wordUpperCase !== CONS.ALIAS_KEYWORD)
  193. previousWord = word;
  194. });
  195. if (table) break;
  196. }
  197. return table;
  198. }
  199. CodeMirror.registerHelper("hint", "sql", function(editor, options) {
  200. tables = parseTables(options && options.tables)
  201. var defaultTableName = options && options.defaultTable;
  202. var disableKeywords = options && options.disableKeywords;
  203. defaultTable = defaultTableName && getTable(defaultTableName);
  204. keywords = getKeywords(editor);
  205. if (defaultTableName && !defaultTable)
  206. defaultTable = findTableByAlias(defaultTableName, editor);
  207. defaultTable = defaultTable || [];
  208. if (defaultTable.columns)
  209. defaultTable = defaultTable.columns;
  210. var cur = editor.getCursor();
  211. var result = [];
  212. var token = editor.getTokenAt(cur), start, end, search;
  213. if (token.end > cur.ch) {
  214. token.end = cur.ch;
  215. token.string = token.string.slice(0, cur.ch - token.start);
  216. }
  217. if (token.string.match(/^[.`\w@]\w*$/)) {
  218. search = token.string;
  219. start = token.start;
  220. end = token.end;
  221. } else {
  222. start = end = cur.ch;
  223. search = "";
  224. }
  225. if (search.charAt(0) == "." || search.charAt(0) == "`") {
  226. start = nameCompletion(cur, token, result, editor);
  227. } else {
  228. addMatches(result, search, tables, function(w) {return w;});
  229. addMatches(result, search, defaultTable, function(w) {return w;});
  230. if (!disableKeywords)
  231. addMatches(result, search, keywords, function(w) {return w.toUpperCase();});
  232. }
  233. return {list: result, from: Pos(cur.line, start), to: Pos(cur.line, end)};
  234. });
  235. });