[home.git] / .bin / r

#!/bin/bash
g() { git "$@"; }
err() { printf '%b\n' "$*" 1>&2; exit 1; }

case $1 in
""|-*) ;;
*)
	cmd=$1
	shift
	acmd=$(git config --get "alias.${cmd}")
	;;
esac

case ${acmd:-${cmd}} in
rebase)
	if [[ $1 == "all" ]] ; then
		shift
		if [[ $# -eq 0 ]] ; then
			exec r forall -c 'r rb all .'
		fi

		branches=$(g b | awk '
			{
				if ($0 ~ "^[*] *[(]no branch[)]") {
					next
				} else if ($1 == "*") {
					b = $NF
				} else {
					list = list $NF " "
				}
			}
			END { print list b }')
		[[ -z ${branches} ]] && exit 0

		eval $(bash-colors --env)
		echo "${GOOD}### ${PWD}${NORMAL}"
		for b in ${branches} ; do
			echo " ${HILITE}### $b${NORMAL}"
			g co $b || exit 1
			if ! r rb "$@" ; then
				g rb-a
			fi
		done
		exit 0
	fi
	;;
sb-push)
	sync_branch="v"

	root=${PWD}
	while [[ ! -d ${root}/.repo && ${root} != "/" ]] ; do
		root=${root%/*}
	done
	cd "${root}" || exit 1

	if [[ ! -e .repo/sandbox-url ]] ; then
		err "Please configure remote url base in ${root}/.repo/sandbox-url"
	fi
	remote=$(<.repo/sandbox-url) || exit 1

	echo "pushing projects from ${root}"

	pids=()
	rlist=$(r list)
	tcnt=$(echo "${rlist}" | wc -l)
	cnt=1
	while read line ; do
		line=( ${line} )
		path=${line[0]}
		export GIT_DIR=${path}/.git
		proj=${line[2]}
		printf '### (%*i/%i %3i%%) %s\n' \
			${#tcnt} $((cnt++)) ${tcnt} $(( cnt * 100 / tcnt )) ${proj}
		src="${sync_branch}"
		g l -1 ${src} >& /dev/null || src=
		( g push --force ${remote}/${proj} ${src}:refs/sandbox/${USER}/${sync_branch} >/dev/null ) &

		# ssh servers do not like it when you hammer them :)
		#	Received disconnect from 74.125.248.80: 7: Too many concurrent connections
		#	fatal: The remote end hung up unexpectedly
		pids+=( $! )
		if [[ ${#pids[@]} -eq 20 ]] ; then
			wait ${pids[@]:0:3}
			pids=( ${pids[@]:3} )
		fi
	done <<<"$(r list)"
	wait

	exit 0
	;;
upload)
	args=()
	while [[ $# -gt 0 ]] ; do
		case $1 in
		--re)
			if [[ $2 == *"OWNERS" ]] ; then
				owners=$(awk -F'@' '{list = list "," $1} END {print substr(list, 2)}' "$2")
				if [[ -z ${owners} ]] ; then
					err "cannot find OWNERS list"
				else
					echo "Auto setting reviewers to: ${owners}"
				fi
				args+=( --re "${owners}" )
				shift 2
				continue
			fi
			;;
		esac
		args+=( "$1" )
		shift
	done
	set -- "${args[@]}"
	;;
esac

exec repo ${acmd:-${cmd}} "$@"
Commit	Line	Data
35c220c5 MF	1	#!/bin/bash
35c220c5 MF	2	g() { git "$@"; }
8f20a215	3	err() { printf '%b\n' "$*" 1>&2; exit 1; }
955eb9e8	4
765bb138 MF	5	case $1 in
	6	""\|-*) ;;
	7	*)
	8	cmd=$1
	9	shift
	10	acmd=$(git config --get "alias.${cmd}")
	11	;;
	12	esac
35c220c5	13
955eb9e8	14	case ${acmd:-${cmd}} in
35c220c5 MF	15	rebase)
35c220c5 MF	16	if [[ $1 == "all" ]] ; then
e947e1df	17	shift
e051dd74 MF	18	if [[ $# -eq 0 ]] ; then
	19	exec r forall -c 'r rb all .'
	20	fi
	21
e947e1df MF	22	branches=$(g b \| awk '
	23	{
	24	if ($0 ~ "^[] [(]no branch[)]") {
	25	next
	26	} else if ($1 == "*") {
	27	b = $NF
	28	} else {
	29	list = list $NF " "
	30	}
	31	}
	32	END { print list b }')
e051dd74 MF	33	[[ -z ${branches} ]] && exit 0
	34
	35	eval $(bash-colors --env)
	36	echo "${GOOD}### ${PWD}${NORMAL}"
e947e1df	37	for b in ${branches} ; do
e051dd74	38	echo " ${HILITE}### $b${NORMAL}"
35c220c5	39	g co $b \|\| exit 1
e051dd74 MF	40	if ! r rb "$@" ; then
	41	g rb-a
	42	fi
35c220c5 MF	43	done
	44	exit 0
	45	fi
	46	;;
955eb9e8 MF	47	sb-push)
	48	sync_branch="v"
	49
	50	root=${PWD}
	51	while [[ ! -d ${root}/.repo && ${root} != "/" ]] ; do
	52	root=${root%/*}
	53	done
	54	cd "${root}" \|\| exit 1
	55
	56	if [[ ! -e .repo/sandbox-url ]] ; then
8f20a215	57	err "Please configure remote url base in ${root}/.repo/sandbox-url"
955eb9e8 MF	58	fi
	59	remote=$(<.repo/sandbox-url) \|\| exit 1
	60
	61	echo "pushing projects from ${root}"
	62
	63	pids=()
	64	rlist=$(r list)
	65	tcnt=$(echo "${rlist}" \| wc -l)
	66	cnt=1
	67	while read line ; do
	68	line=( ${line} )
	69	path=${line[0]}
	70	export GIT_DIR=${path}/.git
	71	proj=${line[2]}
	72	printf '### (%*i/%i %3i%%) %s\n' \
	73	${#tcnt} $((cnt++)) ${tcnt} $(( cnt * 100 / tcnt )) ${proj}
	74	src="${sync_branch}"
	75	g l -1 ${src} >& /dev/null \|\| src=
	76	( g push --force ${remote}/${proj} ${src}:refs/sandbox/${USER}/${sync_branch} >/dev/null ) &
	77
	78	# ssh servers do not like it when you hammer them :)
	79	# Received disconnect from 74.125.248.80: 7: Too many concurrent connections
	80	# fatal: The remote end hung up unexpectedly
	81	pids+=( $! )
	82	if [[ ${#pids[@]} -eq 20 ]] ; then
	83	wait ${pids[@]:0:3}
	84	pids=( ${pids[@]:3} )
	85	fi
	86	done <<<"$(r list)"
	87	wait
	88
	89	exit 0
	90	;;
8f20a215 MF	91	upload)
	92	args=()
	93	while [[ $# -gt 0 ]] ; do
	94	case $1 in
	95	--re)
1d000f52 MF	96	if [[ $2 == *"OWNERS" ]] ; then
1d000f52 MF	97	owners=$(awk -F'@' '{list = list "," $1} END {print substr(list, 2)}' "$2")
8f20a215 MF	98	if [[ -z ${owners} ]] ; then
	99	err "cannot find OWNERS list"
	100	else
	101	echo "Auto setting reviewers to: ${owners}"
	102	fi
	103	args+=( --re "${owners}" )
	104	shift 2
	105	continue
	106	fi
	107	;;
	108	esac
	109	args+=( "$1" )
	110	shift
	111	done
	112	set -- "${args[@]}"
	113	;;
35c220c5 MF	114	esac
	115
	116	exec repo ${acmd:-${cmd}} "$@"