Fix completion build.rs

1 files changed, 10 insertions, 6 deletions

diff --git a/build.rs b/build.rs
index b2522708..8a7c4900 100644
--- a/build.rs
+++ b/build.rs
@@ -3,6 +3,7 @@ extern crate clap;
 #[macro_use]
 extern crate lazy_static;
 
+use std::env;
 use std::fs;
 
 use clap::Shell;
@@ -12,12 +13,15 @@ use clap::Shell;
 mod app;
 
 fn main() {
-    fs::create_dir_all(env!("OUT_DIR")).unwrap();
+    let outdir = match env::var_os("OUT_DIR") {
+        None => return,
+        Some(outdir) => outdir,
+    };
+    fs::create_dir_all(&outdir).unwrap();
 
     let mut app = app::app_short();
-    app.gen_completions("rg", Shell::Bash, env!("OUT_DIR"));
-    app.gen_completions("rg", Shell::Fish, env!("OUT_DIR"));
-    // Zsh seems to fail with a panic.
-    // app.gen_completions("rg", Shell::Zsh, env!("OUT_DIR"));
-    app.gen_completions("rg", Shell::PowerShell, env!("OUT_DIR"));
+    app.gen_completions("rg", Shell::Bash, &outdir);
+    app.gen_completions("rg", Shell::Fish, &outdir);
+    app.gen_completions("rg", Shell::Zsh, &outdir);
+    app.gen_completions("rg", Shell::PowerShell, &outdir);
 }