pkgs/sw: Support a nix specific swallow mode

1 files changed, 23 insertions, 6 deletions

diff --git a/pkgs/by-name/sw/sw/sw.sh b/pkgs/by-name/sw/sw/sw.sh
index 68dd6a75..a62bfcbb 100755
--- a/pkgs/by-name/sw/sw/sw.sh
+++ b/pkgs/by-name/sw/sw/sw.sh
@@ -14,15 +14,32 @@ throwup() {
         riverctl focus-previous-tags
 }
 
-if [ -z "$*" ]; then
-    printf "ERROR: No arguments supplied\n"
-elif ! command -v "$1" >/dev/null; then
-    printf "ERROR: Command '%s' does not exist\n" "$1"
-else
+swallow() {
     # Make sure that we actually `throwup`, when the command fails
     set +e
 
     eat && "$@"
-
     throwup
+
+    set -e
+}
+
+if [ -z "$*" ]; then
+    printf "sw: No arguments supplied\n"
+    exit 2
+else
+    case "$1" in
+    "-n" | "--nix")
+        shift 1
+        [ -z "$1" ] && {
+            printf "sw: --nix requires a nix installible\n"
+            exit 2
+        }
+        swallow nix run "n#$1"
+        ;;
+    *)
+        swallow "$@"
+        ;;
+    esac
+
 fi