When inferring a module map for a framework, infer subframework
modules for each of its subframeworks.
git-svn-id: https://llvm.org/svn/llvm-project/cfe/trunk@145957 91177308-0d34-0410-b5e6-96231b3b80d8
diff --git a/lib/Lex/HeaderSearch.cpp b/lib/Lex/HeaderSearch.cpp
index f090e03..1b969b9 100644
--- a/lib/Lex/HeaderSearch.cpp
+++ b/lib/Lex/HeaderSearch.cpp
@@ -849,7 +849,7 @@
}
// Try to infer a module map.
- return ModMap.inferFrameworkModule(Name, Dir);
+ return ModMap.inferFrameworkModule(Name, Dir, /*Parent=*/0);
}
diff --git a/lib/Lex/ModuleMap.cpp b/lib/Lex/ModuleMap.cpp
index 581a5d5..ac5ae1e 100644
--- a/lib/Lex/ModuleMap.cpp
+++ b/lib/Lex/ModuleMap.cpp
@@ -20,6 +20,7 @@
#include "clang/Basic/TargetInfo.h"
#include "clang/Basic/TargetOptions.h"
#include "llvm/Support/Allocator.h"
+#include "llvm/Support/FileSystem.h"
#include "llvm/Support/Host.h"
#include "llvm/Support/PathV2.h"
#include "llvm/Support/raw_ostream.h"
@@ -219,17 +220,19 @@
Module *
ModuleMap::inferFrameworkModule(StringRef ModuleName,
- const DirectoryEntry *FrameworkDir) {
+ const DirectoryEntry *FrameworkDir,
+ Module *Parent) {
// Check whether we've already found this module.
- if (Module *Module = findModule(ModuleName))
- return Module;
+ if (Module *Mod = lookupModuleQualified(ModuleName, Parent))
+ return Mod;
+
+ FileManager &FileMgr = SourceMgr->getFileManager();
// Look for an umbrella header.
llvm::SmallString<128> UmbrellaName = StringRef(FrameworkDir->getName());
llvm::sys::path::append(UmbrellaName, "Headers");
llvm::sys::path::append(UmbrellaName, ModuleName + ".h");
- const FileEntry *UmbrellaHeader
- = SourceMgr->getFileManager().getFile(UmbrellaName);
+ const FileEntry *UmbrellaHeader = FileMgr.getFile(UmbrellaName);
// FIXME: If there's no umbrella header, we could probably scan the
// framework to load *everything*. But, it's not clear that this is a good
@@ -237,8 +240,14 @@
if (!UmbrellaHeader)
return 0;
- Module *Result = new Module(ModuleName, SourceLocation(),
- /*IsFramework=*/true);
+ Module *Result = new Module(ModuleName, SourceLocation(), Parent,
+ /*IsFramework=*/true, /*IsExplicit=*/false);
+
+ if (Parent)
+ Parent->SubModules[ModuleName] = Result;
+ else
+ Modules[ModuleName] = Result;
+
// umbrella "umbrella-header-name"
Result->UmbrellaHeader = UmbrellaHeader;
Headers[UmbrellaHeader] = Result;
@@ -251,9 +260,24 @@
Result->InferSubmodules = true;
Result->InferExportWildcard = true;
- // FIXME: Look for subframeworks.
+ // Look for subframeworks.
+ llvm::error_code EC;
+ llvm::SmallString<128> SubframeworksDirName = StringRef(FrameworkDir->getName());
+ llvm::sys::path::append(SubframeworksDirName, "Frameworks");
+ for (llvm::sys::fs::directory_iterator Dir(SubframeworksDirName.str(), EC),
+ DirEnd;
+ Dir != DirEnd && !EC; Dir.increment(EC)) {
+ if (!StringRef(Dir->path()).endswith(".framework"))
+ continue;
+
+ if (const DirectoryEntry *SubframeworkDir
+ = FileMgr.getDirectory(Dir->path())) {
+ // FIXME: Do we want to warn about subframeworks without umbrella headers?
+ inferFrameworkModule(llvm::sys::path::stem(Dir->path()), SubframeworkDir,
+ Result);
+ }
+ }
- Modules[ModuleName] = Result;
return Result;
}