From 31f9ef36d5d8f4804d551624830bfac9740b4625 Mon Sep 17 00:00:00 2001
From: Postmodern <postmodern.mod3@gmail.com>
Date: Thu, 4 Jan 2024 02:25:58 -0800
Subject: [PATCH] Added `command_kit-completion` (issue #89).

---
 .gitignore  | 1 +
 Gemfile     | 2 ++
 Rakefile    | 7 +++++++
 gemspec.yml | 1 +
 4 files changed, 11 insertions(+)

diff --git a/.gitignore b/.gitignore
index ecc41cc8..15828763 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,4 +1,5 @@
 /coverage
+/data/completions/ronin-web
 /doc
 /pkg
 /man/*.[1-9]
diff --git a/Gemfile b/Gemfile
index ee62960b..8fe2f6f4 100644
--- a/Gemfile
+++ b/Gemfile
@@ -58,4 +58,6 @@ group :development do
   gem 'stackprof',       require: false, platform: :mri
   gem 'rubocop',         require: false, platform: :mri
   gem 'rubocop-ronin',   require: false, platform: :mri
+
+  gem 'command_kit-completion', '~> 0.1', require: false
 end
diff --git a/Rakefile b/Rakefile
index f75cb505..39c89b5d 100644
--- a/Rakefile
+++ b/Rakefile
@@ -37,3 +37,10 @@ YARD::Rake::YardocTask.new
 
 require 'kramdown/man/task'
 Kramdown::Man::Task.new
+
+require 'command_kit/completion/task'
+CommandKit::Completion::Task.new(
+  class_file:  'ronin/web/cli',
+  class_name:  'Ronin::Web::CLI',
+  output_file: 'data/completions/ronin-web'
+)
diff --git a/gemspec.yml b/gemspec.yml
index d87c57a1..392275f9 100644
--- a/gemspec.yml
+++ b/gemspec.yml
@@ -20,6 +20,7 @@ metadata:
   rubygems_mfa_required: 'true'
 
 generated_files:
+  - data/completions/ronin-web
   - man/ronin-web.1
   - man/ronin-web-browser.1
   - man/ronin-web-diff.1