From f92227a6791c78f212f153679f54aad5e3f18e71 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Matej=20Jane=C5=BEi=C4=8D?= <janezic.mj@gmail.com>
Date: Mon, 2 Mar 2026 14:50:43 +0100
Subject: [PATCH] chore: generalize build.sh

---
 build.rs | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/build.rs b/build.rs
index cb4c17a..c5cd678 100644
--- a/build.rs
+++ b/build.rs
@@ -5,15 +5,17 @@ use std::{env, io::Error};
 include!("src/cli/mod.rs");
 
 fn main() -> Result<(), Error> {
-    let outdir = match env::var_os("OUT_DIR") {
-        None => return Ok(()),
-        Some(outdir) => outdir,
+    let outdir = env::var_os("SHELL_COMPLETIONS_DIR")
+        .or_else(|| env::var_os("OUT_DIR"));
+
+    let Some(outdir) = outdir else {
+        return Ok(());
     };
 
     let mut cmd = ahab::Ahab::command();
 
     for shell in Shell::value_variants() {
-        let path = generate_to(*shell, &mut cmd, "ahab", &outdir)?;
+        let path = generate_to(*shell, &mut cmd, env!("CARGO_PKG_NAME"), &outdir)?;
 
         println!("cargo:warning=completion file is generated: {path:?}");
     }