[IDE] Ignore unspported constructs in result builders in code completion mode

Author: ahoppen

lib/Sema/BuilderTransform.cpp

@@ -264,7 +264,12 @@ class ResultBuilderTransform
     for (auto element : braceStmt->getElements()) {
       if (auto unsupported =
               transformBraceElement(element, newBody, buildBlockArguments)) {
-        return failTransform(*unsupported);
+        // When in code completion mode, simply ignore unsported constructs to
+        // get results for anything that's unrelated to the unsupported
+        // constructs.
+        if (!ctx.CompletionCallback) {
+          return failTransform(*unsupported);
+        }
       }
     }
 
@@ -1171,7 +1176,7 @@ ConstraintSystem::matchResultBuilder(AnyFunctionRef fn, Type builderType,
     auto *body = transform.apply(fn.getBody());
 
     if (auto unsupported = transform.getUnsupportedElement()) {
-      assert(!body);
+      assert(!body || getASTContext().CompletionCallback);
 
       // If we aren't supposed to attempt fixes, fail.
       if (!shouldAttemptFixes()) {

lib/Sema/CSStep.cpp

@@ -959,7 +959,7 @@ StepResult ConjunctionStep::resume(bool prevFailed) {
 
       if (Solutions.size() == 1) {
         auto score = Solutions.front().getFixedScore();
-        if (score.Data[SK_Fix] > 0)
+        if (score.Data[SK_Fix] > 0 && !CS.getASTContext().CompletionCallback)
           Producer.markExhausted();
       }
     } else if (Solutions.size() != 1) {