summaryrefslogtreecommitdiff
path: root/xmake/modules/utils/archive/archive_xmz.lua
blob: 70d3152fe77aecbfdda2ed124430ea28f01f0d60 (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
--!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        archive_xmz.lua
--

-- imports
import("core.base.option")
import("core.base.bytes")
import("core.compress.lz4")

-- archive files
function _archive_files(archivefile, inputfiles, opt)
    local curdir = opt.curdir
    local outputfile = io.open(archivefile, "wb")
    for _, inputfile in ipairs(inputfiles) do
        local filepath = inputfile
        if curdir then
            filepath = path.relative(filepath, curdir)
        end
        outputfile:write(bytes(2):u16be_set(1, #filepath))
        outputfile:write(filepath)
        local data = io.readfile(inputfile, {encoding = "binary"})
        vprint("archiving %s, %d bytes", inputfile, data and #data or 0)
        outputfile:write(bytes(4):u32be_set(1, #data))
        outputfile:write(data)
    end
    outputfile:close()
end

-- compress file
function _compress_file(archivefile, outputfile, opt)
    lz4.compress_file(archivefile, outputfile)
end

-- archive file
--
-- @param archivefile   the archive file. e.g. *.tar.gz, *.zip, *.7z, *.tar.bz2, ..
-- @param inputfiles    the input file or directory or list
-- @param options       the options, e.g.. {curdir = "/tmp", recurse = true, compress = "fastest|faster|default|better|best", excludes = {"*/dir/*", "dir/*"}}
--
function main(archivefile, inputfiles, opt)
    opt = opt or {}
    local files = {}
    for _, inputfile in ipairs(inputfiles) do
        if os.isdir(inputfile) then
            table.join2(files, os.files(path.join(inputfile, opt.recurse and "**" or "*")))
        elseif os.isfile(inputfile) then
            table.insert(files, inputfile)
        end
    end
    inputfiles = files

    local archivefile_tmp = os.tmpfile({ramdisk = false})
    _archive_files(archivefile_tmp, inputfiles, opt)
    _compress_file(archivefile_tmp, archivefile, opt)
    os.tryrm(archivefile_tmp)
end