#!/bin/bash

install_tools() {
sudo apt install -y curl unzip

if [ -f "awscliv2.zip" ]; then rm -rf awscliv2.zip aws; fi
curl "https://awscli.amazonaws.com/awscli-exe-linux-x86_64.zip" -o "awscliv2.zip"
unzip awscliv2.zip
sudo ./aws/install

curl -Lo kops https://github.com/kubernetes/kops/releases/download/$(curl -s https://api.github.com/repos/kubernetes/kops/releases/latest | grep tag_name | cut -d '"' -f 4)/kops-linux-amd64
chmod +x ./kops
sudo mv ./kops /usr/local/bin/

curl -Lo kubectl https://storage.googleapis.com/kubernetes-release/release/$(curl -s https://storage.googleapis.com/kubernetes-release/release/stable.txt)/bin/linux/amd64/kubectl
chmod +x ./kubectl
sudo mv ./kubectl /usr/local/bin/kubectl

if [ -f "gethelm.sh" ]; then rm gethelm.sh; fi
curl -fsSL -o get_helm.sh https://raw.githubusercontent.com/helm/helm/main/scripts/get-helm-3
chmod 700 get_helm.sh
./get_helm.sh

# aws --version
# kops --version
# kubectl --version
# helm version
}

setup_aws_cli() {
  echo -e "Make sure to get the following documents:\n
  Account ID
  Username
  Password
  Access Key
  Secret Access Key"

  export PATH=/usr/local/bin/:$PATH
  source ~/.bash_profile || source ~/.bashrc|| source ~/.profile
  complete -C '/usr/local/bin/aws_completer' aws

  aws configure
}

create_k8s_cluster() {
echo State S3 bucket name:
read bucketname
aws s3api create-bucket --bucket "$bucketname" --region us-east-1
export KOPS_STATE_STORE=s3://"$bucketname"

echo Assign a cluster name:
read clustername
export NAME="$clustername".k8s.local

ssh-keygen
kops create secret sshpublickey admin -i cluster.$NAME --name $NAME --state $KOPS_STATE_STORE


REGION=`sed -n 'x;$p' ~/.aws/config | cut -d" " -f3`

export ZONES=$(aws ec2 describe-availability-zones --region $REGION | grep ZoneName | awk '{print $2}' | head -n1 | tr -d '"' | sed 's/.$//')

kops create cluster \
--name "$NAME" \
--zones "$ZONES" \
--state "$KOPS_STATE_STORE" \
--yes

kops validate cluster --wait 15m

kubectl config set-context $(kubectl config current-context) --namespace $NAME

cat << EOF > storageclass.yml
kind: StorageClass
apiVersion: storage.k8s.io/v1
metadata:
  annotations:
    storageclass.beta.kubernetes.io/is-default-class: "true"
  name: gp2
provisioner: kubernetes.io/aws-ebs
parameters:
  type: gp2
  encrypted: "true"
EOF
kubectl apply -f storageclass.yml
}

install_jupyterhub() {
cat <<EOF >config.yaml
# This file can update the JupyterHub Helm chart's default configuration values.
#
# For reference see the configuration reference and default values, but make
# sure to refer to the Helm chart version of interest to you!
#
# Introduction to YAML:     https://www.youtube.com/watch?v=cdLNKUoMc6c
# Chart config reference:   https://zero-to-jupyterhub.readthedocs.io/en/stable/resources/reference.html
# Chart default values:     https://github.com/jupyterhub/zero-to-jupyterhub-k8s/blob/HEAD/jupyterhub/values.yaml
# Available chart versions: https://jupyterhub.github.io/helm-chart/
#
EOF

helm repo add jupyterhub https://jupyterhub.github.io/helm-chart/
helm repo update
helm search repo jupyterhub

# kubectl get storageclass
# kubectl patch storageclass gp2 -p '{"metadata": {"annotations":{"storageclass.kubernetes.io/is-default-class":"false"}}}'

echo "State release name:"
read releasename

echo "State namespace:"
read namespace
export NAMESPACE="$namespace"

helm upgrade --cleanup-on-fail \
  --install "$releasename" jupyterhub/jupyterhub \
  --namespace $NAMESPACE \
  --create-namespace \
  --version=2.0.0 \
  --values config.yaml

echo "Awaiting External-IP...Takes 2-5 minutes"
time until kubectl --namespace $NAMESPACE get service proxy-public --output json | jq .spec.ports[0].nodePort; do sleep 45; done
kubectl get pod --namespace $NAMESPACE
kubectl --namespace $NAMESPACE get service proxy-public #--output jsonpath='{.status.loadBalancer.ingress[].ip}'
}

install_tools
setup_aws_cli
create_k8s_cluster
install_jupyterhub


echo "All Done!"