scope.rb in ronin-recon-0.1.0.rc2

- old
+ new

@@ -21,10 +21,12 @@
 require 'ronin/recon/values/wildcard'
 require 'ronin/recon/values/domain'
 require 'ronin/recon/values/host'
 require 'ronin/recon/values/ip_range'
 require 'ronin/recon/values/ip'
+require 'ronin/recon/values/url'
+require 'ronin/recon/values/email_address'
 
 module Ronin
   module Recon
     #
     # Defines which domains, hosts, IP addresses are considered "in scope".
@@ -44,11 +46,11 @@
       attr_reader :ignore
 
       #
       # Initializes the scope.
       #
-      # @param [Array<Values::Wildcard, Values::Domain, Values::Host, Values::IPRange, Values::IP>] values
+      # @param [Array<Values::Wildcard, Values::Domain, Values::Host, Values::Website, Values::IPRange, Values::IP>] values
       #   The list of "in scope" values.
       #
       # @param [Array<Value>] ignore
       #   The recon values to ignore and are not "in scope".
       #
@@ -62,11 +64,11 @@
         @host_values = []
         @ip_values   = []
 
         values.each do |value|
           case value
-          when Values::Wildcard, Values::Domain, Values::Host
+          when Values::Wildcard, Values::Domain, Values::Host, Values::Website
             @host_values << value
           when Values::IP, Values::IPRange
             @ip_values << value
           else
             raise(NotImplementedError,"scope value type not supported: #{value.inspect}")
@@ -89,10 +91,12 @@
       def include?(value)
         scope_values = case value
                        when Values::Wildcard,
                             Values::Domain,
                             Values::Host,
-                            Values::URL
+                            Values::Website,
+                            Values::URL,
+                            Values::EmailAddress
                          @host_values
                        when Values::IP,
                             Values::IPRange
                          @ip_values
                        end