From a22485d8f117c82b300eee81fef64b858bf73bb5 Mon Sep 17 00:00:00 2001
From: Daniel Bevenius <daniel.bevenius@gmail.com>
Date: Mon, 14 May 2018 13:09:27 +0200
Subject: [PATCH] tools: add bash completion for node

This commit adds a --completion-bash option to node which can be
sourced to provide bash code completion for node options.

Usage:
$ node --completion-bash  > node_bash_completion
$ source node_bash_completion
$ node --[tab]

PR-URL: https://github.com/nodejs/node/pull/20713
Reviewed-By: Anna Henningsen <anna@addaleax.net>
Reviewed-By: James M Snell <jasnell@gmail.com>
Reviewed-By: Denys Otrishko <shishugi@gmail.com>
---
 doc/api/cli.md                  | 11 +++++++++++
 doc/node.1                      |  3 +++
 lib/internal/bash_completion.js | 25 +++++++++++++++++++++++++
 lib/internal/bootstrap/node.js  |  8 +++++++-
 node.gyp                        |  1 +
 src/node_options.cc             |  3 +++
 src/node_options.h              |  1 +
 7 files changed, 51 insertions(+), 1 deletion(-)
 create mode 100644 lib/internal/bash_completion.js

diff --git a/doc/api/cli.md b/doc/api/cli.md
index 8adb33efb7b06e..3f5441b82e7ce8 100644
--- a/doc/api/cli.md
+++ b/doc/api/cli.md
@@ -52,6 +52,17 @@ If this flag is passed, the behavior can still be set to not abort through
 [`process.setUncaughtExceptionCaptureCallback()`][] (and through usage of the
 `domain` module that uses it).
 
+### `--completion-bash`
+<!-- YAML
+added: REPLACEME
+-->
+
+Print source-able bash completion script for Node.js.
+```console
+$ node --completion-bash > node_bash_completion
+$ source node_bash_completion
+```
+
 ### `--enable-fips`
 <!-- YAML
 added: v6.0.0
diff --git a/doc/node.1 b/doc/node.1
index 8e6616ad969ecc..ad46d58d1f39fd 100644
--- a/doc/node.1
+++ b/doc/node.1
@@ -75,6 +75,9 @@ the next argument will be used as a script filename.
 .It Fl -abort-on-uncaught-exception
 Aborting instead of exiting causes a core file to be generated for analysis.
 .
+.It Fl -completion-bash
+Print source-able bash completion script for Node.js.
+.
 .It Fl -enable-fips
 Enable FIPS-compliant crypto at startup.
 Requires Node.js to be built with
diff --git a/lib/internal/bash_completion.js b/lib/internal/bash_completion.js
new file mode 100644
index 00000000000000..7d2ffbe901d285
--- /dev/null
+++ b/lib/internal/bash_completion.js
@@ -0,0 +1,25 @@
+'use strict';
+const { internalBinding } = require('internal/bootstrap/loaders');
+const { getOptions } = internalBinding('options');
+
+function print(stream) {
+  const { options, aliases } = getOptions();
+  const all_opts = [...options.keys(), ...aliases.keys()];
+
+  stream.write(`_node_complete() {
+  local cur_word options
+  cur_word="\${COMP_WORDS[COMP_CWORD]}"
+  if [[ "\${cur_word}" == -* ]] ; then
+    COMPREPLY=( $(compgen -W '${all_opts.join(' ')}' -- "\${cur_word}") )
+    return 0
+  else
+    COMPREPLY=( $(compgen -f "\${cur_word}") )
+    return 0
+  fi
+}
+complete -F _node_complete node node_g`);
+}
+
+module.exports = {
+  print
+};
diff --git a/lib/internal/bootstrap/node.js b/lib/internal/bootstrap/node.js
index daa7fd5cbc0b79..223930f6983cd8 100644
--- a/lib/internal/bootstrap/node.js
+++ b/lib/internal/bootstrap/node.js
@@ -120,11 +120,17 @@
       NativeModule.require('internal/inspector_async_hook').setup();
     }
 
-    if (internalBinding('options').getOptions('--help')) {
+    const options = internalBinding('options');
+    if (options.getOptions('--help')) {
       NativeModule.require('internal/print_help').print(process.stdout);
       return;
     }
 
+    if (options.getOptions('--completion-bash')) {
+      NativeModule.require('internal/bash_completion').print(process.stdout);
+      return;
+    }
+
     if (isMainThread) {
       mainThreadSetup.setupChildProcessIpcChannel();
     }
diff --git a/node.gyp b/node.gyp
index 3e05d8e36b8926..910f93206d323d 100644
--- a/node.gyp
+++ b/node.gyp
@@ -85,6 +85,7 @@
       'lib/zlib.js',
       'lib/internal/assert.js',
       'lib/internal/async_hooks.js',
+      'lib/internal/bash_completion.js',
       'lib/internal/buffer.js',
       'lib/internal/cli_table.js',
       'lib/internal/child_process.js',
diff --git a/src/node_options.cc b/src/node_options.cc
index 46f529cff2e961..06cce7cf77ccba 100644
--- a/src/node_options.cc
+++ b/src/node_options.cc
@@ -223,6 +223,9 @@ PerProcessOptionsParser::PerProcessOptionsParser() {
             kAllowedInEnvironment);
 
   AddOption("--security-reverts", "", &PerProcessOptions::security_reverts);
+  AddOption("--completion-bash",
+            "print source-able bash completion script",
+            &PerProcessOptions::print_bash_completion);
   AddOption("--help",
             "print node command line options",
             &PerProcessOptions::print_help);
diff --git a/src/node_options.h b/src/node_options.h
index 71615cf0839aa4..4b9ee000367405 100644
--- a/src/node_options.h
+++ b/src/node_options.h
@@ -112,6 +112,7 @@ class PerProcessOptions {
   bool zero_fill_all_buffers = false;
 
   std::vector<std::string> security_reverts;
+  bool print_bash_completion = false;
   bool print_help = false;
   bool print_v8_help = false;
   bool print_version = false;