Then /^I should receive valid JSON$/ do
  yajl = Yajl::Parser.new
  lambda {
    @response = yajl.parse(response_body)
  }.should_not raise_error

  case
  when @response.class == Array
    next
  when @response.keys.first == "hosts"
    @response["hosts"].should respond_to(:size)
  when @response[@response.keys.first].respond_to?(:size)
    host    = @response.keys.first
    plugins = @response[host]
    plugins.size.should > 0
  else
    host   = @response.keys.first
    plugin = @response[host].keys.first
    metric = @response[host][plugin].keys.first

    host.should_not be_nil
    plugin.should_not be_nil
    metric.should_not be_nil

    data = @response[host][plugin][metric]["data"]
  end

end

Then /^I should receive JSON wrapped in a callback named "([^\"]*)"$/ do |callback|
  response_body.should =~ /^#{callback}\(.+\)$/
end

Then /^the JSON should have a plugin instance named "([^\"]*)"$/ do |plugin_instance|
  yajl = Yajl::Parser.new
  data = yajl.parse(response_body)

  data.values.map { |k,v| k.values }.map {|k,v| k.keys }.flatten.include?(plugin_instance).should be_true
end

Then /^the JSON should have a plugin named "([^\"]*)"$/ do |plugin|
  yajl = Yajl::Parser.new
  data = yajl.parse(response_body)

  data.values.map { |k,v| k.keys }.flatten.include?(plugin).should be_true
end

Then /^the JSON should have multiple plugin instances under the "([^\"]*)" plugin$/ do |arg1|
  yajl = Yajl::Parser.new
  data = yajl.parse(response_body)

  data.values.map { |k,v| k[arg1] }.map { |k,v| k.keys }.flatten.size.should > 1
end

Then /^each plugin instance should have a different color$/ do
  yajl = Yajl::Parser.new
  data = yajl.parse(response_body)

  @colours = []
  data.values.map { |k,v| k.values }.map {|k,v| k.values }.map {|k,v| k.values }.map do |a|
    a.each do |b|
      string = b["color"]
      string.should =~ /^#[0-9a-fA-F]+$/
      @colours << string
    end
  end

  @colours.uniq.size.should == @colours.size

end

Then /^the plugin instance should have a color$/ do
  Then "each plugin instance should have a different color"
end


Given /^I have the "([^\"]*)" plugin collecting data on multiple ports$/ do |plugin|
  Dir.glob("/var/lib/collectd/rrd/theodor/tcpconns*").size.should > 1
end

Then /^I should see multiple plugins$/ do
  @response.should_not be_nil
  host = @response.keys.first
  @response[host].keys.size.should > 1
end

Then /^I should see multiple hosts$/ do
  @response.should_not be_nil
  @response.keys.size.should > 1
end

Then /^the JSON should have a list of hosts$/ do
  @response["hosts"].size.should > 0
end

Given /^a list of hosts exist$/ do
  step 'I go to /data'
  step 'the request should succeed'
  step 'I should receive valid JSON'
  step 'the JSON should have a list of hosts'
end

When /^I visit "([^"]*)" on the first available host$/ do |glob|
  host = @response["hosts"].first
  url  = "/data/#{host}/#{glob}"
  step "I go to #{url}"
end


When /^I visit the first available host$/ do
  step 'I go to /data'
  step 'the request should succeed'
  step 'I should receive valid JSON'
  step 'the JSON should have a list of hosts'

  host = @response["hosts"].first
  url  = "/data/#{host}"
  step "I go to #{url}"
end

When /^I visit the first two available hosts$/ do
  step 'I go to /data'
  step 'the request should succeed'
  step 'I should receive valid JSON'
  step 'the JSON should have a list of hosts'

  host = @response["hosts"][0..1].join(',')
  url  = "/data/#{host}"
  step "I go to #{url}"
end

Then /^the JSON should have a list of plugins$/ do
  host    = @response.keys.first
  plugins = @response[host]
  plugins.size.should > 0
end

Then /^the JSON should have a list of types$/ do
  @response.size.should > 0
  @response.each do |type|
    %w(dataset datasource type min max).each do |attr|
      type[attr].should_not be_nil
    end
  end
end


When /^I visit "([^"]*)" on the first available host with the following query parameters:$/ do |glob, table|
  host = @response["hosts"].first
  url  = "/data/#{host}/#{glob}"

  params = Hash[table.hashes.map { |hash| [hash["parameter"], hash["value"]] }]
  params.each do |key, value|
    if value == "1 hour ago"
      params[key] = (Time.now() - 3600).to_i.to_s
    end
    if value == "now"
      params[key] = Time.now().to_i.to_s
    end
  end
  query  = params.map{|k,v| "#{CGI.escape(k)}=#{CGI.escape(v)}"}.join("&")
  url   += "?#{query}"

  step "I go to #{url}"
end

Then /^I should see the following parameters for each plugin instance:$/ do |table|
  params = Hash[table.hashes.map { |hash| [hash["parameter"], hash["value"]] }]

  @response.should_not be_nil

  @response.each_pair do |host, plugin|
    plugin.each_pair do |instance, metric|
      metric.each_pair do |k, series|
        series.each_pair do |k, data|
          params.each do |key, value|
            data[key].should == value.to_i
          end
        end
      end
    end
  end

end

Then /^I should see a 95th percentile value for each plugin instance$/ do
  @response.should_not be_nil

  @response.each_pair do |host, plugin|
    plugin.each_pair do |instance, metric|
      metric.each_pair do |k, series|
        series.each_pair do |k, data|
          data['percentile_95'].should >= 0
        end
      end
    end
  end

end