blob: c48663d944e82f484703392fb84fd36db6878f93 [file] [log] [blame]
# Creates a shell where kubectl is hooked up to the selected cluster.
# Record the directory of this file.
REL=$(dirname "$0")
# Check argument count is valid.
if [ $# == 0 ]; then
echo "$0 <cluster> [<optional commmand line to run.>]"
echo ""
echo -n "Valid cluster names: "
cat ${REL}/../kube/clusters/config.json | jq -r ".clusters | keys | @csv"
exit 1
CLUSTER=$1; shift
# What type of cluster are we connecting to?
TYPE=$(cat ${REL}/../kube/clusters/config.json | jq -r ".clusters.\"${CLUSTER}\".type")
if [ "$TYPE" == "null" ]; then
echo "$1 is not a valid cluster name."
exit 1
# Where we will store the config.
# Create dir to store config.
mkdir -p ${DIR}
# Make kubectl use that config.
export KUBECONFIG=${DIR}/config
if [ "${TYPE}" == "gke" ]; then
# Since we've set KUBECONFIG at this point the following commands will
# change that file, not the default one at ~/.kube/config.
PROJECT=$(cat ${REL}/../kube/clusters/config.json | jq -r ".clusters.\"${CLUSTER}\".project")
ZONE=$(cat ${REL}/../kube/clusters/config.json | jq -r ".clusters.\"${CLUSTER}\".zone")
gcloud container clusters get-credentials ${CLUSTER} --zone ${ZONE} --project ${PROJECT}
gcloud config set project ${PROJECT}
else # Type == "k3s".
IP=$(cat ${REL}/../kube/clusters/config.json | jq -r ".clusters.\"${CLUSTER}\".ip")
PORT=$(cat ${REL}/../kube/clusters/config.json | jq -r ".clusters.\"${CLUSTER}\".port")
# Grab config from the kubernetes cluster and store in the config file.
ssh chrome-bot@${IP} "sudo kubectl config view --raw" > ${DIR}/config
# Set up port-forward to the k83 control endpoint and record the PID of the
# background task.
ssh -N -L ${PORT}:localhost:6443 chrome-bot@${IP} &
# Wait until the port is available.
until nc -z localhost ${PORT}
sleep 1
echo "Waiting for port-forward to come up."
# Change the name so we can track which cluster we are talking to.
kubectl config rename-context default ${CLUSTER}
kubectl config set-cluster default --server=${PORT}
echo ${PID} > ${DIR}/pid
# Protect the config file.
chmod 600 ${DIR}/config
if [ $# -ne 0 ] ; then
printf -v command_line '%q ' "$@"
/bin/bash -c "$command_line"
echo "Remember to exit this shell to disconnect from the cluster."
# Clean up on exit.
if [ "${PID}" != "" ]; then
kill ${PID}