From d9d5968947b09d33074b4158c64435c28f48ba8e Mon Sep 17 00:00:00 2001
From: Jordan Liggitt <liggitt@google.com>
Date: Fri, 10 May 2019 14:56:50 -0400
Subject: [PATCH] Prune matching replace directives in staging repos more
 effectively

---
 hack/update-vendor.sh | 19 +++++++++++++++++--
 1 file changed, 17 insertions(+), 2 deletions(-)

diff --git a/hack/update-vendor.sh b/hack/update-vendor.sh
index 6f4ee965d0c..f351f6c9eb9 100755
--- a/hack/update-vendor.sh
+++ b/hack/update-vendor.sh
@@ -66,7 +66,9 @@ function ensure_require_replace_directives_for_all_dependencies() {
   go mod edit -json | jq -r ".Require // [] | sort | .[] | select(${require_filter})" > "${require_json}"
   go mod edit -json | jq -r ".Replace // [] | sort | .[] | select(${replace_filter})" > "${replace_json}"
 
-  # 1. Ensure require directives have a corresponding replace directive pinning a version
+  # 1a. Ensure replace directives have an explicit require directive
+  cat "${replace_json}" | jq -r '"-require \(.Old.Path)@\(.New.Version)"'             | xargs -L 100 go mod edit -fmt
+  # 1b. Ensure require directives have a corresponding replace directive pinning a version
   cat "${require_json}" | jq -r '"-replace \(.Path)=\(.Path)@\(.Version)"'            | xargs -L 100 go mod edit -fmt
   cat "${replace_json}" | jq -r '"-replace \(.Old.Path)=\(.New.Path)@\(.New.Version)"'| xargs -L 100 go mod edit -fmt
 
@@ -238,7 +240,20 @@ kube::log::status "go.mod: tidying"
 for repo in $(tsort "${TMP_DIR}/tidy_deps.txt"); do
   pushd "${KUBE_ROOT}/staging/src/${repo}" >/dev/null 2>&1
     echo "=== tidying ${repo}" >> "${LOG_FILE}"
-    go mod tidy >>"${LOG_FILE}" 2>&1
+
+    # prune replace directives that pin to the naturally selected version.
+    # do this before tidying, since tidy removes unused modules that
+    # don't provide any relevant packages, which forgets which version of the 
+    # unused transitive dependency we had a require directive for,
+    # and prevents pruning the matching replace directive after tidying.
+    go list -m -json all |
+      jq -r 'select(.Replace != null) | 
+             select(.Path == .Replace.Path) | 
+             select(.Version == .Replace.Version) | 
+             "-dropreplace \(.Replace.Path)"' |
+    xargs -L 100 go mod edit -fmt
+
+    go mod tidy -v >>"${LOG_FILE}" 2>&1
 
     # disallow transitive dependencies on k8s.io/kubernetes
     loopback_deps="$(go list all 2>/dev/null | grep k8s.io/kubernetes/ || true)"