#!/usr/bin/env bash
#  vim:ts=4:sts=4:sw=4:et
#
#  Author: Hari Sekhon
#  Date: 2020-08-28 11:06:23 +0100 (Fri, 28 Aug 2020)
#
#  https://github.com/HariSekhon/DevOps-Bash-tools
#
#  License: see accompanying Hari Sekhon LICENSE file
#
#  If you're using my code you're welcome to connect with me on LinkedIn and optionally send me feedback to help steer this or other code I publish
#
#  https://www.linkedin.com/in/HariSekhon
#

set -euo pipefail
[ -n "${DEBUG:-}" ] && set -x
srcdir="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"

# shellcheck disable=SC1090
. "$srcdir/lib/utils.sh"

# shellcheck disable=SC2034,SC2154
usage_description="
Kubectl exec's to the first pod matching the given filter criteria

Useful to not have to first look up deployment pod names which have random string suffixes
and quickly jump in to any pod in a namespace/deployment to debug a web farm etc.

Shows the full auto-generated 'kubectl exec' command for clarity

Execs to bash if available, otherwise /bin/sh

First arg is the optional pod container name (if no container is specified we'll pick the first one and show you in the kubectl output)
Subsequent args from the first dash are passed straight to 'kubectl get pods' to set namespace, label filters etc.

Examples:

    ${0##*/} -n prod -l app=nginx

    ${0##*/} sidecar-container -n prod -l app=nginx


See also:

    kubectl_exec.sh

for a different approaching using just a partial pod name and optional partial container name, auto-determines the namespace, it's simpler and less typing in most cases
"

# used by usage() in lib/utils.sh
# shellcheck disable=SC2034
usage_args="[<container_name>] [<pod_filters>]"

help_usage "$@"

min_args 1 "$@"

container=""

if [ $# -gt 0 ]; then
    if ! [[ "$1" =~ ^- ]]; then
        container="$1"
        shift
    fi
fi

kubectl cluster-info &>/dev/null || die "Failed to connect to Kubernetes cluster!"

pod="$(kubectl get pods "$@" -o 'jsonpath={.items[0].metadata.name}' 2>/dev/null || :)"

if [ -z "$pod" ]; then
    die "No matching pods found, perhaps you forgot to pass the --namespace? (tip: specify -A / --all-namespaces if lazy, it'll filter by pod name)"
fi

# auto-determine namespace because specifying it is annoying
namespace="$(kubectl get pods --all-namespaces | grep -E "^[^[:space:]]+[[:space:]]+[^[:space:]]*${pod}" | awk '{print $1; exit}' || :)"

if [ -z "$container" ]; then
    # auto-determine first container to show explicitly what we're connecting to
    container="$(kubectl get pods -n "$namespace" "$pod" -o 'jsonpath={.spec.containers[*].name}' | grep -m 1 "." | awk '{print $1}' || :)"
fi

cmd="kubectl exec -ti --namespace \"$namespace\" \"$pod\" --container \"$container\" -- /bin/sh -c 'if type bash >/dev/null 2>&1; then exec bash; else exec sh; fi'"
echo "$cmd"
eval "$cmd"