zsh auto completion

completion.zsh.hbs

@@ -0,0 +1,17 @@
+###-begin-{{app_name}}-completions-###
+#
+# yargs command completion script
+#
+# Installation: {{app_path}} {{completion_command}} >> ~/.zshrc
+#    or {{app_path}} {{completion_command}} >> ~/.zsh_profile on OSX.
+#
+_{{app_name}}_yargs_completions()
+{
+  local reply
+  local si=$IFS
+  IFS=$'\n' reply=($(COMP_CWORD="$((CURRENT-1))" COMP_LINE="$BUFFER" COMP_POINT="$CURSOR" {{app_path}} --get-yargs-completions "$\{words[@]\}"))
+  IFS=$si
+  _describe 'values' reply
+}
+compdef _{{app_name}}_yargs_completions {{app_name}}
+###-end-{{app_name}}-completions-###

lib/completion.js

@@ -79,8 +79,10 @@ module.exports = function completion (yargs, usage, command) {
 
   // generate the completion script to add to your .bashrc.
   self.generateCompletionScript = function generateCompletionScript ($0, cmd) {
+    let zshShell = process.env.SHELL && process.env.SHELL.indexOf('zsh') !== -1
+
     let script = fs.readFileSync(
-      path.resolve(__dirname, '../completion.sh.hbs'),
+      path.resolve(__dirname, zshShell ? '../completion.zsh.hbs' : '../completion.sh.hbs'),
       'utf-8'
     )
     const name = path.basename($0)

test/completion.js

@@ -196,10 +196,21 @@ describe('Completion', () => {
   })
 
   describe('generateCompletionScript()', () => {
-    it('replaces application variable with $0 in script', () => {
+    it('replaces application variable with $0 in script (bash)', () => {
+      process.env.SHELL = '/bin/bash'
       const r = checkUsage(() => yargs([])
         .showCompletionScript(), ['ndm'])
 
+      r.logs[0].should.match(/bashrc/)
+      r.logs[0].should.match(/ndm --get-yargs-completions/)
+    })
+
+    it('replaces application variable with $0 in script (zsh)', () => {
+      process.env.SHELL = '/bin/zsh'
+      const r = checkUsage(() => yargs([])
+        .showCompletionScript(), ['ndm'])
+
+      r.logs[0].should.match(/zshrc/)
       r.logs[0].should.match(/ndm --get-yargs-completions/)
     })