Remove invalid errors

Signed-off-by: worksofliam <mrliamallan@live.co.uk>
codefori · Jan 9, 2025 · 7282ae4 · 7282ae4
1 parent 3bf0cd6
commit 7282ae4
Showing 1 changed file with 13 additions and 3 deletions.
diff --git a/src/language/providers/problemProvider.ts b/src/language/providers/problemProvider.ts
@@ -136,8 +136,8 @@ async function validateSqlDocument(document: TextDocument, specificStatement?: n
           if (specificStatement) {
             // If specificStatement is outside this group, continue
             if (
-              specificStatement < group.range.start ||
-              (specificStatement > (allGroups[i + 1] ? allGroups[i + 1].range.start : group.range.end))
+              (specificStatement < group.range.start && i !== 0) ||
+              (specificStatement > (allGroups[i + 1] ? allGroups[i + 1].range.start : group.range.end) && i !== allGroups.length - 1)
             ) {
               continue;
             }
@@ -170,7 +170,7 @@ async function validateSqlDocument(document: TextDocument, specificStatement?: n
 
           if (validStatements.length > MAX_STATEMENT_COUNT) {
             window.showWarningMessage(`${basename}: the SQL syntax checker cannot run because the statement limit has been reached (${validStatements.length} of ${MAX_STATEMENT_COUNT} max).`);
-            
+
           } else {
 
             const invalidStatements = statementRanges.filter(r => !r.validate);
@@ -243,6 +243,16 @@ async function validateSqlDocument(document: TextDocument, specificStatement?: n
                     }
                   }
 
+                  const lastGroup = allGroups[allGroups.length - 1];
+
+                  if (lastGroup) {
+                    for (let i = currentErrors.length - 1; i >= 0; i--) {
+                      if (document.offsetAt(currentErrors[i].range.start) > lastGroup.range.end) {
+                        currentErrors.splice(i, 1);
+                      }
+                    }
+                  }
+
                   sqlDiagnosticCollection.set(document.uri, currentErrors);
                 }
               }