Skip to content

Commit

Permalink
Added the ronin completion command (closes #183).
Browse files Browse the repository at this point in the history
  • Loading branch information
postmodern committed Jan 4, 2024
1 parent 99fa17a commit 3f50f7a
Show file tree
Hide file tree
Showing 6 changed files with 225 additions and 14 deletions.
22 changes: 11 additions & 11 deletions Gemfile
Original file line number Diff line number Diff line change
Expand Up @@ -14,8 +14,8 @@ if RUBY_VERSION >= '3.1.0'
gem 'net-imap', '~> 0.1', group: :net, platform: :mri
end

# gem 'command_kit', '~> 0.4', github: 'postmodern/command_kit.rb',
# branch: '0.4.0'
# gem 'command_kit', '~> 0.5', github: 'postmodern/command_kit.rb',
# branch: 'main'

group :database do
gem 'sqlite3', '~> 1.0', platform: :mri
Expand All @@ -25,10 +25,10 @@ end
# Library dependencies
gem 'ronin-support', '~> 1.1', github: "ronin-rb/ronin-support",
branch: '1.1.0'
gem 'ronin-core', '~> 0.2', github: "ronin-rb/ronin-core",
gem 'ronin-core', '~> 0.2', github: 'ronin-rb/ronin-core',
branch: '0.2.0'
gem 'ronin-repos', '~> 0.1', github: 'ronin-rb/ronin-repos',
branch: '0.2.0'
# gem 'ronin-repos', '~> 0.1', github: "ronin-rb/ronin-repos",
# branch: 'main'
gem 'ronin-db-activerecord', '~> 0.2', github: "ronin-rb/ronin-db-activerecord",
branch: '0.2.0'
gem 'ronin-db', '~> 0.2', github: "ronin-rb/ronin-db",
Expand All @@ -39,18 +39,18 @@ gem 'ronin-listener-http', '~> 0.1', github: "ronin-rb/ronin-listener-http"
branch: 'main'
gem 'ronin-listener', '~> 0.1', github: "ronin-rb/ronin-listener",
branch: 'main'
# gem 'ronin-fuzzer', '~> 0.1', github: 'ronin-rb/ronin-fuzzer',
# branch: 'main'
gem 'ronin-fuzzer', '~> 0.2', github: 'ronin-rb/ronin-fuzzer',
branch: '0.2.0'
# gem 'ronin-post_ex', '~> 0.1', github: 'ronin-rb/ronin-post_ex',
# branch: 'main'
# gem 'ronin-code-asm', '~> 1.0', github: 'ronin-rb/ronin-code-asm',
# branch: 'main'
# gem 'ronin-code-sql', '~> 2.0', github: 'ronin-rb/ronin-code-sql',
# branch: 'main'
# gem 'ronin-payloads', '~> 0.1', github: 'ronin-rb/ronin-payloads',
# branch: 'main'
# gem 'ronin-exploits', '~> 1.0', github: 'ronin-rb/ronin-exploits',
# branch: 'main'
gem 'ronin-payloads', '~> 0.2', github: 'ronin-rb/ronin-payloads',
branch: '0.2.0'
gem 'ronin-exploits', '~> 1.1', github: 'ronin-rb/ronin-exploits',
branch: '1.1.0'
gem 'ronin-vulns', '~> 0.2', github: 'ronin-rb/ronin-vulns',
branch: '0.2.0'
# gem 'ronin-web-server', '~> 0.1', github: 'ronin-rb/ronin-web-server',
Expand Down
1 change: 1 addition & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -79,6 +79,7 @@ Commands:
cert-dump
cert-gen
cert-grab
completion
decode, dec
decrypt
dns
Expand Down
7 changes: 4 additions & 3 deletions gemspec.yml
Original file line number Diff line number Diff line change
Expand Up @@ -46,6 +46,7 @@ generated_files:
- man/ronin-cert-dump.1
- man/ronin-cert-gen.1
- man/ronin-cert-grab.1
- man/ronin-completion.1
- man/ronin-decode.1
- man/ronin-decrypt.1
- man/ronin-dns.1
Expand Down Expand Up @@ -104,13 +105,13 @@ dependencies:
ronin-support: ~> 1.1
ronin-dns-proxy: ~> 0.1
ronin-core: ~> 0.2
ronin-repos: ~> 0.1
ronin-db: ~> 0.1
ronin-repos: ~> 0.2
ronin-db: ~> 0.2
ronin-listener: ~> 0.1
ronin-nmap: ~> 0.1
ronin-masscan: ~> 0.1
ronin-recon: ~> 0.1
ronin-fuzzer: ~> 0.1
ronin-fuzzer: ~> 0.2
ronin-web: ~> 2.0
ronin-code-asm: ~> 1.0
ronin-code-sql: ~> 2.0
Expand Down
113 changes: 113 additions & 0 deletions lib/ronin/cli/commands/completion.rb
Original file line number Diff line number Diff line change
@@ -0,0 +1,113 @@
# frozen_string_literal: true
#
# Copyright (c) 2006-2023 Hal Brodigan (postmodern.mod3 at gmail.com)
#
# Ronin is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# Ronin is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with Ronin. If not, see <https://www.gnu.org/licenses/>.
#

require 'ronin/core/cli/completion_command'
require 'ronin/repos/cli/commands/completion'
require 'ronin/db/cli/commands/completion'
require 'ronin/fuzzer/cli/commands/completion'
require 'ronin/web/cli/commands/completion'
require 'ronin/vulns/cli/commands/completion'
require 'ronin/payloads/cli/commands/completion'
require 'ronin/exploits/cli/commands/completion'
require 'ronin/listener/cli/commands/completion'
require 'ronin/nmap/cli/commands/completion'
require 'ronin/masscan/cli/commands/completion'
require 'ronin/recon/cli/commands/completion'
require 'ronin/root'

module Ronin
class CLI
module Commands
#
# Manages the shell completion rule for `ronin` and all other `ronin-*`
# commands.
#
# ## Usage
#
# ronin completion [options]
#
# ## Options
#
# --print Prints the shell completion file
# --install Installs the shell completion file
# --uninstall Uninstalls the shell completion file
# -h, --help Print help information
#
# ## Examples
#
# ronin completion --print
# ronin completion --install
# ronin completion --uninstall
#
# @since 2.1.0
#
class Completion < Core::CLI::CompletionCommand

man_dir File.join(ROOT,'man')
man_page 'ronin-completion.1'

description 'Manages the shell completion rules for ronin and all other ronin-* commands'

# All shell completion files for `ronin` and the other `ronin-*`
# commands.
COMPLETION_FILES = [
File.join(ROOT,'data','completions','ronin'),

Repos::CLI::Commands::Completion.completion_file,
DB::CLI::Commands::Completion.completion_file,
Fuzzer::CLI::Commands::Completion.completion_file,
Web::CLI::Commands::Completion.completion_file,
Vulns::CLI::Commands::Completion.completion_file,
Payloads::CLI::Commands::Completion.completion_file,
Exploits::CLI::Commands::Completion.completion_file,
Listener::CLI::Commands::Completion.completion_file,
Nmap::CLI::Commands::Completion.completion_file,
Masscan::CLI::Commands::Completion.completion_file,
Recon::CLI::Commands::Completion.completion_file
]

#
# Prints all completion files.
#
def print_completion_file
COMPLETION_FILES.each do |completion_file|
super(completion_file)
end
end

#
# Installs all completion files.
#
def install_completion_file
COMPLETION_FILES.each do |completion_file|
super(completion_file)
end
end

#
# Uninstall all completion files.
#
def uninstall_completion_file
COMPLETION_FILES.each do |completion_file|
super(completion_file)
end
end
end
end
end
end
80 changes: 80 additions & 0 deletions man/ronin-completion.1.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,80 @@
# ronin-completion 1 "2024-01-01" Ronin "User Manuals"

## NAME

ronin-completion - Manages shell completion rules for `ronin-repos`

## SYNOPSIS

`ronin-repos completion` [*options*]

## DESCRIPTION

The `ronin completion` command can print, install, or uninstall shell
completion rules for the `ronin` command and all other `ronin-*` commands.

Supports installing completion rules for Bash or Zsh shells.
Completion rules for the Fish shell is currently not supported.

### ZSH SUPPORT

Zsh users will have to add the following lines to their `~/.zshrc` file in
order to enable Zsh's Bash completion compatibility layer:

autoload -Uz +X compinit && compinit
autoload -Uz +X bashcompinit && bashcompinit

## OPTIONS

`--print`
: Prints the shell completion file.

`--install`
: Installs the shell completion file.

`--uninstall`
: Uninstalls the shell completion file.

`-h`, `--help`
: Prints help information.

## ENVIRONMENT

*PREFIX*
: Specifies the root prefix for the file system.

*HOME*
: Specifies the home directory of the user. Ronin will search for the
`~/.cache/ronin-repos` cache directory within the home directory.

*XDG_DATA_HOME*
: Specifies the data directory to use. Defaults to `$HOME/.local/share`.

## FILES

`~/.local/share/bash-completion/completions/`
: The user-local installation directory for Bash completion files.

`/usr/local/share/bash-completion/completions/`
: The system-wide installation directory for Bash completions files.

`/usr/local/share/zsh/site-functions/`
: The installation directory for Zsh completion files.

## EXAMPLES

`ronin completion --print`
: Prints all shell completion rules instead of installing them.

`ronin completion --install`
: Installs all shell completion rules for `ronin` and the other `ronin-*`
commands.

`ronin-repos completion --uninstall`
: Uninstalls all shell completion rules for `ronin` and the other `ronin-*`
commands.

## AUTHOR

Postmodern <[email protected]>

16 changes: 16 additions & 0 deletions spec/cli/commands/completion_spec.rb
Original file line number Diff line number Diff line change
@@ -0,0 +1,16 @@
require 'spec_helper'
require 'ronin/cli/commands/completion'
require_relative 'man_page_example'

describe Ronin::CLI::Commands::Completion do
it "must inherit from Ronin::Core::CLI::CompletionCommand" do
expect(described_class).to be < Ronin::Core::CLI::CompletionCommand
end

it "must set man_dir" do
expect(described_class.man_dir).to_not be(nil)
expect(File.directory?(described_class.man_dir)).to be(true)
end

include_examples "man_page"
end

0 comments on commit 3f50f7a

Please sign in to comment.