#!/bin/bash

set -o nounset
set -o errexit
set -o pipefail
#set -o xtrace

BASE=$(cd `dirname $0`; cd ../../; pwd)

gpusplit=${1:-4}
memsplit=${2:-1}

shift 2

param=$@

echo "[Step 1]: Load and push hami related images"
docker load -i $BASE/down/hami.tar
docker tag registry.cn-hangzhou.aliyuncs.com/google_containers/kube-scheduler:v1.28.1 easzlab.io.local:5000/google_containers/kube-scheduler:v1.28.1
docker push easzlab.io.local:5000/google_containers/kube-scheduler:v1.28.1
docker tag hami:v2.4.2 easzlab.io.local:5000/hami:v2.4.2
docker push easzlab.io.local:5000/hami:v2.4.2
docker tag liangjw/kube-webhook-certgen:v1.1.1 easzlab.io.local:5000/liangjw/kube-webhook-certgen:v1.1.1
docker push easzlab.io.local:5000/liangjw/kube-webhook-certgen:v1.1.1
docker tag jettech/kube-webhook-certgen:v1.5.2 easzlab.io.local:5000/jettech/kube-webhook-certgen:v1.5.2
docker push easzlab.io.local:5000/jettech/kube-webhook-certgen:v1.5.2

echo "[Step 2]: Install hami resource"
helm install --debug hami $BASE/hami \
	     --set devicePlugin.deviceSplitCount=$gpusplit \
	     --set devicePlugin.deviceMemoryScaling=$memsplit $param -n kube-system
