[CSApply] changes.

Author: Amritpan Kaur

lib/Sema/CSApply.cpp

@@ -8151,6 +8151,11 @@ namespace {
       // with already-type-checked expressions. Don't walk into them.
       return false;
     }
+    
+    /// Check if there are any closures or tap expressions left to process.
+    bool shouldPrintFullyChecked() {
+      return ClosuresToTypeCheck.empty() && TapsToTypeCheck.empty();
+    }
 
     /// Process delayed closure bodies and `Tap` expressions.
     ///
@@ -8995,14 +9000,31 @@ ExprWalker::rewriteTarget(SolutionApplicationTarget target) {
     if (cs.isDebugMode()) {
       // If target is a multi-statement closure or
       // a tap expression, expression will not be fully
-      // type checked until these types are visited in
+      // type checked until these expressions are visited in
       // processDelayed().
+      bool isPartial = false;
+      resultExpr->forEachChildExpr([&](Expr *child) -> Expr * {
+        if (auto *closure = dyn_cast<ClosureExpr>(child)) {
+          if (!closure->hasSingleExpressionBody()) {
+            isPartial = true;
+            return nullptr;
+          }
+        }
+        if (isa<TapExpr>(child)) {
+          isPartial = true;
+          return nullptr;
+        }
+      return child;
+      });
+      
       auto &log = llvm::errs();
-      if (!ClosuresToTypeCheck.empty() || !TapsToTypeCheck.empty()) {
+      if (isPartial) {
         log << "---Partially type-checked expression---\n";
-        resultExpr->dump(log);
-        log << "\n";
+      } else {
+        log << "---Type-checked expression---\n";
       }
+      resultExpr->dump(log);
+      log << "\n";
     }
   }
 
@@ -9052,7 +9074,7 @@ Optional<SolutionApplicationTarget> ConstraintSystem::applySolution(
   auto resultTarget = walker.rewriteTarget(target);
   if (!resultTarget)
     return None;
-
+  
   // Visit closures that have non-single expression bodies, tap expressions,
   // and possibly other types of AST nodes which could only be processed
   // after contextual expression.
@@ -9061,16 +9083,18 @@ Optional<SolutionApplicationTarget> ConstraintSystem::applySolution(
   // If any of them failed to type check, bail.
   if (hadError)
     return None;
-
-  rewriter.finalize();
 
-  if (isDebugMode()) {
+  // If there are no more closures or tap expressions to check,
+  // print expression.
+  if (walker.shouldPrintFullyChecked()) {
     auto &log = llvm::errs();
     log << "---Fully type-checked expression---\n";
     resultTarget->getAsExpr()->dump(log);
     log << "\n";
   }
 
+  rewriter.finalize();
+
   return resultTarget;
 }