#!/bin/bash
# Kubernetes cluster node
# Version 1.0 - 02/06/2015

# Define master host IP address
KUBE_MASTER="X.X.X.X"

# Set proxy if needed
PROXY=""

# Git repo 
GO_DL="https://storage.googleapis.com/golang/go1.4.2.linux-amd64.tar.gz"
KUBERNETES_GIT="https://github.com/GoogleCloudPlatform/kubernetes.git"
FLANNEL_DL="https://github.com/coreos/flannel/releases/download/v0.4.1/flannel-0.4.1-linux-amd64.tar.gz"

# Suggest not to change the following 
GIT_HOME="/root"
FLANNEL_NETWORK="172.16.0.0/16"
KUBE_ROOT="${GIT_HOME}/kubernetes"
API_HOST=${KUBE_MASTER}
API_PORT="8080"
KUBELET_PORT="10250"
LOG_LEVEL="3"
CHAOS_CHANCE="0.0"
GO_OUT="${KUBE_ROOT}/_output/local/bin/linux/amd64"
LOG_DIR="/var/log/kubernetes"; mkdir -p ${LOG_DIR} 

if [ ! -f ${GIT_HOME}/.kubeinstalled ]; then
        echo "Installing software..."  
        yum -y install git curl gcc docker
	systemctl stop docker.service
	systemctl disable docker.service
        
	# Remove the default docker0 bridge
	ip link set dev docker0 down
	brctl delbr docker0

	# User zzdocker0 as the bridge
	cat > /usr/lib/systemd/system/docker.service << EOF
	[Unit]
	Description=Docker Application Container Engine
	Documentation=http://docs.docker.com
	After=network.target

	[Service]
	Type=notify
	Environment="BRIDGE=zzdocker0"
	EnvironmentFile=-/etc/sysconfig/docker
	EnvironmentFile=-/etc/sysconfig/docker-storage
	EnvironmentFile=-/etc/sysconfig/docker-network
	EnvironmentFile=-/run/flannel/subnet.env
	ExecStart=/usr/bin/docker -d \$OPTIONS \
			\$DOCKER_STORAGE_OPTIONS \
            		\$DOCKER_NETWORK_OPTIONS \
			\$ADD_REGISTRY \
			\$BLOCK_REGISTRY \
			\$INSECURE_REGISTRY \
			--bridge=\${BRIDGE} \
			--mtu=\${FLANNEL_MTU}
	LimitNOFILE=1048576
	LimitNPROC=1048576
	LimitCORE=infinity
	MountFlags=slave

	# set up the bridge
	ExecStartPre=/usr/sbin/brctl addbr \${BRIDGE}
	ExecStartPre=/usr/sbin/ip addr add \${FLANNEL_SUBNET} dev \${BRIDGE}
	ExecStartPre=/usr/sbin/ip link set dev \${BRIDGE} up
	#
	# clean up bridge afterwards
	ExecStopPost=/usr/sbin/ip link set dev \${BRIDGE} down
	ExecStopPost=/usr/sbin/brctl delbr \${BRIDGE}

	[Install]
	WantedBy=multi-user.target
EOF

	systemctl daemon-reload

        # Configure proxy if there is one
	if [ ! -z $PROXY ]; then
	        echo "Adding proxy"
	        X="-x $PROXY"
	        git config --global http.proxy "$PROXY"
	        if [[ ! $(grep ^HTTP_PROXY /etc/sysconfig/docker) ]]; then
			echo HTTP_PROXY="$PROXY" >> /etc/sysconfig/docker
		fi
	fi

	# Install dependencies 
        cd $GIT_HOME
	rm -rf go.tar.gz flannel.tar.gz flannel kubernetes
        curl $X -L $GO_DL -o go.tar.gz; tar -C /usr/local -xzf go.tar.gz
        curl $X -L $FLANNEL_DL -o flannel.tar.gz; tar -xzf flannel.tar.gz; mv flannel-0.4.1 flannel
        git clone $KUBERNETES_GIT kubernetes
 
 	# Update PATH
	sed -i "/kubernetes/d" /root/.bash_profile
        sed -i "$ i PATH=$PATH:/usr/local/go/bin:${GIT_HOME}/kubernetes/cluster:${GIT_HOME}/kubernetes/hack:${GIT_HOME}/flannel" /root/.bash_profile

	# Add to startup
	if [[ ! $(grep kube /etc/rc.local) ]]; then echo "${GIT_HOME}/kube-node.sh > /tmp/kube-node.log 2>&1"; fi
	chmod a+x /etc/rc.local
	systemctl start rc-local
fi

# Terminate the runing processes if any
pkill kubelet
pkill kube-proxy
pkill docker
pkill flanneld

# Initialize 
echo "Initializing..."
cd "${KUBE_ROOT}"
source /root/.bash_profile 
source "${KUBE_ROOT}/hack/lib/init.sh"
"${KUBE_ROOT}/hack/build-go.sh"

# Start flanneld
echo "Starting flanneld..."
${GIT_HOME}/flannel/flanneld -etcd-endpoints="http://${API_HOST}:4001" -etcd-prefix="/coreos.com/network" -iface="ens32" > ${LOG_DIR}/flanneld.log 2>&1 &

# Waiting flanned to finish
while [ ! -f /run/flannel/subnet.env ]; do sleep 1; echo "waiting flaneld to be ready..."; done

# Start docker
echo "Starting docker..."
systemctl restart docker.service

# Start Kubelet
echo "Starting kubelet..."
"${GO_OUT}/kubelet" \
  --v=${LOG_LEVEL} \
  --chaos_chance="${CHAOS_CHANCE}" \
  --address="0.0.0.0" \
  --api_servers="${API_HOST}:${API_PORT}" \
  --port="$KUBELET_PORT" >"${LOG_DIR}/kubelet.log" 2>&1 &
echo kubelet pid is $!

# Start Kubelet-proxy
echo "Starting kubelet-proxy..."
"${GO_OUT}/kube-proxy" \
  --v=${LOG_LEVEL} \
  --master="http://${API_HOST}:${API_PORT}" >"${LOG_DIR}/kube-proxy.log" 2>&1 &
echo kube-proxy pid is $!

date > ${GIT_HOME}/.kubeinstalled