grafana/hack/update-codegen.sh
2024-08-14 11:06:37 -04:00

91 lines
2.9 KiB
Bash
Executable File

#!/usr/bin/env bash
# SPDX-License-Identifier: AGPL-3.0-only
# Provenance-includes-location: https://github.com/kubernetes/sample-apiserver/blob/master/hack/update-codegen.sh
# Provenance-includes-license: Apache-2.0
# Provenance-includes-copyright: The Kubernetes Authors.
set -o errexit
set -o nounset
set -o pipefail
SCRIPT_ROOT=$(dirname "${BASH_SOURCE[0]}")/..
pushd "${SCRIPT_ROOT}/hack" && GO111MODULE=on go mod tidy && popd
CODEGEN_PKG=${CODEGEN_PKG:-$(cd "${SCRIPT_ROOT}"; ls -d -1 ./vendor/k8s.io/code-generator 2>/dev/null || echo $(go env GOPATH)/pkg/mod/k8s.io/code-generator@v0.31.0)}
OUTDIR="${HOME}/go/src"
OPENAPI_VIOLATION_EXCEPTIONS_FILENAME="zz_generated.openapi_violation_exceptions.list"
source "${CODEGEN_PKG}/kube_codegen.sh"
source "$(dirname "${BASH_SOURCE[0]}")/openapi-codegen.sh"
selected_pkg="${1-}"
grafana::codegen:run() {
local generate_root=$1
local skipped="true"
for api_pkg in $(grafana:codegen:lsdirs ./${generate_root}/apis); do
if [[ "${selected_pkg}" != "" && ${api_pkg} != $selected_pkg ]]; then
continue
fi
echo "Generating code for ${generate_root}/apis/${api_pkg}..."
echo "============================================="
skipped="false"
include_common_input_dirs=$([[ ${api_pkg} == "common" ]] && echo "true" || echo "false")
kube::codegen::gen_helpers \
--boilerplate "${SCRIPT_ROOT}/hack/boilerplate.go.txt" \
${generate_root}/apis/${api_pkg}
for pkg_version in $(grafana:codegen:lsdirs ./${generate_root}/apis/${api_pkg}); do
grafana::codegen::gen_openapi \
--input-pkg-single ${generate_root}/apis/${api_pkg}/${pkg_version} \
--output-base "${OUTDIR}" \
--report-filename "${OPENAPI_VIOLATION_EXCEPTIONS_FILENAME}" \
--update-report \
--boilerplate "${SCRIPT_ROOT}/hack/boilerplate.go.txt" \
--include-common-input-dirs ${include_common_input_dirs}
violations_file="${generate_root}/apis/${api_pkg}/${pkg_version}/${OPENAPI_VIOLATION_EXCEPTIONS_FILENAME}"
if [ ! -f "${violations_file}" ]; then
continue
fi
# delete violation exceptions file, if empty
if ! grep -q . "${violations_file}"; then
echo "Deleting ${violations_file} since it is empty"
rm ${violations_file}
fi
echo ""
done
done
if [[ "${skipped}" == "true" ]]; then
echo "no apis matching ${selected_pkg}. skipping..."
echo
return 0
fi
echo "Generating client code..."
echo "-------------------------"
kube::codegen::gen_client \
--with-watch \
--with-applyconfig \
--output-dir ${generate_root}/generated \
--output-pkg github.com/grafana/grafana/${generate_root}/generated \
--boilerplate "${SCRIPT_ROOT}/hack/boilerplate.go.txt" \
${generate_root}/apis
echo ""
}
grafana:codegen:lsdirs() {
ls -d $1/*/ | xargs basename -a
}
grafana::codegen:run pkg
grafana::codegen:run pkg/apimachinery
grafana::codegen:run pkg/aggregator
echo "done."