yle-newrelic_api_v2 0.0.2

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 9151a614abbee9d5c5f34c66570b1a4d8601b528
+  data.tar.gz: 25aacb884bc37810b8cad27979887879a915ed5c
+SHA512:
+  metadata.gz: 712a889db5bb3a5534c288918945dd0d48875da5074dda022cde892fe63611ed2b7ed45624bb397407692881b8c39c22b30f7f44c62f81eefbbbec6efb9b05d1
+  data.tar.gz: 43c1d5fa0bd08e93cc84278f457eb33897644f968be2826eae208b04991f762e776b634b8028c1c166afad961dac2dd5699a99f00ca739e7fe2dc214bf35fe36

data/lib/yle-newrelic_api_v2.rb

@@ -0,0 +1,102 @@
+#!/usr/bin/env ruby
+
+require 'net/http'
+require 'json'
+require 'yaml'
+
+require_relative 'yle-newrelic_api_v2/yle-newrelic_api_v2_user'
+require_relative 'yle-newrelic_api_v2/yle-newrelic_api_v2_notification_channel'
+require_relative 'yle-newrelic_api_v2/yle-newrelic_api_v2_alert_policy'
+require_relative 'yle-newrelic_api_v2/yle-newrelic_api_v2_servers'
+
+module Yle
+  module NewRelicApi
+    def get(params = {})
+      @endpoint.query = URI.encode_www_form(params)
+      Net::HTTP.start(@endpoint.host, @endpoint.port,
+                      use_ssl: @endpoint.scheme == 'https') do |http|
+        request = Net::HTTP::Get.new @endpoint
+        request.add_field('X-Api-Key', @api_key)
+        response = http.request request
+        # raise HTTP Error, if response code is not 2XX
+        response.value
+        process_response_body(response)
+      end
+    end
+
+    def get_all_pages(params = {})
+      count = pages(params)
+      page = 1
+      while page <= count do
+        params['page'] = page
+        get(params)
+        page += 1
+      end
+    end
+
+    def pages(params = {})
+      response = nil
+      @endpoint.query = URI.encode_www_form(params)
+      Net::HTTP.start(@endpoint.host, @endpoint.port,
+                      use_ssl: @endpoint.scheme == 'https') do |http|
+        request = Net::HTTP::Get.new @endpoint
+        request.add_field('X-Api-Key', @api_key)
+        response = http.request request
+        # raise HTTP Error, if response code is not 2XX
+        response.value
+      end
+      last_page(response)
+    end
+
+    def put(params, endpoint)
+      uri = URI("https://api.newrelic.com/v2/#{endpoint}.json")
+      Net::HTTP.start(uri.host, uri.port, use_ssl: uri.scheme == 'https') do |http|
+        request = Net::HTTP::Put.new uri
+        request.add_field('X-Api-Key', @api_key)
+        request.add_field('Content-Type', 'application/json')
+        request.body = params.to_json
+        response = http.request request
+        # raise HTTP Error, if response code is not 2XX
+        response.value
+      end
+    end
+
+    def clear
+      @data.clear
+    end
+
+    def last_page(response)
+      return 1 unless response.key?('Link')
+      m = response['Link'].match(/<http.*\?page=(\d+)>; rel="last"/)
+      return 1 unless m
+      m[1].to_i
+    end
+
+    def get_next_page_number(response)
+      m = response['Link'].match(/<http.*\?page=(\d+)>; rel="next"/)
+      m[1].to_i
+    end
+
+    def delete(endpoint)
+      uri = URI(endpoint)
+      Net::HTTP.start(uri.host, uri.port, use_ssl: uri.scheme == 'https') do |http|
+        request = Net::HTTP::Delete.new uri
+        request.add_field('X-Api-Key', @api_key)
+        response = http.request request
+        # raise HTTP Error, if response code is not 2XX
+        response.value
+      end
+    end
+
+    private
+
+    def process_response_body(response)
+      # check that response has a body
+      if response.class.body_permitted?
+        @data << response.body
+      else
+        raise "Empty response from New Relic API:\n\n #{response}"
+      end
+    end
+  end
+end

data/lib/yle-newrelic_api_v2/yle-newrelic_api_v2_alert_policy.rb

@@ -0,0 +1,118 @@
+require_relative '../yle-newrelic_api_v2'
+require_relative 'yle-newrelic_api_v2_notification_channel'
+
+module Yle
+  module NewRelicApi
+    class AlertPolicy
+      include Yle::NewRelicApi
+
+      def initialize(api_key)
+        @api_key = api_key
+        @data = []
+        @endpoint = URI('https://api.newrelic.com/v2/alert_policies.json')
+      end
+
+      def get_alert_policy_id(name)
+        hits = []
+        @data.each do |d|
+          j = JSON.parse(d)
+          matches = j['alert_policies'].select { |policy| policy['name'] == name }
+          hits << matches unless matches.empty?
+        end
+        hits[0][0]['id']
+      end
+
+      def get_alert_policy(policy_id)
+        hits = []
+        @data.each do |d|
+          j = JSON.parse(d)
+          matches = j['alert_policies'].select { |policy| policy['id'] == policy_id }
+          hits << matches unless matches.empty?
+        end
+        hits[0][0]
+      end
+
+      def generate_alert_policy(config, policy_id)
+        policy = get_alert_policy(policy_id)
+
+        config['conditions'].each do |item|
+          policy['conditions'].each do |alert|
+            update_alert(item, alert)
+          end
+        end
+
+        update_notification_channel(policy, config)
+        policy
+      end
+
+      private
+
+      def update_alert(item, alert)
+        type = item[0]
+
+        if alert['type'] == 'server_downtime' && type == 'server_downtime'
+          downtime = item[1]['downtime']
+          alert['trigger_minutes'] = downtime['trigger_minutes']
+          alert['enabled'] = downtime['enabled']
+        elsif alert['type'] == type
+          critical = item[1]['critical']
+          caution = item[1]['caution']
+
+          if alert['severity'] == 'critical'
+            alert['threshold'] = critical['threshold']
+            alert['trigger_minutes'] = critical['trigger_minutes']
+          elsif alert['severity'] == 'caution'
+            alert['threshold'] = caution['threshold']
+            alert['trigger_minutes'] = caution['trigger_minutes']
+          else
+            raise "Unknown severity level #{alert['severity']}"
+          end
+        end
+      end
+
+      def update_notification_channel(policy, config)
+        notification_channels_params = { 'format' => 'JSON' }
+        channel_client = Yle::NewRelicApi::NotificationChannel.new(@api_key)
+        channel_client.get_all_pages(notification_channels_params)
+
+        # empty notification channels
+        policy['links']['notification_channels'].clear
+
+        config['notification_channels'].each do |item|
+          type = item[0]
+          values = item[1]
+
+          case type
+          when 'email'
+            values.each do |v|
+              policy['links']['notification_channels'] << channel_client.get_email_id(v)
+            end
+          when 'webhook'
+            values.each do |v|
+              policy['links']['notification_channels'] << channel_client.get_webhook_id(v)
+            end
+          when 'notification_group'
+            values.each do |v|
+              policy['links']['notification_channels'] << channel_client.get_group_id(v)
+            end
+          when 'pager_duty'
+            values.each do |v|
+              policy['links']['notification_channels'] << channel_client.get_pager_duty_id(v)
+            end
+          when 'hipchat'
+            values.each do |v|
+              policy['links']['notification_channels'] << channel_client.get_hipchat_id(v)
+            end
+          when 'user'
+            values.each do |v|
+              policy['links']['notification_channels'] << channel_client.get_user_id(v)
+            end
+          else
+            raise "Unknown notification channel type #{type}"
+          end
+        end
+        policy
+      end
+    end
+  end
+end

data/lib/yle-newrelic_api_v2/yle-newrelic_api_v2_notification_channel.rb

@@ -0,0 +1,132 @@
+require_relative '../yle-newrelic_api_v2'
+require_relative 'yle-newrelic_api_v2_user'
+
+module Yle
+  module NewRelicApi
+    class NotificationChannel
+      include Yle::NewRelicApi
+
+      def initialize(api_key)
+        @api_key = api_key
+        @data = []
+        @endpoint = URI('https://api.newrelic.com/v2/notification_channels.json')
+      end
+
+      def get_group_id(name)
+        matches = get_id(name)
+        if matches.nil?
+          raise "Did not find any notification groups with name #{name}"
+        elsif matches.length > 1
+          raise "Found multiple matches for notification group name: #{matches}"
+        else
+          return matches[0]['id']
+        end
+      end
+
+      def get_email_id(name)
+        matches = get_id_for_email(name)
+        if matches.empty?
+          raise "Did not find any emails with name #{name}"
+        elsif matches.length > 1
+          raise "Found multiple matches for email: #{matches}"
+        else
+          return matches[0]['id']
+        end
+      end
+
+      def get_webhook_id(name)
+        matches = get_id(name)
+        if matches.nil?
+          raise "Did not find any webhooks with name #{name}"
+        elsif matches.length > 1
+          raise "Found multiple matches for webhook: #{matches}"
+        else
+          return matches[0]['id']
+        end
+      end
+
+      def get_pager_duty_id(name)
+        matches = get_id_for_pagerduty(name)
+        if matches.nil?
+          raise "Did not find any PagerDuty channels with name #{name}"
+        elsif matches.length > 1
+          raise "Found multiple matches for PagerDuty: #{matches}"
+        else
+          return matches[0]['id']
+        end
+      end
+
+      def get_hipchat_id(name)
+        matches = get_id(name)
+        if matches.nil?
+          raise "Did not find any HipChat channels with name #{name}"
+        elsif matches.length > 1
+          raise "Found multiple matches for HipChat: #{matches}"
+        else
+          return matches[0]['id']
+        end
+      end
+
+      def get_user_id(email)
+        params = {
+          'filter[email]' => email,
+          'format' => 'JSON'
+        }
+
+        client = Yle::NewRelicApi::User.new(@api_key)
+        client.get(params)
+        user_id = client.get_user_id(email)
+        user_notification_channel_id = get_id_for_user(user_id)
+        user_notification_channel_id
+      end
+
+      private
+
+      def get_id(name)
+        hits = []
+        @data.each do |d|
+          j = JSON.parse(d)
+          matches = j['notification_channels'].select { |group| group['name'] == name }
+          hits << matches unless matches.empty?
+        end
+        hits[0]
+      end
+
+      def get_id_for_email(email)
+        hits = []
+        @data.each do |d|
+          j = JSON.parse(d)
+          matches = j['notification_channels'].select { |group| group['email'] == email }
+          hits << matches unless matches.empty?
+        end
+        hits[0]
+      end
+
+      def get_id_for_pagerduty(service)
+        hits = []
+        @data.each do |d|
+          j = JSON.parse(d)
+          matches = j['notification_channels'].select { |group| group['service'] == service }
+          hits << matches unless matches.empty?
+        end
+        hits[0]
+      end
+
+      def get_id_for_user(user_id)
+        @data.each do |d|
+          j = JSON.parse(d)
+          j['notification_channels'].each do |c|
+            if c.key?('links')
+              if c['links'].key?('user')
+                if c['links']['user'] == user_id
+                  # found user id, return notification channel id
+                  return c['id']
+                end
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/yle-newrelic_api_v2/yle-newrelic_api_v2_servers.rb

@@ -0,0 +1,74 @@
+require 'logger'
+
+require_relative '../yle-newrelic_api_v2'
+
+module Yle
+  module NewRelicApi
+    class Servers
+      include Yle::NewRelicApi
+
+      def initialize(api_key)
+        @api_key = api_key
+        @data = []
+        @endpoint = URI('https://api.newrelic.com/v2/servers.json')
+        @pages = nil
+      end
+
+      def servers
+        servers = []
+        @data.each do |d|
+          j = JSON.parse(d)
+          j['servers'].each do |s|
+            servers << s['id']
+          end
+        end
+        servers
+      end
+
+      def healthy?(server_id)
+        @data.each do |d|
+          j = JSON.parse(d)
+          j['servers'].each do |s|
+            return s['reporting'] if s['id'] == server_id
+          end
+        end
+      end
+
+      def delete_non_reporting(server_id)
+        if !healthy?(server_id)
+          endpoint = "https://api.newrelic.com/v2/servers/#{server_id}.json"
+          if hostname(server_id).start_with?('ip-10-')
+            puts "Deleting #{hostname(server_id)}"
+            delete(endpoint)
+          else
+            require 'time'
+            now = Time.new
+            not_reported_for = ((now - Time.parse(last_reported(server_id))) / 60 / 60 / 24).round
+            if not_reported_for > 5
+              puts "Deleting #{hostname(server_id)} - not reported for #{not_reported_for} days"
+              delete(endpoint)
+            end
+          end
+        end
+      end
+
+      def last_reported(server_id)
+        @data.each do |d|
+          j = JSON.parse(d)
+          j['servers'].each do |s|
+            return s['last_reported_at'] if s['id'] == server_id
+          end
+        end
+      end
+
+      def hostname(server_id)
+        @data.each do |d|
+          j = JSON.parse(d)
+          j['servers'].each do |s|
+            return s['host'] if s['id'] == server_id
+          end
+        end
+      end
+    end
+  end
+end

data/lib/yle-newrelic_api_v2/yle-newrelic_api_v2_user.rb

@@ -0,0 +1,36 @@
+require_relative '../yle-newrelic_api_v2'
+
+module Yle
+  module NewRelicApi
+    class User
+      include Yle::NewRelicApi
+
+      def initialize(api_key)
+        @api_key = api_key
+        @data = []
+        @endpoint = URI('https://api.newrelic.com/v2/users.json')
+      end
+
+      def get_user_id(email)
+        matches = get_id_by_email(email)
+        if matches.empty?
+          raise "Did not find any users with email #{email}"
+        elsif matches.length > 1
+          raise "Found multiple matches for email: #{matches}"
+        else
+          return matches[0]['id']
+        end
+      end
+
+      private
+
+      def get_id_by_email(email)
+        @data.each do |d|
+          j = JSON.parse(d)
+          matches = j['users'].select { |user| user['email'] == email }
+          return matches
+        end
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,48 @@
+--- !ruby/object:Gem::Specification
+name: yle-newrelic_api_v2
+version: !ruby/object:Gem::Version
+  version: 0.0.2
+platform: ruby
+authors:
+- Aki Hänninen
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2016-05-26 00:00:00.000000000 Z
+dependencies: []
+description: 
+email:
+- aki.hanninen@affecto.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/yle-newrelic_api_v2.rb
+- lib/yle-newrelic_api_v2/yle-newrelic_api_v2_alert_policy.rb
+- lib/yle-newrelic_api_v2/yle-newrelic_api_v2_notification_channel.rb
+- lib/yle-newrelic_api_v2/yle-newrelic_api_v2_servers.rb
+- lib/yle-newrelic_api_v2/yle-newrelic_api_v2_user.rb
+homepage: 
+licenses: []
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.4.5.1
+signing_key: 
+specification_version: 4
+summary: New Relic API v2 implementation for Ruby
+test_files: []