.20_kubernetes.bash

__kube_completions() {
    # This lazy-loads completions for common Kube commands. This is because a number of them are
    # minorly slow. Completions won't be available until the first time the command is invoked.

    # NOTE: `autoload -Uz compinit && compinit` will need to happen before sourcing any of these!

    if command -v helm &> /dev/null; then
        helm() {
            unset -f "$0"
            # shellcheck disable=SC1090
            source <(helm completion "$(basename "${SHELL}")")
            $0 "$@"
        }
    fi

    if command -v kops &> /dev/null; then
        kops() {
            unset -f "$0"
            # shellcheck disable=SC1090
            source <(kops completion "$(basename "${SHELL}")")
            $0 "$@"
        }
    fi

    if command -v kubectl &> /dev/null; then
        kubectl() {
            unset -f "$0"
            # shellcheck disable=SC1090
            source <(kubectl completion "$(basename "${SHELL}")")
            $0 "$@"
        }
    fi

    if command -v minikube &> /dev/null; then
        minikube() {
            unset -f "$0"
            # shellcheck disable=SC1090
            source <(minikube completion "$(basename "${SHELL}")")
            $0 "$@"
        }
    fi

    if command -v stern &> /dev/null; then
        stern() {
            unset -f "$0"
            # shellcheck disable=SC1090
            source <(stern --completion "$(basename "${SHELL}")")
            $0 "$@"
        }
    fi
}
__kube_completions


__kube_minikube() {
    kminikube() {
        minikube update-context || kubectl config use-context minikube
    }
    alias kmini=kminikube
}
__kube_minikube


__kube_funcs() {
    # Execute `bash` interactively in the Kubernetes pod
    # @param {string} $1 App label
    kbash() {
        kubectl exec --stdin --tty "$(kpod "$1" | head -1)" -- bash
    }

    # List all Kubernetes config maps, optionally filtering to an application
    # @param {string=} $1 App label
    # @param {...string} Additional kubectl options
    kconfigmaps() {
        kubectl get configmaps ${1:+--selector="app=$1"} "${@:2}"
    }

    # List all Kubernetes container names, optionally filtering to an application
    # @param {string=} $1 App label
    # @param {...string} Additional kubectl options
    kcontainers() {
        kubectl get pods --output=jsonpath="{.items[*].spec.containers[*].name}" ${1:+--selector="app=$1"} "${@:2}" | tr -s '[[:space:]]' '\n' | sort -u
    }

    # Change to a kubectl context
    # @param {string} $1 Context name
    # @param {...string} Additional kubectl options
    alias kcontext="kubectl config use-context"

    # List all Kubernetes cron jobs, optionally filtering to an application
    # @param {string=} $1 App label
    # @param {...string} Additional kubectl options
    kcrons() {
        kubectl get cronjob ${1:+--selector="app=$1"} "${@:2}"
    }

    # List all Kubernetes daemon sets
    # @param {...string} Additional kubectl options
    alias kdaemonsets="kubectl get daemonsets"
    alias kds=kdaemonsets

    # List all Kubernetes container names given a deployment name
    # @param {string} $1 Deployment name
    # @param {...string} Additional kubectl options
    kdcontainers() {
        kubectl get deployment --output=jsonpath="{.spec.template.spec.containers[*].name}" "$1" "${@:2}" | tr -s '[[:space:]]' '\n' | sort -u
    }

    # Delete a Kubernetes pod
    # @param {string} $1 Pod name
    # @param {...string} Additional kubectl options
    kdel() {
        kubectl delete pod "$1" "${@:2}"
    }

    # List all Kubernetes deployment names, optionally filtering to an application
    # @param {string=} $1 App label
    # @param {...string} Additional kubectl options
    kdeployments() {
        kubectl get deployments --output=jsonpath="{.items[*].metadata.name}" ${1:+--selector="app=$1"} "${@:2}" | tr -s '[[:space:]]' '\n' | sort -u
    }

    # Open a port forward session to a remote Kubernetes deployment
    # @param {string} $1 Deployment name
    # @param {number} $2 Local+remote OR local port number
    # @param {number=} $3 Remote port number
    # @param {...string} Additional kubectl options
    kforward() {
        kubectl port-forward "deployment/$1" "$2${3:+:$3}" "${@:4}"
    }

    # Show the Kubernetes rollout history for a deployment
    # @param {string} $1 Deployment name
    # @param {number=} $2 Revision number
    # @param {...string} Additional kubectl options
    khistory() {
        kubectl rollout history "deployment/$1" ${2:+--revision "$2"} "${@:3}"
    }

    # List all Kubernetes container image names, optionally filtering to an application
    # @param {string=} $1 App label
    # @param {...string} Additional kubectl options
    kimages() {
        kubectl get pods --output=jsonpath="{.items[*].spec.containers[*].image}" ${1:+--selector="app=$1"} "${@:2}" | tr -s '[[:space:]]' '\n' | sort -u
    }

    # List all Kubernetes ingresses, optionally filtering to an application
    # @param {string=} $1 App label
    # @param {...string} Additional kubectl options
    kingress() {
        kubectl get ingress ${1:+--selector="app=$1"} "${@:2}"
    }

    # List all Kubernetes jobs, optionally filtering to an application
    # @param {string=} $1 App label
    # @param {...string} Additional kubectl options
    kjobs() {
        kubectl get jobs --label-columns="app" ${1:+--selector="app=$1"} "${@:2}"
    }

    # Follow the logs from all Kubernetes containers with a given app label
    # @param {string} $1 App label
    # @param {number=} $2 Tail length
    # @param {...string} Additional kubectl options
    klogs() {
        if command -v stern &> /dev/null; then
            stern --timestamps --tail "${2:-0}" --selector "app=$1"
        else
            kubectl logs --all-containers --timestamps --follow --max-log-requests=9999 "--tail=${2:-0}" --selector="app=$1"
        fi
    }

    # List all Kubernetes namespaces
    # @param {...string} Additional kubectl options
    alias knamespaces="kubectl get namespaces"
    alias kns=knamespaces

    # List all Kubernetes nodes
    # @param {...string} Additional kubectl options
    alias knodes="kubectl get nodes --label-columns='nodegroup-name'"

    # Get the pod name of the newest running Kubernetes containers given an app label
    # @param {string} $1 App label
    # @param {...string} Additional kubectl options
    kpod() {
        kubectl get pods --selector="app=$1" --field-selector=status.phase=Running --sort-by=".metadata.creationTimestamp" --show-labels "${@:2}" | tail -n +2 | sed 's/[ ][ ]*/ /g' | sort -u -k6,6 | awk '{print $1}'
    }

    # List all Kubernetes pods, optionally filtering to an application
    # @param {string=} $1 App label
    # @param {...string} Additional kubectl options
    kpods() {
        kubectl get pods --label-columns="app" ${1:+--selector="app=$1"} "${@:2}"
    }

    # Reboot all Kubernetes deployments with a given app label
    # @param {string} $1 App label
    # @param {...string} Additional kubectl options
    kreboot() {
        if [[ -z "$1" ]]; then
            return 1
        fi
        for DEPLOYMENT in $(kdeployments "$1"); do
            kubectl rollout restart "deployment/${DEPLOYMENT}" "${@:2}"
        done
    }

    # Get the most recent revision number for a Kubernetes deployment
    # @param {string} $1 Deployment name
    # @param {...string} Additional kubectl options
    krevision() {
        khistory "$1" "${@:2}" | grep -E '^[0-9]\+' | sort --sort=numeric --reverse | head -1 | awk '{print $1}'
    }

    # Roll back a Kubernetes deployment
    # @param {string} $1 Deployment name
    # @param {number=} $1 Revision number
    # @param {...string} Additional kubectl options
    krollback() {
        kubectl rollout undo "deployment/$1" ${2:+--to-revision=$2} "${@:3}"
    }

    # List all Kubernetes replica sets, optionally filtering to an application
    # @param {string=} $1 App label
    # @param {...string} Additional kubectl options
    kreplicasets() {
        kubectl get rs ${1:+--selector="app=$1"} "${@:2}"
    }
    alias krs=kreplicasets

    # List all Kubernetes secrets, optionally filtering to an application
    # @param {string=} $1 App label
    # @param {...string} Additional kubectl options
    ksecrets() {
        kubectl get secrets ${1:+--selector="app=$1"} "${@:2}"
    }

    # Describe a Kubernetes service to get info such as labels, IP, and load balancer ingress
    # @param {string=} $1 App label
    # @param {...string} Additional kubectl options
    kservices() {
        kubectl get services ${1:+--selector="app=$1"} "${@:2}"
    }

    # Execute `sh` interactively in the Kubernetes pod
    # @param {string} $1 App label
    ksh() {
        kubectl exec --stdin --tty "$(kpod "$1" | head -1)" -- sh
    }

    # List all Kubernetes stateful sets
    # @param {...string} Additional kubectl options
    alias kstatefulsets="kubectl get statefulsets"
    alias kss=kstatefulsets

    # Emulate a `top` command for Kubernetes pods, optionally filtering to an application
    # @param {string=} $1 App label
    # @param {...string} Additional kubectl options
    ktop() {
        watch "kubectl top pods --containers ${1:+--selector="app=$1"} \"${*:2}\" | awk 'NR == 1; NR > 1 {print \$0 | \"sort -n -k3 -r\"}'"
    }
}
__kube_funcs