From 0e23b898a2b7cd8ff2c05741bc477f30fee03c8e Mon Sep 17 00:00:00 2001
From: Nick Travers <travers@cockroachlabs.com>
Date: Wed, 2 Feb 2022 16:01:51 -0800
Subject: [PATCH] scripts: per-branch bump-pebble.sh script [21.1 variant]

This is the `release-21.1` variant of #75908.

Release note: None
---
 scripts/bump-pebble.sh | 126 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 126 insertions(+)
 create mode 100755 scripts/bump-pebble.sh

diff --git a/scripts/bump-pebble.sh b/scripts/bump-pebble.sh
new file mode 100755
index 000000000000..e098702bc11e
--- /dev/null
+++ b/scripts/bump-pebble.sh
@@ -0,0 +1,126 @@
+#!/usr/bin/env bash
+
+# This script may be used to produce a branch bumping the Pebble
+# version. The storage team bumps CockroachDB's Pebble dependency
+# frequently, and this script automates some of that work.
+#
+# To bump the Pebble dependency to the corresponding pebble branch HEAD, run:
+#
+#   ./scripts/bump-pebble.sh
+#
+# Note that this script has different behaviour based on the branch on which it
+# is run.
+
+set -euo pipefail
+
+echoerr() { printf "%s\n" "$*" >&2; }
+pushd() { builtin pushd "$@" > /dev/null; }
+popd() { builtin popd "$@" > /dev/null; }
+
+# NOTE: After a new release has been cut, update this to the appropriate
+# Cockroach branch name (i.e. release-22.2, etc.), and corresponding Pebble
+# branch name (e.g. crl-release-21.2, etc.).
+BRANCH=release-21.1
+PEBBLE_BRANCH=crl-release-21.1
+
+# The script can only be run from a specific branch.
+if [ "$(git rev-parse --abbrev-ref HEAD)" != "$BRANCH" ]; then
+  echo "This script must be run from the $BRANCH branch."
+  exit 1
+fi
+
+COCKROACH_DIR="$(go env GOPATH)/src/github.com/cockroachdb/cockroach"
+PEBBLE_DIR="$(go env GOPATH)/src/github.com/cockroachdb/pebble"
+VENDORED_DIR="$COCKROACH_DIR/vendor"
+
+# Make sure that the cockroachdb remotes match what we expect. The
+# `upstream` remote must point to github.com/cockroachdb/cockroach.
+pushd "$COCKROACH_DIR"
+git submodule update --init --recursive
+set +e
+COCKROACH_UPSTREAM_URL="$(git config --get remote.upstream.url)"
+set -e
+if [ "$COCKROACH_UPSTREAM_URL" != "git@github.com:cockroachdb/cockroach.git" ]; then
+    echoerr "Error: Expected the upstream remote to be the primary cockroachdb repository"
+    echoerr "at git@github.com:cockroachdb/cockroach.git. Found:"
+    echoerr ""
+    echoerr "  $COCKROACH_UPSTREAM_URL"
+    echoerr ""
+    exit 1
+fi
+popd
+
+# Make sure that the cockroachdb remotes match what we expect. The
+# `upstream` remote must point to github.com/cockroachdb/pebble.
+pushd "$PEBBLE_DIR"
+set +e
+PEBBLE_UPSTREAM_URL="$(git config --get remote.upstream.url)"
+set -e
+if [ "$PEBBLE_UPSTREAM_URL" != "git@github.com:cockroachdb/pebble.git" ]; then
+    echoerr "Error: Expected the upstream remote to be the cockroachdb/pebble repository"
+    echoerr "at git@github.com:cockroachdb/pebble.git. Found:"
+    echoerr ""
+    echoerr "  $PEBBLE_UPSTREAM_URL"
+    echoerr ""
+    exit 1
+fi
+popd
+
+# Ensure the local CockroachDB release branch is up-to-date with
+# upstream and grab the current Pebble SHA.
+pushd "$COCKROACH_DIR"
+git fetch upstream "$BRANCH"
+git checkout "$BRANCH"
+git rebase "upstream/$BRANCH"
+OLD_SHA=$(grep 'github.com/cockroachdb/pebble' go.mod | grep -o -E '[a-f0-9]{12}$')
+popd
+
+# Ensure the local Pebble release branch is up-to-date with upstream,
+# and grab the desired Pebble SHA.
+pushd "$PEBBLE_DIR"
+git fetch upstream "$PEBBLE_BRANCH"
+NEW_SHA=$(git rev-parse "upstream/$PEBBLE_BRANCH")
+COMMITS=$(git log --pretty='format:%h %s' "$OLD_SHA..$NEW_SHA" | grep -v 'Merge pull request')
+echo "$COMMITS"
+popd
+
+VENDORED_BRANCH="$USER/pebble-${BRANCH}-${NEW_SHA:0:12}"
+COCKROACH_BRANCH="$USER/pebble-${BRANCH}-${NEW_SHA:0:12}"
+
+# Pull in the Pebble module at the desired SHA and rebuild the vendor
+# directory.
+pushd "$COCKROACH_DIR"
+go get "github.com/cockroachdb/pebble@${NEW_SHA}"
+go mod tidy
+make -k vendor_rebuild
+popd
+
+# Commit all the pending vendor directory changes to a new
+# github.com/cockroachdb/vendored repository branch, including the
+# commit history.
+pushd "$VENDORED_DIR"
+git branch -D "$VENDORED_BRANCH" || true
+git checkout -b "$VENDORED_BRANCH"
+git add --all
+git commit -m "bump Pebble to ${NEW_SHA:0:12}
+
+$COMMITS"
+git push -u --force origin "$VENDORED_BRANCH"
+popd
+
+# Create the branch and commit on the CockroachDB repository.
+pushd "$COCKROACH_DIR"
+make bazel-generate
+git add go.mod go.sum DEPS.bzl
+git add vendor
+git branch -D "$COCKROACH_BRANCH" || true
+git checkout -b "$COCKROACH_BRANCH"
+git commit -m "vendor: bump Pebble to ${NEW_SHA:0:12}
+
+$COMMITS
+
+Release note:
+"
+# Open an editor to allow the user to set the release note.
+git commit --amend
+popd