mirror of
https://github.com/optim-enterprises-bv/kubernetes.git
synced 2025-11-04 04:08:16 +00:00
4d736a0435
- remove unused tempdir - use mktemp instead of creating tmpfile in repo - normalize indent
143 lines
4.8 KiB
Bash
Executable File
143 lines
4.8 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
|
|
# Copyright 2021 The Kubernetes Authors.
|
|
#
|
|
# Licensed 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.
|
|
|
|
# This script generates mock files using mockgen.
|
|
# Usage: `hack/update-mocks.sh`.
|
|
|
|
set -o errexit
|
|
set -o nounset
|
|
set -o pipefail
|
|
|
|
KUBE_ROOT=$(dirname "${BASH_SOURCE[0]}")/..
|
|
source "${KUBE_ROOT}/hack/lib/init.sh"
|
|
# Explicitly opt into go modules, even though we're inside a GOPATH directory
|
|
export GO111MODULE=on
|
|
|
|
kube::golang::verify_go_version
|
|
kube::golang::setup_env
|
|
|
|
echo 'installing mockgen'
|
|
pushd "${KUBE_ROOT}/hack/tools" >/dev/null
|
|
go install github.com/golang/mock/mockgen
|
|
popd >/dev/null
|
|
|
|
function git_find() {
|
|
# Similar to find but faster and easier to understand. We want to include
|
|
# modified and untracked files because this might be running against code
|
|
# which is not tracked by git yet.
|
|
git ls-files -cmo --exclude-standard \
|
|
':!:vendor/*' `# catches vendor/...` \
|
|
':!:*/vendor/*' `# catches any subdir/vendor/...` \
|
|
':!:third_party/*' `# catches third_party/...` \
|
|
':!:*/third_party/*' `# catches third_party/...` \
|
|
':!:*/testdata/*' \
|
|
"$@"
|
|
}
|
|
|
|
cd "${KUBE_ROOT}"
|
|
|
|
GENERATED_MOCK_FILE_REGEX="^// Code generated by MockGen. DO NOT EDIT.$"
|
|
|
|
# pick a tempfile path for writing to
|
|
tmp=$(mktemp)
|
|
kube::util::trap_add "rm -f ${tmp:?}" EXIT
|
|
|
|
# We use this pattern here rather than `git grep` because we don't really want
|
|
# to encode the pathspec list in multiple places and anything more complicated
|
|
# just isn't worth the effort.
|
|
git_find -z ':(glob)**/*.go' \
|
|
| { xargs -0 grep -l --null "${GENERATED_MOCK_FILE_REGEX}" || true; } \
|
|
| xargs -0 rm -f
|
|
|
|
echo 'executing go generate command on below files'
|
|
|
|
git_find -z ':(glob)**/*.go' | while read -r -d $'\0' file; do
|
|
test -f "$file" || continue
|
|
grep -q "//go:generate mockgen" "$file" || continue
|
|
|
|
temp_file_name="$(kube::realpath "$(mktemp -t "$(basename "$0").XXXXXX")")"
|
|
|
|
# search for build tag used in file
|
|
build_tag_string=$(grep -o '+build.*$' "$file") || true
|
|
|
|
# if the file does not have build string
|
|
if [ -n "$build_tag_string" ]; then
|
|
# write the build tag in the temp file
|
|
echo -n "$build_tag_string" > "$temp_file_name"
|
|
|
|
# if +build tag is defined in interface file
|
|
BUILD_TAG_FILE=$temp_file_name go generate -v "$file"
|
|
else
|
|
# if no +build tag is defined in interface file
|
|
go generate -v "$file"
|
|
fi
|
|
done
|
|
|
|
# get the changed or new mock files
|
|
git ls-files -mo --exclude-standard -z | while read -r -d $'\0' file; do
|
|
# only process files that appear to be mocks
|
|
test -f "$file" || continue
|
|
grep -q "${GENERATED_MOCK_FILE_REGEX}" "$file" || continue
|
|
|
|
# search for build tags used in file
|
|
# //go:build !providerless
|
|
# // +build !providerless
|
|
go_build_tag_string=$(grep -o 'go:build.*$' "$file") || true
|
|
build_tag_string=$(grep -o '+build.*$' "$file") || true
|
|
new_header=''
|
|
|
|
# if the file has both headers
|
|
if [ -n "$build_tag_string" ] && [ -n "$go_build_tag_string" ]
|
|
then
|
|
# create a new header with the build string and the copyright text
|
|
new_header=$(echo -e "//""$go_build_tag_string""\n""//" "$build_tag_string""\n" | cat - hack/boilerplate/boilerplate.generatego.txt)
|
|
|
|
# ignore the first line (build tag) from the file
|
|
tail -n +3 "$file" >"${tmp}"
|
|
fi
|
|
|
|
# if the file has only // +build !providerless header
|
|
if [ -n "$build_tag_string" ] && [ -z "$go_build_tag_string" ]
|
|
then
|
|
# create a new header with the build string and the copyright text
|
|
new_header=$(echo -e "//" "$build_tag_string""\n" | cat - hack/boilerplate/boilerplate.generatego.txt)
|
|
|
|
# ignore the first line (build tag) from the file
|
|
tail -n +2 "$file" >"${tmp}"
|
|
fi
|
|
|
|
# if the file has only //go:build !providerless header
|
|
if [ -z "$build_tag_string" ] && [ -n "$go_build_tag_string" ]
|
|
then
|
|
# create a new header with the build string and the copyright text
|
|
new_header=$(echo -e "//""$go_build_tag_string""\n" | cat - hack/boilerplate/boilerplate.generatego.txt)
|
|
|
|
# ignore the first line (build tag) from the file
|
|
tail -n +2 "$file" >"${tmp}"
|
|
fi
|
|
|
|
# if the header is generated
|
|
if [ -n "$new_header" ]
|
|
then
|
|
# write the newly generated header file to the original file
|
|
echo -e "$new_header" | cat - "${tmp}" > "$file"
|
|
else
|
|
# if no build string insert at the top
|
|
cat hack/boilerplate/boilerplate.generatego.txt "$file" >"${tmp}" && \
|
|
mv "${tmp}" "$file"
|
|
fi
|
|
done
|