Fix for brakcets not taken into account

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

Author: worksofliam

src/language/sql/statement.ts

@@ -330,6 +330,9 @@ export default class Statement {
 					if (sqlObj) {
 						doAdd(sqlObj);
 						i += sqlObj.tokens.length;
+						if (sqlObj.isUDTF || sqlObj.fromLateral) {
+							i += 3; //For the brackets
+						}
 					}
 				}
 			}
@@ -520,6 +523,7 @@ export default class Statement {
 			if (sqlObj) {
 				sqlObj.isUDTF = true;
 				const blockTokens = this.getBlockAt(sqlObj.tokens[0].range.end);
+				
 				sqlObj.tokens = blockTokens;
 				nextIndex = i + 2 + blockTokens.length;
 				nextToken = this.tokens[nextIndex];
@@ -529,12 +533,11 @@ export default class Statement {
 				nextToken = undefined;
 			}
 		} else if (isLateral) {
-			console.log(`HI`);
 			const blockTokens = this.getBlockAt(nextToken.range.end+1);
-			console.log(blockTokens);
 			const newStatement = new Statement(blockTokens, {start: nextToken.range.start, end: blockTokens[blockTokens.length-1].range.end});
 			[sqlObj] = newStatement.getObjectReferences();
-			
+
+			sqlObj.fromLateral = true;
 			nextIndex = i + 2 + blockTokens.length;
 			nextToken = this.tokens[nextIndex];
 

src/language/sql/tests/statements.test.ts

@@ -996,6 +996,24 @@ describe(`Object references`, () => {
     ].join(`\n`);
 
     const document = new Document(lines);
+
+    expect(document.statements.length).toBe(1);
+
+    const statement = document.statements[0];
+
+    expect(statement.type).toBe(StatementType.Select);
+
+    const refs = statement.getObjectReferences();
+
+    console.log(refs);
+    expect(refs.length).toBe(2);
+    expect(refs[0].object.name).toBe(`OBJECT_STATISTICS`);
+    expect(refs[0].object.schema).toBe(`QSYS2`);
+    expect(refs[0].alias).toBe(`a`);
+
+    expect(refs[1].object.name).toBe(`OBJECT_STATISTICS`);
+    expect(refs[1].object.schema).toBe(`QSYS2`);
+    expect(refs[1].alias).toBe(`b`);
   })
 });
 

src/language/sql/types.ts

@@ -84,6 +84,7 @@ export interface ObjectRef {
   alias?: string;
 
 	isUDTF?: boolean;
+	fromLateral?: boolean;
 
 	/** only used within create statements */
 	createType?: string;