From 64101f5ffbcf87463a2fcf6fef20a407c9eef4d8 Mon Sep 17 00:00:00 2001
From: Nicola Squartini <tensor5@gmail.com>
Date: Fri, 6 Dec 2024 06:31:10 +0100
Subject: [PATCH] kopia: add shell completion scripts (#360216)

---
 pkgs/by-name/ko/kopia/package.nix | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/pkgs/by-name/ko/kopia/package.nix b/pkgs/by-name/ko/kopia/package.nix
index f718fb7a15fa..b530dff0065c 100644
--- a/pkgs/by-name/ko/kopia/package.nix
+++ b/pkgs/by-name/ko/kopia/package.nix
@@ -1,4 +1,4 @@
-{ lib, buildGoModule, fetchFromGitHub, gitUpdater, testers, kopia }:
+{ lib, buildGoModule, fetchFromGitHub, gitUpdater, installShellFiles, stdenv, testers, kopia }:
 
 buildGoModule rec {
   pname = "kopia";
@@ -20,6 +20,14 @@ buildGoModule rec {
     "-X github.com/kopia/kopia/repo.BuildInfo=${src.rev}"
   ];
 
+  nativeBuildInputs = [ installShellFiles ];
+
+  postInstall = lib.optionalString (stdenv.buildPlatform.canExecute stdenv.hostPlatform) ''
+    installShellCompletion --cmd kopia \
+      --bash <($out/bin/kopia --completion-script-bash) \
+      --zsh <($out/bin/kopia --completion-script-zsh)
+  '';
+
   passthru = {
     updateScript = gitUpdater { rev-prefix = "v"; };
     tests = {