Support for Quick Assists with the new analysis driver.

pkg/analysis_server/lib/src/analysis_server.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 analysis.server;
 
 import 'dart:async';
 import 'dart:collection';
 import 'dart:core';
 import 'dart:io' as io;
@@ skipping 304 matching lines @@
   ResolverProvider packageResolverProvider;
 
   /**
    * The manager of pub package summaries.
    */
   PubSummaryManager pubSummaryManager;
 
   ByteStore byteStore;
 
   /**
+   * The set of the files that are currently priority.
+   */
+  final Set<String> priorityFiles = new Set<String>();
+
+  /**
+   * The cached results units for [priorityFiles].
+   */
+  final Map<String, nd.AnalysisResult> priorityFileResults = {};
+
+  /**
    * Initialize a newly created server to receive requests from and send
    * responses to the given [channel].
    *
    * If [rethrowExceptions] is true, then any exceptions thrown by analysis are
    * propagated up the call stack.  The default is true to allow analysis
    * exceptions to show up in unit tests, but it should be set to false when
    * running a full analysis server.
    */
   AnalysisServer(
       this.channel,
@@ skipping 232 matching lines @@
    */
   nd.AnalysisDriver getAnalysisDriver(String path) {
     Iterable<nd.AnalysisDriver> drivers = driverMap.values;
     if (drivers.isNotEmpty) {
       return drivers.firstWhere((driver) => driver.isAddedFile(path),
           orElse: () => drivers.first);
     }
     return null;
   }
 
+  /**
+   * Return the analysis result for the file with the given [path].
+   */
+  Future<nd.AnalysisResult> getAnalysisResult(String path) async {
+    nd.AnalysisResult result = priorityFileResults[path];
+    if (result != null) {
+      return result;
+    }
+    nd.AnalysisDriver driver = getAnalysisDriver(path);
+    return driver.getResult(path);
+  }
+
   CompilationUnitElement getCompilationUnitElement(String file) {
     ContextSourcePair pair = getContextSourcePair(file);
     if (pair == null) {
       return null;
     }
     // prepare AnalysisContext and Source
     AnalysisContext context = pair.context;
     Source unitSource = pair.source;
     if (context == null || unitSource == null) {
       return null;
@@ skipping 637 matching lines @@
       prevAnalyzedFiles = null;
     }
     generalAnalysisServices = newServices;
   }
 
   /**
    * Set the priority files to the given [files].
    */
   void setPriorityFiles(String requestId, List<String> files) {
     if (options.enableNewAnalysisDriver) {
+      // Flush results for files that are not priority anymore.
+      priorityFiles
+          .difference(files.toSet())
+          .forEach(priorityFileResults.remove);
+      priorityFiles.clear();
+      priorityFiles.addAll(files);
+      // Set priority files in drivers.
       driverMap.values.forEach((driver) {
         driver.priorityFiles = files;
       });
       return;
     }
     // Note: when a file is a priority file, that information needs to be
     // propagated to all contexts that analyze the file, so that all contexts
     // will be able to do incremental resolution of the file.  See
     // dartbug.com/22209.
     Map<AnalysisContext, List<Source>> sourceMap =
@@ skipping 106 matching lines @@
       operation.perform(this);
     }
   }
 
   /**
    * Implementation for `analysis.updateContent`.
    */
   void updateContent(String id, Map<String, dynamic> changes) {
     if (options.enableNewAnalysisDriver) {
       changes.forEach((file, change) {
+        priorityFileResults.remove(file);
+
         // Prepare the new contents.
         String oldContents = fileContentOverlay[file];
         String newContents;
         if (change is AddContentOverlay) {
           newContents = change.content;
         } else if (change is ChangeContentOverlay) {
           if (oldContents == null) {
             // The client may only send a ChangeContentOverlay if there is
             // already an existing overlay for the source.
             throw new RequestFailure(new Response(id,
@@ skipping 351 matching lines @@
         resourceProvider,
         analysisServer.byteStore,
         analysisServer.fileContentOverlay,
         sourceFactory,
         analysisOptions);
     analysisDriver.name = folder.shortName;
     analysisDriver.status.listen((status) {
       // TODO(scheglov) send server status
     });
     analysisDriver.results.listen((result) {
+      if (analysisServer.priorityFiles.contains(result.path)) {
+        analysisServer.priorityFileResults[result.path] = result;
+      }
       new_sendErrorNotification(analysisServer, result);
       CompilationUnit unit = result.unit;
       if (unit != null) {
         if (analysisServer._hasAnalysisServiceSubscription(
             AnalysisService.HIGHLIGHTS, result.path)) {
           sendAnalysisNotificationHighlights(analysisServer, result.path, unit);
         }
         if (analysisServer._hasAnalysisServiceSubscription(
             AnalysisService.NAVIGATION, result.path)) {
           NavigationCollectorImpl collector = new NavigationCollectorImpl();
@@ skipping 215 matching lines @@
   /**
    * The [PerformanceTag] for time spent in server request handlers.
    */
   static PerformanceTag serverRequests = new PerformanceTag('serverRequests');
 
   /**
    * The [PerformanceTag] for time spent in split store microtasks.
    */
   static PerformanceTag splitStore = new PerformanceTag('splitStore');
 }