threatest: install shell completions

NixOS · May 19, 2023 · f4e1815 · f4e1815
1 parent 5b14e87
commit f4e1815
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/pkgs/tools/security/threatest/default.nix b/pkgs/tools/security/threatest/default.nix
@@ -1,6 +1,7 @@
 { lib
 , buildGoModule
 , fetchFromGitHub
+, installShellFiles
 }:
 
 buildGoModule rec {
@@ -17,6 +18,17 @@ buildGoModule rec {
   proxyVendor = true;
   vendorHash = "sha256-UQ3GPSv7P4+oMvcu4eFlosnw0TQxG33ptlMTQA/5Lkw=";
 
+  nativeBuildInputs = [
+    installShellFiles
+  ];
+
+  postInstall = ''
+    installShellCompletion --cmd threatest \
+      --bash <($out/bin/threatest completion bash) \
+      --fish <($out/bin/threatest completion fish) \
+      --zsh <($out/bin/threatest completion zsh)
+  '';
+
   meta = with lib; {
     description = "Framework for end-to-end testing threat detection rules";
     homepage = "https://github.com/DataDog/threatest";