require 'resolv'

module Serverspec
  module Type
    class Port < Base
      def protocols
        %w(udp tcp tcp6 udp6)
      end

      def options
        @options ||= {}
      end

      def protocol_matcher(protocol)
        protocol = protocol.to_s.downcase
        if protocols.include?(protocol)
          options[:protocol] = protocol
        else
          raise ArgumentError.new("`be_listening` matcher doesn't support #{protocol}")
        end
      end

      def local_address_matcher(local_address)
        if valid_ip_address?(local_address)
          options[:local_address] = local_address
        else
          raise ArgumentError.new("`be_listening` matcher requires valid IPv4 or IPv6 address")
        end
      end

      def listening?(protocol, local_address)
        protocol_matcher(protocol) if protocol
        local_address_matcher(local_address) if local_address
        @runner.check_listening(@name, options)
      end

      def valid_ip_address?(ip_address)
        !!(ip_address =~ Resolv::IPv4::Regex) || !!(ip_address =~ Resolv::IPv6::Regex)
      end
    end
  end
end