Completion with the new analysis driver.

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/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';
+import 'package:analyzer/src/generated/engine.dart' hide AnalysisResult;
 import 'package:analyzer/src/generated/source.dart';
 import 'package:analyzer/src/task/dart.dart';
 import 'package:analyzer/task/dart.dart';
 import 'package:analyzer/task/model.dart';
 
 /**
  * [DartCompletionManager] determines if a completion request is Dart specific
  * and forwards those requests to all [DartCompletionContributor]s.
  */
 class DartCompletionManager implements CompletionContributor {
@@ skipping 59 matching lines @@
     request.checkAborted();
     return suggestions;
   }
 }
 
 /**
  * The information about a requested list of completions within a Dart file.
  */
 class DartCompletionRequestImpl implements DartCompletionRequest {
   @override
+  final AnalysisResult result;
+
+  @override
   final AnalysisContext context;
 
   @override
   final Source source;
 
   @override
   final int offset;
 
   @override
   Expression dotTarget;
@@ skipping 32 matching lines @@
    */
   List<CompilationUnitElement> _resolvedUnits;
 
   OpType _opType;
 
   final CompletionRequest _originalRequest;
 
   final CompletionPerformance performance;
 
   DartCompletionRequestImpl._(
+      this.result,
       this.context,
       this.resourceProvider,
       this.searchEngine,
       this.librarySource,
       this.source,
       this.offset,
       CompilationUnit unit,
       this._originalRequest,
       this.performance) {
     _updateTargets(unit);
@@ skipping 76 matching lines @@
     }
 
     // Gracefully degrade if librarySource cannot be determined
     if (librarySource == null) {
       return;
     }
 
     // Resolve declarations in the target unit
     // TODO(danrubel) resolve the expression or containing method
     // rather than the entire compilation unit
-    CompilationUnit resolvedUnit = await _computeAsync(
-        this,
-        new LibrarySpecificUnit(librarySource, source),
-        RESOLVED_UNIT,
-        performance,
-        'resolve expression');
+    CompilationUnit resolvedUnit;
+    if (result != null) {
+      resolvedUnit = result.unit;
+    } else {
+      resolvedUnit = await _computeAsync(
+          this,
+          new LibrarySpecificUnit(librarySource, source),
+          RESOLVED_UNIT,
+          performance,
+          'resolve expression');
+    }
 
     // TODO(danrubel) determine if the underlying source has been modified
     // in a way that invalidates the completion request
     // and return null
 
     // Gracefully degrade if unit cannot be resolved
     if (resolvedUnit == null) {
       return;
     }
 
@@ skipping 22 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;
+      return _resolvedUnits;
+    }
     LibraryElement libElem = libraryElement;
     if (libElem == null) {
       return null;
     }
     _resolvedUnits = <CompilationUnitElement>[];
     for (CompilationUnitElement unresolvedUnit in libElem.units) {
       CompilationUnit unit = await _computeAsync(
           this,
           new LibrarySpecificUnit(libElem.source, unresolvedUnit.source),
           RESOLVED_UNIT5,
@@ skipping 43 matching lines @@
    * if the completion request has been aborted.
    */
   static Future<DartCompletionRequest> from(CompletionRequest request,
       {ResultDescriptor resultDescriptor}) async {
     request.checkAborted();
     CompletionPerformance performance =
         (request as CompletionRequestImpl).performance;
     const BUILD_REQUEST_TAG = 'build DartCompletionRequest';
     performance.logStartTime(BUILD_REQUEST_TAG);
 
-    Source source = request.source;
-    AnalysisContext context = request.context;
+    Source libSource;
+    CompilationUnit unit;
+    if (request.context == null) {
+      unit = request.result.unit;
+      // TODO(scheglov) support for parts
+      libSource = unit.element.source;
+    } else {
+      Source source = request.source;
+      AnalysisContext context = request.context;
 
-    const PARSE_TAG = 'parse unit';
-    performance.logStartTime(PARSE_TAG);
-    CompilationUnit unit = request.context.computeResult(source, PARSED_UNIT);
-    performance.logElapseTime(PARSE_TAG);
+      const PARSE_TAG = 'parse unit';
+      performance.logStartTime(PARSE_TAG);
+      unit = request.context.computeResult(source, PARSED_UNIT);
+      performance.logElapseTime(PARSE_TAG);
 
-    Source libSource;
-    if (unit.directives.any((d) => d is PartOfDirective)) {
-      List<Source> libraries = context.getLibrariesContaining(source);
-      if (libraries.isNotEmpty) {
-        libSource = libraries[0];
+      if (unit.directives.any((d) => d is PartOfDirective)) {
+        List<Source> libraries = context.getLibrariesContaining(source);
+        if (libraries.isNotEmpty) {
+          libSource = libraries[0];
+        }
+      } else {
+        libSource = source;
       }
-    } else {
-      libSource = source;
-    }
 
-    // Most (all?) contributors need declarations in scope to be resolved
-    if (libSource != null) {
-      unit = await _computeAsync(
-          request,
-          new LibrarySpecificUnit(libSource, source),
-          resultDescriptor ?? RESOLVED_UNIT5,
-          performance,
-          'resolve declarations');
+      // Most (all?) contributors need declarations in scope to be resolved
+      if (libSource != null) {
+        unit = await _computeAsync(
+            request,
+            new LibrarySpecificUnit(libSource, source),
+            resultDescriptor ?? RESOLVED_UNIT5,
+            performance,
+            'resolve declarations');
+      }
     }
 
     DartCompletionRequestImpl dartRequest = new DartCompletionRequestImpl._(
+        request.result,
         request.context,
         request.resourceProvider,
         request.searchEngine,
         libSource,
         request.source,
         request.offset,
         unit,
         request,
         performance);
 
@@ skipping 86 matching lines @@
           if (start <= requestOffset && requestOffset <= end) {
             // Replacement range for import URI
             return new ReplacementRange(start, end - start);
           }
         }
       }
     }
     return new ReplacementRange(requestOffset, 0);
   }
 }