Initial implementation for 'analysis.setSubscriptions' API.

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 'package:analysis_server/src/analysis_logger.dart';
 import 'package:analysis_server/src/channel.dart';
@@ skipping 102 matching lines @@
    * no calls to [performTask] on the event queue.
    */
   final List<AnalysisContext> contextWorkQueue = new List<AnalysisContext>();
 
   /**
    * A set of the [ServerService]s to send notifications for.
    */
   Set<ServerService> serverServices = new Set<ServerService>();
 
   /**
+   * A table mapping [AnalysisService]s to the file paths for which these
+   * notifications should be sent.
+   */
+  Map<AnalysisService, Set<String>> analysisServices = <AnalysisService, Set<String>>{};
+
+  /**
    * Initialize a newly created server to receive requests from and send
    * responses to the given [channel].
    */
   AnalysisServer(this.channel, ResourceProvider resourceProvider) {
     contextDirectoryManager = new AnalysisServerContextDirectoryManager(this, resourceProvider);
     AnalysisEngine.instance.logger = new AnalysisLogger();
     running = true;
     Notification notification = new Notification(NOTIFICATION_CONNECTED);
     channel.sendNotification(notification);
     channel.listen(handleRequest, onDone: done, onError: error);
@@ skipping 104 matching lines @@
   }
 
   /**
    * Send the information in the given list of notices back to the client.
    */
   void sendNotices(List<ChangeNotice> notices) {
     for (int i = 0; i < notices.length; i++) {
       ChangeNotice notice = notices[i];
       Source source = notice.source;
       CompilationUnit dartUnit = notice.compilationUnit;
-      // TODO(scheglov) use subscriptions to determine notifications to send
+      // TODO(scheglov) use default subscriptions
+      String file = source.fullName;
+      if (dartUnit != null) {
+        Set<String> files = analysisServices[AnalysisService.HIGHLIGHTS];
+        if (files != null && files.contains(file)) {
+          sendAnalysisNotificationHighlights(file, dartUnit);
+        }
+      }
       if (!source.isInSystemLibrary) {
         // errors
-        {
-          Notification notification = new Notification(NOTIFICATION_ERRORS);
-          notification.setParameter(FILE, source.fullName);
-          notification.setParameter(ERRORS, notice.errors.map(errorToJson).toList());
-          sendNotification(notification);
-        }
-        // highlights
-        if (dartUnit != null) {
-          Notification notification = new Notification(NOTIFICATION_HIGHLIGHTS);
-          notification.setParameter(FILE, source.fullName);
-          notification.setParameter(
-              REGIONS,
-              new DartUnitHighlightsComputer(dartUnit).compute());
-          sendNotification(notification);
-        }
+        sendAnalysisNotificationErrors(file, notice.errors);
       }
     }
   }
 
+  void sendAnalysisNotificationErrors(String file, List<AnalysisError> errors) {
+    Notification notification = new Notification(NOTIFICATION_ERRORS);
+    notification.setParameter(FILE, file);
+    notification.setParameter(ERRORS, errors.map(errorToJson).toList());
+    sendNotification(notification);
+  }
+
+  void sendAnalysisNotificationHighlights(String file, CompilationUnit dartUnit) {
+    Notification notification = new Notification(NOTIFICATION_HIGHLIGHTS);
+    notification.setParameter(FILE, file);
+    notification.setParameter(
+        REGIONS,
+        new DartUnitHighlightsComputer(dartUnit).compute());
+    sendNotification(notification);
+  }
+
   /**
    * Send status notification to the client. The `contextId` indicates
    * the current context being analyzed or `null` if analysis is complete.
    */
   void sendStatusNotification(String contextId) {
     if (contextId == lastStatusNotificationContextId) {
       return;
     }
     lastStatusNotificationContextId = contextId;
     Notification notification = new Notification(NOTIFICATION_STATUS);
@@ skipping 46 matching lines @@
         } else {
           analysisContext.setChangedContents(source, change.content,
               change.offset, change.oldLength, change.newLength);
         }
         addContextToWorkQueue(analysisContext);
       }
     });
   }
 
   /**
+   * Implementation for `analysis.setSubscriptions`.
+   */
+  void setAnalysisSubscriptions(Map<AnalysisService, Set<String>> subscriptions) {
+    // send notifications for already analyzed sources
+    subscriptions.forEach((service, Set<String> newFiles) {
+      Set<String> oldFiles = analysisServices[service];
+      Set<String> todoFiles = oldFiles != null ? newFiles.difference(oldFiles) : newFiles;
+      for (String file in todoFiles) {
+        if (service == AnalysisService.ERRORS) {
+          Source source = _getSource(file);
+          AnalysisContext analysisContext = _getAnalysisContext(file);
+          List<AnalysisError> errors = analysisContext.getErrors(source).errors;
+          sendAnalysisNotificationErrors(file, errors);
+        }
+        if (service == AnalysisService.HIGHLIGHTS) {
+          CompilationUnit dartUnit = test_getResolvedCompilationUnit(file);
+          if (dartUnit != null) {
+            sendAnalysisNotificationHighlights(file, dartUnit);
+          }
+        }
+      }
+    });
+    // remember new subscriptions
+    this.analysisServices = subscriptions;
+  }
+
+  /**
    * Return the [AnalysisContext] that is used to analyze the given [path].
    * Return `null` if there is no such context.
    */
   AnalysisContext _getAnalysisContext(String path) {
     for (Folder folder in folderMap.keys) {
       if (path.startsWith(folder.path)) {
         return folderMap[folder].context;
       }
     }
     return null;
@@ skipping 129 matching lines @@
 /**
  * An enumeration of the services provided by the server domain.
  */
 class ServerService extends Enum2<ServerService> {
   static const ServerService STATUS = const ServerService('STATUS', 0);
 
   static const List<ServerService> VALUES = const [STATUS];
 
   const ServerService(String name, int ordinal) : super(name, ordinal);
 }