fix: add semantics check for table locator db2 host variable

Author: ap891843

server/engine/src/main/antlr4/org/eclipse/lsp/db2/parser/Db2SqlParser.g4

@@ -36,7 +36,14 @@ binary_host_variable_varbinary_size: T=dbs_integerliteral_expanded {validateInte
 
 host_variable_usage: (USAGE IS?)? SQL TYPE IS;
 
-sql_host_variables: dbs_level_01 entry_name host_variable_usage (result_set_locator | tableLocators);
+sql_host_variables: result_set_locator_variable
+                  | tableLocators_variable ;
+
+result_set_locator_variable: dbs_level_01 entry_name host_variable_usage result_set_locator;
+
+tableLocators_variable: dbs_host_var_levels entry_name host_variable_usage tableLocators;
+
+dbs_host_var_levels: dbs_level_01 | T=dbs_integer {validateIntegerRange($T.text, 2, 48);};
 
 result_set_locator: RESULT_SET_LOCATOR VARYING;
 

server/engine/src/main/java/org/eclipse/lsp/cobol/implicitDialects/sql/Db2SqlVisitor.java

@@ -82,33 +82,31 @@ public List<Node> visitExecRule(Db2SqlParser.ExecRuleContext ctx) {
     }
 
     @Override
-    public List<Node> visitSql_host_variables(
-            Db2SqlParser.Sql_host_variablesContext ctx) {
+    public List<Node> visitResult_set_locator_variable(Db2SqlParser.Result_set_locator_variableContext ctx) {
+        return createHostVariableDefinitionNode(ctx, ctx.dbs_level_01(), ctx.entry_name());
+    }
+
+    @Override
+    public List<Node> visitTableLocators_variable(Db2SqlParser.TableLocators_variableContext ctx) {
+        return createHostVariableDefinitionNode(ctx, ctx.dbs_host_var_levels(), ctx.entry_name());
+    }
+
+    private List<Node> createHostVariableDefinitionNode(ParserRuleContext ctx, ParserRuleContext levelCtx, ParserRuleContext nameCtx) {
         addReplacementContext(ctx);
-        Locality statementLocality =
-                getLocality(this.context.getExtendedDocument().mapLocation(constructRange(ctx)));
+        Locality statementLocality = getLocality(this.context.getExtendedDocument().mapLocation(constructRange(ctx)));
 
-        // semantics node just checks that the statement is present at right location
-        Db2WorkingAndLinkageSectionNode semanticsNode =
-                new Db2WorkingAndLinkageSectionNode(statementLocality);
+        Db2WorkingAndLinkageSectionNode semanticsNode = new Db2WorkingAndLinkageSectionNode(statementLocality);
+
+        VariableDefinitionNode variableDefinitionNode = VariableDefinitionNode.builder()
+                .level(Integer.parseInt(levelCtx.getText()))
+                .levelLocality(getLocality(this.context.getExtendedDocument().mapLocation(constructRange(levelCtx))))
+                .statementLocality(statementLocality)
+                .variableNameAndLocality(new VariableNameAndLocality(
+                        VisitorHelper.getName(nameCtx),
+                        getLocality(this.context.getExtendedDocument().mapLocation(constructRange(nameCtx)))))
+                .usageClauses(ImmutableList.of(UsageFormat.DISPLAY))
+                .build();
 
-        // variable definition node
-        VariableDefinitionNode variableDefinitionNode =
-                VariableDefinitionNode.builder()
-                        .level(Integer.parseInt(ctx.dbs_level_01().getText()))
-                        .levelLocality(
-                                getLocality(
-                                        this.context.getExtendedDocument().mapLocation(constructRange(ctx.dbs_level_01()))))
-                        .statementLocality(statementLocality)
-                        .variableNameAndLocality(
-                                new VariableNameAndLocality(
-                                        VisitorHelper.getName(ctx.entry_name()),
-                                        getLocality(
-                                                this.context
-                                                        .getExtendedDocument()
-                                                        .mapLocation(constructRange(ctx.entry_name())))))
-                        .usageClauses(ImmutableList.of(UsageFormat.DISPLAY))
-                        .build();
         variableDefinitionNode.addChild(semanticsNode);
         return ImmutableList.of(variableDefinitionNode);
     }