Correctly seperate statements when parsing document

Signed-off-by: worksofliam <[email protected]>

Author: worksofliam

src/language/sql/document.ts

@@ -51,6 +51,7 @@ export default class Document {
         case `keyword`:
           switch (tokens[i].value?.toUpperCase()) {
             case `BEGIN`:
+            case `DO`:
               // We include BEGIN in the current statement
               // then the next statement beings
               const statementTokens = tokens.slice(statementStart, i+1);

src/language/sql/statement.ts

@@ -43,7 +43,7 @@ export default class Statement {
 		// These statements can end with BEGIN, which signifies a block starter
 		if ([StatementType.Create, StatementType.Declare].includes(this.type)) {
 			const last = this.tokens[this.tokens.length-1];
-			if (tokenIs(last, `keyword`, `BEGIN`)) {
+			if (tokenIs(last, `keyword`, `BEGIN`) || tokenIs(last, `keyword`, `DO`)) {
 				return true;
 			}
 		}

src/language/sql/tokens.ts

@@ -74,7 +74,7 @@ export default class SQLTokeniser {
     {
       name: `KEYWORD`,
       match: [{ type: `word`, match: (value: string) => {
-        return [`AS`, `FOR`, `OR`, `REPLACE`, `BEGIN`, `END`, `CURSOR`, `DEFAULT`, `HANDLER`, `REFERENCES`, `ON`, `UNIQUE`, `SPECIFIC`, `EXTERNAL`].includes(value.toUpperCase());
+        return [`AS`, `FOR`, `OR`, `REPLACE`, `BEGIN`, `DO`, `END`, `CURSOR`, `DEFAULT`, `HANDLER`, `REFERENCES`, `ON`, `UNIQUE`, `SPECIFIC`, `EXTERNAL`].includes(value.toUpperCase());
       } }],
       becomes: `keyword`,
     },