1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
|
--!A cross-platform build utility based on Lua
--
-- Licensed under the Apache License, Version 2.0 (the "License");
-- you may not use this file except in compliance with the License.
-- You may obtain a copy of the License at
--
-- http://www.apache.org/licenses/LICENSE-2.0
--
-- Unless required by applicable law or agreed to in writing, software
-- distributed under the License is distributed on an "AS IS" BASIS,
-- WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-- See the License for the specific language governing permissions and
-- limitations under the License.
--
-- Copyright (C) 2015-present, Xmake Open Source Community.
--
-- @author ruki
-- @file main.lua
--
-- imports
import("core.base.option")
import("core.base.hashset")
import("core.project.config")
import("core.project.project")
import("lib.detect.find_tool")
import("async.runjobs")
import("utils.progress")
import("private.action.require.impl.packagenv")
import("private.action.require.impl.install_packages")
import("private.action.utils", {alias = "action_utils"})
-- match source files
function _match_sourcefiles(sourcefile, filepatterns)
for _, filepattern in ipairs(filepatterns) do
if sourcefile:match(filepattern.pattern) == sourcefile then
if filepattern.excludes then
if filepattern.rootdir and sourcefile:startswith(filepattern.rootdir) then
sourcefile = sourcefile:sub(#filepattern.rootdir + 2)
end
for _, exclude in ipairs(filepattern.excludes) do
if sourcefile:match(exclude) == sourcefile then
return false
end
end
end
return true
end
end
end
-- convert all sourcefiles to lua pattern
function _get_file_patterns(sourcefiles)
local patterns = {}
for _, sourcefile in ipairs(path.splitenv(sourcefiles)) do
-- get the excludes
local pattern = sourcefile:trim()
local excludes = pattern:match("|.*$")
if excludes then excludes = excludes:split("|", {plain = true}) end
-- translate excludes
if excludes then
local _excludes = {}
for _, exclude in ipairs(excludes) do
local exclude = path.translate(exclude)
exclude = path.pattern(exclude)
table.insert(_excludes, exclude)
end
excludes = _excludes
end
-- translate path and remove some repeat separators
pattern = path.translate((pattern:gsub("|.*$", "")))
-- remove "./" or '.\\' prefix
if pattern:sub(1, 2):find('%.[/\\]') then
pattern = pattern:sub(3)
end
-- get the root directory
local rootdir = pattern
local startpos = pattern:find("*", 1, true)
if startpos then
rootdir = rootdir:sub(1, startpos - 1)
end
rootdir = path.directory(rootdir)
-- convert to lua path pattern
pattern = path.pattern(pattern)
table.insert(patterns, {pattern = pattern, excludes = excludes, rootdir = rootdir})
end
return patterns
end
-- get all the targets that match the group or targetname
function _get_targets(targetname, group_pattern)
local targets = {}
if targetname then
table.insert(targets, project.target(targetname))
else
for _, target in pairs(project.targets()) do
local group = target:get("group")
if (target:is_default() and not group_pattern) or option.get("all") or (group_pattern and group and group:match(group_pattern)) then
table.insert(targets, target)
end
end
end
return targets
end
-- tell if the source batch is a c/c++/objc/objc++/cuda source batch
function _source_batch_should_format(sourcebatch)
local rulename = sourcebatch.rulename
local matched_rules = {"c.build", "c++.build", "c++.build.modules", "cuda.build", "objc.build", "objc++.build"}
return table.contains(matched_rules, rulename)
end
-- main
function main()
-- load configuration
config.load()
-- enter the environments of llvm
local oldenvs = packagenv.enter("llvm")
-- find clang-format
local packages = {}
local clang_format = find_tool("clang-format")
if not clang_format then
table.join2(packages, install_packages("llvm"))
end
-- enter the environments of installed packages
for _, instance in ipairs(packages) do
instance:envs_enter()
end
-- we need to force detect and flush detect cache after loading all environments
if not clang_format then
clang_format = find_tool("clang-format", {force = true})
end
assert(clang_format, "clang-format not found!")
-- create style file
local argv = {}
local projectdir = project.directory()
if option.get("create") then
table.insert(argv, "--style=" .. (option.get("style") or "Google"))
table.insert(argv, "--dump-config")
os.execv(clang_format.program, argv, {stdout = path.join(projectdir, ".clang-format"), curdir = projectdir})
return
end
-- set style file
if option.get("style") then
table.insert(argv, "--style=" .. option.get("style"))
end
if option.get("dry-run") then
-- do not make any changes, just show the files that would be formatted
table.insert(argv, "--dry-run")
else
-- inplace flag
table.insert(argv, "-i")
end
-- changes formatting warnings to errors
if option.get("error") then
table.insert(argv, "--Werror")
end
-- print verbose information
if option.get("verbose") then
table.insert(argv, "--verbose")
end
-- collect sourcefiles
local sourcefiles = {}
local targetname, group_pattern = action_utils.get_target_and_group()
local targets = _get_targets(targetname, group_pattern)
if option.get("files") then
local filepatterns = _get_file_patterns(option.get("files"))
for _, target in ipairs(targets) do
for _, source in ipairs(target:sourcefiles()) do
if _match_sourcefiles(source, filepatterns) then
table.insert(sourcefiles, path.absolute(source, projectdir))
end
end
for _, header in ipairs(target:headerfiles()) do
if _match_sourcefiles(header, filepatterns) then
table.insert(sourcefiles, path.absolute(header, projectdir))
end
end
end
else
for _, target in ipairs(targets) do
for _, sourcebatch in pairs(target:sourcebatches()) do
if _source_batch_should_format(sourcebatch) then
for _, source in ipairs(sourcebatch.sourcefiles) do
table.insert(sourcefiles, path.absolute(source, projectdir))
end
end
end
for _, header in ipairs(target:headerfiles()) do
table.insert(sourcefiles, path.absolute(header, projectdir))
end
end
end
-- format files in parallel
if #sourcefiles > 0 then
local jobs = tonumber(option.get("jobs"))
if not jobs or jobs <= 0 then
jobs = os.default_njob()
end
local format_time = os.mclock()
local runjobs_opt = {
total = #sourcefiles,
comax = jobs,
showtips = false,
progress_refresh = true
}
runjobs("clang-format", function (index, total, opt)
local sourcefile = sourcefiles[index]
local format_argv = table.join(argv, {sourcefile})
progress.show(opt.progress, "clang-format.formatting %s", sourcefile)
os.execv(clang_format.program, format_argv, {curdir = projectdir})
end, runjobs_opt)
format_time = os.mclock() - format_time
progress.show(100, "${color.success}clang-format formatted %d files, spent %.3fs", #sourcefiles, format_time / 1000)
end
os.setenvs(oldenvs)
end
|
