| // Copyright (c) 2018, the Dart project authors. Please see the AUTHORS file |
| // for details. All rights reserved. Use of this source code is governed by a |
| // BSD-style license that can be found in the LICENSE file. |
| |
| import 'package:analysis_server/lsp_protocol/protocol_generated.dart'; |
| import 'package:analysis_server/lsp_protocol/protocol_special.dart'; |
| import 'package:analysis_server/src/lsp/handlers/handlers.dart'; |
| import 'package:analysis_server/src/lsp/lsp_analysis_server.dart'; |
| import 'package:analysis_server/src/lsp/mapping.dart'; |
| import 'package:analysis_server/src/protocol_server.dart' show SearchResult; |
| import 'package:analysis_server/src/protocol_server.dart' show NavigationTarget; |
| import 'package:analysis_server/src/search/element_references.dart'; |
| import 'package:analyzer/dart/analysis/results.dart'; |
| import 'package:analyzer/dart/ast/ast.dart'; |
| import 'package:analyzer/dart/element/element.dart'; |
| import 'package:analyzer_plugin/src/utilities/navigation/navigation.dart'; |
| import 'package:analyzer_plugin/utilities/navigation/navigation_dart.dart'; |
| import 'package:collection/collection.dart'; |
| |
| class ReferencesHandler |
| extends MessageHandler<ReferenceParams, List<Location>?> { |
| ReferencesHandler(LspAnalysisServer server) : super(server); |
| @override |
| Method get handlesMessage => Method.textDocument_references; |
| |
| @override |
| LspJsonHandler<ReferenceParams> get jsonHandler => |
| ReferenceParams.jsonHandler; |
| |
| @override |
| Future<ErrorOr<List<Location>?>> handle( |
| ReferenceParams params, CancellationToken token) async { |
| if (!isDartDocument(params.textDocument)) { |
| return success(const []); |
| } |
| |
| final pos = params.position; |
| final path = pathOfDoc(params.textDocument); |
| final unit = await path.mapResult(requireResolvedUnit); |
| final offset = await unit.mapResult((unit) => toOffset(unit.lineInfo, pos)); |
| return offset.mapResult( |
| (offset) => _getRefererences(path.result, offset, params, unit.result)); |
| } |
| |
| List<Location> _getDeclarations(CompilationUnit unit, int offset) { |
| final collector = NavigationCollectorImpl(); |
| computeDartNavigation(server.resourceProvider, collector, unit, offset, 0); |
| |
| return convert(collector.targets, (NavigationTarget target) { |
| final targetFilePath = collector.files[target.fileIndex]; |
| final lineInfo = server.getLineInfo(targetFilePath); |
| return lineInfo != null |
| ? navigationTargetToLocation(targetFilePath, target, lineInfo) |
| : null; |
| }).whereNotNull().toList(); |
| } |
| |
| Future<ErrorOr<List<Location>?>> _getRefererences(String path, int offset, |
| ReferenceParams params, ResolvedUnitResult unit) async { |
| var element = await server.getElementAtOffset(path, offset); |
| if (element is ImportElement) { |
| element = element.prefix; |
| } |
| if (element is FieldFormalParameterElement) { |
| element = element.field; |
| } |
| if (element is PropertyAccessorElement) { |
| element = element.variable; |
| } |
| if (element == null) { |
| return success(null); |
| } |
| |
| final computer = ElementReferencesComputer(server.searchEngine); |
| final results = await computer.compute(element, false); |
| |
| Location? toLocation(SearchResult result) { |
| final lineInfo = server.getLineInfo(result.location.file); |
| return searchResultToLocation(result, lineInfo); |
| } |
| |
| final referenceResults = |
| convert(results, toLocation).whereNotNull().toList(); |
| |
| final compilationUnit = unit.unit; |
| if (compilationUnit != null && params.context.includeDeclaration == true) { |
| // Also include the definition for the symbol at this location. |
| referenceResults.addAll(_getDeclarations(compilationUnit, offset)); |
| } |
| |
| return success(referenceResults); |
| } |
| } |