From d0b7ad46cd7400160a5d7e909b932ec73b8bb7e1 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Torma=20Krist=C3=B3f?= <tormakristof@tormakristof.eu>
Date: Wed, 24 Apr 2019 13:30:12 +0200
Subject: [PATCH] Update kubernetes_install.sh

---
 deploy/kubernetes_install.sh | 74 +++++++++++++++++++++++++++---------
 1 file changed, 56 insertions(+), 18 deletions(-)

diff --git a/deploy/kubernetes_install.sh b/deploy/kubernetes_install.sh
index 911be3d..5a85159 100644
--- a/deploy/kubernetes_install.sh
+++ b/deploy/kubernetes_install.sh
@@ -1,11 +1,24 @@
 #!/bin/bash
 
-DOCKER_VERSION=$1
-KUBERNETES_VERSION=$2
-CLIENT=$3
-IP=$4
-TOKEN=$5
-HASH=$6
+# Setting all parameters
+NODE_TYPE=$1
+INTERNAL=!$2
+MASTER_IP=$3
+
+## Parameters for master node installation
+if [ "$NODE_TYPE" == "master" ]
+then
+	if [ "$#" -lt 4 ]; then
+		POD_NETWORK_ARG=""
+	else
+		POD_NETWORK_ARG="--pod-network-cidr=$4"
+	fi
+# Parameters for worker node installation
+elif [ "$NODE_TYPE" == "worker" ]
+then
+	TOKEN=$4
+	HASH=$5
+fi
 
 #Installing Docker
 DOCKER_INSTALLED=$(which docker)
@@ -17,7 +30,7 @@ then
 	curl -fsSL https://download.docker.com/linux/ubuntu/gpg | apt-key add -
 	add-apt-repository "deb [arch=amd64] https://download.docker.com/linux/ubuntu $(lsb_release -cs) stable"
 	apt-get update
-	apt-get install -y docker-ce=$DOCKER_VERSION
+	apt-get install -y docker-ce
 fi
 
 
@@ -30,26 +43,51 @@ then
 	chmod 666 /etc/apt/sources.list.d/kubernetes.list
 	echo "deb http://apt.kubernetes.io/ kubernetes-xenial main" > /etc/apt/sources.list.d/kubernetes.list
 	apt-get update
-	apt-get install -y kubelet=$KUBERNETES_VERSION kubeadm=$KUBERNETES_VERSION kubectl=$KUBERNETES_VERSION kubernetes-cni=$KUBERNETES_VERSION
+	apt-get install -y kubelet kubeadm kubectl kubernetes-cni
 fi
 
 #Disabling swap for Kubernetes
 sysctl net.bridge.bridge-nf-call-iptables=1 > /dev/null
 swapoff -a
 
-if [ -z "$CLIENT" ]
+# Initialize Kubernetes as Master node
+if [ "$NODE_TYPE" == "master" ]
 then
-#	kubeadm init --ignore-preflight-errors=SystemVerification
-#	mkdir -p $HOME/.kube
-#	cp -i /etc/kubernetes/admin.conf $HOME/.kube/config
-#	chown $(id -u):$(id -g) $HOME/.kube/config
-	:
+	## Set master node for internal network
+	if [ $INTERNAL ]; then
+		touch /etc/default/kubelet
+		echo "KUBELET_EXTRA_ARGS=--node-ip=$MASTER_IP" > /etc/default/kubelet
+	fi
+	## Init Kubernetes
+	kubeadm init --ignore-preflight-errors=SystemVerification \
+		--apiserver-advertise-address=$MASTER_IP $POD_NETWORK_ARG
+	mkdir -p $HOME/.kube
+	cp -i /etc/kubernetes/admin.conf $HOME/.kube/config
+	chown $(id -u):$(id -g) $HOME/.kube/config
 
-elif [ "$CLIENT" = "true" ]
+	echo "[master:$(hostname -s)] Node is up and running on $MASTER_IP"
+
+# Initialize Kubernetes as Worker node
+elif [ "$NODE_TYPE" = "worker" ]
 then
-	kubeadm join $IP --token $TOKEN --discovery-token-ca-cert-hash $HASH
-	echo "Client ($IP) joined to Master"
+	## Set worker node for internal network
+	if [ $INTERNAL ]; then
+		IP=$(grep -oP \
+				'(?<=src )[^ ]*' \
+				<(grep  -f <(ls -l /sys/class/net | grep pci | awk '{print $9}') \
+					<(ip ro sh) |
+				grep -v $(ip ro sh | grep default | awk '{print $5}')) |
+			head -1)
+		touch /etc/default/kubelet
+		echo "KUBELET_EXTRA_ARGS=--node-ip=$IP" > /etc/default/kubelet
+	else
+		IP=$(grep -oP '(?<=src )[^ ]*' <(ip ro sh | grep default))
+	fi
+	## Join to Kubernetes Master node
+	kubeadm join $MASTER_IP --token $TOKEN --discovery-token-ca-cert-hash $HASH \
+		--ignore-preflight-errors=SystemVerification
+
+	echo "[worker:$(hostname -s)] Client ($IP) joined to Master ($MASTER_IP)"
 else
 	echo "Invalid argument"
 fi
-