1 // Copyright (c) 2013 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "gn/ninja_toolchain_writer.h"
6
7 #include <fstream>
8
9 #include "base/files/file_util.h"
10 #include "base/strings/stringize_macros.h"
11 #include "gn/build_settings.h"
12 #include "gn/builtin_tool.h"
13 #include "gn/c_tool.h"
14 #include "gn/filesystem_utils.h"
15 #include "gn/general_tool.h"
16 #include "gn/ninja_utils.h"
17 #include "gn/pool.h"
18 #include "gn/settings.h"
19 #include "gn/substitution_writer.h"
20 #include "gn/target.h"
21 #include "gn/toolchain.h"
22 #include "gn/trace.h"
23
24 namespace {
25
26 const char kIndent[] = " ";
27
28 } // namespace
29
NinjaToolchainWriter(const Settings * settings,const Toolchain * toolchain,std::ostream & out)30 NinjaToolchainWriter::NinjaToolchainWriter(const Settings* settings,
31 const Toolchain* toolchain,
32 std::ostream& out)
33 : settings_(settings),
34 toolchain_(toolchain),
35 out_(out),
36 path_output_(settings_->build_settings()->build_dir(),
37 settings_->build_settings()->root_path_utf8(),
38 ESCAPE_NINJA) {}
39
40 NinjaToolchainWriter::~NinjaToolchainWriter() = default;
41
Run(const std::vector<NinjaWriter::TargetRulePair> & rules)42 void NinjaToolchainWriter::Run(
43 const std::vector<NinjaWriter::TargetRulePair>& rules) {
44 std::string rule_prefix = GetNinjaRulePrefixForToolchain(settings_);
45
46 for (const auto& tool : toolchain_->tools()) {
47 if (tool.second->name() == GeneralTool::kGeneralToolAction ||
48 tool.second->AsBuiltin()) {
49 continue;
50 }
51 WriteToolRule(tool.second.get(), rule_prefix);
52 }
53 out_ << std::endl;
54
55 for (const auto& pair : rules)
56 out_ << pair.second;
57 }
58
59 // static
RunAndWriteFile(const Settings * settings,const Toolchain * toolchain,const std::vector<NinjaWriter::TargetRulePair> & rules)60 bool NinjaToolchainWriter::RunAndWriteFile(
61 const Settings* settings,
62 const Toolchain* toolchain,
63 const std::vector<NinjaWriter::TargetRulePair>& rules) {
64 base::FilePath ninja_file(settings->build_settings()->GetFullPath(
65 GetNinjaFileForToolchain(settings)));
66 ScopedTrace trace(TraceItem::TRACE_FILE_WRITE_NINJA,
67 FilePathToUTF8(ninja_file));
68
69 base::CreateDirectory(ninja_file.DirName());
70
71 std::ofstream file;
72 file.open(FilePathToUTF8(ninja_file).c_str(),
73 std::ios_base::out | std::ios_base::binary);
74 if (file.fail())
75 return false;
76
77 NinjaToolchainWriter gen(settings, toolchain, file);
78 gen.Run(rules);
79 return true;
80 }
81
WriteToolRule(Tool * tool,const std::string & rule_prefix)82 void NinjaToolchainWriter::WriteToolRule(Tool* tool,
83 const std::string& rule_prefix) {
84 out_ << "rule " << rule_prefix << tool->name() << std::endl;
85
86 // Rules explicitly include shell commands, so don't try to escape.
87 EscapeOptions options;
88 options.mode = ESCAPE_NINJA_PREFORMATTED_COMMAND;
89
90 WriteCommandRulePattern("command", tool->command_launcher(), tool->command(),
91 options);
92
93 WriteRulePattern("description", tool->description(), options);
94 WriteRulePattern("rspfile", tool->rspfile(), options);
95 WriteRulePattern("rspfile_content", tool->rspfile_content(), options);
96
97 if (CTool* c_tool = tool->AsC()) {
98 if (c_tool->depsformat() == CTool::DEPS_GCC) {
99 // GCC-style deps require a depfile.
100 if (!c_tool->depfile().empty()) {
101 WriteRulePattern("depfile", tool->depfile(), options);
102 out_ << kIndent << "deps = gcc" << std::endl;
103 }
104 } else if (c_tool->depsformat() == CTool::DEPS_MSVC) {
105 // MSVC deps don't have a depfile.
106 out_ << kIndent << "deps = msvc" << std::endl;
107 }
108 } else if (!tool->depfile().empty()) {
109 WriteRulePattern("depfile", tool->depfile(), options);
110 out_ << kIndent << "deps = gcc" << std::endl;
111 }
112
113 // Use pool is specified.
114 if (tool->pool().ptr) {
115 std::string pool_name =
116 tool->pool().ptr->GetNinjaName(settings_->default_toolchain_label());
117 out_ << kIndent << "pool = " << pool_name << std::endl;
118 }
119
120 if (tool->restat())
121 out_ << kIndent << "restat = 1" << std::endl;
122 }
123
WriteRulePattern(const char * name,const SubstitutionPattern & pattern,const EscapeOptions & options)124 void NinjaToolchainWriter::WriteRulePattern(const char* name,
125 const SubstitutionPattern& pattern,
126 const EscapeOptions& options) {
127 if (pattern.empty())
128 return;
129 out_ << kIndent << name << " = ";
130 SubstitutionWriter::WriteWithNinjaVariables(pattern, options, out_);
131 out_ << std::endl;
132 }
133
WriteCommandRulePattern(const char * name,const std::string & launcher,const SubstitutionPattern & command,const EscapeOptions & options)134 void NinjaToolchainWriter::WriteCommandRulePattern(
135 const char* name,
136 const std::string& launcher,
137 const SubstitutionPattern& command,
138 const EscapeOptions& options) {
139 CHECK(!command.empty()) << "Command should not be empty";
140 out_ << kIndent << name << " = ";
141 if (!launcher.empty())
142 out_ << launcher << " ";
143 SubstitutionWriter::WriteWithNinjaVariables(command, options, out_);
144 out_ << std::endl;
145 }
146