lib/rogue_one/cli.rb in rogue_one-0.1.4 vs lib/rogue_one/cli.rb in rogue_one-0.2.0

@@ -4,14 +4,18 @@
 require "json"
 
 module RogueOne
   class CLI < Thor
     desc "report [DNS_SERVER]", "Show a report of a given DNS server"
+    method_option :custom_list, type: :string, desc: "A path to a custom list of domains"
+    method_option :verbose, type: :boolean
     def report(dns_server)
       with_error_handling do
         Ping.pong? dns_server
 
-        detector = Detector.new(target: dns_server)
+        custom_list = options["custom_list"]
+        verbose = options["verbose"]
+        detector = Detector.new(target: dns_server, custom_list: custom_list, verbose: verbose)
         puts JSON.pretty_generate(detector.report)
       end
     end
 
     no_commands do