settings are removed before trying to make an ast (we do not need set…

…tings with the ast)

src/data/ast.test.ts

@@ -1,4 +1,5 @@
-import { getFields } from './ast';
+import { getFields, sqlToStatement } from './ast';
+import { toSql } from 'pgsql-ast-parser'
 
 describe('ast', () => {
   describe('getFields', () => {
@@ -7,4 +8,12 @@ describe('ast', () => {
       expect(stm.length).toBe(1);
     });
   });
+  describe('sqlToStatement', () => {
+    it('settings parse correctly', () => {
+      const sql = 'SELECT count(*) FROM mytable SETTINGS setting1=stuff setting2=stuff';
+      const stm = sqlToStatement(sql);
+      // this is formatted like this to match how pgsql generates its sql
+      expect(toSql.statement(stm)).toEqual('SELECT (count (*) )  FROM mytable');
+    });
+  });
 });

src/data/ast.ts

@@ -7,17 +7,26 @@ export function sqlToStatement(sql: string): Statement {
     replacementName: string;
   }> = [];
   //default is a key word in this grammar, but it can be used in CH
-  const re = /(\$__|\$|default)/gi;
+  const re = /(\$__|\$|default|settings)/gi;
   let regExpArray: RegExpExecArray | null;
   while ((regExpArray = re.exec(sql)) !== null) {
-    replaceFuncs.push({ startIndex: regExpArray.index, name: regExpArray[0], replacementName: '' });
+    let n = regExpArray[0];
+    if (n.toLowerCase() === "settings") {
+      n = sql.substring(regExpArray.index)
+    }
+    replaceFuncs.push({ startIndex: regExpArray.index, name: n, replacementName: '' });
   }
 
   //need to process in reverse so starting positions aren't effected by replacing other things
   for (let i = replaceFuncs.length - 1; i >= 0; i--) {
     const si = replaceFuncs[i].startIndex;
     const replacementName = 'f' + (Math.random() + 1).toString(36).substring(7);
     replaceFuncs[i].replacementName = replacementName;
+    // settings do not parse and we do not need information from them so we will remove them
+    if (replaceFuncs[i].name.toLowerCase().includes("settings")) {
+      sql = sql.substring(0, si)
+      continue;
+    }
     sql = sql.substring(0, si) + replacementName + sql.substring(si + replaceFuncs[i].name.length);
   }
 
@@ -98,3 +107,7 @@ export function getFields(sql: string): string[] {
     }
   });
 }
+
+export function statementToSql(stm: Statement): string {
+  return toSql.statement(stm);
+}