From 1817cd5d1b1bf3d201577dae4174669e1358558f Mon Sep 17 00:00:00 2001
From: ryota2357 <noreply@ryota2357.com>
Date: Thu, 25 Dec 2025 13:50:54 +0900
Subject: [PATCH] pkl: add shell completion

---
 pkgs/by-name/pk/pkl/package.nix | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/pkgs/by-name/pk/pkl/package.nix b/pkgs/by-name/pk/pkl/package.nix
index 9846881b4a85..af961ce14aef 100644
--- a/pkgs/by-name/pk/pkl/package.nix
+++ b/pkgs/by-name/pk/pkl/package.nix
@@ -8,6 +8,7 @@
   nix-update-script,
   replaceVars,
   makeWrapper,
+  installShellFiles,
 }:
 let
   jdk = temurin-bin-21;
@@ -43,6 +44,7 @@ stdenv.mkDerivation (finalAttrs: {
     jdk
     kotlinOverlay
     makeWrapper
+    installShellFiles
   ];
 
   mitmCache = gradle.fetchDeps {
@@ -81,6 +83,13 @@ stdenv.mkDerivation (finalAttrs: {
 
     makeWrapper ${lib.getExe jdk} $out/bin/pkl --add-flags "-jar $out/opt/pkl/jpkl.jar"
 
+    ${lib.optionalString (stdenv.buildPlatform.canExecute stdenv.hostPlatform) ''
+      installShellCompletion --cmd pkl \
+        --bash <($out/bin/pkl shell-completion bash) \
+        --zsh <($out/bin/pkl shell-completion zsh) \
+        --fish <($out/bin/pkl shell-completion fish)
+    ''}
+
     runHook postInstall
   '';