IDE: Expose a utility to collect argument label ranges from either tuple or paren exprs. (#8982)

Author: nkcsgexi

include/swift/IDE/Utils.h

@@ -397,6 +397,9 @@ class SourceEditOutputConsumer : public SourceEditConsumer {
   ~SourceEditOutputConsumer();
   void accept(SourceManager &SM, RegionType RegionType, ArrayRef<Replacement> Replacements) override;
 };
+
+// Get the ranges of argument labels from an Arg, either tuple or paren.
+std::vector<CharSourceRange> getCallArgLabelRanges(SourceManager &SM, Expr *Arg);
 } // namespace ide
 } // namespace swift
 

lib/IDE/SwiftSourceDocInfo.cpp

@@ -912,3 +912,27 @@ void swift::ide::getLocationInfo(const ValueDecl *VD,
                                        DeclarationLoc, Filename);
   }
 }
+
+std::vector<CharSourceRange> swift::ide::
+getCallArgLabelRanges(SourceManager &SM, Expr *Arg) {
+  std::vector<CharSourceRange> Ranges;
+  if (TupleExpr *TE = dyn_cast<TupleExpr>(Arg)) {
+    size_t ElemIndex = 0;
+    for(Expr *Elem: TE->getElements()) {
+      SourceLoc LabelStart(Elem->getStartLoc());
+      SourceLoc LabelEnd(LabelStart);
+
+      auto NameIdentifier = TE->getElementName(ElemIndex);
+      if (!NameIdentifier.empty()) {
+        LabelStart = TE->getElementNameLoc(ElemIndex);
+      }
+      Ranges.push_back(CharSourceRange(SM, LabelStart, LabelEnd));
+      ++ElemIndex;
+    }
+  } else if (ParenExpr *PE = dyn_cast<ParenExpr>(Arg)) {
+    if (PE->getSubExpr())
+      Ranges.push_back(CharSourceRange(PE->getSubExpr()->getStartLoc(), 0));
+  }
+
+  return Ranges;
+}