[Frontend] -Rmodule-loading shows both source path and cached path

Author: xymus

include/swift/AST/DiagnosticsSema.def

@@ -989,9 +989,10 @@ REMARK(cross_import_added,none,
        (Identifier, Identifier, Identifier))
 
 REMARK(module_loaded,none,
-       "loaded module at %0"
-       "%select{|; overlay for a clang dependency}1",
-       (StringRef, unsigned))
+       "loaded module %0"
+       "%select{| (overlay for a clang dependency)}1"
+       "; source: '%2', loaded: '%3'",
+       (Identifier, unsigned, StringRef, StringRef))
 
 REMARK(explicit_interface_build_skipped,none,
        "Skipped rebuilding module at %0 - up-to-date",

include/swift/AST/FileUnit.h

@@ -404,6 +404,11 @@ class LoadedFile : public FileUnit {
   /// This is usually a filesystem path.
   virtual StringRef getFilename() const;
 
+  /// Get the path to the file loaded by the compiler. Usually the binary
+  /// swiftmodule file or a pcm in the cache. Returns an empty string if not
+  /// applicable.
+  virtual StringRef getLoadedFilename() const { return StringRef(); }
+
   virtual StringRef getFilenameForPrivateDecl(const ValueDecl *decl) const {
     return StringRef();
   }

include/swift/AST/Module.h

@@ -840,6 +840,17 @@ class ModuleDecl
   /// string if this is not applicable.
   StringRef getModuleFilename() const;
 
+  /// Get the path to the file defining this module, what we consider the
+  /// source of truth about the module. Usually a swiftinterface file for a
+  /// resilient module, a swiftmodule for a non-resilient module, or the
+  /// modulemap for a clang module. Returns an empty string if not applicable.
+  StringRef getModuleSourceFilename() const;
+
+  /// Get the path to the file loaded by the compiler. Usually the binary
+  /// swiftmodule file or a pcm in the cache. Returns an empty string if not
+  /// applicable.
+  StringRef getModuleLoadedFilename() const;
+
   /// \returns true if this module is the "swift" standard library module.
   bool isStdlibModule() const;
 

include/swift/ClangImporter/ClangModule.h

@@ -110,6 +110,8 @@ class ClangModuleUnit final : public LoadedFile {
 
   virtual StringRef getFilename() const override;
 
+  virtual StringRef getLoadedFilename() const override;
+
   virtual const clang::Module *getUnderlyingClangModule() const override {
     return getClangModule();
   }

include/swift/Serialization/SerializedModuleLoader.h

@@ -451,6 +451,8 @@ class SerializedASTFile final : public LoadedFile {
 
   virtual StringRef getFilename() const override;
 
+  virtual StringRef getLoadedFilename() const override;
+
   virtual StringRef getModuleDefiningPath() const override;
 
   ValueDecl *getMainDecl() const override;

lib/AST/ASTContext.cpp

@@ -2471,8 +2471,10 @@ ASTContext::getModule(ImportPath::Module ModulePath, bool AllowMemoryCached) {
       if (LangOpts.EnableModuleLoadingRemarks) {
         Diags.diagnose(ModulePath.getSourceRange().Start,
                        diag::module_loaded,
-                       M->getModuleFilename(),
-                       /*overlay=*/false);
+                       M->getName(),
+                       /*overlay=*/false,
+                       M->getModuleSourceFilename(),
+                       M->getModuleLoadedFilename());
       }
       return M;
     }
@@ -2504,8 +2506,10 @@ ModuleDecl *ASTContext::getOverlayModule(const FileUnit *FU) {
       if (LangOpts.EnableModuleLoadingRemarks) {
         Diags.diagnose(SourceLoc(),
                        diag::module_loaded,
-                       M->getModuleFilename(),
-                       /*overlay=*/true);
+                       M->getName(),
+                       /*overlay=*/true,
+                       M->getModuleSourceFilename(),
+                       M->getModuleLoadedFilename());
       }
       return M;
     }

lib/AST/Module.cpp

@@ -1973,6 +1973,25 @@ StringRef ModuleDecl::getModuleFilename() const {
   return Result;
 }
 
+StringRef ModuleDecl::getModuleSourceFilename() const {
+  for (auto F : getFiles()) {
+    if (auto *SFU = dyn_cast<SynthesizedFileUnit>(F))
+      continue;
+    return F->getModuleDefiningPath();
+  }
+
+  return StringRef();
+}
+
+StringRef ModuleDecl::getModuleLoadedFilename() const {
+  for (auto F : getFiles()) {
+    if (auto LF = dyn_cast<LoadedFile>(F)) {
+      return LF->getLoadedFilename();
+    }
+  }
+  return StringRef();
+}
+
 bool ModuleDecl::isStdlibModule() const {
   return !getParent() && getName() == getASTContext().StdlibModuleName;
 }

lib/ClangImporter/ClangImporter.cpp

@@ -3625,6 +3625,12 @@ StringRef ClangModuleUnit::getFilename() const {
   return StringRef();
 }
 
+StringRef ClangModuleUnit::getLoadedFilename() const {
+  if (const clang::FileEntry *F = clangModule->getASTFile())
+    return F->getName();
+  return StringRef();
+}
+
 clang::TargetInfo &ClangImporter::getTargetInfo() const {
   return Impl.Instance->getTarget();
 }

lib/Serialization/ModuleFile.h

@@ -718,6 +718,10 @@ class ModuleFile
     return Core->ModuleInputBuffer->getBufferIdentifier();
   }
 
+  StringRef getModuleLoadedFilename() const {
+    return Core->ModuleInputBuffer->getBufferIdentifier();
+  }
+
   StringRef getTargetTriple() const {
     return Core->TargetTriple;
   }

lib/Serialization/SerializedModuleLoader.cpp

@@ -1570,6 +1570,10 @@ StringRef SerializedASTFile::getFilename() const {
   return File.getModuleFilename();
 }
 
+StringRef SerializedASTFile::getLoadedFilename() const {
+  return File.getModuleLoadedFilename();
+}
+
 StringRef SerializedASTFile::getTargetTriple() const {
   return File.getTargetTriple();
 }