[completion] Update CodeCompletionConsumer::handleResults()

Receive 'CodeCompletionContext' so that consumers can get more
information from it.

Author: benlangmuir

include/swift/IDE/CodeCompletion.h

@@ -995,8 +995,7 @@ struct SimpleCachingCodeCompletionConsumer : public CodeCompletionConsumer {
                                DeclContext *DCForModules) override;
 
   /// Clients should override this method to receive \p Results.
-  virtual void handleResults(
-      MutableArrayRef<CodeCompletionResult *> Results) = 0;
+  virtual void handleResults(CodeCompletionContext &context) = 0;
 };
 
 /// A code completion result consumer that prints the results to a
@@ -1018,7 +1017,8 @@ class PrintingCodeCompletionConsumer
        IncludeComments(IncludeComments),
        PrintAnnotatedDescription(PrintAnnotatedDescription) {}
 
- void handleResults(MutableArrayRef<CodeCompletionResult *> Results) override;
+ void handleResults(CodeCompletionContext &context) override;
+ void handleResults(MutableArrayRef<CodeCompletionResult *> Results);
 };
 
 /// Create a factory for code completion callbacks.

lib/IDE/CodeCompletion.cpp

@@ -7376,6 +7376,12 @@ void CodeCompletionCallbacksImpl::doneParsing() {
   deliverCompletionResults(CompletionContext, Lookup, CurDeclContext, Consumer);
 }
 
+void PrintingCodeCompletionConsumer::handleResults(
+    CodeCompletionContext &context) {
+  auto results = context.takeResults();
+  handleResults(results);
+}
+
 void PrintingCodeCompletionConsumer::handleResults(
     MutableArrayRef<CodeCompletionResult *> Results) {
   unsigned NumResults = 0;
@@ -7545,7 +7551,7 @@ void SimpleCachingCodeCompletionConsumer::handleResultsAndModules(
                        &context.getResultSink());
   }
 
-  handleResults(context.takeResults());
+  handleResults(context);
 }
 
 //===----------------------------------------------------------------------===//

lib/IDE/REPLCodeCompletion.cpp

@@ -165,7 +165,8 @@ class REPLCodeCompletionConsumer : public SimpleCachingCodeCompletionConsumer {
   REPLCodeCompletionConsumer(REPLCompletions &Completions)
       : Completions(Completions) {}
 
-  void handleResults(MutableArrayRef<CodeCompletionResult *> Results) override {
+  void handleResults(CodeCompletionContext &context) override {
+    MutableArrayRef<CodeCompletionResult *> Results = context.takeResults();
     CodeCompletionContext::sortCompletionResults(Results);
     for (auto Result : Results) {
       std::string InsertableString = toInsertableString(Result);

tools/SourceKit/lib/SwiftLang/SwiftCompletion.cpp

@@ -97,7 +97,8 @@ struct SwiftCodeCompletionConsumer
   }
   void clearContext() { swiftContext = SwiftCompletionInfo(); }
 
-  void handleResults(MutableArrayRef<CodeCompletionResult *> Results) override {
+  void handleResults(CodeCompletionContext &context) override {
+    MutableArrayRef<CodeCompletionResult *> Results = context.takeResults();
     assert(swiftContext.swiftASTContext);
     handleResultsImpl(Results, swiftContext);
   }