Add Additional TSQL Data Definitions (#608)

Linked to #606 and a duplicate of #607

Author: nene

src/languages/transactsql/transactsql.formatter.ts

@@ -29,7 +29,8 @@ const reservedClauses = expandPhrases([
   'UPDATE SET',
   // Data definition
   'CREATE [OR ALTER] [MATERIALIZED] VIEW',
-  'CREATE TABLE',
+  'CREATE [OR ALTER] TABLE',
+  'CREATE [OR ALTER] PROCEDURE',
 ]);
 
 const onelineClauses = expandPhrases([

test/transactsql.test.ts

@@ -117,4 +117,28 @@ describe('TransactSqlFormatter', () => {
       ALTER COLUMN foo INT NOT NULL DEFAULT 5;
     `);
   });
+
+  it('formats SET NOCOUNT ON; GO CREATE OR ALTER TABLE', () => {
+    const result = format('SET NOCOUNT ON; GO CREATE OR ALTER TABLE t');
+    expect(result).toBe(dedent`
+    SET
+    NOCOUNT ON;
+
+    GO
+    CREATE OR ALTER TABLE
+      t
+    `);
+  });
+
+  it('formats SET NOCOUNT ON; GO CREATE OR ALTER PROCEDURE', () => {
+    const result = format('SET NOCOUNT ON; GO CREATE OR ALTER PROCEDURE p');
+    expect(result).toBe(dedent`
+    SET
+    NOCOUNT ON;
+
+    GO
+    CREATE OR ALTER PROCEDURE
+      p
+    `);
+  });
 });