@waron97/prbot 1.0.2 → 1.1.0

package/index.js

@@ -7,7 +7,9 @@ import path from "path";
 import { execFile } from "child_process";
 import { program } from "commander";
 
-configDotenv({ path: path.join(process.env.HOME, ".prbot") });
+if (process.argv[2] !== "completion") {
+  configDotenv({ path: path.join(process.env.HOME, ".prbot") });
+}
 
 async function getToken() {
   const url = process.env.KC_URL;
@@ -252,6 +254,75 @@ program
       });
   });
 
+program
+  .command("completion [shell]")
+  .description("Print shell completion script")
+  .action((shell = "bash") => {
+    if (shell === "bash") {
+      process.stdout.write(`
+_prbot_completion() {
+    local cur prev
+    COMPREPLY=()
+    cur="\${COMP_WORDS[COMP_CWORD]}"
+    prev="\${COMP_WORDS[COMP_CWORD-1]}"
+
+    if [[ "\$prev" == "-m" || "\$prev" == "--module" ]]; then
+        local addons_path
+        addons_path=\$(grep -E '^ADDONS_PATH=' ~/.prbot 2>/dev/null | head -1 | cut -d= -f2- | sed "s|~|\$HOME|g")
+        if [[ -n "\$addons_path" && -d "\$addons_path/config" ]]; then
+            local modules
+            modules=\$(ls -1 "\$addons_path/config" 2>/dev/null)
+            COMPREPLY=(\$(compgen -W "\$modules" -- "\$cur"))
+        fi
+        return 0
+    fi
+
+    COMPREPLY=(\$(compgen -W "pr ver completion" -- "\$cur"))
+}
+complete -F _prbot_completion prbot
+`);
+    } else if (shell === "zsh") {
+      process.stdout.write(`
+#compdef prbot
+
+_prbot() {
+    local state
+    _arguments \\
+        '1: :->command' \\
+        '*: :->args'
+
+    case \$state in
+        command)
+            _values 'command' 'pr' 'ver' 'completion'
+            ;;
+        args)
+            case \${words[2]} in
+                pr|ver)
+                    case \${words[CURRENT-1]} in
+                        -m|--module)
+                            local addons_path
+                            addons_path=\$(grep -E '^ADDONS_PATH=' ~/.prbot 2>/dev/null | head -1 | cut -d= -f2- | sed "s|~|\$HOME|g")
+                            if [[ -n "\$addons_path" && -d "\$addons_path/config" ]]; then
+                                local -a modules
+                                modules=(\$(ls -1 "\$addons_path/config" 2>/dev/null))
+                                _describe 'module' modules
+                            fi
+                            ;;
+                    esac
+                    ;;
+            esac
+            ;;
+    esac
+}
+
+_prbot
+`);
+    } else {
+      console.error(`Unsupported shell: ${shell}. Use 'bash' or 'zsh'.`);
+      process.exit(1);
+    }
+  });
+
 program
   .command("ver")
   .option("-m, --module <module>")

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@waron97/prbot",
-  "version": "1.0.2",
+  "version": "1.1.0",
   "description": "",
   "main": "index.js",
   "scripts": {