#!/usr/bin/python3
import os
import sys
import yaml
import tempfile
import argparse
import subprocess
from components import CommonUtils

# Capture our command line parameters
parser = argparse.ArgumentParser(description='Utility to run builds for multiple projects on CI')
parser.add_argument('-p','--projects', nargs='+', help='Projects to be built', required=True)
parser.add_argument('--branch', type=str, required=True)
parser.add_argument('--platform', type=str, required=True)
parser.add_argument('--skip-dependencies-fetch', default=False, action='store_true')
parser.add_argument('--publish-to-cache', default=False, action='store_true')
parser.add_argument('--missing-only', default=False, action='store_true')

arguments = parser.parse_args()

if len(arguments.projects) == 1 and ' ' in arguments.projects[0]:
    fixedProjects = arguments.projects[0].split();
    print("Fixing the projects list: {} -> {}", arguments.projects, fixedProjects)
    arguments.projects = fixedProjects

projects = dict.fromkeys(arguments.projects, arguments.branch)
out = [
    {
        'on': ['@all'],
        'require': projects
    }
]
print ("Projects to be generated: {}".format(projects))
print ("Seed file generated:")
print (yaml.dump(out, indent = 2))

seedFile = tempfile.NamedTemporaryFile(delete=False, mode='w')
yaml.dump(out, seedFile, indent = 2)
seedFile.close()

commandToRun = "{0} -u {1}/seed-package-registry.py --seed-file {2} --platform {3}".format(
            sys.executable,
            CommonUtils.scriptsBaseDirectory(),
            seedFile.name,
            arguments.platform
        )

if arguments.skip_dependencies_fetch:
    commandToRun += " --skip-dependencies-fetch"

if arguments.publish_to_cache:
     commandToRun += ' --publish-to-cache'

if arguments.missing_only:
     commandToRun += ' --missing-only'

print('## Run the build for the requested projects: {}'.format(commandToRun))

workingDirectory = os.getcwd()

try:
    # Then run it!
    subprocess.check_call( commandToRun, stdout=sys.stdout, stderr=sys.stderr, shell=True, cwd=workingDirectory )
except:
    os.remove(seedFile.name)
    raise

os.remove(seedFile.name)