From d16b8262c9709d6e09f826316ba4b72e258df98a Mon Sep 17 00:00:00 2001
From: Lukas Cech <lcech@veracode.com>
Date: Mon, 3 Feb 2025 13:09:50 +0100
Subject: [PATCH] Fix klogs

---
 home-manager/common.nix | 32 +++++++++++++++++---------------
 1 file changed, 17 insertions(+), 15 deletions(-)

diff --git a/home-manager/common.nix b/home-manager/common.nix
index f53fe3b..9e13ae7 100644
--- a/home-manager/common.nix
+++ b/home-manager/common.nix
@@ -138,24 +138,26 @@ in
         git-sync-remote = lib.mkDefault "git remote update origin --prune";
 
         klogs = lib.mkDefault "${pkgs.writeShellScript "klogs" ''
-            ctx="$1"
-            namespace="$2"
-            label="$3"
-            echo=${pkgs.coreutils}/bin/echo
+          ctx="$1"
+          shift
+          namespace="$1"
+          shift
+          label="$1"
+          shift
 
-            if [[ "$ctx" == "" || "$namespace" == "" || "$label" == "" ]]; then
-                    echo "Usage: klogs context namespace label"
-                    echo "${"\n"}Contexts:"
-                    kubectl config get-contexts -o name | sed 's/^/\t/g'
+          if [[ "$ctx" == "" || "$namespace" == "" || "$label" == "" ]]; then
+                  echo "Usage: klogs context namespace label"
+                  echo "${"\n"}Contexts:"
+                  kubectl config get-contexts -o name | sed 's/^/\t/g'
 
-                    echo "Label examples:"
-                    echo "${"\t"}app.kubernetes.io/name=..."
-                    echo "${"\t"}eks.amazonaws.com/component=..."
-                    exit 1
-            fi
+                  echo "Label examples:"
+                  echo "${"\t"}app.kubernetes.io/name=..."
+                  echo "${"\t"}eks.amazonaws.com/component=..."
+                  exit 1
+          fi
 
-            kubectl --context "$ctx" logs -f -n "$namespace" -l "$label"
-          ''}";
+          kubectl --context "$ctx" logs -f -n "$namespace" -l "$label" $@
+        ''}";
 
         nixfix = lib.mkDefault "nix fmt ./**/*.nix";