Prepare for decoupling analyzer ASTs from element model.

pkg/analysis_server/lib/src/services/completion/dart/completion_manager.dart

 // 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.
 
 library services.completion.dart.manager;
 
 import 'dart:async';
 
 import 'package:analysis_server/plugin/protocol/protocol.dart';
 import 'package:analysis_server/src/provisional/completion/completion_core.dart'
     show CompletionContributor, CompletionRequest;
 import 'package:analysis_server/src/provisional/completion/dart/completion_dart.dart';
 import 'package:analysis_server/src/provisional/completion/dart/completion_plugin.dart';
 import 'package:analysis_server/src/provisional/completion/dart/completion_target.dart';
 import 'package:analysis_server/src/services/completion/completion_core.dart';
 import 'package:analysis_server/src/services/completion/completion_performance.dart';
 import 'package:analysis_server/src/services/completion/dart/common_usage_sorter.dart';
 import 'package:analysis_server/src/services/completion/dart/contribution_sorter.dart';
 import 'package:analysis_server/src/services/completion/dart/optype.dart';
 import 'package:analysis_server/src/services/search/search_engine.dart';
 import 'package:analyzer/dart/ast/ast.dart';
+import 'package:analyzer/dart/ast/standard_resolution_map.dart';
 import 'package:analyzer/dart/ast/standard_ast_factory.dart';
 import 'package:analyzer/dart/ast/token.dart';
 import 'package:analyzer/dart/element/element.dart';
 import 'package:analyzer/dart/element/type.dart';
 import 'package:analyzer/exception/exception.dart';
 import 'package:analyzer/file_system/file_system.dart';
 import 'package:analyzer/src/context/context.dart' show AnalysisFutureHelper;
 import 'package:analyzer/src/dart/analysis/driver.dart';
 import 'package:analyzer/src/dart/ast/token.dart';
 import 'package:analyzer/src/generated/engine.dart' hide AnalysisResult;
@@ skipping 140 matching lines @@
       this.offset,
       CompilationUnit unit,
       this._originalRequest,
       this.performance) {
     _updateTargets(unit);
   }
 
   @override
   LibraryElement get coreLib {
     if (result != null) {
-      AnalysisContext context = result.unit.element.context;
+      AnalysisContext context =
+          resolutionMap.elementForCompilationUnit(result.unit).context;
       _coreLib = context.typeProvider.objectType.element.library;
     } else {
       Source coreUri = sourceFactory.forUri('dart:core');
       _coreLib = context.computeLibraryElement(coreUri);
     }
     return _coreLib;
   }
 
   @override
   bool get includeIdentifiers {
@@ skipping 125 matching lines @@
     return _resolvedImports;
   }
 
   @override
   Future<List<CompilationUnitElement>> resolveUnits() async {
     checkAborted();
     if (_resolvedUnits != null) {
       return _resolvedUnits;
     }
     if (result != null) {
-      _resolvedUnits = result.unit.element.library.units;
+      _resolvedUnits =
+          resolutionMap.elementForCompilationUnit(result.unit).library.units;
       return _resolvedUnits;
     }
     LibraryElement libElem = libraryElement;
     if (libElem == null) {
       return null;
     }
     _resolvedUnits = <CompilationUnitElement>[];
     for (CompilationUnitElement unresolvedUnit in libElem.units) {
       CompilationUnit unit = await _computeAsync(
           this,
@@ skipping 50 matching lines @@
     CompletionPerformance performance =
         (request as CompletionRequestImpl).performance;
     const BUILD_REQUEST_TAG = 'build DartCompletionRequest';
     performance.logStartTime(BUILD_REQUEST_TAG);
 
     Source libSource;
     CompilationUnit unit;
     if (request.context == null) {
       unit = request.result.unit;
       // TODO(scheglov) support for parts
-      libSource = unit.element.source;
+      libSource = resolutionMap.elementForCompilationUnit(unit).source;
     } else {
       Source source = request.source;
       AnalysisContext context = request.context;
 
       const PARSE_TAG = 'parse unit';
       performance.logStartTime(PARSE_TAG);
       unit = request.context.computeResult(source, PARSED_UNIT);
       performance.logElapseTime(PARSE_TAG);
 
       if (unit.directives.any((d) => d is PartOfDirective)) {
@@ skipping 118 matching lines @@
           if (start <= requestOffset && requestOffset <= end) {
             // Replacement range for import URI
             return new ReplacementRange(start, end - start);
           }
         }
       }
     }
     return new ReplacementRange(requestOffset, 0);
   }
 }