Merge pull request #3697 from 1c-syntax/copilot/resolve-coloring-nstr-strshablon

Add semantic highlighting for NStr and StrTemplate functions

Author: nixel2007

src/main/java/com/github/_1c_syntax/bsl/languageserver/context/symbol/SymbolTree.java

@@ -27,7 +27,9 @@
 import lombok.Getter;
 import lombok.Value;
 import org.antlr.v4.runtime.ParserRuleContext;
+import org.eclipse.lsp4j.Position;
 import org.eclipse.lsp4j.Range;
+import org.eclipse.lsp4j.SymbolKind;
 
 import java.util.ArrayList;
 import java.util.Collections;
@@ -198,6 +200,35 @@ public Optional<VariableSymbol> getVariableSymbol(String variableName, SourceDef
     );
   }
 
+  /**
+   * Поиск самого вложенного символа, содержащего указанную позицию.
+   * <p>
+   * Использует иерархический спуск по дереву символов вместо линейного поиска.
+   *
+   * @param position Позиция в документе.
+   * @return Символ, содержащий позицию, или символ модуля, если позиция вне всех символов.
+   */
+  public SourceDefinedSymbol getSymbolAtPosition(Position position) {
+    return findSymbolAtPosition(module, position);
+  }
+
+  private SourceDefinedSymbol findSymbolAtPosition(SourceDefinedSymbol parent, Position position) {
+    // Ищем среди детей символ, содержащий позицию
+    for (var child : parent.getChildren()) {
+      if (Ranges.containsPosition(child.getRange(), position)) {
+        // Рекурсивно ищем более вложенный символ
+        var found = findSymbolAtPosition(child, position);
+        // Пропускаем Namespace (Region) - ищем дальше или возвращаем parent
+        if (found.getSymbolKind() == SymbolKind.Namespace) {
+          continue;
+        }
+        return found;
+      }
+    }
+    // Если среди детей не нашли — возвращаем текущий символ
+    return parent;
+  }
+
   private List<SourceDefinedSymbol> createChildrenFlat() {
     List<SourceDefinedSymbol> symbols = new ArrayList<>();
     getChildren().forEach(child -> flatten(child, symbols));

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

@@ -43,7 +43,6 @@
 import org.springframework.stereotype.Component;
 
 import java.net.URI;
-import java.util.Collection;
 import java.util.List;
 import java.util.Locale;
 import java.util.Optional;
@@ -276,7 +275,7 @@ private Optional<Reference> buildReference(
 
     return getSourceDefinedSymbol(symbolOccurrence.symbol())
       .map((SourceDefinedSymbol symbol) -> {
-        SourceDefinedSymbol from = getFromSymbol(symbolOccurrence);
+        var from = getFromSymbol(symbolOccurrence);
         return new Reference(from, symbol, uri, range, occurrenceType);
       })
       .filter(ReferenceIndex::isReferenceAccessible);
@@ -307,20 +306,12 @@ private Optional<SourceDefinedSymbol> getSourceDefinedSymbol(Symbol symbolEntity
   }
 
   private SourceDefinedSymbol getFromSymbol(SymbolOccurrence symbolOccurrence) {
-
     var uri = symbolOccurrence.location().getUri();
     var position = symbolOccurrence.location().getRange().getStart();
 
-    var symbolTree = Optional.ofNullable(serverContext.getDocument(uri))
-      .map(DocumentContext::getSymbolTree);
-    return symbolTree
-      .map(SymbolTree::getChildrenFlat)
-      .stream()
-      .flatMap(Collection::stream)
-      .filter(sourceDefinedSymbol -> sourceDefinedSymbol.getSymbolKind() != SymbolKind.Namespace)
-      .filter(symbol -> Ranges.containsPosition(symbol.getRange(), position))
-      .findFirst()
-      .or(() -> symbolTree.map(SymbolTree::getModule))
+    return Optional.ofNullable(serverContext.getDocument(uri))
+      .map(DocumentContext::getSymbolTree)
+      .map(symbolTree -> symbolTree.getSymbolAtPosition(position))
       .orElseThrow();
   }
 

src/main/java/com/github/_1c_syntax/bsl/languageserver/semantictokens/LexicalSemanticTokensSupplier.java

@@ -30,15 +30,14 @@
 import org.springframework.stereotype.Component;
 
 import java.util.ArrayList;
-import java.util.HashSet;
 import java.util.List;
 import java.util.Objects;
 import java.util.Set;
 
 /**
- * Сапплаер семантических токенов для лексических элементов: строк, чисел, операторов и ключевых слов.
+ * Сапплаер семантических токенов для лексических элементов: чисел, операторов и ключевых слов.
  * <p>
- * Исключает строки, которые содержат запросы SDBL (они обрабатываются в {@link QuerySemanticTokensSupplier}).
+ * Строки обрабатываются в {@link StringSemanticTokensSupplier}.
  */
 @Component
 @RequiredArgsConstructor
@@ -53,8 +52,7 @@ public class LexicalSemanticTokensSupplier implements SemanticTokensSupplier {
     BSLLexer.STRING,
     BSLLexer.STRINGPART,
     BSLLexer.STRINGSTART,
-    BSLLexer.STRINGTAIL,
-    BSLLexer.PREPROC_STRING
+    BSLLexer.STRINGTAIL
   );
 
   private static final Set<Integer> OPERATOR_TYPES = Set.of(
@@ -107,14 +105,13 @@ public class LexicalSemanticTokensSupplier implements SemanticTokensSupplier {
   public List<SemanticTokenEntry> getSemanticTokens(DocumentContext documentContext) {
     List<SemanticTokenEntry> entries = new ArrayList<>();
     var tokensFromDefaultChannel = documentContext.getTokensFromDefaultChannel();
-    var stringsWithQueries = collectStringsWithQueries(documentContext);
 
     for (Token token : tokensFromDefaultChannel) {
       var tokenType = token.getType();
       var tokenText = Objects.toString(token.getText(), "");
       if (!tokenText.isEmpty()) {
-        // Skip string tokens that contain SDBL tokens - they'll be handled by QuerySemanticTokensSupplier
-        if (STRING_TYPES.contains(tokenType) && stringsWithQueries.contains(token)) {
+        // Skip STRING tokens - they are handled by StringSemanticTokensSupplier
+        if (STRING_TYPES.contains(tokenType)) {
           continue;
         }
         selectAndAddSemanticToken(entries, token, tokenType);
@@ -124,46 +121,15 @@ public List<SemanticTokenEntry> getSemanticTokens(DocumentContext documentContex
     return entries;
   }
 
-  private Set<Token> collectStringsWithQueries(DocumentContext documentContext) {
-    var queries = documentContext.getQueries();
-    if (queries.isEmpty()) {
-      return Set.of();
-    }
-
-    var stringsToSkip = new HashSet<Token>();
-    for (var query : queries) {
-      for (Token queryToken : query.getTokens()) {
-        if (queryToken.getChannel() != Token.DEFAULT_CHANNEL) {
-          continue;
-        }
-        int queryLine = queryToken.getLine();
-        for (var bslToken : documentContext.getTokensFromDefaultChannel()) {
-          if (!STRING_TYPES.contains(bslToken.getType())) {
-            continue;
-          }
-          if (bslToken.getLine() == queryLine) {
-            var bslRange = Ranges.create(bslToken);
-            int queryStart = queryToken.getCharPositionInLine();
-            int queryEnd = queryStart + queryToken.getText().length();
-            if (queryStart >= bslRange.getStart().getCharacter() && queryEnd <= bslRange.getEnd().getCharacter()) {
-              stringsToSkip.add(bslToken);
-            }
-          }
-        }
-      }
-    }
-    return stringsToSkip;
-  }
-
   private void selectAndAddSemanticToken(List<SemanticTokenEntry> entries, Token token, int tokenType) {
     // Skip '&' and all ANNOTATION_* symbol tokens here to avoid duplicate Decorator emission (handled via AST)
     if (tokenType == BSLLexer.AMPERSAND || ANNOTATION_TOKENS.contains(tokenType)) {
       return;
     }
 
-    if (STRING_TYPES.contains(tokenType)) {
+    if (tokenType == BSLLexer.DATETIME) {
       helper.addRange(entries, Ranges.create(token), SemanticTokenTypes.String);
-    } else if (tokenType == BSLLexer.DATETIME) {
+    } else if (tokenType == BSLLexer.PREPROC_STRING) {
       helper.addRange(entries, Ranges.create(token), SemanticTokenTypes.String);
     } else if (NUMBER_TYPES.contains(tokenType)) {
       helper.addRange(entries, Ranges.create(token), SemanticTokenTypes.Number);