commit b4b85f2033fcbca13fb154979da8895056eb6254
author Argyrios Kyrtzidis <akyrtzi@gmail.com> Mon Apr 24 14:52:00 2017 +0000
committer Argyrios Kyrtzidis <akyrtzi@gmail.com> Mon Apr 24 14:52:00 2017 +0000
tree f761c585d8bf8f3c729bb2d2f52f745154bbea29
parent cab4617132b0342d1774ff71dcbd16fde2409bd4

[index] If the 'external_source_symbol' attribute indicates 'Swift' as the language then report it accordingly

llvm-svn: 301183

clang/lib/Index/IndexSymbol.cpp

diff --git a/clang/lib/Index/IndexSymbol.cpp b/clang/lib/Index/IndexSymbol.cpp
index ea66b70..0bfa193 100644
--- a/clang/lib/Index/IndexSymbol.cpp
+++ b/clang/lib/Index/IndexSymbol.cpp
@@ -318,6 +318,20 @@
   if (Info.Properties & (unsigned)SymbolProperty::Generic)
     Info.Lang = SymbolLanguage::CXX;
 
+  auto getExternalSymAttr = [](const Decl *D) -> ExternalSourceSymbolAttr* {
+    if (auto *attr = D->getAttr<ExternalSourceSymbolAttr>())
+      return attr;
+    if (auto *dcd = dyn_cast<Decl>(D->getDeclContext())) {
+      if (auto *attr = dcd->getAttr<ExternalSourceSymbolAttr>())
+        return attr;
+    }
+    return nullptr;
+  };
+  if (auto *attr = getExternalSymAttr(D)) {
+    if (attr->getLanguage() == "Swift")
+      Info.Lang = SymbolLanguage::Swift;
+  }
+
   return Info;
 }
 
@@ -458,6 +472,7 @@
   case SymbolLanguage::C: return "C";
   case SymbolLanguage::ObjC: return "ObjC";
   case SymbolLanguage::CXX: return "C++";
+  case SymbolLanguage::Swift: return "Swift";
   }
   llvm_unreachable("invalid symbol language kind");
 }