From 45c1a1d1e04417cd96685f8314510abb0ceaedd2 Mon Sep 17 00:00:00 2001
From: Will Owens <ghthor@gmail.com>
Date: Sat, 24 May 2025 18:48:30 -0400
Subject: [PATCH] aws-sso-cli: enable postInstall of shell completions

---
 pkgs/by-name/aw/aws-sso-cli/package.nix | 21 ++++++++++++++++-----
 1 file changed, 16 insertions(+), 5 deletions(-)

diff --git a/pkgs/by-name/aw/aws-sso-cli/package.nix b/pkgs/by-name/aw/aws-sso-cli/package.nix
index c46a6230b9da..af8ceb32ed51 100644
--- a/pkgs/by-name/aw/aws-sso-cli/package.nix
+++ b/pkgs/by-name/aw/aws-sso-cli/package.nix
@@ -2,6 +2,7 @@
   buildGoModule,
   fetchFromGitHub,
   getent,
+  installShellFiles,
   lib,
   makeWrapper,
   stdenv,
@@ -19,17 +20,27 @@ buildGoModule rec {
   };
   vendorHash = "sha256-SNMU7qDfLRGUSLjzrJHtIMgbcRc2DxXwWEUaUEY6PME=";
 
-  nativeBuildInputs = [ makeWrapper ];
+  nativeBuildInputs = [
+    makeWrapper
+    installShellFiles
+  ];
 
   ldflags = [
     "-X main.Version=${version}"
     "-X main.Tag=nixpkgs"
   ];
 
-  postInstall = ''
-    wrapProgram $out/bin/aws-sso \
-      --suffix PATH : ${lib.makeBinPath [ xdg-utils ]}
-  '';
+  postInstall =
+    ''
+      wrapProgram $out/bin/aws-sso \
+        --suffix PATH : ${lib.makeBinPath [ xdg-utils ]}
+    ''
+    + lib.optionalString (stdenv.buildPlatform.canExecute stdenv.hostPlatform) ''
+      installShellCompletion --cmd aws-sso \
+        --bash <($out/bin/aws-sso setup completions --source --shell=bash) \
+        --fish <($out/bin/aws-sso setup completions --source --shell=fish) \
+        --zsh <($out/bin/aws-sso setup completions --source --shell=zsh)
+    '';
 
   nativeCheckInputs = [ getent ];