#!/bin/bash

# github servers where WFR repos are found

server1=git-amr-1.devtools.intel.com:29418

server2=git-amr-2.devtools.intel.com:29418

Usage()

{

	echo "Usage: gitpull.sh [-PSdcv] [-b branch] [-t topdir] [-g gitdir] [repo ...]" >&2

	echo "      -P - skip pulling/cloning from repro" >&2

	echo "      -S - skip building and expanding the srpm" >&2

	echo "      -d - diff against CVS" >&2

	echo "      -c - copy to CVS" >&2

	echo "      -v - verbose output" >&2

	echo "      -b branch - git branch to checkout" >&2

	echo "                  (by default will read branch name from CVS commitids)" >&2

	echo "      -t topdir - top directory in CVS tree (default is $PWD)" >&2

	echo "      -g gitdir - parent directory for git repo(s) (defailt is GIT_REPOS)" >&2

	echo "      repo - list of wfr repos to operate on, default is all repos" >&2

	exit 2

}

# a few aren't working yet

repos="wfr-driver wfr-firmware wfr-psm wfr-diagtools-sw"

repos="$repos wfr-lite wfr-ibacm wfr-libverbs"

repos="$repos wfr-libibumad"

repos="$repos wfr-mvapich2"

repos="$repos wfr-openmpi"

repos="$repos wfr-gasnet"

repos="$repos wfr-openshmem"

repos="$repos wfr-openshmem-tests"

# skip this, linux kernel 3.12.18 for devel

#repos="$repos wfr-linux-devel"

in_branch=

pull=y

makesrpm=y

diff=n

copy=n

topdir=$PWD

gitdir=GIT_REPOS

verbose=n

while getopts PSdcvb:t:g: param

do

	case $param in

	P)	pull=n;;

	S)	makesrpm=n;;

	d)	diff=y;;

	c)	copy=y;;

	v)	verbose=y;;

	b)	in_branch="$OPTARG";;

	t)	topdir="$OPTARG";;

	g)	gitdir="$OPTARG";;

	*)	Usage;;

	esac

done

shift $((OPTIND -1))

if [ $# -gt 0 ]

then

	repos="$*"

fi

if [ ! -d $topdir ]

then

	echo "gitpull: FAILED: topdir Directory Not Found: $topdir" >&2

	Usage

fi

if [ ! -d $gitdir ]

then

	echo "gitpull: FAILED: gitdir Directory Not Found: $gitdir" >&2

	Usage

fi

# convert topdir to an absolute path

if [ `echo $topdir | cut -c 1` != '/' ]

then

	topdir=$PWD/$topdir

fi

# convert gitdir to an absolute path

if [ `echo $gitdir | cut -c 1` != '/' ]

then

	gitdir=$PWD/$gitdir

fi

[ $verbose = y ] && echo "Using topdir= $topdir; gitdir= $gitdir"

# different repo servers

get_server()

{

	case $1 in

	wfr-lite|wfr-openshmem|wfr-oshmem|wfr-openshmem-tests)	echo $server1;;

	*) echo $server2;;

	esac

}

# some repos have names different from Ofed/ tree name, remap

get_ofed_dir()

{

	case $1 in

	wfr-ibacm)	echo "ibacm";;

	wfr-libibumad)	echo "libibumad";;

	wfr-openshmem-tests)	echo "wfr-openshmem-tests/openshmem-test-suite";;

	*)	echo "$1";;

	esac

}

get_git_branch()

{

	if [ ! -e $topdir/Ofed/$1/commitids ]

	then

		echo "FAILED: Unable to get git branch.  Not Found: $topdir/Ofed/$1/commitids" >&2

	else

		tail -1 $topdir/Ofed/$1/commitids|cut -f 1 -d ' '

	fi

}

for i in $repos

do

	branch=$in_branch

	ofed_dir=`get_ofed_dir $i`

	echo "Processing git repo $i against Ofed/$ofed_dir"

	if [ $pull = y ]

	then

		# lookup git branch in commitids file in CVS

		if [ x"$branch" = x ]

		then

			branch=`get_git_branch $ofed_dir`

		fi

		if [ x"$branch" = x ]

		then

			echo "FAILED: unable to determine git branch for $i" >&2

			continue

		fi

		[ $verbose = y ] && echo "Using git branch $branch"

		## get the latest copy of source from the repos

		if [ ! -d $gitdir/$i ]

		then

			server=`get_server $i`

			[ $verbose = y ] && echo "(cd $gitdir; git clone ssh://$USER@$server/$i)"

			(cd $gitdir; git clone ssh://$USER@$server/$i)

		else

			[ $verbose = y ] && echo "(cd $gitdir/$i; git pull)"

			(cd $gitdir/$i; git pull)

		fi

		##Next, checkout the desired branch in each directory

		[ $verbose = y ] && echo "(cd $gitdir/$i; git checkout $branch )"

		(cd $gitdir/$i; git checkout $branch )

	fi

	if [ ! -d $gitdir/$i ]

	then

		echo "FAILED: Need to pull, Directory not found: $gitdir" >&2

		continue

	fi

	##show the branch we are on

	(cd $gitdir/$i; git branch )

	if [ $makesrpm = y ]

	then

		if [ ! -e $gitdir/$i/makesrpm.sh ]

		then

			echo "FAILED: Unable to makesrpm, File not found: $gitdir/$i/makesrpm.sh" >&2

			continue

		fi

			##make the srpm

		[ $verbose = y ] && echo "rm -rf $gitdir/$i/SOURCE_TREE"

		rm -rf $gitdir/$i/SOURCE_TREE

		[ $verbose = y ] && echo "(cd $gitdir/$i/; bash -x ./makesrpm.sh )"

		(cd $gitdir/$i/; bash -x ./makesrpm.sh )

			##expand it

		if [ `ls $gitdir/$i/SRPMS/*.src.rpm|wc -l` != 1 ]

		then

			echo "FAILED: srpm missing or too many, File not found: $gitdir/$i/SRPMS/*.src.rpm" >&2

			continue

		fi

		[ $verbose = y ] && echo "Expanding source into $gitdir/$i/SOURCE_TREE"

		(cd $gitdir/$i/; mkdir SOURCE_TREE; $topdir/MakeTools/expand_source.sh -d SOURCE_TREE SRPMS/*.src.rpm)

	fi

	if [ $diff = y ]

	then

		if [ ! -e $gitdir/$i/SOURCE_TREE/package_list ]

		then

			echo "FAILED: Unable to diff, File not found: $gitdir/$i/SOURCE_TREE/package_list" >&2

			continue

		fi

			## use package_list because srpm name is not necessarily $i

		package=`head -1 $gitdir/$i/SOURCE_TREE/package_list`

		[ $verbose = y ] && echo "Diff $gitdir/$i/SOURCE_TREE/$package/ $topdir/Ofed/$ofed_dir/ ..."

		diff -b -u --no-dereference --exclude CVS --exclude .git --exclude package_list -r $gitdir/$i/SOURCE_TREE/$package/ $topdir/Ofed/$ofed_dir/

	fi

	if [ $copy = y ]

	then

		if [ ! -e $gitdir/$i/SOURCE_TREE/package_list ]

		then

			echo "FAILED: Unable to copy, File not found: $gitdir/$i/SOURCE_TREE/package_list" >&2

			continue

		fi

		[ $verbose = y ] && echo "Replacing source in $topdir/Ofed/$i ..."

		##copy expanded srpm source from SOURCE_TREE to CVS Ofed/ tree

			##remove existing files in Ofed/ checkout

		find $topdir/Ofed/$ofed_dir/src/ ! -type d| grep -v CVS|xargs rm -f

			##replace with new files from srpm

			## use package_list because srpm name is not necessarily $i

		package=`head -1 $gitdir/$i/SOURCE_TREE/package_list`

		# use find to avoid symlinks, make_symlinks will handle them when build

		#cp -r $gitdir/$i/SOURCE_TREE/$package/* $topdir/Ofed/$ofed_dir/

		(cd $gitdir/$i/SOURCE_TREE/$package; find . -type d -o -type f|cpio -pdvum $topdir/Ofed/$ofed_dir/)

	fi

done 2>&1|tee -a gitpull.log