QuasarAppCI/BuildBotLib/asssetsinstaller.py

# This Python file uses the following encoding: utf-8

import BuildBotLib.basemodule as base
from buildbot.plugins import util, steps
import os
import shutil
import subprocess

LAST_FORMAT = [""]


@util.renderer
def NDKDownloadCMD(props):
    link = props.getProperty("link")
    module = props.getProperty("module")

    if os.path.isfile(module):
        shutil.rmtree(module)

    res = []
    format = link[link.rfind('.'):].lower()
    LAST_FORMAT[0] = format

    if module == "AndroidNDK":
        if os.path.isfile(module):
            os.remove("temp" + format)
        res = ["curl", link, "--output", "temp" + format]

    return res


@util.renderer
def ExtractCMD(props):

    format = LAST_FORMAT[0]
    module = props.getProperty("module")

    res = ["echo", "format '" + format + "' not supported"]

    if format == ".zip":
        res = ["unzip", "temp" + format, "-d", module]

    all_subdirs = base.allSubdirsOf("./" + module)
    latest_subdir = max(all_subdirs, key=os.path.getmtime)
    subprocess.getoutput(["ln -sf " + latest_subdir + " ./current"])

    return res


def getFactory():
    factory = base.getFactory()

    factory.addStep(
        steps.ShellCommand(
            command=NDKDownloadCMD,
            name='download new item',
            description='download new item',
        )
    )

    factory.addStep(
        steps.ShellCommand(
            command=ExtractCMD,
            name='extract new item',
            description='extract new item',
        )
    )

    return factory


def getRepo():
    return ""


def getPropertyes():
    return [
        util.ChoiceStringParameter(
            name='module',
            choices=["AndroidNDK", "AndroidSDK"],
            default="AndroidNDK"
        ),

        util.StringParameter(
            name='link',
            label="url to download item",
            default=""
        ),
    ]