#!/bin/bash

NEW_NAMESPACE="$1"
CONTEXT=$(kubectl config current-context)
if [ -z "$NEW_NAMESPACE" ]; then
  NAMESPACES=( $(kubectl get namespace -o name | awk -F "/" '{ print $2 }' | sort) )
  PS3="$(tput bold)$(tput setaf 6)Select namespace: $(tput sgr0)"
  select CHOICE in "${NAMESPACES[@]}"; do
    for NS in "${NAMESPACES[@]}"; do
      if [[ $NS == $CHOICE ]]; then
        NEW_NAMESPACE=$CHOICE
        break 2
      fi
    done
  done
fi
kubectl config set-context $CONTEXT --namespace=$NEW_NAMESPACE
echo "Set namespace to $NEW_NAMESPACE"