| // Copyright (c) 2015, 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/plugin/analysis/occurrences/occurrences_core.dart'; |
| import 'package:analysis_server/src/protocol_server.dart' as protocol; |
| import 'package:analyzer/dart/ast/ast.dart'; |
| import 'package:analyzer/dart/ast/visitor.dart'; |
| import 'package:analyzer/dart/element/element.dart'; |
| import 'package:analyzer/src/dart/element/element.dart'; |
| import 'package:analyzer/src/dart/element/member.dart'; |
| |
| void addDartOccurrences(OccurrencesCollector collector, CompilationUnit unit) { |
| _DartUnitOccurrencesComputerVisitor visitor = |
| new _DartUnitOccurrencesComputerVisitor(); |
| unit.accept(visitor); |
| visitor.elementsOffsets.forEach((engineElement, offsets) { |
| int length = engineElement.nameLength; |
| protocol.Element serverElement = protocol.convertElement(engineElement); |
| protocol.Occurrences occurrences = |
| new protocol.Occurrences(serverElement, offsets, length); |
| collector.addOccurrences(occurrences); |
| }); |
| } |
| |
| class _DartUnitOccurrencesComputerVisitor extends RecursiveAstVisitor { |
| final Map<Element, List<int>> elementsOffsets = <Element, List<int>>{}; |
| |
| @override |
| visitSimpleIdentifier(SimpleIdentifier node) { |
| Element element = node.bestElement; |
| if (element != null) { |
| _addOccurrence(element, node.offset); |
| } |
| return super.visitSimpleIdentifier(node); |
| } |
| |
| void _addOccurrence(Element element, int offset) { |
| element = _canonicalizeElement(element); |
| if (element == null || element == DynamicElementImpl.instance) { |
| return; |
| } |
| List<int> offsets = elementsOffsets[element]; |
| if (offsets == null) { |
| offsets = <int>[]; |
| elementsOffsets[element] = offsets; |
| } |
| offsets.add(offset); |
| } |
| |
| Element _canonicalizeElement(Element element) { |
| if (element is FieldFormalParameterElement) { |
| element = (element as FieldFormalParameterElement).field; |
| } |
| if (element is PropertyAccessorElement) { |
| element = (element as PropertyAccessorElement).variable; |
| } |
| if (element is Member) { |
| element = (element as Member).baseElement; |
| } |
| return element; |
| } |
| } |