Added the --nodejs option to ronin quote (closes #247).

ronin-rb · Nov 23, 2024 · 5887b1a · 5887b1a
1 parent f221500
commit 5887b1a
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 0 deletions.
diff --git a/lib/ronin/cli/commands/quote.rb b/lib/ronin/cli/commands/quote.rb
@@ -38,6 +38,7 @@ module Commands
       #     -c, --c                          Quotes the data as a C string
       #     -J, --java                       Quotes the data as a Java string
       #     -j, --js                         Quotes the data as a JavaScript string
+      #     -n, --nodejs                     Quotes the data as a Node.js string
       #     -S, --shell                      Quotes the data as a Shell string
       #     -P, --powershell                 Quotes the data as a PowerShell string
       #         --perl                       Quotes the data as a Perl string
@@ -70,6 +71,12 @@ class Quote < StringMethodsCommand
                         @method_calls << :java_string
                       end
 
+        option :nodejs, short: '-n',
+                        desc: 'Quotes the data as a Node.js string' do
+                          require 'ronin/support/encoding/node_js'
+                          @method_calls << :node_js_string
+                        end
+
         option :js, short: '-j',
                     desc: 'Quotes the data as a JavaScript string' do
                       require 'ronin/support/encoding/js'

diff --git a/man/ronin-quote.1.md b/man/ronin-quote.1.md
@@ -44,6 +44,9 @@ Produces quoted a string for a variety of programming languages.
 `-j`, `--js`
 : Quotes the data as a JavaScript string.
 
+`-n`, `--nodejs`
+: Quotes the data as a Node.js string.
+
 `-S`, `--shell`
 : Quotes the data as a Shell string.
 

diff --git a/spec/cli/commands/quote_spec.rb b/spec/cli/commands/quote_spec.rb
@@ -56,6 +56,18 @@
       end
     end
 
+    describe "--nodejs" do
+      let(:argv) { %w[--nodejs] }
+
+      it "must require 'ronin/support/encoding/node_js'" do
+        expect(require('ronin/support/encoding/node_js')).to be(false)
+      end
+
+      it "must add :node_js_string to #method_calls" do
+        expect(subject.method_calls.last).to eq(:node_js_string)
+      end
+    end
+
     describe "--shell" do
       let(:argv) { %w[--shell] }