110 lines
4.8 KiB
Bash
110 lines
4.8 KiB
Bash
|
#!/usr/bin/env bash
|
||
|
|
|
||
|
|
# Copyright 2017 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.
|
||
|
|
|
||
|
|
set -o errexit
|
||
|
|
set -o nounset
|
||
|
|
set -o pipefail
|
||
|
|
|
||
|
|
# generate-internal-groups generates everything for a project with internal types, e.g. an
|
||
|
|
# user-provided API server based on k8s.io/apiserver.
|
||
|
|
|
||
|
|
if [ "$#" -lt 5 ] || [ "${1}" == "--help" ]; then
|
||
|
|
cat <<EOF
|
||
|
|
Usage: $(basename $0) <generators> <output-package> <internal-apis-package> <extensiona-apis-package> <groups-versions> ...
|
||
|
|
|
||
|
|
<generators> the generators comma separated to run (deepcopy,defaulter,conversion,client,lister,informer) or "all".
|
||
|
|
<output-package> the output package name (e.g. github.com/example/project/pkg/generated).
|
||
|
|
<int-apis-package> the internal types dir (e.g. github.com/example/project/pkg/apis).
|
||
|
|
<ext-apis-package> the external types dir (e.g. github.com/example/project/pkg/apis or githubcom/example/apis).
|
||
|
|
<groups-versions> the groups and their versions in the format "groupA:v1,v2 groupB:v1 groupC:v2", relative
|
||
|
|
to <api-package>.
|
||
|
|
... arbitrary flags passed to all generator binaries.
|
||
|
|
|
||
|
|
Examples:
|
||
|
|
$(basename $0) all github.com/example/project/pkg/client github.com/example/project/pkg/apis github.com/example/project/pkg/apis "foo:v1 bar:v1alpha1,v1beta1"
|
||
|
|
$(basename $0) deepcopy,defaulter,conversion github.com/example/project/pkg/client github.com/example/project/pkg/apis github.com/example/project/apis "foo:v1 bar:v1alpha1,v1beta1"
|
||
|
|
EOF
|
||
|
|
exit 0
|
||
|
|
fi
|
||
|
|
|
||
|
|
GENS="$1"
|
||
|
|
OUTPUT_PKG="$2"
|
||
|
|
INT_APIS_PKG="$3"
|
||
|
|
EXT_APIS_PKG="$4"
|
||
|
|
GROUPS_WITH_VERSIONS="$5"
|
||
|
|
shift 5
|
||
|
|
|
||
|
|
go install ./$(dirname "${0}")/cmd/{defaulter-gen,conversion-gen,client-gen,lister-gen,informer-gen,deepcopy-gen}
|
||
|
|
function codegen::join() { local IFS="$1"; shift; echo "$*"; }
|
||
|
|
|
||
|
|
# enumerate group versions
|
||
|
|
ALL_FQ_APIS=() # e.g. k8s.io/kubernetes/pkg/apis/apps k8s.io/api/apps/v1
|
||
|
|
INT_FQ_APIS=() # e.g. k8s.io/kubernetes/pkg/apis/apps
|
||
|
|
EXT_FQ_APIS=() # e.g. k8s.io/api/apps/v1
|
||
|
|
for GVs in ${GROUPS_WITH_VERSIONS}; do
|
||
|
|
IFS=: read G Vs <<<"${GVs}"
|
||
|
|
|
||
|
|
if [ -n "${INT_APIS_PKG}" ]; then
|
||
|
|
ALL_FQ_APIS+=("${INT_APIS_PKG}/${G}")
|
||
|
|
INT_FQ_APIS+=("${INT_APIS_PKG}/${G}")
|
||
|
|
fi
|
||
|
|
|
||
|
|
# enumerate versions
|
||
|
|
for V in ${Vs//,/ }; do
|
||
|
|
ALL_FQ_APIS+=("${EXT_APIS_PKG}/${G}/${V}")
|
||
|
|
EXT_FQ_APIS+=("${EXT_APIS_PKG}/${G}/${V}")
|
||
|
|
done
|
||
|
|
done
|
||
|
|
|
||
|
|
if [ "${GENS}" = "all" ] || grep -qw "deepcopy" <<<"${GENS}"; then
|
||
|
|
echo "Generating deepcopy funcs"
|
||
|
|
${GOPATH}/bin/deepcopy-gen --input-dirs $(codegen::join , "${ALL_FQ_APIS[@]}") -O zz_generated.deepcopy --bounding-dirs ${INT_APIS_PKG},${EXT_APIS_PKG} "$@"
|
||
|
|
fi
|
||
|
|
|
||
|
|
if [ "${GENS}" = "all" ] || grep -qw "defaulter" <<<"${GENS}"; then
|
||
|
|
echo "Generating defaulters"
|
||
|
|
${GOPATH}/bin/defaulter-gen --input-dirs $(codegen::join , "${EXT_FQ_APIS[@]}") -O zz_generated.defaults "$@"
|
||
|
|
fi
|
||
|
|
|
||
|
|
if [ "${GENS}" = "all" ] || grep -qw "conversion" <<<"${GENS}"; then
|
||
|
|
echo "Generating conversions"
|
||
|
|
${GOPATH}/bin/conversion-gen --input-dirs $(codegen::join , "${ALL_FQ_APIS[@]}") -O zz_generated.conversion "$@"
|
||
|
|
fi
|
||
|
|
|
||
|
|
if [ "${GENS}" = "all" ] || grep -qw "client" <<<"${GENS}"; then
|
||
|
|
echo "Generating clientset for ${GROUPS_WITH_VERSIONS} at ${OUTPUT_PKG}/clientset"
|
||
|
|
if [ -n "${INT_APIS_PKG}" ]; then
|
||
|
|
${GOPATH}/bin/client-gen --clientset-name internalversion --input-base "" --input $(codegen::join , $(printf '%s/ ' "${INT_FQ_APIS[@]}")) --output-package ${OUTPUT_PKG}/clientset "$@"
|
||
|
|
fi
|
||
|
|
${GOPATH}/bin/client-gen --clientset-name versioned --input-base "" --input $(codegen::join , "${EXT_FQ_APIS[@]}") --output-package ${OUTPUT_PKG}/clientset "$@"
|
||
|
|
fi
|
||
|
|
|
||
|
|
if [ "${GENS}" = "all" ] || grep -qw "lister" <<<"${GENS}"; then
|
||
|
|
echo "Generating listers for ${GROUPS_WITH_VERSIONS} at ${OUTPUT_PKG}/listers"
|
||
|
|
${GOPATH}/bin/lister-gen --input-dirs $(codegen::join , "${ALL_FQ_APIS[@]}") --output-package ${OUTPUT_PKG}/listers "$@"
|
||
|
|
fi
|
||
|
|
|
||
|
|
if [ "${GENS}" = "all" ] || grep -qw "informer" <<<"${GENS}"; then
|
||
|
|
echo "Generating informers for ${GROUPS_WITH_VERSIONS} at ${OUTPUT_PKG}/informers"
|
||
|
|
${GOPATH}/bin/informer-gen \
|
||
|
|
--input-dirs $(codegen::join , "${ALL_FQ_APIS[@]}") \
|
||
|
|
--versioned-clientset-package ${OUTPUT_PKG}/clientset/versioned \
|
||
|
|
--internal-clientset-package ${OUTPUT_PKG}/clientset/internalversion \
|
||
|
|
--listers-package ${OUTPUT_PKG}/listers \
|
||
|
|
--output-package ${OUTPUT_PKG}/informers \
|
||
|
|
"$@"
|
||
|
|
fi
|