Initial implementation for 'search.findElementReferences'.

pkg/analysis_server/lib/src/search/search_domain.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.search;
+library search.domain;
+
+import 'dart:async';
 
 import 'package:analysis_server/src/analysis_server.dart';
+import 'package:analysis_server/src/search/element_references.dart';
+import 'package:analysis_server/src/search/search_result.dart';
 import 'package:analysis_server/src/constants.dart';
 import 'package:analysis_server/src/protocol.dart';
+import 'package:analysis_services/search/search_engine.dart';
+import 'package:analyzer/src/generated/element.dart';
 
 /**
  * Instances of the class [SearchDomainHandler] implement a [RequestHandler]
  * that handles requests in the search domain.
  */
 class SearchDomainHandler implements RequestHandler {
   /**
    * The analysis server that is using this handler to process requests.
    */
   final AnalysisServer server;
 
   /**
+   * The [SearchEngine] for this server.
+   */
+  SearchEngine searchEngine;
+
+  /**
+   * The next searc response id.
+   */
+  int _nextSearchId = 0;
+
+  /**
    * Initialize a newly created handler to handle requests for the given [server].
    */
-  SearchDomainHandler(this.server);
-
-  @override
-  Response handleRequest(Request request) {
-    try {
-      String requestName = request.method;
-      if (requestName == SEARCH_FIND_ELEMENT_REFERENCES) {
-        return findElementReferences(request);
-      } else if (requestName == SEARCH_FIND_MEMBER_DECLARATIONS) {
-        return findMemberDeclarations(request);
-      } else if (requestName == SEARCH_FIND_MEMBER_REFERENCES) {
-        return findMemberReferences(request);
-      } else if (requestName == SEARCH_FIND_TOP_LEVEL_DECLARATIONS) {
-        return findTopLevelDeclarations(request);
-      }
-    } on RequestFailure catch (exception) {
-      return exception.response;
-    }
-    return null;
+  SearchDomainHandler(this.server) {
+    searchEngine = server.searchEngine;
   }
 
   Response findElementReferences(Request request) {
-    // file
-    RequestDatum fileDatum = request.getRequiredParameter(FILE);
-    String file = fileDatum.asString();
-    // offset
-    RequestDatum offsetDatum = request.getRequiredParameter(OFFSET);
-    int offset = offsetDatum.asInt();
-    // includePotential
-    RequestDatum includePotentialDatum = request.getRequiredParameter(LENGTH);
-    bool includePotential = includePotentialDatum.asBool();
-    // TODO(brianwilkerson) implement
-    return null;
+    String file = request.getRequiredParameter(FILE).asString();
+    int offset = request.getRequiredParameter(OFFSET).asInt();
+    bool includePotential =
+        request.getRequiredParameter(INCLUDE_POTENTIAL).asBool();
+    // schedule search
+    String searchId = (_nextSearchId++).toString();
+    List<Element> elements = server.getElementsAtOffset(file, offset);
+    elements.forEach((Element element) {
+      var computer = new ElementReferencesComputer(searchEngine);
+      var future = computer.compute(element, includePotential);
+      return future.then((List<SearchResult> results) {
+        bool isLast = identical(element, elements.last);
+        _sendSearchNotification(searchId, isLast, results);
+      });
+    });
+    if (elements.isEmpty) {
+      new Future.microtask(() {
+        _sendSearchNotification(searchId, true, []);
+      });
+    }
+    // respond
+    return new Response(request.id)..setResult(ID, searchId);
   }
 
   Response findMemberDeclarations(Request request) {
     // name
     RequestDatum nameDatum = request.getRequiredParameter(FILE);
     String name = nameDatum.asString();
     // TODO(brianwilkerson) implement
     return null;
   }
 
   Response findMemberReferences(Request request) {
     // name
     RequestDatum nameDatum = request.getRequiredParameter(FILE);
     String name = nameDatum.asString();
     // TODO(brianwilkerson) implement
     return null;
   }
 
   Response findTopLevelDeclarations(Request request) {
     // pattern
     RequestDatum patternDatum = request.getRequiredParameter(FILE);
     String pattern = patternDatum.asString();
     // TODO(brianwilkerson) implement
     return null;
   }
+
+  @override
+  Response handleRequest(Request request) {
+    try {
+      String requestName = request.method;
+      if (requestName == SEARCH_FIND_ELEMENT_REFERENCES) {
+        return findElementReferences(request);
+      } else if (requestName == SEARCH_FIND_MEMBER_DECLARATIONS) {
+        return findMemberDeclarations(request);
+      } else if (requestName == SEARCH_FIND_MEMBER_REFERENCES) {
+        return findMemberReferences(request);
+      } else if (requestName == SEARCH_FIND_TOP_LEVEL_DECLARATIONS) {
+        return findTopLevelDeclarations(request);
+      }
+    } on RequestFailure catch (exception) {
+      return exception.response;
+    }
+    return null;
+  }
+
+  void _sendSearchNotification(String searchId, bool isLast,
+      List<SearchResult> results) {
+    Notification notification = new Notification(SEARCH_RESULTS);
+    notification.setParameter(ID, searchId);
+    notification.setParameter(LAST, isLast);
+    notification.setParameter(
+        RESULTS,
+        results.map(SearchResult.asJson).toList());
+    server.sendNotification(notification);
+  }
 }