[SourceKit] Return RenameLocUsage in related identifiers response

This will allow us to get syntactic rename ranges from the locations returned by related identifiers.

Author: ahoppen

test/SourceKit/RelatedIdents/related_idents.swift

@@ -109,10 +109,10 @@ func ifLet(test: Int?) {
 
 // RUN: %sourcekitd-test -req=related-idents -pos=6:17 %s -- -module-name related_idents %s | %FileCheck -check-prefix=CHECK1 %s
 // CHECK1: START RANGES
-// CHECK1-NEXT: 1:7 - 2
-// CHECK1-NEXT: 6:11 - 2
-// CHECK1-NEXT: 6:16 - 2
-// CHECK1-NEXT: 9:11 - 2
+// CHECK1-NEXT: 1:7 - 2 - source.syntacticrename.definition
+// CHECK1-NEXT: 6:11 - 2 - source.syntacticrename.reference
+// CHECK1-NEXT: 6:16 - 2 - source.syntacticrename.reference
+// CHECK1-NEXT: 9:11 - 2 - source.syntacticrename.reference
 // CHECK1-NEXT: END RANGES
 
 // RUN: %sourcekitd-test -req=related-idents -pos=5:9 %s -- -module-name related_idents %s | %FileCheck -check-prefix=CHECK2 %s

tools/SourceKit/include/SourceKit/Core/LangSupport.h

@@ -786,6 +786,7 @@ struct SemanticRefactoringInfo {
 struct RelatedIdentInfo {
   unsigned Offset;
   unsigned Length;
+  swift::ide::RenameLocUsage Usage;
 };
 
 /// Represent one branch of an if config.

tools/SourceKit/lib/SwiftLang/SwiftSourceDocInfo.cpp

@@ -2596,7 +2596,8 @@ void SwiftLangSupport::findRelatedIdentifiersInFile(
           }
           unsigned Offset = SrcMgr.getLocOffsetInBuffer(
               ResolvedLoc.range.getStart(), BufferID);
-          Ranges.push_back({Offset, ResolvedLoc.range.getByteLength()});
+          Ranges.push_back(
+              {Offset, ResolvedLoc.range.getByteLength(), RenameLoc.Usage});
         }
       };
       Action();

tools/SourceKit/tools/sourcekitd-test/sourcekitd-test.cpp

@@ -2456,8 +2456,10 @@ static void printRelatedIdents(sourcekitd_variant_t Info, StringRef Filename,
     sourcekitd_variant_t Range = sourcekitd_variant_array_get_value(Res, i);
     int64_t Offset = sourcekitd_variant_dictionary_get_int64(Range, KeyOffset);
     int64_t Length = sourcekitd_variant_dictionary_get_int64(Range, KeyLength);
+    auto Usage = sourcekitd_variant_dictionary_get_uid(Range, KeyNameType);
     auto LineCol = resolveToLineCol(Offset, Filename, VFSFiles);
-    OS << LineCol.first << ':' << LineCol.second << " - " << Length << '\n';
+    OS << LineCol.first << ':' << LineCol.second << " - " << Length << " - "
+       << sourcekitd_uid_get_string_ptr(Usage) << '\n';
   }
   OS << "END RANGES\n";
 }

tools/SourceKit/tools/sourcekitd/lib/Service/Requests.cpp

@@ -2858,6 +2858,19 @@ reportVariableTypeInfo(const RequestResult<VariableTypesInFile> &Result,
 // FindRelatedIdents
 //===----------------------------------------------------------------------===//
 
+static sourcekitd_uid_t renameLocUsageUID(swift::ide::RenameLocUsage Usage) {
+  switch (Usage) {
+  case swift::ide::RenameLocUsage::Definition:
+    return KindDefinition;
+  case swift::ide::RenameLocUsage::Reference:
+    return KindReference;
+  case swift::ide::RenameLocUsage::Call:
+    return KindCall;
+  case swift::ide::RenameLocUsage::Unknown:
+    return KindUnknown;
+  }
+}
+
 static void findRelatedIdents(StringRef PrimaryFilePath,
                               StringRef InputBufferName, int64_t Offset,
                               bool CancelOnSubsequentRequest,
@@ -2882,6 +2895,7 @@ static void findRelatedIdents(StringRef PrimaryFilePath,
           auto Elem = Arr.appendDictionary();
           Elem.set(KeyOffset, R.Offset);
           Elem.set(KeyLength, R.Length);
+          Elem.set(KeyNameType, renameLocUsageUID(R.Usage));
         }
 
         Rec(RespBuilder.createResponse());