Переменные созданные при инициализации циклов

Добавлены в индекс переменные созданные при работе циклов Для...  и Для Каждого...
Объяевление переменной в объявлении цикла работает так же как и объявление переменной присваиванием.
Тип ссылки OccurrenceType.DEFINITION, тип переменной VariableKind.DYNAMIC

Author: qtLex

src/main/java/com/github/_1c_syntax/bsl/languageserver/context/computer/VariableSymbolComputer.java

@@ -35,6 +35,7 @@
 import com.github._1c_syntax.bsl.parser.BSLParserRuleContext;
 import org.antlr.v4.runtime.Token;
 import org.antlr.v4.runtime.tree.ParseTree;
+import org.antlr.v4.runtime.tree.TerminalNode;
 import org.eclipse.lsp4j.Range;
 
 import java.util.ArrayList;
@@ -154,27 +155,34 @@ public ParseTree visitLValue(BSLParser.LValueContext ctx) {
       return ctx;
     }
 
-    var variable = VariableSymbol.builder()
-      .name(ctx.getText().intern())
-      .owner(documentContext)
-      .range(Ranges.create(ctx))
-      .variableNameRange(Ranges.create(ctx))
-      .export(false)
-      .kind(VariableKind.DYNAMIC)
-      .scope(currentMethod)
-      .description(createDescription(ctx))
-      .build();
-    variables.add(variable);
+    updateVariablesCache(ctx.IDENTIFIER(), createDescription(ctx));
+    return ctx;
+  }
 
-    // Если файл с ошибками разбора может возникнуть ситуация, когда по дереву мы будем находиться внутри метода,
-    // но при определении скоупа в модуле. В таких случаях будем сохранять описание в коллекции переменных модуля.
-    if (currentMethod == module) {
-      moduleVariables.put(ctx.getText(), ctx.getText());
-    } else {
-      currentMethodVariables.put(ctx.getText(), ctx.getText());
+  @Override
+  public ParseTree visitForStatement(BSLParser.ForStatementContext ctx) {
+    if (
+      currentMethodVariables.containsKey(ctx.IDENTIFIER().getText())
+      || moduleVariables.containsKey(ctx.IDENTIFIER().getText())
+    ) {
+      return super.visitForStatement(ctx);
     }
 
-    return ctx;
+    updateVariablesCache(ctx.IDENTIFIER(), createDescription(ctx));
+    return super.visitForStatement(ctx);
+  }
+
+  @Override
+  public ParseTree visitForEachStatement(BSLParser.ForEachStatementContext ctx) {
+    if (
+      currentMethodVariables.containsKey(ctx.IDENTIFIER().getText())
+      || moduleVariables.containsKey(ctx.IDENTIFIER().getText())
+    ) {
+      return super.visitForEachStatement(ctx);
+    }
+
+    updateVariablesCache(ctx.IDENTIFIER(), createDescription(ctx));
+    return super.visitForEachStatement(ctx);
   }
 
   private SourceDefinedSymbol getVariableScope(BSLParser.SubVarDeclarationContext ctx) {
@@ -231,4 +239,26 @@ private Optional<VariableDescription> createDescription(BSLParserRuleContext ctx
 
   }
 
+  private void updateVariablesCache(TerminalNode node, Optional<VariableDescription> description) {
+    var variable = VariableSymbol.builder()
+      .name(node.getText().intern())
+      .owner(documentContext)
+      .range(Ranges.create(node))
+      .variableNameRange(Ranges.create(node))
+      .export(false)
+      .kind(VariableKind.DYNAMIC)
+      .scope(currentMethod)
+      .description(description)
+      .build();
+    variables.add(variable);
+
+    // Если файл с ошибками разбора может возникнуть ситуация, когда по дереву мы будем находиться внутри метода,
+    // но при определении скоупа в модуле. В таких случаях будем сохранять описание в коллекции переменных модуля.
+    if (currentMethod == module) {
+      moduleVariables.put(node.getText(), node.getText());
+    } else {
+      currentMethodVariables.put(node.getText(), node.getText());
+    }
+  }
+
 }

src/main/java/com/github/_1c_syntax/bsl/languageserver/references/ReferenceIndexFiller.java

@@ -261,6 +261,46 @@ public BSLParserRuleContext visitComplexIdentifier(BSLParser.ComplexIdentifierCo
       return super.visitComplexIdentifier(ctx);
     }
 
+    @Override
+    public BSLParserRuleContext visitForStatement(BSLParser.ForStatementContext ctx) {
+      if (ctx.IDENTIFIER() == null) {
+        return super.visitForStatement(ctx);
+      }
+
+      findVariableSymbol(ctx.IDENTIFIER().getText()).ifPresent(s -> {
+        if (notVariableInitialization(ctx, s)) {
+          addVariableUsage(
+            s.getRootParent(SymbolKind.Method),
+            ctx.IDENTIFIER().getText(),
+            Ranges.create(ctx.IDENTIFIER()),
+            false
+          );
+        }
+      });
+
+      return super.visitForStatement(ctx);
+    }
+
+    @Override
+    public BSLParserRuleContext visitForEachStatement(BSLParser.ForEachStatementContext ctx) {
+      if (ctx.IDENTIFIER() == null) {
+        return super.visitForEachStatement(ctx);
+      }
+
+      findVariableSymbol(ctx.IDENTIFIER().getText()).ifPresent(s -> {
+        if (notVariableInitialization(ctx, s)) {
+          addVariableUsage(
+            s.getRootParent(SymbolKind.Method),
+            ctx.IDENTIFIER().getText(),
+            Ranges.create(ctx.IDENTIFIER()),
+            false
+          );
+        }
+      });
+
+      return super.visitForEachStatement(ctx);
+    }
+
     private Optional<VariableSymbol> findVariableSymbol(String variableName) {
       var variableSymbol = documentContext.getSymbolTree()
         .getVariableSymbol(variableName, currentScope);
@@ -281,6 +321,14 @@ private boolean notVariableInitialization(BSLParser.ModuleVarDeclarationContext
       return !Ranges.containsRange(variableSymbol.getRange(), Ranges.create(ctx));
     }
 
+    private boolean notVariableInitialization(BSLParser.ForStatementContext ctx, VariableSymbol variableSymbol) {
+      return !Ranges.containsRange(variableSymbol.getRange(), Ranges.create(ctx.IDENTIFIER()));
+    }
+
+    private boolean notVariableInitialization(BSLParser.ForEachStatementContext ctx, VariableSymbol variableSymbol) {
+      return !Ranges.containsRange(variableSymbol.getRange(), Ranges.create(ctx.IDENTIFIER()));
+    }
+
     private void addVariableUsage(Optional<SourceDefinedSymbol> methodSymbol,
                                   String variableName,
                                   Range range,

src/test/java/com/github/_1c_syntax/bsl/languageserver/references/ReferenceIndexFillerTest.java

@@ -100,13 +100,33 @@ void testFindVariables() {
       .getMethodSymbol("ТретийМетод");
     assertThat(scopeMethod).isPresent();
     var references = referenceIndex.getReferencesFrom(scopeMethod.get());
-    assertThat(references).hasSize(11);
+    assertThat(references).hasSize(13);
 
     var targetVariable = documentContext.getSymbolTree().getVariables().get(0);
     var usage = referenceIndex.getReferencesTo(targetVariable);
     assertThat(usage).hasSize(5);
   }
 
+  @Test
+  void testFindVariablesInForStatements() {
+    DocumentContext documentContext = TestUtils.getDocumentContextFromFile(
+      "./src/test/resources/references/ReferenceIndexFillerVariableTest.bsl"
+    );
+    referenceIndexFiller.fill(documentContext);
+
+    var referencedForStatementsSymbol = referenceIndex.getReference(
+      documentContext.getUri(),
+      new Position(38, 25)
+    );
+    assertThat(referencedForStatementsSymbol).isPresent();
+
+    var referencedForEachStatementSymbol = referenceIndex.getReference(
+      documentContext.getUri(),
+      new Position(44, 30)
+    );
+    assertThat(referencedForEachStatementSymbol).isPresent();
+  }
+
   @Test
   void testFindVariablesRangesCallStatement() {
     DocumentContext documentContext = TestUtils.getDocumentContextFromFile(

src/test/resources/references/ReferenceIndexFillerVariableTest.bsl

@@ -34,6 +34,18 @@
     Модуль.Метод(Параметр3);
     Третья = Модуль.Метод();
 
+    Для Каждого ЭлементКоллекции Из Коллекция Цикл
+
+            Метод(ЭлементКоллекции);
+
+    КонецЦикла;
+
+    Для Счетчик = 1 По 100 Цикл
+
+        Коллекция.Добавить(Счетчик);
+
+    КонецЦикла;
+
 КонецФункции
 
 Метод(Модуль, Модуль2);