fix: detect zsh when zsh isnt run as a login prompt (yargs#1395)

moimikey · Aug 20, 2019 · 8792d13 · 8792d13
1 parent 28e600c
commit 8792d13
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lib/completion.js b/lib/completion.js
@@ -8,7 +8,8 @@ module.exports = function completion (yargs, usage, command) {
     completionKey: 'get-yargs-completions'
   }
 
-  const zshShell = process.env.SHELL && process.env.SHELL.indexOf('zsh') !== -1
+  const zshShell = (process.env.SHELL && process.env.SHELL.indexOf('zsh') !== -1) ||
+    (process.env.ZSH_NAME && process.env.ZSH_NAME.indexOf('zsh') !== -1)
   // get a list of completion commands.
   // 'args' is the array of strings from the line to be completed
   self.getCompletion = function getCompletion (args, done) {