feat: add completion command

Author: Zamoca42

README.md

@@ -120,55 +120,32 @@ you can set the configuration variable `skipUpdateWhenPageNotFound` to `true` (d
 
 ## Command-line Autocompletion
 
-Currently we only support command-line autocompletion for zsh
-and bash. Pull requests for other shells are most welcome!
+We currently support command-line autocompletion for zsh and bash.  
+Pull requests for other shells are most welcome!
 
-### zsh
-
-It's easiest for
-[oh-my-zsh](https://github.com/robbyrussell/oh-my-zsh)
-users, so let's start with that.
-
-```zsh
-mkdir -p $ZSH_CUSTOM/plugins/tldr
-ln -s bin/completion/zsh/_tldr $ZSH_CUSTOM/plugins/tldr/_tldr
-```
-
-Then add tldr to your oh-my-zsh plugins,
-usually defined in `~/.zshrc`,
-resulting in something looking like this:
-
-```zsh
-plugins=(git tmux tldr)
-```
+To enable autocompletion for the tldr command, run:
 
-Alternatively, using [zplug](https://github.com/zplug/zplug)
+### zsh
 
 ```zsh
-zplug "tldr-pages/tldr-node-client", use:bin/completion/zsh
+tldr completion zsh
+source ~/.zshrc
 ```
 
-Fret not regular zsh user!
-Copy or symlink `bin/completion/zsh/_tldr` to
-`my/completions/_tldr`
-(note the filename).
-Then add the containing directory to your fpath:
+### bash
 
-```zsh
-fpath=(my/completions $fpath)
+```bash
+tldr completion bash
+source ~/.bashrc
 ```
 
-### Bash
+This command will generate the appropriate completion script and append it to your shell's configuration file (`.zshrc` or `.bashrc`).
 
-```bash
-ln -s bin/completion/bash/tldr ~/.tldr-completion.bash
-```
+After running the completion installation command, restart your shell or reload your configuration file to enable the autocompletion.
 
-Now add the following line to our bashrc file:
+You should now have autocompletion enabled for the tldr command.
 
-```bash
-source ~/.tldr-completion.bash
-```
+If you encounter any issues or need more information about the autocompletion setup, please refer to the [completion.js](https://github.com/tldr-pages/tldr-node-client/blob/master/lib/completion.js) file in the repository.
 
 ## FAQ
 

bin/completion/bash/tldr

@@ -1,5 +1,17 @@
 #!/bin/bash
 
+# tldr bash completion
+
+# Check if bash-completion is already sourced
+if ! type _completion_loader &>/dev/null; then
+  # If not, try to load it
+  if [ -f /usr/share/bash-completion/bash_completion ]; then
+    . /usr/share/bash-completion/bash_completion
+  elif [ -f /etc/bash_completion ]; then
+    . /etc/bash_completion
+  fi
+fi
+
 BUILTIN_THEMES="single base16 ocean"
 
 PLATFORM_TYPES="android freebsd linux netbsd openbsd osx sunos windows"

bin/tldr

@@ -5,10 +5,33 @@ const pkg = require('../package');
 const Tldr = require('../lib/tldr');
 const config = require('../lib/config');
 const platforms = require('../lib/platforms');
+const Completion = require('../lib/completion');
 const { TldrError } = require('../lib/errors');
 
 pkg.version = `v${pkg.version}\nClient Specification: 2.0`;
 
+program
+  .command('completion <shell>')
+  .description('Output shell completion script')
+  .action((shell) => {
+    const completion = new Completion(shell);
+    completion.getScript()
+      .then((script) => {
+        return completion.appendScript(script);
+      })
+      .then(() => {
+        if (shell === 'zsh') {
+          console.log('If completions don\'t work, you may need to rebuild your zcompdump:');
+          console.log('  rm -f ~/.zcompdump; compinit');
+        }
+        process.exit(0);
+      })
+      .catch((err) => {
+        console.error(err.message);
+        process.exit(err.code || 1);
+      });
+  });
+
 program
   .version(pkg.version, '-v, --version', 'Display version')
   .helpOption('-h, --help', 'Show this help message')
@@ -104,7 +127,7 @@ if (program.list) {
 } else if (program.search) {
   program.args.unshift(program.search);
   p = tldr.search(program.args);
-} else if (program.args.length >= 1) {
+} else if (program.args.length >= 1 && program.args[0] !== 'completion') {
   p = tldr.get(program.args, program);
 }
 

lib/completion.js

@@ -0,0 +1,82 @@
+'use strict';
+
+const fs = require('fs');
+const path = require('path');
+const os = require('os');
+const { UnsupportedShellError, CompletionScriptError } = require('./errors');
+
+class Completion {
+  constructor(shell) {
+    this.supportedShells = ['bash', 'zsh'];
+    if (!this.supportedShells.includes(shell)) {
+      throw new UnsupportedShellError(shell, this.supportedShells);
+    }
+    this.shell = shell;
+    this.rcFilename = shell === 'zsh' ? '.zshrc' : '.bashrc';
+  }
+
+  getFilePath() {
+    const homeDir = os.homedir();
+    return path.join(homeDir, this.rcFilename);
+  }
+
+  appendScript(script) {
+    const rcFilePath = this.getFilePath();
+    return new Promise((resolve, reject) => {
+      fs.appendFile(rcFilePath, `\n${script}\n`, (err) => {
+        if (err) {
+          reject((new CompletionScriptError(`Error appending to ${rcFilePath}: ${err.message}`)));
+        } else {
+          console.log(`Completion script added to ${rcFilePath}`);
+          console.log(`Please restart your shell or run 'source ~/${this.rcFilename}' to enable completions`);
+          resolve();
+        }
+      });
+    });
+  }
+
+  getScript() {
+    return new Promise((resolve) => {
+      if (this.shell === 'zsh') {
+        resolve(this.getZshScript());
+      } else if (this.shell === 'bash') {
+        resolve(this.getBashScript());
+      }
+    });
+  }
+
+  getZshScript() {
+    const completionDir = path.join(__dirname, '..', 'bin', 'completion', 'zsh');
+    return `
+# tldr zsh completion
+fpath=(${completionDir} $fpath)
+
+# You might need to force rebuild zcompdump:
+# rm -f ~/.zcompdump; compinit
+
+# If you're using oh-my-zsh, you can force reload of completions:
+# autoload -U compinit && compinit
+
+# Check if compinit is already loaded, if not, load it
+if (( ! $+functions[compinit] )); then
+  autoload -Uz compinit
+  compinit -C
+fi
+    `.trim();
+  }
+
+  getBashScript() {
+    return new Promise((resolve, reject) => {
+      const scriptPath = path.join(__dirname, '..', 'bin', 'completion', 'bash', 'tldr');
+      fs.readFile(scriptPath, 'utf8', (err, data) => {
+        if (err) {
+          reject(new CompletionScriptError(`Error reading bash completion script: ${err.message}`));
+        } else {
+          resolve(data);
+        }
+      });
+    });
+  }
+}
+
+module.exports = Completion;

lib/errors.js

@@ -44,11 +44,31 @@ class MissingRenderPathError extends TldrError {
   }
 }
 
+class UnsupportedShellError extends TldrError {
+  constructor(shell, supportedShells) {
+    super(`Unsupported shell: ${shell}. Supported shells are: ${supportedShells.join(', ')}`);
+    this.name = 'UnsupportedShellError';
+    // eslint-disable-next-line no-magic-numbers
+    this.code = 5;
+  }
+}
+
+class CompletionScriptError extends TldrError {
+  constructor(message) {
+    super(message);
+    this.name = 'CompletionScriptError';
+    // eslint-disable-next-line no-magic-numbers
+    this.code = 6;
+  }
+}
+
 module.exports = {
   TldrError,
   EmptyCacheError,
   MissingPageError,
-  MissingRenderPathError
+  MissingRenderPathError,
+  UnsupportedShellError,
+  CompletionScriptError
 };
 
 function trim(strings, ...values) {

test/completion.spec.js

@@ -0,0 +1,139 @@
+'use strict';
+
+const Completion = require('../lib/completion');
+const { UnsupportedShellError, CompletionScriptError } = require('../lib/errors');
+const sinon = require('sinon');
+const fs = require('fs');
+const os = require('os');
+const should = require('should');
+
+describe('Completion', () => {
+  describe('constructor()', () => {
+    it('should construct with supported shell', () => {
+      const completion = new Completion('zsh');
+      should.exist(completion);
+      completion.shell.should.equal('zsh');
+      completion.rcFilename.should.equal('.zshrc');
+    });
+
+    it('should throw UnsupportedShellError for unsupported shell', () => {
+      (() => {return new Completion('fish');}).should.throw(UnsupportedShellError);
+    });
+  });
+
+  describe('getFilePath()', () => {
+    let homeStub;
+
+    beforeEach(() => {
+      homeStub = sinon.stub(os, 'homedir').returns('/home/user');
+    });
+
+    afterEach(() => {
+      homeStub.restore();
+    });
+
+    it('should return .zshrc path for zsh', () => {
+      const completion = new Completion('zsh');
+      completion.getFilePath().should.equal('/home/user/.zshrc');
+    });
+
+    it('should return .bashrc path for bash', () => {
+      const completion = new Completion('bash');
+      completion.getFilePath().should.equal('/home/user/.bashrc');
+    });
+  });
+
+  describe('appendScript()', () => {
+    let appendFileStub;
+    let getFilePathStub;
+
+    beforeEach(() => {
+      appendFileStub = sinon.stub(fs, 'appendFile').yields(null);
+      getFilePathStub = sinon.stub(Completion.prototype, 'getFilePath').returns('/home/user/.zshrc');
+    });
+
+    afterEach(() => {
+      appendFileStub.restore();
+      getFilePathStub.restore();
+    });
+
+    it('should append script to file', () => {
+      const completion = new Completion('zsh');
+      return completion.appendScript('test script')
+        .then(() => {
+          appendFileStub.calledOnce.should.be.true();
+          appendFileStub.firstCall.args[0].should.equal('/home/user/.zshrc');
+          appendFileStub.firstCall.args[1].should.equal('\ntest script\n');
+        });
+    });
+
+    it('should reject with CompletionScriptError on fs error', () => {
+      const completion = new Completion('zsh');
+      appendFileStub.yields(new Error('File write error'));
+      return completion.appendScript('test script')
+        .should.be.rejectedWith(CompletionScriptError);
+    });
+  });
+
+  describe('getScript()', () => {
+    it('should return zsh script for zsh shell', () => {
+      const completion = new Completion('zsh');
+      return completion.getScript()
+        .then((script) => {
+          script.should.containEql('# tldr zsh completion');
+          script.should.containEql('fpath=(');
+        });
+    });
+
+    it('should return bash script for bash shell', () => {
+      const completion = new Completion('bash');
+      const readFileStub = sinon.stub(fs, 'readFile').yields(null, '# bash completion script');
+
+      return completion.getScript()
+        .then((script) => {
+          script.should.equal('# bash completion script');
+          readFileStub.restore();
+        });
+    });
+  });
+
+  describe('getZshScript()', () => {
+    it('should return zsh completion script', () => {
+      const completion = new Completion('zsh');
+      const script = completion.getZshScript();
+      script.should.containEql('# tldr zsh completion');
+      script.should.containEql('fpath=(');
+      script.should.containEql('compinit');
+    });
+  });
+
+  describe('getBashScript()', () => {
+    let readFileStub;
+
+    beforeEach(() => {
+      readFileStub = sinon.stub(fs, 'readFile');
+    });
+
+    afterEach(() => {
+      readFileStub.restore();
+    });
+
+    it('should return bash completion script', () => {
+      const completion = new Completion('bash');
+      readFileStub.yields(null, '# bash completion script');
+
+      return completion.getBashScript()
+        .then((script) => {
+          script.should.equal('# bash completion script');
+        });
+    });
+
+    it('should reject with CompletionScriptError on fs error', () => {
+      const completion = new Completion('bash');
+      readFileStub.yields(new Error('File read error'));
+
+      return completion.getBashScript()
+        .should.be.rejectedWith(CompletionScriptError);
+    });
+  });
+});