From 0dd1faf8e12a7a89ea00d2414cb37ab1437e0ef8 Mon Sep 17 00:00:00 2001
From: Haseeb Majid <hello@haseebmajid.dev>
Date: Tue, 8 Jul 2025 19:32:02 +0100
Subject: [PATCH] kaf: add shell completions

---
 pkgs/by-name/ka/kaf/package.nix | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/pkgs/by-name/ka/kaf/package.nix b/pkgs/by-name/ka/kaf/package.nix
index 34a542b8ac3a..65fec0f2e434 100644
--- a/pkgs/by-name/ka/kaf/package.nix
+++ b/pkgs/by-name/ka/kaf/package.nix
@@ -2,6 +2,8 @@
   lib,
   buildGoModule,
   fetchFromGitHub,
+  installShellFiles,
+  stdenv,
 }:
 
 buildGoModule rec {
@@ -17,9 +19,18 @@ buildGoModule rec {
 
   vendorHash = "sha256-1QcQeeYQFsStK27NVdyCAb1Y40lyifBf0dlSgzocG3Y=";
 
+  nativeBuildInputs = [ installShellFiles ];
+
   # Many tests require a running Kafka instance
   doCheck = false;
 
+  postInstall = lib.optionalString (stdenv.buildPlatform.canExecute stdenv.hostPlatform) ''
+    installShellCompletion --cmd kaf \
+      --bash <($out/bin/kaf completion bash) \
+      --zsh <($out/bin/kaf completion zsh) \
+      --fish <($out/bin/kaf completion fish)
+  '';
+
   meta = with lib; {
     description = "Modern CLI for Apache Kafka, written in Go";
     mainProgram = "kaf";