[MINOR] Adding scripts to checkout and push to PRs (#2109)
- Tested the checkout_pr.sh locally - Tested a dryrun of pr_push_command.sh
This commit is contained in:
61
scripts/checkout_pr.sh
Executable file
61
scripts/checkout_pr.sh
Executable file
@@ -0,0 +1,61 @@
|
||||
#!/bin/bash
|
||||
|
||||
# Licensed to the Apache Software Foundation (ASF) under one
|
||||
# or more contributor license agreements. See the NOTICE file
|
||||
# distributed with this work for additional information
|
||||
# regarding copyright ownership. The ASF licenses this file
|
||||
# to you under the Apache License, Version 2.0 (the
|
||||
# "License"); you may not use this file except in compliance
|
||||
# with the License. You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
#
|
||||
# Usage: ./scripts/checkout_pr.sh
|
||||
#
|
||||
# Checkout a PR given the PR number into a local branch. PR branches are named
|
||||
# using the convention "pull/<PR_NUMBER>", to enable pr_push_command.sh to work
|
||||
# in tandem.
|
||||
#
|
||||
set -eou pipefail
|
||||
|
||||
function printUsage {
|
||||
echo "Usage: $(basename "${0}") [-r REMOTE] [-f] <PR_NUMBER>" 2>&1
|
||||
echo ' -r REMOTE remote to grab PR from (default: apache)'
|
||||
echo ' -f force overwrite of local branch (default: fail if exists)'
|
||||
exit 1
|
||||
}
|
||||
|
||||
if [[ ${#} -eq 0 ]]; then
|
||||
printUsage
|
||||
fi
|
||||
|
||||
REMOTE="origin"
|
||||
FORCE=""
|
||||
while getopts ":r:f" arg; do
|
||||
case "${arg}" in
|
||||
r)
|
||||
REMOTE="${OPTARG}"
|
||||
;;
|
||||
f)
|
||||
FORCE="--force"
|
||||
;;
|
||||
?)
|
||||
printUsage
|
||||
;;
|
||||
esac
|
||||
done
|
||||
shift "$(($OPTIND -1))"
|
||||
|
||||
# Debug output
|
||||
PR_NUM=$1
|
||||
|
||||
# Checkout the PR into a local branch.
|
||||
git fetch ${REMOTE} pull/${PR_NUM}/head:pull/${PR_NUM} ${FORCE}
|
||||
git checkout pull/${PR_NUM}
|
||||
60
scripts/pr_push_command.sh
Executable file
60
scripts/pr_push_command.sh
Executable file
@@ -0,0 +1,60 @@
|
||||
#!/bin/bash
|
||||
|
||||
# Licensed to the Apache Software Foundation (ASF) under one
|
||||
# or more contributor license agreements. See the NOTICE file
|
||||
# distributed with this work for additional information
|
||||
# regarding copyright ownership. The ASF licenses this file
|
||||
# to you under the Apache License, Version 2.0 (the
|
||||
# "License"); you may not use this file except in compliance
|
||||
# with the License. You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
#
|
||||
# Usage: ./scripts/pr_push_command.sh
|
||||
#
|
||||
# When run from a PR branch checked out by checkout_pr.sh, provides a command
|
||||
# that can push local PR branch to its corresponding remote.
|
||||
# NOTE: Always double check correctness of command, before pushing.
|
||||
#
|
||||
|
||||
set -eou pipefail
|
||||
|
||||
CURR_BRANCH=$(git status | grep "On branch" | awk '{print $NF}')
|
||||
REMOTE="apache"
|
||||
|
||||
# Get PR number from branch
|
||||
if [[ ${CURR_BRANCH} = pull/* ]]
|
||||
then
|
||||
PR_NUM=$(echo "${CURR_BRANCH}" | awk -F'/' '{print $NF}')
|
||||
else
|
||||
echo "Not on a PR branch?"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# Parse the pr's remote, branch & add a remote if needed.
|
||||
PR_RESP=$(curl https://api.github.com/repos/${REMOTE}/hudi/pulls/${PR_NUM})
|
||||
if ! echo ${PR_RESP} | jq '.url' | grep "hudi/pulls/${PR_NUM}" ; then
|
||||
echo "Unable to find PR number ${PR_NUM} in remote ${REMOTE}"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
PR_SSH_URL=$(echo ${PR_RESP} | jq -r '.head.repo.ssh_url')
|
||||
PR_REMOTE_BRANCH=$(echo ${PR_RESP} | jq -r '.head.ref')
|
||||
PR_REMOTE=$(echo ${PR_RESP} | jq -r '.head.repo.owner.login')
|
||||
|
||||
# Add a new remote, if needed.
|
||||
if ! git remote -v | grep ${PR_REMOTE} ; then
|
||||
echo "Adding new remote ${PR_REMOTE} with ssh url ${PR_SSH_URL}"
|
||||
git remote add ${PR_REMOTE} ${PR_SSH_URL}
|
||||
fi
|
||||
|
||||
# Push local branch to PR remote/branch
|
||||
echo "If you are sure, execute the following command to push"
|
||||
echo " git push ${PR_REMOTE} ${CURR_BRANCH}:${PR_REMOTE_BRANCH}"
|
||||
Reference in New Issue
Block a user