feat: port-forward pod (#59)

Co-authored-by: Delice0 <[email protected]>
Co-authored-by: Ramilito <[email protected]>

Author: idelice

ftplugin/k8s_pods.lua

@@ -1,4 +1,5 @@
 local api = vim.api
+local actions = require("kubectl.actions.actions")
 local commands = require("kubectl.actions.commands")
 local container_view = require("kubectl.views.containers")
 local deployment_view = require("kubectl.views.deployments")
@@ -17,6 +18,7 @@ api.nvim_buf_set_keymap(0, "n", "g?", "", {
     hints = hints .. tables.generateHintLine("<d>", "Describe selected pod \n")
     hints = hints .. tables.generateHintLine("<t>", "Show resources used \n")
     hints = hints .. tables.generateHintLine("<enter>", "Opens container view \n")
+    hints = hints .. tables.generateHintLine("<shift-f>", "Port forward \n")
     hints = hints .. tables.generateHintLine("<C-k>", "Kill pod \n")
     view.Hints(hints)
   end,
@@ -104,6 +106,46 @@ api.nvim_buf_set_keymap(0, "n", "<C-k>", "", {
   end,
 })
 
+api.nvim_buf_set_keymap(0, "n", "<S-f>", "", {
+  noremap = true,
+  silent = true,
+  callback = function()
+    local namespace, pod_name = tables.getCurrentSelection(unpack(col_indices))
+    if pod_name and namespace then
+      local current_port_query = "get pod " .. pod_name .. ' -o jsonpath="{.spec.containers[*].ports[*].containerPort}"'
+
+      local current_port_result = commands.execute_shell_command("kubectl", current_port_query)
+
+      vim.ui.input({ prompt = "Port forward " .. current_port_result }, function(input)
+        if input ~= nil then
+          actions.confirmation_buffer(
+            "Are you sure that you want to port forward to " .. input .. "?",
+            nil,
+            function(confirm)
+              if confirm then
+                local port_forward_query = "port-forward pods/"
+                  .. pod_name
+                  .. " "
+                  .. input
+                  .. ":"
+                  .. current_port_result
+
+                commands.shell_command_async("kubectl", port_forward_query, function(response)
+                  vim.schedule(function()
+                    vim.notify(response)
+                  end)
+                end)
+              end
+            end
+          )
+        end
+      end)
+    else
+      api.nvim_err_writeln("Failed to select pod.")
+    end
+  end,
+})
+
 if not loop.is_running() then
   loop.start_loop(pod_view.Pods)
 end

lua/kubectl/views/pods/init.lua

@@ -15,6 +15,7 @@ function M.Pods(cancellationToken)
           { key = "<d>", desc = "describe" },
           { key = "<t>", desc = "top" },
           { key = "<enter>", desc = "containers" },
+          { key = "<shift-f>", desc = "port forward" },
           { key = "<C-k>", desc = "kill pod" },
         }, true, true)
         :display("k8s_pods", "Pods", cancellationToken)