From 3555936f7712f80b4c556748d993d9de93995d20 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Kalle=20Ahlstr=C3=B6m?= <kalle.ahlstrom@iki.fi>
Date: Thu, 5 Jun 2025 04:11:20 +0300
Subject: [PATCH] dive: add shell completion

Co-authored-by: Kenichi Kamiya <kachick1@gmail.com>
---
 pkgs/by-name/di/dive/package.nix | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/pkgs/by-name/di/dive/package.nix b/pkgs/by-name/di/dive/package.nix
index 46a3a530c126..d051c1771386 100644
--- a/pkgs/by-name/di/dive/package.nix
+++ b/pkgs/by-name/di/dive/package.nix
@@ -4,6 +4,7 @@
   buildGoModule,
   fetchFromGitHub,
   pkg-config,
+  installShellFiles,
   btrfs-progs,
   gpgme,
   lvm2,
@@ -21,7 +22,10 @@ buildGoModule rec {
 
   vendorHash = "sha256-egsFnnHZMPRTJeFw6uByE9OJH06zqKRTvQi9XhegbDI=";
 
-  nativeBuildInputs = [ pkg-config ];
+  nativeBuildInputs = [
+    pkg-config
+    installShellFiles
+  ];
 
   buildInputs = lib.optionals stdenv.hostPlatform.isLinux [
     btrfs-progs
@@ -35,6 +39,13 @@ buildGoModule rec {
     "-X main.version=${version}"
   ];
 
+  postInstall = lib.optionalString (stdenv.buildPlatform.canExecute stdenv.hostPlatform) ''
+    installShellCompletion --cmd dive \
+      --bash <($out/bin/dive completion bash) \
+      --fish <($out/bin/dive completion fish) \
+      --zsh <($out/bin/dive completion zsh)
+  '';
+
   meta = {
     description = "Tool for exploring each layer in a docker image";
     mainProgram = "dive";