SQL Server typos and keyword completion modifier

demo/kitchen-sink/docs/sqlserver.sqlserver

@@ -1,17 +1,17 @@
 -- =============================================
 -- Author:		Morgan Yarbrough
 -- Create date: 4/27/2015
--- Description:	Test Procedure that shows off language features.
--- 				Includes non-standard folding using region comments using either
--- 				line comments or block comments (both are demonstrated below)
+-- Description:	Test procedure that shows off language features.
+-- 				Includes non-standard folding with region comments using either
+-- 				line comments or block comments (both are demonstrated below).
 --				This mode imitates SSMS and it designed to be used with SQL Server theme.
 -- =============================================
 CREATE PROCEDURE dbo.TestProcedure
 
 --#region parameters
 	@vint INT = 1
 	,@vdate DATE = NULL
-	,@vdatetime DATETIME = DATEADD (dd, 1, GETDATE())
+	,@vdatetime DATETIME = DATEADD(dd, 1, GETDATE())
 	,@vvarchar VARCHAR(MAX) = ''
 --#endregion
 
@@ -38,16 +38,16 @@ BEGIN
 	-- another folding demonstration
 	IF @vint = 1 
 	BEGIN
-		SET @vvarchar='one'
-		SET @vint = DATEDIFFT(dd, @vdate, @vdatetime)
+		SET @vvarchar = 'one'
+		SET @vint = DATEDIFF(dd, @vdate, @vdatetime)
 	END
 
 	-- this mode handles strings properly
 	DECLARE @sql NVARCHAR(4000) = N'SELECT TOP(1) OrderID 
 									FROM Orders
 									WHERE @OrderDate > GETDATE()'
 
-	-- this mode is aware of build in stored procedures 
+	-- this mode is aware of built in stored procedures 
 	EXECUTE sp_executesql @sql
 
 	-- demonstrating some syntax highlighting
@@ -60,4 +60,9 @@ BEGIN
 	WHERE CompanyName NOT LIKE '%something'
 		OR CompanyName IS NULL
 		OR CompanyName IN ('bla', 'nothing')
+
+	-- this mode includes snippets
+	-- place your cusor at the end of the line below and trigger auto complete (Ctrl+Space)
+	createpr
+
 END

lib/ace/mode/sqlserver.js

@@ -46,6 +46,33 @@ oop.inherits(Mode, TextMode);
 (function() {
     this.lineCommentStart = "--";
     this.blockComment = {start: "/*", end: "*/"};
+
+    /**
+     * Override keyword completions to ensure proper case for completions
+     * and use smart meta tags instead of 'keyword' for all completions. 
+     */
+    this.getCompletions = function(state, session, pos, prefix) {
+        var keywords = this.$keywordList || this.$createKeywordList();
+        var types = session.$mode.$highlightRules.keywordTypes;
+
+        return keywords.map(function(word) {
+            var meta = 'keyword';
+            if (types.builtInStoredProcedures.indexOf(word) !== -1) meta = "procedure";
+            else {
+                word = word.toUpperCase(); //all others are upper case
+                if (types.builtinFunctions.indexOf(word) !== -1) meta = "function";
+                else if (types.dataTypes.indexOf(word) !== -1) meta = "type";
+                else if (types.setStatements.indexOf(word) !== -1) meta = "statement";
+                else if (types.logicalOperators.indexOf(word) !== -1) meta = "operator";
+            }
+            return {
+                name: word,
+                value: word,
+                score: 0,
+                meta: meta
+            };
+        });
+    };
 
     this.$id = "ace/mode/sql";
 }).call(Mode.prototype);

lib/ace/mode/sqlserver_highlight_rules.js

@@ -77,7 +77,9 @@ var SqlServerHighlightRules = function() {
         /* https://msdn.microsoft.com/en-us/library/ms177520.aspx */
         "@@CONNECTIONS|@@CPU_BUSY|@@IDLE|@@IO_BUSY|@@PACKET_ERRORS|@@PACK_RECEIVED|@@PACK_SENT|@@TIMETICKS|@@TOTAL_ERRORS|@@TOTAL_READ|@@TOTAL_WRITE|FN_VIRTUALFILESTATS|" +
         /* https://msdn.microsoft.com/en-us/library/ms188353.aspx */
-        "PATINDEX|TEXTPTR|TEXTVALID"
+        "PATINDEX|TEXTPTR|TEXTVALID|" +
+        /* other */
+        "COALESCE|NULLIF"
     );
 
 
@@ -104,30 +106,22 @@ var SqlServerHighlightRules = function() {
     keywords += "|TYPE";
 
 
-    //remove any other built in things from key word list
+    //remove specific built in types from keyword list
     keywords = keywords.split('|');
     keywords = keywords.filter(function(value, index, self) {
         return logicalOperators.split('|').indexOf(value) === -1 && builtinFunctions.split('|').indexOf(value) === -1 && dataTypes.split('|').indexOf(value) === -1;
     });
     keywords = keywords.sort().join('|');
+
+
     var keywordMapper = this.createKeywordMapper({
         "constant.language": logicalOperators,
         "storage.type": dataTypes,
         "support.function": builtinFunctions,
         "support.storedprocedure": builtInStoredProcedures,
         "keyword": keywords,
     }, "identifier", true);
-
-
-    // createKeywordMapper ignores case which we want because SqlServer keywords are not case sensitive which causes our keywords to get changed to lowercase.
-    // However, the preferred standard for keywords is uppercase, so this transforms them back to uppercase for code completion
-    //      EXCEPTION: build in stored procedures are lower case
-    for (var i = 0; i < this.$keywordList.length; i++) {
-        var keyword = this.$keywordList[i];
-        if (builtInStoredProcedures.indexOf(keyword) !== -1) continue;
-        this.$keywordList[i] = keyword.toUpperCase();
-    }
-
+
 
     //https://msdn.microsoft.com/en-us/library/ms190356.aspx
     var setStatements = "SET ANSI_DEFAULTS|SET ANSI_NULLS|SET ANSI_NULL_DFLT_OFF|SET ANSI_NULL_DFLT_ON|SET ANSI_PADDING|SET ANSI_WARNINGS|SET ARITHABORT|SET ARITHIGNORE|SET CONCAT_NULL_YIELDS_NULL|SET CURSOR_CLOSE_ON_COMMIT|SET DATEFIRST|SET DATEFORMAT|SET DEADLOCK_PRIORITY|SET FIPS_FLAGGER|SET FMTONLY|SET FORCEPLAN|SET IDENTITY_INSERT|SET IMPLICIT_TRANSACTIONS|SET LANGUAGE|SET LOCK_TIMEOUT|SET NOCOUNT|SET NOEXEC|SET NUMERIC_ROUNDABORT|SET OFFSETS|SET PARSEONLY|SET QUERY_GOVERNOR_COST_LIMIT|SET QUOTED_IDENTIFIER|SET REMOTE_PROC_TRANSACTIONS|SET ROWCOUNT|SET SHOWPLAN_ALL|SET SHOWPLAN_TEXT|SET SHOWPLAN_XML|SET STATISTICS IO|SET STATISTICS PROFILE|SET STATISTICS TIME|SET STATISTICS XML|SET TEXTSIZE|SET XACT_ABORT".split('|');
@@ -147,7 +141,7 @@ var SqlServerHighlightRules = function() {
             regex: "'",
             next: [{
                 token: "constant.language.escape",
-                regex: /\\'/
+                regex: /''/
             }, {
                 token: "string.end",
                 next: "start",
@@ -171,7 +165,7 @@ var SqlServerHighlightRules = function() {
             regex: "[+-]?\\d+(?:(?:\\.\\d*)?(?:[eE][+-]?\\d+)?)?\\b"
         }, {
             token: keywordMapper,
-            regex: "@{0,2}[a-zA-Z_$][a-zA-Z0-9_$]*\\b" //up to 2 @symbols for some build in functions
+            regex: "@{0,2}[a-zA-Z_$][a-zA-Z0-9_$]*\\b" //up to 2 @symbols for some built in functions
         }, {
             token: "constant.class",
             regex: "@@?[a-zA-Z_$][a-zA-Z0-9_$]*\\b"
@@ -216,6 +210,16 @@ var SqlServerHighlightRules = function() {
     this.embedRules(DocCommentHighlightRules, "doc-", [DocCommentHighlightRules.getEndRule("start")]);
 
     this.normalizeRules();
+
+    // export types for overriding get completions
+    this.keywordTypes = {
+        builtInStoredProcedures: builtInStoredProcedures.split('|'),
+        logicalOperators: logicalOperators.split('|'),
+        builtinFunctions: builtinFunctions.split('|'),
+        dataTypes: dataTypes.split('|'),
+        setStatements: setStatements,
+    };
+
 };
 
 oop.inherits(SqlServerHighlightRules, TextHighlightRules);

lib/ace/snippets/sqlserver.snippets

@@ -22,6 +22,16 @@ snippet dateadd
 # DATEFROMPARTS 
 snippet datefromparts
 	DATEFROMPARTS(${1:year}, ${2:month}, ${3:day})
+# OBJECT_DEFINITION
+snippet objectdef
+	SELECT OBJECT_DEFINITION(OBJECT_ID('${1:sys.server_permissions /*object name*/}'))
+# STUFF XML
+snippet stuffxml
+	STUFF((SELECT ', ' + ${1:ColumnName}
+		FROM ${2:TableName}
+		WHERE ${3:WhereClause}
+		FOR XML PATH('')), 1, 1, '') AS ${4:Alias}
+	${5:/*https://msdn.microsoft.com/en-us/library/ms188043.aspx*/}
 # Create Procedure
 snippet createproc
 	-- =============================================
@@ -30,12 +40,14 @@ snippet createproc
 	-- Description:	${3:Description}
 	-- =============================================
 	CREATE PROCEDURE ${4:Procedure_Name}
-		-- Add the parameters for the stored procedure here
+		${5:/*Add the parameters for the stored procedure here*/}
 	AS
 	BEGIN
 		-- SET NOCOUNT ON added to prevent extra result sets from interfering with SELECT statements.
 		SET NOCOUNT ON;
 
+		${6:/*Add the T-SQL statements to compute the return value here*/}
+
 	END
 	GO
 # Create Scalar Function
@@ -52,5 +64,7 @@ snippet createfn
 	BEGIN
 		DECLARE @Result ${5:Function_Data_Type}
 
+		${6:/*Add the T-SQL statements to compute the return value here*/}
+
 	END
 	GO