#!/usr/bin/env python3
#
# Generate an auto.conf and associated other config files for a given autobuilder configuration
#
# Called with $1 - The 'nightly' target the autobuilder is running
#             $2 - The autobuilder step number (a given target can run multiple steps with different configurations)
#             $3 - The target build directory to configure
#             $4 - The poky branch name the build is running on
#             $5 - The name of the repository the build is running on
#             $6 - The directory to publish sstate into
#             $7 - A build-appliance SRCREV to use
#

import json
import os
import sys
import subprocess
import errno
import copy
import utils

if len(sys.argv) != 8:
    print("Incorrect number of parameters, please call as %s <nightly-target> <stepnumber> <target-builddir> <branch-name> <repo-name>")
    sys.exit(1)

target = sys.argv[1]
stepnum = int(sys.argv[2]) + 1 # Our step numbering is 1 2 3 etc., not 0 of buildbot
builddir = sys.argv[3]
branchname = sys.argv[4]
reponame = sys.argv[5]
sstate_release = sys.argv[6]
buildappsrcrev = sys.argv[7]

scriptsdir = os.path.dirname(os.path.realpath(__file__))

with open(os.path.join(scriptsdir, '..', 'config.json')) as f:
    ourconfig = json.load(f)

variables = []

# Expand templates in the configuration
ourconfig = utils.expandtemplates(ourconfig)

autoconf = os.path.join(builddir, "conf", "auto.conf")
if os.path.exists(autoconf):
    os.remove(autoconf)

# Ensure autoconf's directory exists
try:
    os.makedirs(os.path.dirname(autoconf))
except OSError as e:
    if e.errno != errno.EEXIST:
        # Do not complain if the directory exists
        raise e

sdkextraconf = os.path.join(builddir, "conf", "sdk-extra.conf")
if os.path.exists(sdkextraconf):
    os.remove(sdkextraconf)

for v in ["MACHINE", "DISTRO", "SDKMACHINE"]:
    value = utils.getconfigvar(v, ourconfig, target, stepnum)
    if value and value != "None":
        variables.append(v + ' = "%s"' % value)

distro = utils.getconfigvar("DISTRO", ourconfig, target, stepnum)

for v in ["DLDIR", "PRSERV"]:
    value = utils.getconfigvar(v, ourconfig, target, stepnum)
    if value:
        variables.append(value)

# Use a separate SSTATE_DIR with the primary
# SSTATE_DIR configured as a mirror so that we
# have a directory of symlinks to sstate objects
# that can be published for the release
key = "SSTATEDIR"
if sstate_release != "None":
    key = "SSTATEDIR_RELEASE"
value = utils.getconfiglist(key, ourconfig, target, stepnum)
for v in value:
    v = v.replace("@RELEASENUM@", sstate_release)
    variables.append(v)

if buildappsrcrev != "None" and buildappsrcrev != "DEFAULT":
    if buildappsrcrev == "AUTOREV":
        buildappsrcrev = "${AUTOREV}"
    value = utils.getconfiglist("BUILDAPP_SRCREV", ourconfig, target, stepnum)
    for v in value:
        v = v.replace("@SRCREV@", buildappsrcrev)
        variables.append(v)

if utils.getconfigvar("BUILDINFO", ourconfig, target, stepnum):
    infovars = utils.getconfiglist("BUILDINFOVARS", ourconfig, target, stepnum)
    variables.extend(infovars)

extravars = utils.getconfiglist("extravars", ourconfig, target, stepnum)
if extravars:
    variables.extend(extravars)

if utils.contains(["BUILD_HISTORY_DIR", "build-history-targets", "BUILD_HISTORY_REPO"], ourconfig):
    if target in ourconfig["build-history-targets"]:
        base = None
        if (reponame + ":" + branchname) in ourconfig["BUILD_HISTORY_DIRECTPUSH"]:
            base = reponame + ":" + branchname
        if (reponame + ":" + branchname) in ourconfig["BUILD_HISTORY_FORKPUSH"]:
            base = ourconfig["BUILD_HISTORY_FORKPUSH"][reponame + ":" + branchname]
        if base:
            baserepo, basebranch = base.split(":")
            bh_path = os.path.join(ourconfig["BUILD_HISTORY_DIR"], distro, branchname, target)
            remoterepo = ourconfig["BUILD_HISTORY_REPO"]
            remotebranch = reponame + "/" + branchname + "/" + target
            baseremotebranch = baserepo + "/" + basebranch + "/" + target
            if stepnum == 1:
                utils.runcmd([os.path.join(scriptsdir, "buildhistory-init"), bh_path, remoterepo, remotebranch, baseremotebranch])
            variables.append('INHERIT += "buildhistory"')
            variables.append('BUILDHISTORY_DIR = "%s"' % bh_path)
            variables.append('BUILDHISTORY_PUSH_REPO = "%s %s:%s"' % (remoterepo, remotebranch, remotebranch))
            variables.append("BUILDHISTORY_COMMIT = '1'")
            variables.append('ERROR_QA_remove = "version-going-backwards"')

utils.printheader("Writing %s with contents:" % autoconf)
with open(autoconf, "w") as f:
    for v in variables:
        print("  " + v)
        f.write(v + "\n")

utils.printheader("Writing %s with contents:" % sdkextraconf)
with open(sdkextraconf, "w") as f:
    for v in utils.getconfiglist("SDKEXTRAS", ourconfig, target, stepnum):
        replace = ""
        if sstate_release != "None":
            replace = sstate_release + "/"
        v = v.replace("@RELEASENUM@", replace)
        print("  " + v)
        f.write(v + "\n")