X-Git-Url: https://git.wh0rd.org/?a=blobdiff_plain;f=.bin%2Fr;h=a969597655a52ff8dec24afdd286d64a705cbfab;hb=a5f629de0d2669846d3b30fe8a38dfca79d92aae;hp=6a86466c38ecdedb3e53527b7699c15e4699ac62;hpb=d3bdab7a0f168e65849b2bc8863d9d7a8ef123aa;p=home.git

diff --git a/.bin/r b/.bin/r
index 6a86466..a969597 100755
--- a/.bin/r
+++ b/.bin/r
@@ -3,6 +3,10 @@ g() { git "$@"; }
 err() { printf '%b\n' "$*" 1>&2; exit 1; }
 vr() { echo "$@"; "$@"; }
 
+case $1 in
+-x) set -x; shift;;
+esac
+
 case $1 in
 ""|-*) ;;
 l)
@@ -13,6 +17,11 @@ s)
 	cmd=sync
 	shift
 	;;
+rb-all)
+	cmd=rebase
+	shift
+	set -- all "$@"
+	;;
 *)
 	cmd=$1
 	shift
@@ -50,12 +59,79 @@ mj_finish() {
 
 repo_root() {
 	local root=${PWD}
-	while [[ ! -d ${root}/.repo && ${root} != "/" ]] ; do
+	while [[ ! -d ${root}/.repo && ${root:-/} != "/" ]] ; do
 		root=${root%/*}
 	done
 	echo "${root}"
 }
 
+process_reviewers() {
+	local r arr=()
+	for r in ${*//,/ } ; do
+		case ${r} in
+		OWNERS*|*/OWNERS*)
+			local owners=$(
+				awk -F'@' '
+					($2 == "chromium.org" || $2 == "google.com") {list = list "," $1}
+					END {print substr(list, 2)}
+				' "${r}"
+			)
+			if [[ -z ${owners} ]] ; then
+				err "cannot find OWNERS list"
+			else
+				echo "Auto setting reviewers to: ${owners}"
+			fi
+			arr+=( ${owners} )
+			;;
+		*)
+			arr+=( "${r}" )
+			;;
+		esac
+	done
+	reviewers=$(printf '%s,' "${arr[@]}")
+	reviewers=${reviewers%,}
+}
+
+find_repo() {
+	# Diff projects have diff versions of repo.  Find a compatible one.
+	local root=$(repo_root)
+
+	# Use the manifest repo URL.
+	local manifest_dir="${root}/.repo/manifests.git"
+	local d
+
+	# Default to the local repo if it's there.
+	local search=(
+		"${root}/.repo/repo"
+	)
+
+	case $(g --git-dir="${manifest_dir}" config remote.origin.url) in
+	*android*)
+		search+=(
+			/usr/local/src/repo
+			~/src/repo
+		)
+		;;
+	*chromium*|*chrome*)
+		search+=(
+			~/depot_tools
+			~/chromiumos/depot_tools
+			/usr/local/src/depot_tools
+			~/src/depot_tools
+		)
+		;;
+	esac
+	for d in "${search[@]}" ; do
+		if [[ -x ${d}/repo ]] ; then
+			echo "${d}"/repo
+			return
+		fi
+	done
+
+	# Fallback: use $PATH.
+	type -P repo
+}
+
 case ${acmd:-${cmd}} in
 rebase)
 	if [[ $1 == "all" ]] ; then
@@ -67,18 +143,26 @@ rebase)
 			while read -a line ; do
 				dir=${line[0]}
 				proj=${line[2]}
-				cd "${root}/${dir}"
+				if ! cd "${root}/${dir}" ; then
+					echo "bad ${proj}"
+					continue
+				fi
 				(
 				out=$(env _proj=${proj} r rb all . 2>&1)
 				if [[ -n ${out} ]] ; then
 					while read line ; do
-						if [[ ${line} == "# "* ]] ; then
+						case ${line} in
+						"Fast-forwarded "*) ;;
+						"# "*)
 							line="${line#[#] }"
 							printf '%s### %s%-40s%s: %s\n' \
 								"${BRACKET}" "${GOOD}" "${line%%:*}" "${NORMAL}" "${line#*:}"
-						else
-							echo "${line}"
-						fi
+							;;
+						*)
+							printf '%s### %s%-40s%s: ERROR: %s\n' \
+								"${BRACKET}" "${BAD}" "${dir}" "${NORMAL}" "${line}"
+							;;
+						esac
 					done < <(echo "${out}")
 				fi
 				_mj_child
@@ -94,7 +178,7 @@ rebase)
 			{
 				if ($0 ~ "^[*] *[(]no branch[)]") {
 					next
-				} else if ($0 ~ "^[*] *[(]detached from ") {
+				} else if ($0 ~ "^[*] *[(](HEAD )?detached (from|at) ") {
 					next
 				} else if ($1 == "*") {
 					b = $2
@@ -143,7 +227,15 @@ clean)
 	mj_finish
 	exit
 	;;
-sb-push)
+sb)
+	sb_cmd=$1
+	case ${sb_cmd} in
+	pull) ;;
+	push) ;;
+	f|fetch) sb_cmd="fetch" ;;
+	*) err "unknown sandbox command: $1"
+	esac
+
 	sync_branch="v"
 
 	root=$(repo_root)
@@ -172,8 +264,17 @@ sb-push)
 		printf '### (%*i/%i %3i%%) %s\n' \
 			${#tcnt} $((cnt++)) ${tcnt} $(( cnt * 100 / tcnt )) ${proj}
 		src="${sync_branch}"
-		g l -1 ${src} >& /dev/null || src=
-		mj_child g push --force ${remote}/${proj} ${src}:refs/sandbox/${USER}/${sync_branch} >/dev/null
+		case ${sb_cmd} in
+		push)
+			g l -1 ${src} >& /dev/null || src=
+			mj_child g push --force ${remote}/${proj} ${src}:refs/sandbox/${USER}/${sync_branch} >/dev/null
+			;;
+		pull)
+			;;
+		fetch)
+			mj_child g fetch ${remote}/${proj} refs/sandbox/${USER}/${sync_branch}:refs/remotes/sb/${sync_branch} >/dev/null
+			;;
+		esac
 	done < <(echo "${rlist}")
 	mj_finish
 
@@ -202,7 +303,7 @@ g-push)
 	while [[ $# -gt 0 ]] ; do
 		case $1 in
 		--re)
-			reviewers=$2
+			process_reviewers "$2"
 			shift
 			;;
 		-n|--dry-run|--draft)
@@ -237,17 +338,10 @@ upload)
 	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
+			process_reviewers "$2"
+			args+=( --re "${reviewers}" )
+			shift 2
+			continue
 			;;
 		esac
 		args+=( "$1" )
@@ -257,9 +351,14 @@ upload)
 	;;
 email)
 	email=${1:-${USER}@chromium.org}
+
+	root=$(repo_root)
+	git --git-dir="${root}/.repo/manifests.git" cfg user.email "${email}"
+	git --git-dir="${root}/.repo/repo/.git" cfg user.email "${email}"
+
 	cmd='forall'
-	set -- -c "git cfg user.email ${email}"
+	set -- -c "git cfg user.email '${email}'"
 	;;
 esac
 
-exec repo ${acmd:-${cmd}} "$@"
+exec python2 $(find_repo) ${acmd:-${cmd}} "$@"