#!/bin/bash

usage()
{
cat << EOF
usage: $0 -d <distribution> -s <signame(s)> -t <tag(s)>

This script generate new build target in koji for SIGS.

OPTIONS:
   -d           Distribution             : 5 6 7
   -s		SIG name                 : cloud
   -p		SIG project name         : cloud6-<openstack>, sclo-<mariadb100>, etc...
   -r		SIG project release name : cloud6-openstack-<juno>
   -t		DISTTAGS                 : "el7 el7.centos el7_0"
   -c		COLLECTION               : mariadb100 (a single collection can be used at this time)
EOF
}

optiond=false
options=false
optiont=false
optionc=false
optionp=false
optionr=false

while getopts ":hd:s:t:c:p:r:" OPTION
do
        case $OPTION in
        h)
             usage
             exit 0
             ;;
	d)
	     optiond=true
	     DISTS="${OPTARG}"
	     ;;
	s)
	     options=true
	     SIGS="${OPTARG}"
	     ;;
	r)
	     optionr=true
	     RELEASES="${OPTARG}"
	     ;;
	p)
	     optionp=true
	     PROJECTS="${OPTARG}"
	     ;;
	t)
	     optiont=true
	     TAGS="${OPTARG}"
	     ;;
	c)
	     optionc=true
	     COLLECTIONS="${OPTARG}"
	     ;;
        ?)
             exit 0
             ;;
        :)
             echo "Option -$OPTARG requires an argument."
             exit 1
             ;;
        esac
done

shift $(($OPTIND - 1))

if ! ( $optiond && $options && $optiont  )
then
	usage
	exit 0	
fi

# Check if user is allowed to send command to koji and has 'admin' permission
KOJI=`which koji`
PERM=`$KOJI list-permissions --mine`
if [ "$PERM" != "admin"  ]
then
	echo "[ERROR] Koji misconfigure/missing admin privilege for creating SIG tags"
	exit 1
fi

for DIST in $DISTS
do

	case $DIST in
		5) echo "* Checking distribution el$DIST configuration..."; ARCHES="i386 x86_64"; DEFAULT_DISTTAG="el5"
		;;
		6) echo "* Checking distribution el$DIST configuration..."; ARCHES="i686 x86_64"; DEFAULT_DISTTAG="el6"
		;;
		7) echo "* Checking distribution el$DIST configuration..."; ARCHES="x86_64"; DEFAULT_DISTTAG="el7.centos"
		;;
		*) echo "It seems your distribution el${DIST} is unsupported" && continue
		;;
	esac

	$KOJI list-tags | grep buildsys${DIST} &> /dev/null
	[ $? -gt 0 ] && echo " -> [ERROR] Something is wrong buildsys${DIST} tag not found." && continue

	$KOJI list-external-repos | grep ^centos${DIST}-os &> /dev/null
	[ $? -gt 0 ] && echo " -> [ERROR] centos${DIST}-os external repo not configured in koji." && continue

	$KOJI list-external-repos | grep ^centos${DIST}-updates &> /dev/null
	[ $? -gt 0 ] && echo " -> [ERROR] centos${DIST}-updates external repo not configured in koji." && continue

	for SIG in $SIGS
	do
		echo " -> Checking $SIG config..."
		$KOJI add-user $SIG &> /dev/null
		[ $? -eq 0 ] && echo "Creating user : ${SIG}"
		SIG="${SIG}${DIST}"
		for PROJECT in $PROJECTS
		do
			P_SIG="${SIG}-${PROJECT}"
			# Check for -common
			#$KOJI list-tags | grep $P_SIG-common-candidate &> /dev/null
			#if [ $? -gt 0 ]
			#then
			#	echo "Creating tag  : ${P_SIG}-common"
			#	$KOJI add-tag $P_SIG-common-candidate
			#	 
			#fi	
			for RELEASE in $RELEASES
			do
				R_SIG="${P_SIG}-${RELEASE}"
				$KOJI list-tags | grep $R_SIG-candidate &> /dev/null
				[ $? -gt 0 ] && echo "Creating tag  : ${R_SIG}-candidate" && $KOJI add-tag $R_SIG-testing 
				$KOJI list-tags | grep $R_SIG-testing &> /dev/null
				[ $? -gt 0 ] && echo "Creating tag  : ${R_SIG}-testing" && $KOJI add-tag $R_SIG-testing 
				$KOJI list-tags | grep $R_SIG-release &> /dev/null
				[ $? -gt 0 ] && echo "Creating tag  : ${R_SIG}-release" && $KOJI add-tag $R_SIG-release

				for TAG in $TAGS
				do
					# Add collection support abuse TAG variable
					REALTAG=$TAG
					if ( $optionc )
					then
						TAG="$TAG-$COLLECTIONS"
					fi
					$KOJI list-tags | grep $R_SIG-$TAG-build &> /dev/null 
					[ $? -eq 0 ] && echo " -> $R_SIG-$TAG target already exists. skipping." && continue
					echo " -> creating $R_SIG-$TAG"
					$KOJI add-tag --arches "$ARCHES" $R_SIG-$TAG-build
					$KOJI add-target $R_SIG-$TAG $R_SIG-$TAG-build $R_SIG-candidate
					$KOJI add-external-repo --tag=$R_SIG-$TAG-build centos${DIST}-updates
					$KOJI add-external-repo --tag=$R_SIG-$TAG-build centos${DIST}-os
					$KOJI add-group $R_SIG-$TAG-build build
					$KOJI add-group $R_SIG-$TAG-build srpm-build
					# Add <collection>-build to the buildroot if collection enabled
					if ( $optionc )
                        		then
						$KOJI add-group-pkg $R_SIG-$TAG-build build bash bzip2 coreutils cpio diffutils redhat-release findutils gawk gcc gcc-c++ grep gzip info make patch redhat-rpm-config rpm-build sed shadow-utils tar unzip util-linux-ng which buildsys-macros-$REALTAG tar buildsys-tools $COLLECTIONS-build
				$KOJI add-group-pkg $R_SIG-$TAG-build srpm-build bash buildsys-macros curl cvs redhat-release gnupg make redhat-rpm-config rpm-build shadow-utils buildsys-macros-$REALTAG tar buildsys-tools $COLLECTIONS-build
					else
						$KOJI add-group-pkg $R_SIG-$TAG-build build bash bzip2 coreutils cpio diffutils redhat-release findutils gawk gcc gcc-c++ grep gzip info make patch redhat-rpm-config rpm-build sed shadow-utils tar unzip util-linux-ng which buildsys-macros-$REALTAG tar buildsys-tools
                                		$KOJI add-group-pkg $R_SIG-$TAG-build srpm-build bash buildsys-macros curl cvs redhat-release gnupg make redhat-rpm-config rpm-build shadow-utils buildsys-macros-$REALTAG tar buildsys-tools
					fi
					$KOJI add-tag-inheritance --priority 5 $R_SIG-$TAG-build buildsys${DIST}
					$KOJI add-tag-inheritance --priority 10 $R_SIG-$TAG-build $R_SIG-candidate
					# If -common exists for the project add it
					$KOJI list-tags | grep $P_SIG-common-candidate &> /dev/null
					if [ "x$RELEASE" != "xcommon" ]
					then
                                		[ $? -gt 0 ] && echo "Adding $P_SIG-common-candidate as inheritance" && $KOJI add-tag-inheritance --priority 15 $R_SIG-$TAG-build $P_SIG-common-candidate
					fi
					$KOJI list-tagged buildsys${DIST} | grep buildsys-macros-$REALTAG &> /dev/null
					if [ $? -gt 0 ] 
					then
						if [ "x$REALTAG" != "x$DEFAULT_DISTTAG" ]
						then
							echo " -> [WARN] buildsys-macros-$REALTAG rpm not found. Please build it within koji."
							[ -f $PWD/etc/buildsys.spec.template ] && echo " -> Generating buildsys-macros-$TAG.spec"
						fi
					fi
				done
			done
		done
	done
done