From b8edb1d257e71e8c2e11a06b9227b6b0a4de5b9c Mon Sep 17 00:00:00 2001
From: Pavan Gudiwada <25551553+pavangudiwada@users.noreply.github.com>
Date: Fri, 16 Aug 2024 04:39:44 +0530
Subject: [PATCH] Adding HolmesGPT plugin (#2822)

* Adding HolmesGPT plugin

* Updated to use a temporary file
---
 plugins/README.md                     |  2 +-
 plugins/ai-incident-investigaton.yaml | 61 +++++++++++++++++++++++++++
 2 files changed, 62 insertions(+), 1 deletion(-)
 create mode 100644 plugins/ai-incident-investigaton.yaml

diff --git a/plugins/README.md b/plugins/README.md
index 4afe2cf166..8d48715471 100644
--- a/plugins/README.md
+++ b/plugins/README.md
@@ -15,6 +15,6 @@ Following is an example of some of plugin files in this directory. Other files a
 | log_stern.yml      | View resource logs using stern                                   | pods                     | Ctrl-l   |                                                                                       |
 | log_jq.yml         | View resource logs using jq                                      | pods                     | Ctrl-j   | kubectl-plugins/kubectl-jq                                                            |
 | log_full.yml       | get full logs from pod/container                                 | pods/containers          | Ctrl-l   |                                                                                       |
-
+| ai-incident-investigation.yaml | Run AI investigation on application issues to find the root cause in seconds | all | Shift-h/o | [HolmesGPT](https://github.com/robusta-dev/holmesgpt) |
 [1]: https://kubernetes.io/docs/tasks/debug/debug-application/debug-running-pod/#ephemeral-container
 [2]: https://github.com/nicolaka/netshoot
diff --git a/plugins/ai-incident-investigaton.yaml b/plugins/ai-incident-investigaton.yaml
new file mode 100644
index 0000000000..eed0a1df30
--- /dev/null
+++ b/plugins/ai-incident-investigaton.yaml
@@ -0,0 +1,61 @@
+plugins:
+# Author: Pavan Gudiwada
+# Investigate incidents in your cluster to quickly find the root cause using HolmesGPT 
+# Requires HolmesGPT to be installed and configured (https://github.com/robusta-dev/holmesgpt) on your system
+# Open any K9s view, then:
+# Shift+H to run an investigation with default ask command
+# Shift+O to customize the question before running an investigation.
+  holmesgpt:
+    shortCut: Shift-H 
+    description: Ask HolmesGPT 
+    scopes:
+      - all 
+    command: bash
+    background: false
+    confirm: false
+    args:
+      - -c
+      - |
+        holmes ask "why is $NAME of $RESOURCE_NAME in -n $NAMESPACE not working as expected"
+        
+        echo "Press 'q' to exit"
+        while : ; do
+        read -n 1 k <&1
+        if [[ $k = q ]] ; then
+        break
+        fi
+        done
+  custom-holmesgpt:
+    shortCut: Shift-Q
+    description: Custom HolmesGPT Ask
+    scopes:
+      - all 
+    command: bash
+    background: false
+    confirm: false
+    args:
+      - -c
+      - |
+        INSTRUCTIONS="# Edit the line below. Lines starting with '#' will be ignored."
+        DEFAULT_ASK_COMMAND="why is $NAME of $RESOURCE_NAME in -n $NAMESPACE not working as expected"
+
+        QUESTION_FILE=$(mktemp)
+
+        echo "$INSTRUCTIONS" > "$QUESTION_FILE"
+        echo "$DEFAULT_ASK_COMMAND" >> "$QUESTION_FILE"
+
+        # Open the line in the default text editor
+        ${EDITOR:-nano} "$QUESTION_FILE"
+
+        # Read the modified line, ignoring lines starting with '#'
+        user_input=$(grep -v '^#' "$QUESTION_FILE")
+
+        echo running: holmes ask "\"$user_input\""
+        holmes ask "$user_input"
+        echo "Press 'q' to exit"
+        while : ; do
+        read -n 1 k <&1
+        if [[ $k = q ]] ; then
+        break
+        fi
+        done
\ No newline at end of file