summaryrefslogtreecommitdiff
path: root/xmake/modules/private/tools/vstool.lua
blob: c4b22883deaba4a14d430ea7f52bf38efe00de7a (plain)
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
--!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        vstool.lua
--

-- quietly run command with arguments list
function runv(program, argv, opt)

    -- init options
    opt = opt or {}

    -- if has VS_BINARY_OUTPUT dont enable unicode output
    local envs = opt.envs or {}
    if envs.VS_BINARY_OUTPUT then
        return os.runv(program, argv, opt)
    end

    -- make temporary output and error file
    local outpath = os.tmpfile()
    local errpath = os.tmpfile()
    local outfile = io.open(outpath, 'w')

    -- enable unicode output for vs toolchains, e.g. cl.exe, link.exe and etc.
    -- @see https://github.com/xmake-io/xmake/issues/528
    if is_host("windows") then
        opt.envs = table.join(envs, {VS_UNICODE_OUTPUT = outfile:rawfd()})
    end

    -- execute it
    local ok, syserrors = os.execv(program, argv, table.join(opt, {try = true, stdout = outfile, stderr = errpath}))

    -- close outfile first
    outfile:close()

    -- failed?
    if ok ~= 0 then

        -- read errors
        local encoding = is_host("windows") and "utf16le" or nil
        local outdata = os.isfile(outpath) and io.readfile(outpath, {encoding = encoding}) or nil
        local errdata = os.isfile(errpath) and io.readfile(errpath) or nil
        local errors = errdata or ""
        if #errors:trim() == 0 then
            errors = outdata or ""
        end

        -- make the default errors
        if not errors or #errors == 0 then

            -- get command
            local cmd = program
            if argv then
                cmd = cmd .. " " .. os.args(argv)
            end

            -- get errors
            if ok ~= nil then
                errors = string.format("vstool.runv(%s) failed(%d)", cmd, ok)
            else
                errors = string.format("vstool.runv(%s), %s", cmd, syserrors and syserrors or "unknown reason")
            end
        end

        -- remove the files
        os.tryrm(outpath)
        os.tryrm(errpath)

        -- raise errors
        os.raise({errors = errors, stderr = errdata, stdout = outdata})
    end

    -- remove the files
    os.tryrm(outpath)
    os.tryrm(errpath)
end

-- run command and return output and error data
function iorunv(program, argv, opt)

    -- init options
    opt = opt or {}

    -- if has VS_BINARY_OUTPUT dont enable unicode output
    local envs = opt.envs or {}
    if envs.VS_BINARY_OUTPUT then
        return os.runv(program, argv, opt)
    end

    -- make temporary output and error file
    local outpath = os.tmpfile()
    local errpath = os.tmpfile()
    local outfile = io.open(outpath, 'w')

    -- enable unicode output for vs toolchains, e.g. cl.exe, link.exe and etc.
    -- @see https://github.com/xmake-io/xmake/issues/528
    if is_host("windows") then
        opt.envs = table.join(envs, {VS_UNICODE_OUTPUT = outfile:rawfd()})
    end

    -- run command
    local ok, syserrors = os.execv(program, argv, table.join(opt, {try = true, stdout = outfile, stderr = errpath}))

    -- get output and error data
    outfile:close()
    local encoding = is_host("windows") and "utf16le" or nil
    local outdata = os.isfile(outpath) and io.readfile(outpath, {encoding = encoding}) or nil
    local errdata = os.isfile(errpath) and io.readfile(errpath) or nil

    -- remove the temporary output and error file
    os.tryrm(outpath)
    os.tryrm(errpath)

    -- failed?
    if ok ~= 0 then

        -- get errors
        local errors = errdata or ""
        if #errors:trim() == 0 then
            errors = outdata or ""
        end

        -- make the default errors
        if not errors or #errors == 0 then

            -- get command
            local cmd = program
            if argv then
                cmd = cmd .. " " .. os.args(argv)
            end

            -- get errors
            if ok ~= nil then
                errors = string.format("vstool.iorunv(%s) failed(%d)", cmd, ok)
            else
                errors = string.format("vstool.iorunv(%s), %s", cmd, syserrors and syserrors or "unknown reason")
            end
        end

        -- raise errors
        os.raise({errors = errors, stderr = errdata, stdout = outdata})
    end
    return outdata, errdata
end