CICDScripts/compress.py at main · PipeRift/CICDScripts · GitHub

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
import sys
import os
import shutil

from helpers.util import *
from helpers import env, util

install('click')
import click  # NOQA


@click.group()
def compress():
    pass


class InvalidCompressError(Exception):
    pass


@click.command()
@click.option('-n', '--name', envvar="CI_NAME", required=True, help="Name of the plugin (without .uplugin)")
@click.option('-p', '--path', envvar="CI_PATH", type=click.Path(), help="Path that contains all plugin files")
@click.option('-b', '--build-path', envvar="CI_BUILD_PATH", type=click.Path(exists=True), help="Destination of the packaged plugin (default: {path}/Build)")
@click.option('-z', '--zip-path', envvar="CI_ZIP_PATH", type=click.Path(exists=True), help="Folder that will contain the vault (default: {path}/Package)")
def plugin(name, path, build_path, zip_path):
    """Compresses a plugin for release. """
    plugin = env.Plugin(name, path, build_path)
    if not os.path.isdir(plugin.build_path):
        raise InvalidCompressError(
            f"Build path doesn't exist ('{plugin.build_path}')")

    if not zip_path:
        zip_path = os.path.join(plugin.path, "Package")

    if not os.path.isabs(zip_path):
        zip_path = os.path.abspath(zip_path)

    temp_path = os.path.join(zip_path, plugin.name)
    temp_path_rel = os.path.join('.', plugin.name)

    install('py7zr')
    import py7zr

    print("Clean destination folder")
    if not os.path.exists(zip_path):
        os.makedirs(zip_path)

    print("Remove old packaged Plugin")
    file = os.path.join(zip_path, f'{plugin.name}_v{plugin.get_version()}_{plugin.get_short_engine_version()}.zip')
    file_bin = os.path.join(zip_path, f'{plugin.name}_v{plugin.get_version()}_{plugin.get_short_engine_version()}_Bin.zip')
    remove(file)
    remove(file_bin)

    print("Copy built plugin")
    remove(temp_path)
    shutil.copytree(plugin.build_path, temp_path,
                    ignore=shutil.ignore_patterns('Intermediate', 'Docs', 'Build', 'Vault'))

    print("Cleanup Online Documentation")
    docs_path = os.path.join(temp_path, 'Docs')
    if os.path.exists(docs_path):
        shutil.rmtree(docs_path)

    print("Copy Config")
    shutil.copytree(os.path.join(plugin.path, 'Config'),
                    os.path.join(temp_path, 'Config'))

    plugin_build = util.import_from_path(plugin.name, os.path.join(plugin.path, "build.py"))
    if plugin_build and plugin_build.compress_additional_plugins:
        print("Compress Additional Plugins")
        additional_plugins = plugin_build.compress_additional_plugins()
        for addt_plugin in additional_plugins:
            addt_plugin = os.path.join(temp_path, addt_plugin)
            for f in ["Docs", "Binaries", "Intermediate"]: shutil.rmtree(os.path.join(addt_plugin, f), True) # Remove unwanted folders
            os.chdir(os.path.dirname(addt_plugin))
            with py7zr.SevenZipFile(f'{os.path.basename(addt_plugin)}.zip', 'w') as f:
                f.writeall(os.path.basename(addt_plugin))
            shutil.rmtree(addt_plugin, True)

    print("Compress Plugin with Binaries")
    os.chdir(zip_path)
    with py7zr.SevenZipFile(file_bin, 'w') as f:
        f.writeall(temp_path_rel)

    print("Cleanup Plugin Binaries")
    shutil.rmtree(os.path.join(temp_path, 'Binaries'))

    print("Compress Plugin without Binaries")
    os.chdir(zip_path)
    with py7zr.SevenZipFile(file, 'w') as f:
        f.writeall(temp_path_rel)

    print("Clean temporaries")
    remove(temp_path)

    return 0


compress.add_command(plugin)

if __name__ == '__main__':
    sys.exit(compress())