@vexify-org/yaggs 5.3.0 → 6.0.0

package/index.js

@@ -105,6 +105,10 @@ class Yaggs {
         return this;
     }
 
+    generateCompletion(shell = 'bash') {
+        return this.parser.generateCompletion(shell);
+    }
+
     check(fn) {
         this.middleware(fn);
         return this;

package/lib/parser.js

@@ -536,6 +536,144 @@ class ArgumentParser {
         return null;
     }
 
+    generateCompletion(shell = 'bash') {
+        if (shell === 'bash') {
+            return this._generateBashCompletion();
+        } else if (shell === 'zsh') {
+            return this._generateZshCompletion();
+        } else if (shell === 'fish') {
+            return this._generateFishCompletion();
+        }
+        return '';
+    }
+
+    _generateBashCompletion() {
+        let completion = `#!/usr/bin/env bash
+
+_${this.scriptName}_completions() {
+    local cur prev opts
+    COMPREPLY=()
+    cur="\${COMP_WORDS[COMP_CWORD]}"
+    prev="\${COMP_WORDS[COMP_CWORD-1]}"
+    
+    opts=""
+    
+    OPTIONS="`;
+        
+        for (const [name, opt] of Object.entries(this.options)) {
+            if (!opt.hidden) {
+                completion += `--${name} `;
+                if (opt.alias && opt.alias.length > 0) {
+                    const aliases = Array.isArray(opt.alias) ? opt.alias : [opt.alias];
+                    for (const alias of aliases) {
+                        completion += `-${alias} `;
+                    }
+                }
+            }
+        }
+        
+        completion += `"
+    
+    COMMANDS="`;
+    
+        for (const [name, cmd] of Object.entries(this.commands)) {
+            if (!cmd.hidden) {
+                completion += `${name} `;
+                if (cmd.aliases) {
+                    completion += `${cmd.aliases.join(' ')} `;
+                }
+            }
+        }
+        
+        completion += `"
+    
+    if [[ "\${cur}" == -* ]]; then
+        COMPREPLY=( \$(compgen -W "\${OPTIONS}" -- "\${cur}") )
+    else
+        COMPREPLY=( \$(compgen -W "\${COMMANDS}" -- "\${cur}") )
+    fi
+    
+    return 0
+}
+
+complete -F _${this.scriptName}_completions ${this.scriptName}
+`;
+        
+        return completion;
+    }
+
+    _generateZshCompletion() {
+        let completion = `#compdef ${this.scriptName}
+
+local commands options
+
+commands=('`;
+        
+        for (const [name, cmd] of Object.entries(this.commands)) {
+            if (!cmd.hidden) {
+                completion += `${name}:${cmd.description} `;
+            }
+        }
+        
+        completion += `')
+
+options=('`;
+        
+        for (const [name, opt] of Object.entries(this.options)) {
+            if (!opt.hidden) {
+                completion += `--${name}[${opt.description}]:`;
+                if (opt.choices) {
+                    completion += ':(' + opt.choices.join(' ') + ') ';
+                } else {
+                    completion += ' ';
+                }
+                if (opt.alias && opt.alias.length > 0) {
+                    const aliases = Array.isArray(opt.alias) ? opt.alias : [opt.alias];
+                    for (const alias of aliases) {
+                        completion += `-${alias}[${opt.description}]: `;
+                    }
+                }
+            }
+        }
+        
+        completion += `')
+
+_arguments \${options[@]} \\
+    '1:command:\${commands}' \\
+    '*::args:_normal'
+`;
+        
+        return completion;
+    }
+
+    _generateFishCompletion() {
+        let completion = `#!/usr/bin/env fish
+
+complete -c ${this.scriptName} `;
+        
+        for (const [name, opt] of Object.entries(this.options)) {
+            if (!opt.hidden) {
+                completion += `-l ${name} -d "${opt.description}" `;
+                if (opt.alias && opt.alias.length > 0) {
+                    const aliases = Array.isArray(opt.alias) ? opt.alias : [opt.alias];
+                    for (const alias of aliases) {
+                        completion += `-s ${alias} `;
+                    }
+                }
+            }
+        }
+        
+        completion += '\n\n';
+        
+        for (const [name, cmd] of Object.entries(this.commands)) {
+            if (!cmd.hidden) {
+                completion += `complete -c ${this.scriptName} -a "${name}" -d "${cmd.description}"\n`;
+            }
+        }
+        
+        return completion;
+    }
+
     getHelp() {
         let help = `\u001b[36m${this.scriptName}\u001b[0m`;
         

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@vexify-org/yaggs",
-    "version": "5.3.0",
+    "version": "6.0.0",
     "description": "A powerful CLI argument parser, better than yargs",
     "main": "index.js",
     "scripts": {