require 'net/http'
require "net/https"
require 'timeout'
require 'etc'
require 'uri'

class TDIPlan < TDI

  def _parse(uri,params)

    # Normalizing
    if not uri =~ /^https?:\/\//
        uri = 'http://' + uri.to_s
    end

    # URI
    _uri = URI(uri)
    ssl  = _uri.scheme.eql?("https")
    host = _uri.host
    port = _uri.port
    path = _uri.path.empty? ? '/' : _uri.path

    # Params
    code = params['code'].nil? ? 200 : params['code'].to_i
    match = params['match']
    timeout_limit = params['timeout'].nil? ? 2 : params['timeout'].to_i

    if not params['proxy'].nil?
        proxy_addr, proxy_port = params['proxy'].split(/:/)
        proxy_port = 3128 unless not proxy_port.nil?
    end

    return host, port, path, proxy_addr, proxy_port, code, match, ssl, timeout_limit
    
  end

  def http(plan)
    plan.select { |key, val|
      val.is_a?(Hash)
    }.each_pair do |case_name,case_content|

      host, port, path, proxy_addr, proxy_port, code, match, ssl, timeout_limit = _parse(case_name,case_content)

      # User.
      user = Etc.getpwuid(Process.euid).name
     
      response = nil

      if not proxy_addr.nil? and not proxy_port.nil?

        http = Net::HTTP::Proxy(proxy_addr, proxy_port)

        begin
          timeout(timeout_limit) do
            begin
              http.start(host,port,:use_ssl => ssl, :verify_mode => OpenSSL::SSL::VERIFY_NONE) { |http|
                response = http.get(path)
              }
            rescue Errno::ECONNREFUSED, Errno::ECONNRESET
              warning "HTTP (#{user}): #{case_name} - Connection reset or refused."
            end
          end
        rescue Timeout::Error
          failure "HTTP (#{user}): #{case_name} - Timed out (#{timeout_limit}s)."
        end
        
      else

        http = Net::HTTP.new(host, port)

        if ssl
            http.use_ssl = true
            http.verify_mode = OpenSSL::SSL::VERIFY_NONE
        end 

        begin        
          timeout(timeout_limit) do
            begin
              http.start() { |http|
                response = http.get(path)
              }
            rescue Errno::ECONNREFUSED, Errno::ECONNRESET
              warning "HTTP (#{user}): #{case_name} - Connection reset or refused."
            end
          end
        rescue Timeout::Error
          failure "HTTP (#{user}): #{case_name} - Timed out."
        end
        
      end

      if not response.nil?
          if not match.nil? and not response.body.chomp.include?(match.chomp)
            failure "HTTP (#{user}): #{case_name} - Expected string '#{match.chomp}'."
          elsif not code.nil? and (response.code.to_i != code)
            failure "HTTP (#{user}): #{case_name} - Expected HTTP #{code}."
          else
            success "HTTP (#{user}): #{case_name}"
          end
      end
      
    end
  end
end