#!/usr/bin/python3
"""
Allow buildbot to run jobs dynamically defined a a project repo
Requires docker
"""

# TODO: change "ghcontext" in master.cfg to interpolate the job name
# TODO: write upload script
# TODO: write hook (perhaps the default hook is ok), authentication for hook
# TODO: write hook job, maybe also a dockerfile?
# TODO: what to do about non-docker jobs

from os import listdir, walk
from os.path import exists, isfile, join
import requests
import re
from buildbot.plugins import steps, util

from .lib.renderers import *

addStep(steps.ShellCommand(
    name="download worker",
    command=["sudo", "wget", "-O", "lib/worker_multibuild.py", "worker_multibuild.py"]
))


def find_artifacts(directory="out"):
    """
    find artifacts (any file) in a directory
    """
    for _ in listdir(directory):
        if not isfile(join(directory, _)):
            continue
        return join(directory, _)


@util.renderer
def get_dockerfile_contents(props):
    return _get_dockerfile_contents(props, props.getProperty('jobname', default=None), util.Interpolate("%(prop:os_codename)s"))


def add_parent_step(build_factory):
    """
    Add a step to the parent build factory that will trigger the child hooks
    """
    build_factory.addStep(
        steps.SetPropertyFromCommand(
            name="Get OS codename",
            command="grep ^VERSION_CODENAME= /etc/os-release | cut -d= -f2",
            property="os_codename",
        )
    )
    
    build_factory.addStep(
        steps.ShellCommand(
            name="Execute multibuild script",
            command=[
                "python",
                "multibuild.py",
                util.Interpolate("%(prop:jobname)s"),
                util.Property('repository'),
                util.Property('branch'),
                "https://buildbot.bitmessage.org",
                util.Interpolate("%(prop:os_codename)s"),
            ],
        )
    )


def add_child_build_sh_step(build_factory, directory=".buildbot"):
    """
    Add a step to the build factory
    """
    build_factory.addStep(
        steps.ShellCommand(
            name=util.Interpolate("build_%(prop:jobname)s"),
            command=util.Interpolate("%(kw:directory)s/%(prop:jobname)s/test.sh", directory=directory),
            doStepIf=is_build_script_available,
            hideStepIf=isnt_build_script_available,
        )
    )


def add_child_test_sh_step(build_factory, directory=".buildbot"):
    """
    Add a step to the build factory
    """
    build_factory.addStep(
        steps.ShellCommand(
            name= util.Interpolate("test_%(prop:jobname)s"),
            command=util.Interpolate("%(kw:directory)s/%(prop:jobname)s/test.sh", directory=directory),
            doStepIf=is_test_script_available,
            hideStepIf=isnt_test_script_available,
        )
    )


if __name__ == "__main__":
    # expect jobname, repository, branch, buildbotUrl, os_codename from command line args
    import sys

    if len(sys.argv) == 6:
        jobname = sys.argv[1]
        repository = sys.argv[2]
        branch = sys.argv[3]
        buildbotUrl = sys.argv[4]
        os_codename = sys.argv[5]

        trigger_child_hooks(buildbotUrl, os_codename, repository, branch, jobname)
    else:
        print(
            "Usage: python3 multibuild.py <jobname> <repository> <branch> <buildbotUrl> <os_codename>"
        )