Contents

module Fog
  module Bouncer
    class Source
      attr_reader :group, :source
      attr_writer :local, :remote

      def initialize(source, group, &block)
        @source = source
        @group = group
        if block_given?
          @local = true
          instance_eval(&block)
        end
      end

      def extras
        protocols.select { |protocol| !protocol.local? }
      end

      def add_protocol(type, port)
        protocol = protocols.find { |p| p.match(type, port) }
        if protocol.nil?
          protocol = case type.to_sym
          when :icmp
            Fog::Bouncer::Protocols::ICMP.new(port, self)
          when :tcp
            Fog::Bouncer::Protocols::TCP.new(port, self)
          when :udp
            Fog::Bouncer::Protocols::UDP.new(port, self)
          end

          protocols << protocol
        end

        protocol
      end

      def local
        @local ||= false
      end

      def local?
        !!local
      end

      def missing
        protocols.select { |protocol| protocol.local? && !protocol.remote? }
      end

      def protocols
        @protocols ||= []
      end

      def remote
        @remote ||= false
      end

      def remote?
        !!remote
      end

      def ==(other)
        source == other.source &&
        group == other.group &&
        protocols.sort! == other.protocols.sort!
      end

      def inspect
        "<#{self.class.name} @source=#{source.inspect} @local=#{local} @remote=#{remote} @protocols=#{protocols.inspect}>"
      end

      private

      def icmp(*ports)
        ports.each { |port| p = add_protocol(:icmp, port); p.local = true }
      end

      def tcp(*ports)
        ports.each { |port| p = add_protocol(:tcp, port); p.local = true }
      end

      def udp(*ports)
        ports.each { |port| p = add_protocol(:udp, port); p.local = true }
      end
    end
  end
end

Version data entries

12 entries across 12 versions & 1 rubygems

Version	Path
fog-bouncer-0.2.7	lib/fog/bouncer/source.rb
fog-bouncer-0.2.6	lib/fog/bouncer/source.rb
fog-bouncer-0.2.5	lib/fog/bouncer/source.rb
fog-bouncer-0.2.4	lib/fog/bouncer/source.rb
fog-bouncer-0.2.3	lib/fog/bouncer/source.rb
fog-bouncer-0.2.2	lib/fog/bouncer/source.rb
fog-bouncer-0.2.1	lib/fog/bouncer/source.rb
fog-bouncer-0.2.0	lib/fog/bouncer/source.rb
fog-bouncer-0.1.1	lib/fog/bouncer/source.rb
fog-bouncer-0.1.0	lib/fog/bouncer/source.rb
fog-bouncer-0.0.8	lib/fog/bouncer/source.rb
fog-bouncer-0.0.6	lib/fog/bouncer/source.rb