Add support for PostgreSQL DO statements

src/barrel/statements.ts

@@ -8,6 +8,7 @@ export * from "../lexer/statements/mysql/rename";
 
 export * from "../lexer/statements/postgres/alter";
 export * from "../lexer/statements/postgres/create";
+export * from "../lexer/statements/postgres/do";
 export * from "../lexer/statements/postgres/drop";
 export * from "../lexer/statements/postgres/select";
 export * from "../lexer/statements/postgres/truncate";

src/lexer/statementFactory.ts

@@ -7,6 +7,7 @@ import {
   Drop,
   Truncate,
   Rename,
+  Do,
 } from "../barrel/statements";
 
 class StatementFactory {
@@ -21,6 +22,7 @@ class StatementFactory {
       alter: new Alter(),
       truncate: new Truncate(),
       rename: new Rename(),
+      do: new Do(),
     };
 
     if (Object.keys(statementMap).includes(statement)) {

src/lexer/statements/postgres/do.ts

@@ -0,0 +1,36 @@
+import { Query } from "../../../reader/query";
+import { ILexer } from "../../interface";
+import { cleanUnquotedIdentifier } from "../../lexer";
+import { Keyword } from "../../../syntax/keywords";
+import { Types } from "../../types";
+import { Token } from "../../token";
+
+class Do implements ILexer {
+  public options: string[] = [];
+
+  public tokenise(query: Query): Query {
+    let lastToken = "";
+
+    query.lines.forEach((line) => {
+      line.content.split(" ").forEach((word) => {
+        let item = word.toLowerCase().trim();
+        if (item === Keyword.Do) {
+          line.tokens.push(new Token(Types.Keyword, item));
+        } else if (lastToken === Keyword.Do) {
+          item = cleanUnquotedIdentifier(item);
+
+          if (item.length > 0) {
+            line.tokens.push(
+              new Token(Types.Option, cleanUnquotedIdentifier(item))
+            );
+          }
+        }
+        lastToken = item;
+      });
+    });
+
+    return query;
+  }
+}
+
+export { Do };

src/syntax/keywords.ts

@@ -8,6 +8,7 @@ export enum Keyword {
   Declare = "declare",
   Delete = "delete",
   Delimiter = "delimiter",
+  Do = "do",
   Drop = "drop",
   Else = "else",
   End = "end",