pagerduty_to_graphite.rb

#!/usr/bin/env ruby

require 'json'
require 'date'
require 'time'
require 'socket'
require 'optparse'
require 'net/https'
require 'uri'


options = { :start_date => '2012-01-01' }
OptionParser.new do |opts|
  opts.banner = "Usage: pagerduty_to_graphite.rb [options]"
  opts.on("--url URL", "--pagerduty_url URL", "Pagerduty URL") do |v|
    options[:pagerduty_url] = v
  end
  opts.on("-u USERNAME", "--pagerduty_user USERNAME", "Pagerduty username") do |v|
    options[:pagerduty_user] = v
  end
  opts.on("-p PASSWORD", "--pagerduty_pass PASSWORD", "Pagerduty password") do |v|
    options[:pagerduty_pass] = v
  end
  opts.on("-c HOST:PORT", "--carbon_socket HOST:PORT", "Graphite Carbon socket") do |v|
    options[:carbon_socket] = v
  end
  opts.on("-k API_KEY", "--key API_KEY", "Pagerduty API Key") do |v|
    options[:pagerduty_apikey] = v
  end
  opts.on("-s YYYY-MM-DD", "--start_date YYYY-MM-DD", "Earliest date to retrieve from Pagerduty") do |v|
    options[:start_date] = v
  end
end.parse!

required_opts = [:pagerduty_url, :carbon_socket, :start_date]
required_opts.each do |o|
  unless options[o]
    puts "Missing required parameter --#{o.to_s}"
    exit 2
  end
end

unless options.include?(:pagerduty_apikey) or (options.include?(:pagerduty_user) and options.include?(:pagerduty_pass))
  puts "Missing authentication parameters, either apikey or username and password"
  exit 2
end

# Connect to Graphite (Carbon) listener
if options[:carbon_socket] =~ /^(carbon:\/\/)?([^:]+):([0-9]+)$/
  host, port = $2, $3
  carbon = TCPSocket.new host, port
end

uri = URI.parse(options[:pagerduty_url])
unless uri.is_a? URI::HTTPS
  puts "Invalid Pagerduty URL: #{options[:pagerduty_url]}"
  puts "Must be a complete https:// URL."
  exit 1
end

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

# force SSL since we are passing creds
http.use_ssl = true

last_date = options[:start_date]
last_incident_number = 0
offset = 0
while true
  # Connect to PagerDuty Incidents API
  # Make sure to get the results in ascending order of creation date
  request = Net::HTTP::Get.new("#{uri.request_uri}/api/v1/incidents?since=#{last_date}&sort_by=created_on:asc&offset=#{offset}")

  # default to api if specified
  if options[:pagerduty_apikey]
    request['Authorization'] = 'Token token=' + options[:pagerduty_apikey]
  else
    request.basic_auth(options[:pagerduty_user], options[:pagerduty_pass])
  end

  response = http.request(request)

  # Retrieve as many incidents as we can
  # PagerDuty caps us at 100 per request
  incidents = JSON.parse(response.body)["incidents"]
  puts "FOUND #{incidents.count} INCIDENTS SINCE #{last_date}"

  incidents.each do |alert|
    # Don't double-count incidents on the since date
    # This assumes incidents numbers are monotonically increasing
    next if alert['incident_number'].to_i <= last_incident_number

    # Customize based on your use case
    case alert['service']['name']
    when 'Pingdom'
      metric = "pingdom.#{alert['incident_key'].gsub(/\./, '_').gsub(/[\(\)]/, '').gsub(/\s+/, '.')}"
    when 'nagios'
      data = alert['trigger_summary_data']
      outage = data['SERVICEDESC'] === '' ? 'host_down' : data['SERVICEDESC']
      begin
        metric = "nagios.#{data['HOSTNAME'].gsub(/\./, '_')}.#{outage}"
      rescue
        puts "UNKNOWN ALERT: #{alert.to_json}"
      end
    when 'Enterprise Zendesk'
      metric = "enterprise.zendesk.#{alert['service']['id']}"
    when 'Sensu'
      metric = "sensu.#{alert['service']['id']}"
    else
      puts "UNKNOWN ALERT: #{alert.to_json}"
    end

    # If have a valid metric string push it to Graphite
    if metric
      carbon.puts "alerts.#{metric} 1 #{Time.parse(alert['created_on']).to_i}"
    end
  end
  
  # Stop if we retrieved the last 100-page of incidents
  break if incidents.count < 100
  
  # Bump our start date to the last known date
  if last_date == Time.parse(incidents.last['created_on']).to_date.to_s
    offset += 100
  else
    last_date = Time.parse(incidents.last['created_on']).to_date.to_s
    offset = 0
  end
  last_incident_number = incidents.last['incident_number'].to_i
end

puts "FINISHED"