From 9331708da6f6596e1e702a98c43dd7685156683a Mon Sep 17 00:00:00 2001
From: Janik H <janik@aq0.de>
Date: Thu, 27 Jul 2023 14:36:34 +0200
Subject: [PATCH] containerlab: add shell completions

---
 pkgs/tools/networking/containerlab/default.nix | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/pkgs/tools/networking/containerlab/default.nix b/pkgs/tools/networking/containerlab/default.nix
index b90ee096c4cbc..d6e78b2cfb792 100644
--- a/pkgs/tools/networking/containerlab/default.nix
+++ b/pkgs/tools/networking/containerlab/default.nix
@@ -1,6 +1,7 @@
 { lib
 , buildGoModule
 , fetchFromGitHub
+, installShellFiles
 }:
 
 buildGoModule rec {
@@ -14,6 +15,8 @@ buildGoModule rec {
     hash = "sha256-Wo6Dfcdax8EjZRP5JBKrnxpFTVx0GV+7uaDgBY/zg6I=";
   };
 
+  nativeBuildInputs = [ installShellFiles ];
+
   vendorHash = "sha256-uaYeFzG74VTUrtql6mSi+LNh6IjDomixrWwGQMgmONc=";
 
   ldflags = [
@@ -24,6 +27,14 @@ buildGoModule rec {
     "-X" "github.com/srl-labs/containerlab/cmd.date=1970-01-01T00:00:00Z"
   ];
 
+  postInstall = ''
+    local INSTALL="$out/bin/containerlab"
+    installShellCompletion --cmd containerlab \
+      --bash <($out/bin/containerlab completion bash) \
+      --fish <($out/bin/containerlab completion fish) \
+      --zsh <($out/bin/containerlab completion zsh)
+  '';
+
   meta = with lib; {
     description = "Container-based networking lab";
     homepage = "https://containerlab.dev/";