module EDIS

# File lib/edis_client.rb, line 314
def build_date_params(options, optional_date_params)
  optional_date_params.inject({}) do |params, date_param|
    if options[date_param]
      comparison = options[date_param]
      case comparison[:type]
        when :between
          params[camelize(date_param)] =
            "#BETWEEN:#{date_param[:from_date]}:#{date_param[:to_date]}"
        when :before, :after, :exact
          params[camelize(date_param)] = 
            "#{comparison[:type].uppercase}:#{comparision[:date]}"
        else
          raise ArgumentError, 
            "Unknown comparison type #{comparison[:comparison_type]}"
      end
    end
    params
  end
end

# File lib/edis_client.rb, line 304
def build_params(options, optional_params)
  optional_params.inject({}) do |params, param|
    params[camelize(param.to_s, false)] = options[param] if options[param]
    params
  end
end

# File lib/edis_client.rb, line 294
def build_path(root, options, optional_resources)
  optional_resources.inject(root) do |path, resource|
    path << "/#{options[resource]}" if options[resource]
    path
  end
end

# File lib/edis_client.rb, line 337
def camelize(s, first_letter_in_uppercase = true)
  if first_letter_in_uppercase
    s.to_s.gsub(/\/(.?)/) { "::#{$1.upcase}" }.gsub(/(?:^|_)(.)/) { $1.upcase }
  else
    s.to_s[0].chr.downcase + camelize(s)[1..-1]
  end
end

# File lib/edis_client.rb, line 249
def connect
  uri  = URI.parse('https://edis.usitc.gov/')
  http = net_http_class.new(uri.host, uri.port)
  http.use_ssl = true 
  http.read_timeout = @config[:timeout] and http
end

# File lib/edis_client.rb, line 147
def document_date_params
  [:official_received_date,  :modified_date]
end

# File lib/edis_client.rb, line 160
def download_paths
  [:document_id, :attachment_id]
end

# File lib/edis_client.rb, line 207
def get_resource(path, options, params = {})
  connect.start do |http|
    path = path_with_params(path, params) unless params.empty?
    xml  = http.get("/data/#{path}", header(options) || {}).body
    RecursiveOpenStruct.new Crack::XML.parse(xml)
  end
end

# File lib/edis_client.rb, line 261
def header(options)
  digest = if options[:digest]
    options[:digest]
  elsif @config[:digest]
    @config[:digest]
  else
    false
  end
  {'Authorization' => "Basic #{digest}"} if digest
end

# File lib/edis_client.rb, line 155
def investigation_params
  [:page, :investigation_type, :investigation_status]
end

# File lib/edis_client.rb, line 152
def investigation_paths
  [:investigation_number, :investigation_phase]
end

# File lib/edis_client.rb, line 282
def net_http_class
  if proxy?
    proxy = @config[:proxy]
    Net::HTTP::Proxy(proxy[:uri].host, proxy[:uri].port, proxy[:user], proxy[:password])
  else
    Net::HTTP
  end
end

# File lib/edis_client.rb, line 241
def path_with_params(path, params)
  "#{path}?".concat \
    params.collect { |k,v| "#{k}=#{CGI::escape(v.to_s)}" }.reverse.join('&')
end

# File lib/edis_client.rb, line 230
def post_resource(path, params)
  connect.start do |http|
    req = Net::HTTP::Post.new("/data/#{path}") and req.set_form_data params
    xml = http.request(req).body
    RecursiveOpenStruct.new Crack::XML.parse(xml)
  end
end

# File lib/edis_client.rb, line 275
def proxy?
  @config.key? :proxy
end

# File lib/edis_client.rb, line 219
def stream_resource(path, options)
  connect.start do |http|
    http.get("/data/#{path}", header(options) || {}) do |chunk|
      yield chunk
    end
  end
end

# File lib/edis_client.rb, line 187
def validate_digest(options)
  unless options[:digest] || @config[:digest]
    raise ArgumentError, "A digest is required.  Please use gen_digest."
  end
end

# File lib/edis_client.rb, line 179
def validate_download(options)
  validate_digest options 
  validate_presence_of [:document_id, :attachment_id], options
end

# File lib/edis_client.rb, line 197
def validate_investigation_options(options)
  if options[:investigation_phase]
    validate_presence_of [:investigation_number], options,
      ":investigation_number is required when :investigation_phase is specified."
  end
end

# File lib/edis_client.rb, line 168
def validate_presence_of(requires, options, msg = nil)
  requires.each do |required|
    unless options[required]
      raise ArgumentError, msg || "Missing one or more required options #{requires}"
    end
  end
end