From 5ddda189833ede47b7f64e3095c8fbd3f0c8c917 Mon Sep 17 00:00:00 2001
From: aleksana <me@aleksana.moe>
Date: Sun, 9 Mar 2025 19:10:58 +0800
Subject: [PATCH] amber-lang: generate bash completion

---
 pkgs/by-name/am/amber-lang/package.nix | 18 ++++++++++++++----
 1 file changed, 14 insertions(+), 4 deletions(-)

diff --git a/pkgs/by-name/am/amber-lang/package.nix b/pkgs/by-name/am/amber-lang/package.nix
index 45b7e19cc14b..13a669ddd26c 100644
--- a/pkgs/by-name/am/amber-lang/package.nix
+++ b/pkgs/by-name/am/amber-lang/package.nix
@@ -5,6 +5,8 @@
   bc,
   util-linux,
   makeWrapper,
+  installShellFiles,
+  stdenv,
   runCommand,
   amber-lang,
   nix-update-script,
@@ -29,7 +31,10 @@ rustPlatform.buildRustPackage rec {
       --replace-fail 'Command::new("/usr/bin/env")' 'Command::new("env")'
   '';
 
-  nativeBuildInputs = [ makeWrapper ];
+  nativeBuildInputs = [
+    makeWrapper
+    installShellFiles
+  ];
 
   nativeCheckInputs = [
     bc
@@ -43,9 +48,14 @@ rustPlatform.buildRustPackage rec {
     "--skip=tests::formatter::all_exist"
   ];
 
-  postInstall = ''
-    wrapProgram "$out/bin/amber" --prefix PATH : "${lib.makeBinPath [ bc ]}"
-  '';
+  postInstall =
+    ''
+      wrapProgram "$out/bin/amber" --prefix PATH : "${lib.makeBinPath [ bc ]}"
+    ''
+    + lib.optionalString (stdenv.buildPlatform.canExecute stdenv.hostPlatform) ''
+      installShellCompletion --cmd amber \
+        --bash <($out/bin/amber completion)
+    '';
 
   passthru = {
     updateScript = nix-update-script { };