1 //===--- ModuleDependencyCollector.cpp - Collect module dependencies ------===//
3 // The LLVM Compiler Infrastructure
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 // Collect the dependencies of a set of modules.
12 //===----------------------------------------------------------------------===//
14 #include "clang/Frontend/Utils.h"
15 #include "clang/Serialization/ASTReader.h"
16 #include "llvm/ADT/StringSet.h"
17 #include "llvm/ADT/iterator_range.h"
18 #include "llvm/Support/FileSystem.h"
19 #include "llvm/Support/Path.h"
20 #include "llvm/Support/raw_ostream.h"
22 using namespace clang;
25 /// Private implementation for ModuleDependencyCollector
26 class ModuleDependencyListener : public ASTReaderListener {
27 ModuleDependencyCollector &Collector;
29 std::error_code copyToRoot(StringRef Src);
31 ModuleDependencyListener(ModuleDependencyCollector &Collector)
32 : Collector(Collector) {}
33 bool needsInputFileVisitation() override { return true; }
34 bool needsSystemInputFileVisitation() override { return true; }
35 bool visitInputFile(StringRef Filename, bool IsSystem, bool IsOverridden,
36 bool IsExplicitModule) override;
40 void ModuleDependencyCollector::attachToASTReader(ASTReader &R) {
41 R.addListener(llvm::make_unique<ModuleDependencyListener>(*this));
44 void ModuleDependencyCollector::writeFileMap() {
48 SmallString<256> Dest = getDest();
49 llvm::sys::path::append(Dest, "vfs.yaml");
52 llvm::raw_fd_ostream OS(Dest, EC, llvm::sys::fs::F_Text);
60 std::error_code ModuleDependencyListener::copyToRoot(StringRef Src) {
61 using namespace llvm::sys;
63 // We need an absolute path to append to the root.
64 SmallString<256> AbsoluteSrc = Src;
65 fs::make_absolute(AbsoluteSrc);
66 // Canonicalize to a native path to avoid mixed separator styles.
67 path::native(AbsoluteSrc);
68 // TODO: We probably need to handle .. as well as . in order to have valid
69 // input to the YAMLVFSWriter.
70 path::remove_dots(AbsoluteSrc);
72 // Build the destination path.
73 SmallString<256> Dest = Collector.getDest();
74 path::append(Dest, path::relative_path(AbsoluteSrc));
76 // Copy the file into place.
77 if (std::error_code EC = fs::create_directories(path::parent_path(Dest),
78 /*IgnoreExisting=*/true))
80 if (std::error_code EC = fs::copy_file(AbsoluteSrc, Dest))
82 // Use the absolute path under the root for the file mapping.
83 Collector.addFileMapping(AbsoluteSrc, Dest);
84 return std::error_code();
87 bool ModuleDependencyListener::visitInputFile(StringRef Filename, bool IsSystem,
89 bool IsExplicitModule) {
90 if (Collector.insertSeen(Filename))
91 if (copyToRoot(Filename))
92 Collector.setHasErrors();