#!/usr/bin/env ruby
 
require 'open-uri'
require 'rexml/document'
require 'rexml/xpath'
 
url = 'http://rpm.newrelic.com/accounts.xml?include=application_health'
 
headers = {'x-license-key' => 'YOUR LICENSE KEY'}
 
name = $0.split('_').last.split('.').first.downcase
 
sections = {
  'cpu' => {:title => 'CPU', :metric => '%'},
  'memory' => {:title => 'Memory', :metric => 'MB'},
  'errors' => {:title => 'Errors', :metric => 'errors per minute'},
  'response' => {:title => 'Response Time', :metric => 'ms'},
  'throughput' => {:title => 'Throughput', :metric => 'calls per minute'},
  'db' => {:title => 'DB', :metric => '%'}
}
 
unless sections.keys.member? name
  puts "ERROR: specify section in #{sections.keys.inspect}"
  puts "example: ln -s /usr/share/munin/plugins/relic.rb relic_db"
  exit 1
end
 
section = sections[name]
 
if ARGV.first == 'autoconf'
  puts 'yes'
  exit 0
end
 
if ARGV.first == 'config'
  puts "graph_category New Relic"
  puts "graph_title #{section[:title]}"
  puts "graph_vlabel #{section[:metric]}"
 
  puts "#{name}.label #{name}"
  exit 0
end
 
 
open(url, headers) do |f|
  doc = REXML::Document.new(f.read)
  REXML::XPath.each(doc, "//threshold_value") do |metric|
    if metric.attributes['name'].downcase.match(name)
      puts "#{name}.value #{metric.attributes['metric_value']}"
    end
  end
end