# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.

# SPDX-License-Identifier: Apache-2.0.

import argparse

import boto3

# Parse required options

parser = argparse.ArgumentParser(description='Creates all required AWS CodeBuild projects for a repo')

parser.add_argument('project', type=str, help='The name of the repo to create the projects for')

parser.add_argument('--github-account', type=str, dest='github_account', default='awslabs', help='The GitHub account that owns the repo')

parser.add_argument('--profile', type=str, default='default', help='The profile in ~/.aws/credentials to use when creating the jobs')

args = parser.parse_args()

# The template for the arguments to be passed to create_project

CREATE_PARAM_TEMPLATE = {

}

# The common enviroment objects to feed to CodeBuild

ENVIRONMENTS = {

}

# The list of all of our build configs paired with their environments

BUILD_CONFIGS = [

]

# Fully populate the BUILDS list with all final build objects

BUILDS = {}

for config in BUILD_CONFIGS:

build_name = config['build']

build = dict(CREATE_PARAM_TEMPLATE)

env = dict(ENVIRONMENTS[config['env']])

if 'privileged' in config:

env['privilegedMode'] = config['privileged']

build['environment'] = env

sub_params = {

'project': args.project,

'build': build_name,

'account': args.github_account,

}

# Replace all templates with the values above

def do_replace(obj):

return obj

do_replace(build)

BUILDS['{}-{}'.format(args.project, build_name)] = build

# Connect to codebuild

session = boto3.Session(profile_name=args.profile, region_name='us-east-1')

codebuild = session.client('codebuild')

# Find out which projects already exist and should be updated, and which must be created

all_project_names = list(BUILDS.keys())

existing_projects = codebuild.batch_get_projects(names=all_project_names)

new_projects = existing_projects['projectsNotFound']

existing_projects = [project['name'] for project in existing_projects['projects']]

# Actually create the projects

for build_name, build in BUILDS.items():

if build_name in new_projects:

print('{}: Creating'.format(build_name))

codebuild.create_project(**build)

codebuild.create_webhook(projectName=build_name)

elif build_name in existing_projects:

print('{}: Updating'.format(build_name))

codebuild.update_project(**build)

else:

assert False