Completion with the new analysis driver.

pkg/analysis_server/lib/src/domain_completion.dart

 // Copyright (c) 2014, 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 domain.completion;
 
 import 'dart:async';
 
 import 'package:analysis_server/plugin/protocol/protocol.dart';
 import 'package:analysis_server/src/analysis_server.dart';
 import 'package:analysis_server/src/constants.dart';
 import 'package:analysis_server/src/provisional/completion/completion_core.dart';
 import 'package:analysis_server/src/services/completion/completion_core.dart';
 import 'package:analysis_server/src/services/completion/completion_performance.dart';
-import 'package:analyzer/src/generated/engine.dart';
+import 'package:analyzer/src/dart/analysis/driver.dart';
+import 'package:analyzer/src/generated/engine.dart' hide AnalysisResult;
+import 'package:analyzer/src/source/source_resource.dart';
 import 'package:analyzer/src/generated/source.dart';
 
 /**
  * Instances of the class [CompletionDomainHandler] implement a [RequestHandler]
  * that handles requests in the search domain.
  */
 class CompletionDomainHandler implements RequestHandler {
   /**
    * The maximum number of performance measurements to keep.
    */
@@ skipping 69 matching lines @@
     // TODO (danrubel) if request is obsolete
     // (processAnalysisRequest returns false)
     // then send empty results
 
     return new CompletionResult(
         request.replacementOffset, request.replacementLength, suggestions);
   }
 
   @override
   Response handleRequest(Request request) {
-    if (server.options.enableNewAnalysisDriver) {
-      // TODO(scheglov) implement for the new analysis driver
-      String completionId = (_nextCompletionId++).toString();
-      new Future(() {
-        sendCompletionNotification(completionId, 0, 0, []);
-      });
-      return new CompletionGetSuggestionsResult(completionId)
-          .toResponse(request.id);
-    }
     if (server.searchEngine == null) {
       return new Response.noIndexGenerated(request);
     }
-    return runZoned(() {
+    runZoned(() {
       try {
         String requestName = request.method;
         if (requestName == COMPLETION_GET_SUGGESTIONS) {
-          return processRequest(request);
+          processRequest(request);
         }
       } on RequestFailure catch (exception) {
         return exception.response;
       }
       return null;
     }, onError: (exception, stackTrace) {
       server.sendServerErrorNotification(
           'Failed to handle completion domain request: ${request.toJson()}',
           exception,
           stackTrace);
     });
+    return Response.DELAYED_RESPONSE;
   }
 
   /**
    * Process a `completion.getSuggestions` request.
    */
-  Response processRequest(Request request) {
+  Future<Null> processRequest(Request request) async {
     performance = new CompletionPerformance();
 
     // extract and validate params
     CompletionGetSuggestionsParams params =
         new CompletionGetSuggestionsParams.fromRequest(request);
-    ContextSourcePair contextSource = server.getContextSourcePair(params.file);
-    AnalysisContext context = contextSource.context;
-    Source source = contextSource.source;
-    if (context == null || !context.exists(source)) {
-      return new Response.unknownSource(request);
-    }
-    TimestampedData<String> contents = context.getContents(source);
-    if (params.offset < 0 || params.offset > contents.data.length) {
-      return new Response.invalidParameter(
-          request,
-          'params.offset',
-          'Expected offset between 0 and source length inclusive,'
-          ' but found ${params.offset}');
+
+    AnalysisResult result;
+    AnalysisContext context;
+    Source source;
+    if (server.options.enableNewAnalysisDriver) {
+      result = await server.getAnalysisResult(params.file);
+
+      if (result == null) {
+        server.sendResponse(new Response.unknownSource(request));
+        return;
+      }
+
+      if (params.offset < 0 || params.offset > result.content.length) {
+        server.sendResponse(new Response.invalidParameter(
+            request,
+            'params.offset',
+            'Expected offset between 0 and source length inclusive,'
+                ' but found ${params.offset}'));
+        return;
+      }
+
+      source = new FileSource(
+          server.resourceProvider.getFile(result.path), result.uri);
+    } else {
+      ContextSourcePair contextSource =
+          server.getContextSourcePair(params.file);
+
+      context = contextSource.context;
+      source = contextSource.source;
+      if (context == null || !context.exists(source)) {
+        server.sendResponse(new Response.unknownSource(request));
+        return;
+      }
+
+      TimestampedData<String> contents = context.getContents(source);
+      if (params.offset < 0 || params.offset > contents.data.length) {
+        server.sendResponse(new Response.invalidParameter(
+            request,
+            'params.offset',
+            'Expected offset between 0 and source length inclusive,'
+            ' but found ${params.offset}'));
+        return;
+      }
     }
 
     recordRequest(performance, context, source, params.offset);
 
     CompletionRequestImpl completionRequest = new CompletionRequestImpl(
+        result,
         context,
         server.resourceProvider,
         server.searchEngine,
         source,
         params.offset,
         performance);
     String completionId = (_nextCompletionId++).toString();
 
     _abortCurrentRequest();
     _currentRequest = completionRequest;
 
+    // initial response without results
+    server.sendResponse(new CompletionGetSuggestionsResult(completionId)
+        .toResponse(request.id));
+
     // Compute suggestions in the background
     computeSuggestions(completionRequest).then((CompletionResult result) {
       const SEND_NOTIFICATION_TAG = 'send notification';
       performance.logStartTime(SEND_NOTIFICATION_TAG);
       sendCompletionNotification(completionId, result.replacementOffset,
           result.replacementLength, result.suggestions);
       performance.logElapseTime(SEND_NOTIFICATION_TAG);
 
       performance.notificationCount = 1;
       performance.logFirstNotificationComplete('notification 1 complete');
       performance.suggestionCountFirst = result.suggestions.length;
       performance.suggestionCountLast = result.suggestions.length;
       performance.complete();
     }).whenComplete(() {
       if (_currentRequest == completionRequest) {
         _currentRequest = null;
       }
     });
-
-    // initial response without results
-    return new CompletionGetSuggestionsResult(completionId)
-        .toResponse(request.id);
   }
 
   /**
    * If tracking code completion performance over time, then
    * record addition information about the request in the performance record.
    */
   void recordRequest(CompletionPerformance performance, AnalysisContext context,
       Source source, int offset) {
     performance.source = source;
     if (performanceListMaxLength == 0 || context == null || source == null) {
@@ skipping 51 matching lines @@
   final int replacementOffset;
 
   /**
    * The suggested completions.
    */
   final List<CompletionSuggestion> suggestions;
 
   CompletionResult(
       this.replacementOffset, this.replacementLength, this.suggestions);
 }