From f6e7d2a2a5f573bb47327a2cfabd350862583019 Mon Sep 17 00:00:00 2001
From: Alex Manning <alexander.manning@stfc.ac.uk>
Date: Thu, 25 Jan 2024 14:00:23 +0000
Subject: [PATCH] Add kubectl ssh

---
 home/.local/bin/kubectl-ssh   | 145 ++++++++++++++++++++++++++++++++++
 home/.vimrc.d/dev-environ.vim |   2 +-
 home/.vimrc.d/plugins.vim     |   2 +-
 3 files changed, 147 insertions(+), 2 deletions(-)
 create mode 100755 home/.local/bin/kubectl-ssh

diff --git a/home/.local/bin/kubectl-ssh b/home/.local/bin/kubectl-ssh
new file mode 100755
index 0000000..aefd0e4
--- /dev/null
+++ b/home/.local/bin/kubectl-ssh
@@ -0,0 +1,145 @@
+#!/usr/bin/env bash
+
+set -e
+
+ssh_node() {
+  node=$1
+  shift
+  if [ "$node" = "" ]; then
+    node=$(kubectl get node -o name "$@" | sed 's/node\///' | tr '\n' ' ')
+    node=${node::-1}
+
+    if [[ "$node" =~ " " ]]; then
+      echo "Node name must be specified. Choose one of: [$node]"
+      exit 1
+    else
+      echo "Single-node cluster detected. Defaulting to node $node"
+    fi
+  fi
+
+  if [ $# -eq 1 ]; then
+    command="$1"
+    shift
+  fi
+
+  ns=default
+  pod=$(
+    kubectl create -n "$ns" -o name "$@" -f - <<EOF
+apiVersion: v1
+kind: Pod
+metadata:
+  generateName: ssh-node-
+  labels:
+    plugin: ssh-node
+spec:
+  nodeName: $node
+  containers:
+  - name: ssh-node
+    image: busybox
+    imagePullPolicy: IfNotPresent
+    command: ["chroot", "/host"]
+    tty: true
+    stdin: true
+    stdinOnce: true
+    securityContext:
+      privileged: true
+    volumeMounts:
+    - name: host
+      mountPath: /host
+  volumes:
+  - name: host
+    hostPath:
+      path: /
+  hostNetwork: true
+  hostIPC: true
+  hostPID: true
+  restartPolicy: Never
+  tolerations:
+    - operator: "Exists"
+EOF
+  )
+
+  deletePod() {
+    kubectl delete -n "$ns" "$pod" "$@" --wait=false
+  }
+  trap deletePod EXIT
+
+  echo "Created $pod"
+  echo "Waiting for container to start..."
+  kubectl wait -n "$ns" --for=condition=Ready --timeout 60s "$pod" "$@" >/dev/null
+
+  if [ -z "$command" ]; then
+    kubectl attach -n "$ns" -it "$pod" -c ssh-node "$@"
+  else
+    kubectl exec -n "$ns" -it "$pod" ssh-node -- "$command"
+  fi
+}
+
+ssh_pod() {
+  # TODO: improve this
+  if [ "$1" == "" ]; then
+    echo "Pod name must be specified."
+    exit 1
+  fi
+  # Use sh as a default and switch to bash if it's available
+  # shellcheck disable=SC2016
+  kubectl exec -it "$@" -- sh -c 'exec "$( command -v bash || echo sh )"'
+}
+
+print_usage() {
+  echo "Provider-agnostic way of opening a remote shell to a Kubernetes node."
+  echo
+  echo "Enables you to access a node even when it doesn't run an SSH server or"
+  echo "when you don't have the required credentials. Also, the way you log in"
+  echo "is always the same, regardless of what provides the Kubernetes cluster"
+  echo "(e.g. Minikube, Kind, Docker Desktop, GKE, AKS, EKS, ...)"
+  echo
+  echo "You must have cluster-admin rights to use this plugin."
+  echo
+  echo "The primary focus of this plugin is to provide access to nodes, but it"
+  echo "also provides a quick way of running a shell inside a pod."
+  echo
+  echo "Examples: "
+  echo "  # Open a shell to node of a single-node cluster (e.g. Docker Desktop)"
+  echo "  kubectl ssh node"
+  echo
+  echo "  # Open a shell to node of a multi-node cluster (e.g. GKE)"
+  echo "  kubectl ssh node my-worker-node-1"
+  echo
+  echo "  # Execute the command ls on a node my-worker-node-1"
+  echo "  kubectl ssh node my-worker-node-1 ls"
+  echo
+  echo "  # Open a shell to a pod"
+  echo "  kubectl ssh pod my-pod"
+  echo
+  echo "Usage:"
+  echo "  kubectl ssh node [nodeName [command]]"
+  echo "  kubectl ssh pod [podName] [-n namespace] [-c container]"
+  exit 0
+}
+
+if [ "$1" == "--help" ]; then
+  print_usage
+fi
+
+if [[ "$1" == node/* ]]; then
+  nodeName=${1:5}
+  shift
+  ssh_node "$nodeName" "$@"
+elif [ "$1" == "node" ]; then
+  shift
+  case "$1" in
+  -*) nodeName="";;
+  "") ;;
+  *) nodeName="$1"; shift ;;
+  esac
+
+  ssh_node "$nodeName" "$@"
+elif [[ "$1" == pod/* ]]; then
+  ssh_pod "$@"
+elif [ "$1" == "pod" ]; then
+  shift
+  ssh_pod "$@"
+else
+  print_usage
+fi
diff --git a/home/.vimrc.d/dev-environ.vim b/home/.vimrc.d/dev-environ.vim
index 0faade0..5bb1066 100644
--- a/home/.vimrc.d/dev-environ.vim
+++ b/home/.vimrc.d/dev-environ.vim
@@ -45,7 +45,7 @@ let g:ale_json_jq_options = '--sort-keys'
 
 """Base Settings for COC
 let g:coc_config_home = '~/.alex-environ/home/.vimrc.d/coc'
-let g:coc_global_extensions = ['coc-jedi', 'coc-rls', 'coc-tsserver', 'coc-json', 'coc-htmldjango']
+let g:coc_global_extensions = ['coc-jedi', 'coc-rust-analyzer', 'coc-tsserver', 'coc-json', 'coc-htmldjango']
 """Language Specific Options.
 "Remeber that COC has it's own settings file too, so most of it's settings are
 "there.
diff --git a/home/.vimrc.d/plugins.vim b/home/.vimrc.d/plugins.vim
index 398a050..119d8c6 100644
--- a/home/.vimrc.d/plugins.vim
+++ b/home/.vimrc.d/plugins.vim
@@ -26,7 +26,7 @@ function! LoadPlugins()
 
     Plug 'amanning9/django-plus.vim'                    " Improvements for identifying files in django projects.
     Plug 'towolf/vim-helm'
-    Plug 'Glench/Vim-Jinja2-Syntax'                     " Jinja syntax detection.
+    Plug 'https://gitlab.com/HiPhish/jinja.vim.git'                     " Jinja syntax detection.
 endfunction
 
 " PLUGIN INITIALISATION