schatz 0.0.4

data/Gemfile

@@ -0,0 +1,8 @@
+# A sample Gemfile
+source "http://rubygems.org"
+require "rubygems"
+# windows color output
+gem "windows-pr" 
+gem "win32console"
+gem "rainbow"
+gem "nokogiri"

data/Gemfile.lock

@@ -0,0 +1,21 @@
+GEM
+  remote: http://rubygems.org/
+  specs:
+    nokogiri (1.5.0-x86-mingw32)
+    rainbow (1.1.3)
+    win32-api (1.4.8-x86-mingw32)
+    win32console (1.3.0-x86-mingw32)
+    windows-api (0.4.0)
+      win32-api (>= 1.4.5)
+    windows-pr (1.2.1)
+      win32-api (>= 1.4.5)
+      windows-api (>= 0.3.0)
+
+PLATFORMS
+  x86-mingw32
+
+DEPENDENCIES
+  nokogiri
+  rainbow
+  win32console
+  windows-pr

data/README.txt

@@ -0,0 +1,14 @@
+= Tools for connectivity checking
+== Description
+Gem provides means for testing TCP direct connections as well as HTTP web services through HTTP proxy.
+How to install gem:
+* $ gem install schatz
+Testing INCRs
+
+schatz -f file_name [-po] [-do] [-g 123456]
+
+-f    name of the export(.html ior .htm extension) or name of .txt file
+-po   test only http/https links via proxies
+-do   test only direct connections
+-p    test via specified proxy
+-h    help 

data/bin/schatz

@@ -0,0 +1,12 @@
+#!/usr/bin/env ruby
+$:.unshift(File.dirname(__FILE__) + '/../lib') unless $:.include?(File.dirname(__FILE__) + '/../lib')
+require File.expand_path(File.dirname(__FILE__)+ "/../lib/schatz")
+begin
+  Schatz::Cli::Runner.run(ARGV.dup, STDOUT, STDERR)
+rescue SystemExit => e
+  Kernel.exit(e.status)
+rescue Exception => e
+  STDERR.puts("#{e.message} (#{e.class})")
+  STDERR.puts(e.backtrace.join("\n"))
+  Kernel.exit(1)
+end

data/config/config.yml

@@ -0,0 +1,10 @@
+#please add necessary proxies
+---
+proxies:
+ 1: "http://10.233.46.129:8080"
+ 2: "http://10.233.46.130:8080"
+ 3: "http://10.233.46.131:8080"
+ 4: "http://10.233.46.132:8080"
+ 5: "http://10.237.46.130:8080"
+ 6: "http://10.237.46.131:8080"
+ 7: "http://10.237.46.132:8080"

data/install.rb

@@ -0,0 +1,5 @@
+system "gem install schatz"
+system "gem install windows-pr"
+system "gem install win32console"
+system "gem install rainbow"
+system "gem install nokogiri"

data/lib/schatz.rb

@@ -0,0 +1,31 @@
+$:.push File.expand_path(File.dirname(__FILE__) + "/../lib")
+require 'rubygems'
+require 'optparse'
+require 'open-uri'
+require 'schatz/cli/runner'
+require 'schatz/cli/configuration'
+require 'schatz/input/export'
+require 'schatz/testers/incr_test'
+require 'schatz/input/incr_export'
+require 'schatz/input/text_export'
+require 'schatz/input/connection'
+require 'schatz/input/connection_result'
+require 'schatz/input/txt_line_parser'
+require 'schatz/input/incr_line_parser'
+require 'schatz/output/out'
+require 'schatz/output/html'
+require 'schatz/output/cmd'
+require 'schatz/schemes/direct_tcp'
+require 'schatz/schemes/proxy_http'
+require 'schatz/schemes/proxy_https'
+require 'schatz/input/parser'
+require 'timeout'
+require 'rainbow'
+module Schatz
+  exceptions = %w(IncorrentParameters IncorrectIncrFormat ExportedIncrFieldErrorPorts FileNotExistInCurrentDirectory IncorrectFileFormat SheetNotFound ExportedIncrFieldErrorUnrecognized
+  IncorrectTextFormat NotImplementedMethod)
+  exceptions.each do |exc| 
+    const_set exc, Class.new(StandardError)
+  end
+end
+Sickill::Rainbow.enabled = true

data/lib/schatz/cli/configuration.rb

@@ -0,0 +1,45 @@
+module Schatz
+  module Cli
+    class Configuration
+      attr_accessor :opt, :options, :stdout, :stderr
+      def initialize(output=STDOUT, error=STDERR,options={})
+        @stdout = output
+        @stderr = error
+        @options =options
+        @opt = OptionParser.new
+        @opt.banner = "Usage: schatz [file_name]"
+        @opt.separator "Options:"
+        @opt.on("-f", "--file [file]") 		{|file|   	@options[:file_name]	= file}
+        @opt.on("-h", "--help") 		{|ignore| 	@options[:help]	   	= true}
+        @opt.on("--po") 			{|ignore| 	@options[:proxy_only]	= true}
+        @opt.on("--do") 			{|ignore| 	@options[:direct_only]	= true}
+        @opt.on("-g", "--gcm [gcm]")		{|gcm|		@options[:gcm]	   	= "gcm000000" +gcm}
+        @opt.on("-p", "--prx [proxy]")		{|proxy|  	@options[:proxies]   	= {1 => proxy}}
+      end
+      
+      def parse!(argv)
+        raise ArgumentError, "expected array of argv" unless argv.is_a?(Array)
+        raise ArgumentError, "you should provide at least a name of the incr" if argv.empty? || argv.first=="-f" && argv.length==1
+        @opt.parse(argv)
+        @options[:file_name] = full_file_name(@options[:file_name]) if @options[:file_name]
+        @options[:proxies] ||= read_config
+        show_info
+      end
+		 
+      def show_info
+        puts @opt if @options[:help]
+        puts "GCM number is not provided so will check all links. If you want to test only those implemented after specified GCM please mention it".foreground(:yellow) unless @options[:gcm]
+      end
+        
+      def full_file_name(file_name)
+        raise FileNotExistInCurrentDirectory until File.exist?(file_name)
+        raise IncorrectFileFormat until file_name =~ /\.xlsx?/ || file_name =~ /\.html?/ || file_name =~ /\.txt?/
+        File.expand_path(file_name)
+      end
+		
+      def read_config
+        YAML::load_file(File.expand_path(File.dirname(__FILE__)+"./../../../config/config.yml"))["proxies"]
+      end
+    end
+  end
+end

data/lib/schatz/cli/runner.rb

@@ -0,0 +1,35 @@
+require 'yaml'
+module Schatz
+  module Cli
+    class Runner
+      attr_accessor :options, :output, :error, :configuration
+      def self.run(argv, output, error, options={})
+       new(argv, output, error, options).run
+      end
+      def initialize(argv, output, error, options={})
+        @output = output
+        @error = error
+        @configuration = Configuration.new(output, error, options)
+        @configuration.parse!(argv)
+        @options = @configuration.options
+      end 
+      def run
+        trap_interrupt
+        source = case @options[:file_name]
+          when /\.htm/
+           Input::IncrExport.new(@options)
+          when /\.txt/
+            Input::TextExport.new(@options)
+          else
+            raise IncorrectFileFormat
+          end
+        outputters = [Output::CMD.new(@options), Output::HTML.new(@options)]
+        tester = Testers::IncrTest.new(@options, outputters)
+        tester.check_connections(source.get_conversations + source.get_links )
+      end
+      def trap_interrupt
+        trap("INT") {exit!}
+      end		
+    end
+  end 
+end

data/lib/schatz/input/connection.rb

@@ -0,0 +1,94 @@
+module Schatz
+  module Input
+    class Connection
+      attr_accessor :scheme, :host, :port, :gcm, :result, :color, :description
+      def initialize
+        @port = 80
+        @scheme = "http"
+        yield self
+        if @port=="" || @port==0
+          case @scheme
+          when "http"
+            @port = 80
+          when "https"
+            @port = 443
+          end
+        end 
+      end
+      def eql?(b)
+        self.scheme == b.scheme && self.host   == b.host && self.port == b.port && self.gcm    == b.gcm ? true : false   
+      end
+      
+      def result=(value)
+        @result=value
+        mark
+      end
+      def scheme=(value)
+        @scheme = value.downcase
+      end
+      def gcm=(value)
+        @gcm    = value.downcase
+      end
+      
+      def http?
+        @scheme="http"
+      end
+      
+      def https?
+        @scheme="https"
+      end
+      
+      def mark
+        direct? ? mark_hosts : mark_links
+      end
+      
+      def direct?
+        scheme.downcase == "tcp" || scheme.downcase == "udp"
+      end
+      
+      def link?
+        scheme.downcase == "http" || scheme.downcase == "https"
+      end
+      
+      def mark_hosts
+        self.color = result == "open" ? "green" : "red"
+      end
+      
+      def mark_links
+        self.color, self.description= case result 
+          when /30[12]/ then ["green", "Redirection to websso"]
+          when "200" then ["green", "OK"]
+          when /40[13]/ then ["green", "Error authorization"]
+          else ["red", result]
+        end
+      end
+      
+      def check
+        direct? ? host_check : link_check
+      end
+      def link_check
+        raise ArgumentError unless @proxy
+        begin
+          self.result = case scheme
+            when "http" 
+              Schemes::ProxyHttp.http_test(self, @proxy)
+            when "https"
+              Schemes::ProxyHttps.https_test(self, @proxy)
+          end
+        rescue Exception => e
+          self.result = e.message
+        end
+        self          
+      end
+
+      def host_check
+        begin
+          self.result = Schemes::DirectTcp.direct_test(self.host, self.port, 0.5) ? "open" : "closed"
+        rescue Exception => e
+          self.result = "closed"
+        end
+        self
+      end    
+    end
+  end
+end

data/lib/schatz/input/connection_result.rb

@@ -0,0 +1,14 @@
+module Schatz
+  module Input
+    class ConnectionResult < Connection
+      attr_accessor :proxy
+      def initialize(connection, proxy=nil)
+        @scheme = connection.scheme
+        @host   = connection.host
+        @port   = connection.port
+        @gcm    = connection.gcm
+        @proxy  = proxy
+      end
+    end
+  end
+end

data/lib/schatz/input/export.rb

@@ -0,0 +1,14 @@
+module  Schatz
+  module Input
+    class Export
+      def get_links
+        
+      end 
+      
+      
+      def get_conversations
+      
+      end
+    end
+  end
+end

data/lib/schatz/input/incr_export.rb

@@ -0,0 +1,81 @@
+require 'nokogiri'
+require 'net/http'
+require 'net/https'
+require 'schatz/input/parser'
+require 'timeout'
+module Schatz
+  module Input
+    class IncrExport < Export
+      include ParserHelper
+      attr_accessor :options, :file		
+
+      def initialize(options)
+        @options        = options
+        @file = Nokogiri::HTML(File.open(@options[:file_name], 'r'))
+      end
+
+      def get_links
+        Schatz::Input::IncrLineParser.new(rows_in_links_table).links
+      end
+
+      def get_conversations
+        Schatz::Input::IncrLineParser.new(rows_in_conversation_table).connections
+      end
+      
+      # eliminate the first row(names of the fields)
+      def rows_in_conversation_table
+        get_rid_of_header(find_conv_table).select {|row| row =~ CONVERSATION_ROW }
+      end
+
+      def rows_in_links_table
+        get_rid_of_header(find_links_table)
+      end
+
+      def get_rid_of_header(table_number)
+        rows = @file.xpath("//table[#{table_number+1}]//tr")
+        rows.map {|row| delete_mess(row) }
+      end
+      
+      def delete_mess(row)
+        row.text.gsub(/[\n\t]/, "").gsub(/\u00A0/, " ").gsub(/\r\r/, " ").gsub(/\r/, "")
+      end
+
+      def get_table options={}
+        allowed_keys= ["table_id"]
+        raise ArgumentError if options.empty?
+        options.keys.each {|key| raise IncorrentParameters unless allowed_keys.include? key.to_s}			
+        @file.xpath("/html/body/table")[options[:table_id].to_i]
+      end
+
+      def find_conv_table
+        reg1 = Regexp.new 'Conversation Type'
+        reg2 = Regexp.new 'CR/GCM Number'
+        reg3 = Regexp.new 'Destination Host'
+        find_table([reg1, reg2, reg3])
+      end
+
+      def find_links_table
+        reg1 = Regexp.new 'Intranet Site'
+        reg2 = Regexp.new 'Reason and Activity'
+        reg3 = Regexp.new 'CR/GCM Number'
+        find_table([reg1, reg2, reg3])
+      end
+
+      def find_table(regs)
+        tables = @file.xpath("/html/body/table")
+        conv_table = tables.select {|t| regs.all?{ |regexp| t.text =~ regexp} }
+        raise IncorrectIncrFormat if conv_table.size > 1
+        tables.index(conv_table.first)
+      end		
+			
+      def delete_source_subnets(row)
+        row.match(SOURCE_SUBNET).to_a.each{|reg| row.gsub!(/#{reg}/, "")} while row =~ SOURCE_SUBNET
+        row
+      end
+			
+      def scan_gcm_numbers(rows)
+        rows.map{ |row| row.scan(GCM_NUMBER).first }.compact.map(&:downcase)
+      end
+    end
+  end
+end

data/lib/schatz/input/incr_line_parser.rb

@@ -0,0 +1,35 @@
+module Schatz
+  module Input
+    class IncrLineParser < Parser
+      include ParserHelper
+      def initialize(lines)
+        super(lines)
+      end
+      
+      def parse(line)
+        case line
+        when IncrExport::LINK_HTML
+          @links << Connection.new do |c|
+            c.gcm    = $1.downcase
+            c.scheme = $2
+            c.host   = $3
+            c.port   = $4.to_i if $4
+          end
+        when IncrExport::CONVERSATION_HTML
+          gcm    = $1
+          scheme = $4
+          line.scan(IP_ADDRESS).map do |address|
+            parse_ports_field(line.match(IncrExport::PORTS_WITH_PROTOCOL).captures.first).map do |port|
+              @connections << Connection.new do |c|
+                c.gcm    = gcm
+                c.scheme = scheme
+                c.host   = address.first
+                c.port   = port.to_i
+              end
+            end
+          end
+        end       
+      end  
+    end
+  end
+end

data/lib/schatz/input/parser.rb

@@ -0,0 +1,53 @@
+module Schatz
+  module Input  
+    class Parser
+      attr_accessor :connections, :links
+      def initialize(lines)
+        @connections = []
+        @links       = []
+        lines.map { |line| parse(line)}
+      end
+      
+      def parse(line)
+        raise NotImplementedMethod
+      end    
+    end
+    module ParserHelper
+    
+      LINK               = Regexp.new '(https?):\/\/([a-zA-Z0-9\.\-]+\.?d?b?.com):?(\d{1,5}*)'
+      LINK_HTML          = Regexp.new '(gcm\d{12}|GCM\d{12}).*(https?):\/\/([a-zA-Z0-9\.\-]+\.?d?b?.com):?(\d{1,5}*).*'      
+      SOURCE_SUBNET      = Regexp.new '\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}\ *\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}'
+      GCM_NUMBER         = Regexp.new 'gcm\d{12}|GCM\d{12}'
+      REGEXP_HOST        = Regexp.new '^\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}.+(\d{1,5}[\-\,]?)+$'
+      REGEXP_HOST_PORTS  = Regexp.new '^(\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3})\ +(\d{1,5}\-\d{1,5}|\d{1,5}(\,\d{1,5})*)$'
+      ONE_PORT           = Regexp.new '^\d{1,5}$'
+      ONE_HYPHEN         = Regexp.new '^\d{1,5}(\-)+\d{1,5}$'
+      ONE_OR_MORE_COMMA  = Regexp.new'^\d{1,5}(,\d{1,5})+$'
+      IP_ADDRESS         = Regexp.new '(10\.\d{1,3}\.\d{1,3}\.\d{1,3})(?#matches ip address)(?!.*255\.\d{1,3}\.\d{1,3}\.\d{1,3})'
+      PORTS_WITH_PROTOCOL= Regexp.new '([\d\-\,]{1,450}) ?(TCP|UDP)'
+      CONVERSATION_HTML  = Regexp.new "(#{GCM_NUMBER.source}).*#{IP_ADDRESS.source}.*#{PORTS_WITH_PROTOCOL.source}.*"
+      CONVERSATION_ROW   = Regexp.new ".*#{IP_ADDRESS.source}.*#{PORTS_WITH_PROTOCOL.source}.*"
+      
+      def parse_ports_field(ports)
+        raise ArgumentError, "received nil instead of string for ports" unless ports 
+        raise ArgumentError, "received blank instead of string for ports" if ports.empty?        
+        ports.gsub!(/ /, "")
+      case ports
+        when ONE_HYPHEN
+          start_range, end_range = ports.split("-")
+          [*start_range..end_range].map(&:to_i)
+        when ONE_OR_MORE_COMMA
+          ports.split(",").map(&:to_i)
+        when ONE_PORT
+          [ports.to_i]
+        else ports
+          raise ExportedIncrFieldErrorPorts, "for #{ports}"
+        end
+      end
+	  
+      def push_connection(host, port, protocol, gcm, hosts)
+        hosts << {:host => host, :port => port.to_i, :protocol => protocol, :gcm => gcm}
+      end
+    end
+  end
+end

data/lib/schatz/input/result.rb

@@ -0,0 +1,7 @@
+module Schatz
+  module Input
+    class Result
+      
+    end
+  end
+end

data/lib/schatz/input/text_export.rb

@@ -0,0 +1,35 @@
+require 'schatz/input/parser'
+module Schatz
+  module Input
+    class TextExport < Export
+      include ParserHelper
+      attr_accessor :file, :options
+      def initialize(options)
+        @options = options
+        @addresses_with_ports = []
+        @conversations 	= []
+        @links_with_ports = []
+        @file = File.open(@options[:file_name], 'r')
+        check_format
+      end
+			
+      def get_links
+        TxtLineParser.new(split_lines).links
+      end
+      
+      def get_conversations
+        TxtLineParser.new(split_lines).connections
+      end
+      
+      # split txt file into lines
+      def split_lines
+        @file.pos=0
+        @file.read.split(/\n/).map{ |line| line.gsub /\r/, "" }
+      end
+      
+      def check_format
+        raise IncorrectTextFormat, "format of txt file is incorrect" unless split_lines.all? { |line| line =~ REGEXP_HOST || line =~ LINK }
+      end
+    end
+  end
+end

data/lib/schatz/input/txt_line_parser.rb

@@ -0,0 +1,29 @@
+module Schatz
+  module Input
+    class TxtLineParser < Parser
+      include ParserHelper
+      def initialize(lines)
+        super(lines)
+      end
+        
+      def parse(line)
+        case line
+        when REGEXP_HOST_PORTS
+          parse_ports_field($2).map do |port| 
+            @connections << Connection.new do |c|
+              c.host = $1
+              c.port = port
+              c.scheme = "tcp"
+            end
+          end
+        when LINK
+          @links << Connection.new do |c|
+            c.scheme = $1
+            c.host   = $2
+            c.port   = $3.to_i if $3           
+          end
+        end
+      end
+    end
+  end
+end

data/lib/schatz/output/cmd.rb

@@ -0,0 +1,30 @@
+module Schatz
+  module Output
+    class CMD < Out
+      def initialize(options={})
+        @options = options
+      end
+      def print(connection)
+        case connection.scheme
+            when "tcp" 
+              direct(connection)
+            when "http"
+              link(connection)
+            when "https"
+              link(connection)
+        end
+      end
+      
+      def direct(connection)
+        color = connection.result == "open" ? :green : :red
+        puts (connection.host+" "+connection.port.to_s+" "+connection.scheme).foreground(color)
+       end
+      
+      def link(connection)
+        codes = %w(403 302 200)
+        color = codes.include?(connection.result) ? :green : :red
+        puts (connection.scheme+"://"+connection.host+" "+connection.port.to_s+" via "+ connection.proxy).foreground(color)
+      end
+    end
+  end
+end

data/lib/schatz/output/html.rb

@@ -0,0 +1,69 @@
+module Schatz
+  module Output
+    class HTML < Out
+      attr_accessor :html, :connections
+      def initialize(options={})
+        @html=""
+        @options = options
+        @connections = []
+      end
+      def print(connection)
+        @connections << connection
+      end
+      
+      def generate
+        File.open(file_name,"w+") do |file|
+          print_head
+          print_hosts
+          print_links
+          print_end
+          file << @html
+          @html
+        end
+      end
+			
+      def file_name
+        @options[:file_name].to_s+"_test_"+ Time.now.strftime("%d-%m-%y-%s")+".html"
+      end
+      
+      def print_hosts
+        @html << "<h3>Conversation Map</h3>"
+        @html << "<table>"
+        @html << "<tr><th>Number</th><th> IP address</th><th> Port </th><th> Protocol </th><th> Result </th></tr>"
+        @connections.select{ |connection| connection.direct? }.each_with_index do |host, index|
+          @html << "<tr><td>#{index+1}</td><td>#{host.host}</td><td>#{host.port}</td><td>#{host.scheme}</td><td class='#{host.color}'>#{host.result}</td></tr>"
+        end
+        @html << "</table><br/><br/>"
+      end
+
+      def print_links
+        @connections.select{ |connection| connection.link? }.group_by { |connection| connection.proxy}.each do |proxy, links|
+          print_links_table(proxy, links)
+        end        
+      end
+      
+      def print_links_table(proxy, links)
+        @html << "<h3>Intranet Access #{proxy}</h3>"
+        @html << "<table>"
+        @html << "<tr><th>Number</th><th> Link to page</th><th> Result </th></tr>"
+        links.each_with_index do |link, index|
+          @html << "<tr><td>#{index+1}</td><td><a href='#{link.scheme}://#{link.host}:#{link.port}'> #{link.scheme}://#{link.host}:#{link.port} </td><td class='#{link.color}'>#{link.result}</td></tr>"
+        end
+        @html << "</table>"
+      end
+      def print_head
+        @html << "<!DOCTYPE HTML PUBLIC '-//W3C//DTD HTML 4.01//EN'><html lang='en'>
+        <head>
+        <title>Connection Test Results</title>
+        <style type='text/css'>
+          .green{background-color:SeaGreen}
+          .red	{background-color:Crimson}
+        </style>
+        </head><body>"
+      end
+      def print_end
+        @html << "</body></html>"
+      end
+    end
+  end
+end

data/lib/schatz/output/out.rb

@@ -0,0 +1,9 @@
+module Schatz
+  module Output
+    class Out
+      attr_accessor :options
+      def generate
+      end
+    end
+  end
+end

data/lib/schatz/schemes/direct_tcp.rb

@@ -0,0 +1,17 @@
+require 'timeout'
+require 'socket'
+module Schatz
+  module Schemes
+    module DirectTcp 
+      def self.direct_test(host, port, timeout =0.1)
+        addr = Socket.pack_sockaddr_in(port, host)
+        s = Socket.new(:AF_INET, :SOCK_STREAM, 0)
+        begin
+          s.connect_nonblock(addr)
+        rescue Errno::EINPROGRESS
+          IO.select(nil, [s], nil, timeout) ? true : (raise Timeout::Error)
+        end
+      end
+    end
+  end
+end 

data/lib/schatz/schemes/proxy_http.rb

@@ -0,0 +1,19 @@
+require 'uri'
+require 'net/http'
+module Schatz
+  module Schemes
+    module ProxyHttp
+      def self.http_test(link, proxy)
+        Proxy.connect(proxy).start(link.host, link.port, :proxy => proxy).get("/").code
+      end
+    end
+    module Proxy
+      def self.connect(proxy)
+        uri = URI(proxy)
+        Net::HTTP::Proxy(uri.host, uri.port)
+      end
+    end
+  end
+end
+
+

data/lib/schatz/schemes/proxy_https.rb

@@ -0,0 +1,18 @@
+require 'uri'
+module Schatz
+	module Schemes
+		module ProxyHttps
+			def self.https_test(link, proxy)
+				response = Proxy.connect(proxy).start(link.host, link.port, :use_ssl => true, :verify_mode => OpenSSL::SSL::VERIFY_NONE, :open_timeout => 1.5).get("/")
+				case response 
+					when Net::HTTPFound
+						"302" if response['location'] =~ /https\:\/\/login\-intranet\.isso\.db\.com/ ? "redirected to WebSSO" : "redirected to #{response['location']}"
+					when Net::HTTPOK
+						"200"
+					else 
+						[response.code, response.msg].join(" ")
+				end
+			end
+		end
+	end
+end

data/lib/schatz/testers/incr_test.rb

@@ -0,0 +1,46 @@
+module Schatz
+  module Testers
+    class IncrTest
+      attr_accessor :options, :output, :connections
+      def  initialize(options={}, outputters=[])
+        @options = options
+        @outputters = outputters
+      end
+        
+      def check_connections(connections)
+        @connections = connections
+        if @options[:direct_only]
+          direct_check
+        elsif @options[:proxy_only]
+          links_check
+        else
+          direct_check
+          links_check
+        end
+        @outputters.each { |outputter| outputter.generate }
+      end
+      
+      def links_check
+        match_gcm_connections.select { |c| c.link? }.map do |connection|
+          proxies.map do |proxy|
+            @outputters.each { |out| out.print(Input::ConnectionResult.new(connection, proxy).check) }
+          end                  
+        end
+      end
+      
+      def direct_check
+       match_gcm_connections.select { |c| c.direct? }.map do |connection|
+          @outputters.each { |out| out.print(Input::ConnectionResult.new(connection).check) }
+        end        
+      end
+      
+      def match_gcm_connections
+       @options[:gcm] ? @connections.select { |c| c.gcm == @options[:gcm] } : @connections
+      end
+           
+      def proxies
+        @options[:proxies].values
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,71 @@
+--- !ruby/object:Gem::Specification
+name: schatz
+version: !ruby/object:Gem::Version
+  version: 0.0.4
+  prerelease: 
+platform: ruby
+authors:
+- Sergiy Shevchik
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-03-03 00:00:00.000000000 Z
+dependencies: []
+description: Tool for checking the connectivity
+email: berrouz@gmail.com
+executables:
+- schatz
+extensions: []
+extra_rdoc_files:
+- README.txt
+files:
+- lib/schatz.rb
+- lib/schatz/input/connection.rb
+- lib/schatz/input/connection_result.rb
+- lib/schatz/input/export.rb
+- lib/schatz/input/incr_export.rb
+- lib/schatz/input/incr_line_parser.rb
+- lib/schatz/input/parser.rb
+- lib/schatz/input/result.rb
+- lib/schatz/input/text_export.rb
+- lib/schatz/input/txt_line_parser.rb
+- lib/schatz/cli/runner.rb
+- lib/schatz/cli/configuration.rb
+- lib/schatz/output/html.rb
+- lib/schatz/output/out.rb
+- lib/schatz/output/cmd.rb
+- lib/schatz/schemes/direct_tcp.rb
+- lib/schatz/schemes/proxy_http.rb
+- lib/schatz/schemes/proxy_https.rb
+- lib/schatz/testers/incr_test.rb
+- install.rb
+- config/config.yml
+- Gemfile
+- Gemfile.lock
+- README.txt
+- bin/schatz
+homepage: https://github.com/berrouz/schatz
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: 1.9.3
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.11
+signing_key: 
+specification_version: 3
+summary: TCP,UDP
+test_files: []