#!/bin/bash

# Copyright 2014 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

setglobal KUBE_ROOT = "$[dirname $(BASH_SOURCE)]/.."
source "$(KUBE_ROOT)/hack/lib/init.sh"

kube::golang::setup_env

setglobal BUILD_TARGETS = ''(
  vendor/k8s.io/code-generator/cmd/client-gen
  vendor/k8s.io/code-generator/cmd/lister-gen
  vendor/k8s.io/code-generator/cmd/informer-gen
)
make -C $(KUBE_ROOT) WHAT="$(BUILD_TARGETS[*])"

setglobal clientgen = $[kube::util::find-binary "client-gen]
setglobal listergen = $[kube::util::find-binary "lister-gen]
setglobal informergen = $[kube::util::find-binary "informer-gen]

# Please do not add any logic to this shell script. Add logic to the go code
# that generates the set-gen program.
#

setglobal GROUP_VERSIONS = ''(${KUBE_AVAILABLE_GROUP_VERSIONS})
setglobal GV_DIRS = ''()
for gv in [$(GROUP_VERSIONS[@])] {
	# add items, but strip off any leading apis/ you find to match command expectations
	setglobal api_dir = $[kube::util::group-version-to-pkg-path $(gv)]
	setglobal nopkg_dir = $(api_dir#pkg/)
	setglobal nopkg_dir = $(nopkg_dir#vendor/k8s.io/api/)
	setglobal pkg_dir = $(nopkg_dir#apis/)

	# skip groups that aren't being served, clients for these don't matter
    if [[ " ${KUBE_NONSERVER_GROUP_VERSIONS} " == *" ${gv} "* ]] {
      continue
    }

	setglobal GV_DIRS = ''("${pkg_dir}")
}
# delimit by commas for the command
setglobal GV_DIRS_CSV = $[setglobal IFS = '','';echo $(GV_DIRS[*]// /,);setglobal IFS = '$']

# This can be called with one flag, --verify-only, so it works for both the
# update- and verify- scripts.
$(clientgen) @Argv
$(clientgen) --output-base "$(KUBE_ROOT)/vendor" --clientset-path="k8s.io/client-go" --clientset-name="kubernetes" --input-base="k8s.io/kubernetes/vendor/k8s.io/api" --input="$(GV_DIRS_CSV)" @Argv
# Clientgen for federation clientset.
$(clientgen) --clientset-name=federation_clientset --clientset-path=k8s.io/kubernetes/federation/client/clientset_generated --input-base="k8s.io/kubernetes/vendor/k8s.io/api" --input="../../../federation/apis/federation/v1beta1","core/v1","extensions/v1beta1","batch/v1","autoscaling/v1" --included-types-overrides="core/v1/Service,core/v1/Namespace,extensions/v1beta1/ReplicaSet,core/v1/Secret,extensions/v1beta1/Ingress,extensions/v1beta1/Deployment,extensions/v1beta1/DaemonSet,core/v1/ConfigMap,core/v1/Event,batch/v1/Job,autoscaling/v1/HorizontalPodAutoscaler"   @Argv

setglobal listergen_internal_apis = ''(
pkg/api
$(
  cd ${KUBE_ROOT}
  find pkg/apis -maxdepth 2 -name types.go | xargs -n1 dirname | sort 
)
)
setglobal listergen_internal_apis = ''(${listergen_internal_apis[@]/#/k8s.io/kubernetes/})
setglobal listergen_internal_apis_csv = $[setglobal IFS = ','; echo $(listergen_internal_apis[*])]
$(listergen) --input-dirs $(listergen_internal_apis_csv) @Argv

setglobal listergen_external_apis = ''(
$(
  cd ${KUBE_ROOT}/staging/src
  # because client-gen doesn't do policy/v1alpha1, we have to skip it too
  find k8s.io/api -name types.go | xargs -n1 dirname | sort | grep -v pkg.apis.policy.v1alpha1
)
)
setglobal listergen_external_apis_csv = $[setglobal IFS = ','; echo $(listergen_external_apis[*])]
$(listergen) --output-base "$(KUBE_ROOT)/vendor" --output-package "k8s.io/client-go/listers" --input-dirs $(listergen_external_apis_csv) @Argv

setglobal informergen_internal_apis = ''(
pkg/api
$(
  cd ${KUBE_ROOT}
  find pkg/apis -maxdepth 2 -name types.go | xargs -n1 dirname | sort 
)
)
setglobal informergen_internal_apis = ''(${informergen_internal_apis[@]/#/k8s.io/kubernetes/})
setglobal informergen_internal_apis_csv = $[setglobal IFS = ','; echo $(informergen_internal_apis[*])]
$(informergen) \
  --input-dirs $(informergen_internal_apis_csv) \
  --internal-clientset-package k8s.io/kubernetes/pkg/client/clientset_generated/internalclientset \
  --listers-package k8s.io/kubernetes/pkg/client/listers \
  @Argv

setglobal informergen_external_apis = ''(
$(
  cd ${KUBE_ROOT}/staging/src
  # because client-gen doesn't do policy/v1alpha1, we have to skip it too
  find k8s.io/api -name types.go | xargs -n1 dirname | sort | grep -v pkg.apis.policy.v1alpha1
)
)

setglobal informergen_external_apis_csv = $[setglobal IFS = ','; echo $(informergen_external_apis[*])]

$(informergen) \
  --output-base "$(KUBE_ROOT)/vendor" \
  --output-package "k8s.io/client-go/informers" \
  --single-directory \
  --input-dirs $(informergen_external_apis_csv) \
  --versioned-clientset-package k8s.io/client-go/kubernetes \
  --listers-package k8s.io/client-go/listers \
  @Argv

# You may add additional calls of code generators like set-gen above.

# call generation on sub-project for now
env CODEGEN_PKG=./vendor/k8s.io/code-generator vendor/k8s.io/code-generator/hack/update-codegen.sh
env CODEGEN_PKG=./vendor/k8s.io/code-generator vendor/k8s.io/kube-aggregator/hack/update-codegen.sh
env CODEGEN_PKG=./vendor/k8s.io/code-generator vendor/k8s.io/sample-apiserver/hack/update-codegen.sh
env CODEGEN_PKG=./vendor/k8s.io/code-generator vendor/k8s.io/apiextensions-apiserver/hack/update-codegen.sh
env CODEGEN_PKG=./vendor/k8s.io/code-generator vendor/k8s.io/metrics/hack/update-codegen.sh