mackerel-client 0.3.0 → 0.4.0

data/lib/mackerel/organization.rb

@@ -0,0 +1,33 @@
+module Mackerel
+  class Organization
+    attr_accessor :name
+    def initialize(args = {})
+      @name               = args["name"]
+    end
+
+    def to_h
+      instance_variables.flat_map do |name|
+        respond_to?(name[1..-1]) ? [name[1..-1]] : []
+      end.each_with_object({}) do |name, hash|
+        hash[name] = public_send(name)
+      end.delete_if { |key, val| val == nil }
+    end
+
+    def to_json(options = nil)
+      return to_h.to_json(options)
+    end
+
+  end
+
+  module REST
+    module Organization
+
+      def get_organization()
+        command = ApiCommand.new(:get, '/api/v0/org', @api_key)
+        data = command.execute(client)
+        Mackerel::Organization.new(data)
+      end
+
+    end
+  end
+end

data/lib/mackerel/role.rb

@@ -0,0 +1,24 @@
+module Mackerel
+  class Role 
+
+    attr_accessor :name, :memo
+
+    def initialize(args = {})
+      @name                            = args["name"]
+      @memo                            = args["memo"]
+    end
+
+    def to_h
+      instance_variables.flat_map do |name|
+        respond_to?(name[1..-1]) ? [name[1..-1]] : []
+      end.each_with_object({}) do |name, hash|
+        hash[name] = public_send(name)
+      end.delete_if { |key, val| val == nil }
+    end
+
+    def to_json(options = nil)
+      return to_h.to_json(options)
+    end
+  end
+end
+

data/lib/mackerel/service.rb

@@ -0,0 +1,51 @@
+module Mackerel
+
+  class Service
+
+    attr_accessor :name, :memo, :roles
+
+    def initialize(args = {})
+      @name                            = args["name"]
+      @memo                            = args["memo"]
+      @roles                           = args["roles"]
+    end
+
+    def to_h
+      instance_variables.flat_map do |name|
+        respond_to?(name[1..-1]) ? [name[1..-1]] : []
+      end.each_with_object({}) do |name, hash|
+        hash[name] = public_send(name)
+      end.delete_if { |key, val| val == nil }
+    end
+
+    def to_json(options = nil)
+      return to_h.to_json(options)
+    end
+
+  end
+
+  module REST
+    module Service
+
+      def get_services()
+        command = ApiCommand.new(:get, '/api/v0/services', @api_key)
+        data = command.execute(client)
+        data['services'].map {|s| Mackerel::Service.new(s) }
+      end
+
+      def get_roles(serviceName)
+        command = ApiCommand.new(:get, "/api/v0/services/#{serviceName}/roles", @api_key)
+        data = command.execute(client)
+        data['roles'].map {|s| Mackerel::Role.new(s) }
+      end
+
+      def get_service_metric_names(serviceName)
+        command = ApiCommand.new(:get, "/api/v0/services/#{serviceName}/metric-names", @api_key)
+        data = command.execute(client)
+        data['names']
+      end
+
+    end
+  end
+end
+

data/lib/mackerel/user.rb

@@ -0,0 +1,42 @@
+module Mackerel
+
+  class User
+    attr_accessor :id, :screenName, :email
+    def initialize(args = {})
+      @id                              = args["id"]
+      @screenName                      = args["screenName"]
+      @email                           = args["email"]
+    end
+
+    def to_h
+      instance_variables.flat_map do |name|
+        respond_to?(name[1..-1]) ? [name[1..-1]] : []
+      end.each_with_object({}) do |name, hash|
+        hash[name] = public_send(name)
+      end.delete_if { |key, val| val == nil }
+    end
+
+    def to_json(options = nil)
+      return to_h.to_json(options)
+    end
+
+  end
+
+  module REST
+    module User
+
+      def get_users()
+        command = ApiCommand.new(:get, '/api/v0/users', @api_key)
+        data = command.execute(client)
+        data['users'].map{|u| Mackerel::User.new(u)}
+      end
+  
+      def remove_user(user_id)
+        command = ApiCommand.new(:delete, "/api/v0/users/#{user_id}", @api_key)
+        data = command.execute(client)
+        Mackerel::User.new(data)
+      end
+
+    end
+  end
+end

data/lib/mackerel/version.rb

@@ -0,0 +1,3 @@
+module Mackerel
+  VERSION = "0.2.0"
+end

data/lib/mackerel.rb

@@ -1,5 +1,4 @@
 require 'mackerel/client'
-require 'mackerel/monitor'
-require 'mackerel/metadata'
+require 'mackerel/api_command'
 require 'mackerel/runner'
-
+require 'mackerel/version'

data/spec/mackerel/alert_spec.rb

@@ -0,0 +1,103 @@
+require 'mackerel'
+require 'mackerel/host'
+require 'json'
+
+describe Mackerel::Client do
+  let(:api_key) { 'xxxxxxxx' }
+  let(:client) { Mackerel::Client.new(:mackerel_api_key => api_key) }
+
+  describe '#get_alerts' do
+    let(:stubbed_response) {
+      [
+        200,
+        {},
+        JSON.dump(response_object)
+      ]
+    }
+
+    let(:test_client) {
+      Faraday.new do |builder|
+        builder.adapter :test do |stubs|
+          stubs.get(api_path) { stubbed_response }
+        end
+      end
+    }
+
+    let(:api_path) { '/api/v0/alerts' }
+
+    let(:alertId) { 'abcdefg' }
+    let(:monitorId) { 'hijklmnopqr' }
+    let(:reason) { 'Nantonaku' }
+    let(:alerts) {
+      [
+        {
+          'id' => alertId,
+          'status' => 'OK',
+          'reason' => reason,
+          'monitorId' => monitorId,
+          'type' => 'check',
+          'openedAt' => 1234567890
+        }
+      ]
+    }
+
+    let(:response_object) {
+      { 'alerts' => alerts }
+    }
+
+    before do
+      allow(client).to receive(:http_client).and_return(test_client)
+    end
+
+    it "successfully get alerts" do
+      expect(client.get_alerts().map(&:to_h)).to eq(alerts)
+    end
+  end
+
+
+  describe '#close_alert' do
+    let(:stubbed_response) {
+      [
+        200,
+        {},
+        JSON.dump(response_object)
+      ]
+    }
+
+    let(:test_client) {
+      Faraday.new do |builder|
+        builder.adapter :test do |stubs|
+          stubs.post(api_path) { stubbed_response }
+        end
+      end
+    }
+
+    let(:alertId) { 'abcdefg' }
+    let(:monitorId) { 'hijklmnopqr' }
+    let(:reason) { 'Nantonaku' }
+    let(:api_path) { "/api/v0/alerts/#{alertId}/close"}
+
+    let(:alert) {
+      {
+        'id' => alertId,
+        'status' => 'OK',
+        'monitorId' => monitorId,
+        'type' => 'check',
+        'openedAt' => 1234567890
+      }
+    }
+
+    let(:response_object) {
+      Mackerel::Alert.new(alert)
+    }
+
+    before do
+      allow(client).to receive(:http_client).and_return(test_client)
+    end
+
+    it "successfully close alert" do
+      expect(client.close_alert(alertId, reason).to_h).to eq(alert)
+    end
+  end
+
+end

data/spec/mackerel/annotation_spec.rb

@@ -0,0 +1,208 @@
+require 'mackerel'
+require 'mackerel/host'
+require 'json'
+
+describe Mackerel::Client do
+  let(:api_key) { 'xxxxxxxx' }
+  let(:client) { Mackerel::Client.new(:mackerel_api_key => api_key) }
+
+  describe '#post_graph_annotation' do
+    let(:stubbed_response) {
+      [
+        200,
+        {},
+        JSON.dump(response_object)
+      ]
+    }
+
+    let(:test_client) {
+      Faraday.new do |builder|
+        builder.adapter :test do |stubs|
+          stubs.post(api_path) { stubbed_response }
+        end
+      end
+    }
+
+    let(:api_path) { '/api/v0/graph-annotations' }
+
+    let(:annotationId) { 'bllotation' }
+    let(:annotation) {
+      {
+        "title" => "deploy application",
+        "description" => "link: https://example.com/",
+        "from" => 1484000000,
+        "to" => 1484000030,
+        "service" => "ExampleService",
+        "roles" => [ "ExampleRole1", "ExampleRole2" ]
+      }
+    }
+
+    let(:response_object) {
+      Mackerel::Annotation.new( annotation.merge({ 'id' => annotationId }) )
+    }
+
+    before do
+      allow(client).to receive(:http_client).and_return(test_client)
+    end
+
+    it "successfully post annotation" do
+      expect(client.post_graph_annotation(annotation).to_h).to eq( annotation.merge({ 'id' => annotationId }) )
+    end
+  end
+
+
+  describe '#get_graph_annotations' do
+    let(:stubbed_response) {
+      [
+        200,
+        {},
+        JSON.dump(response_object)
+      ]
+    }
+
+    let(:test_client) {
+      Faraday.new do |builder|
+        builder.adapter :test do |stubs|
+          stubs.get(api_path) { stubbed_response }
+        end
+      end
+    }
+
+    let(:api_path) { "/api/v0/graph-annotations"}
+
+    let(:service) { "ExampleServic" }
+    let(:from) { 1484020459 }
+    let(:to) { 1484020759 }
+    let(:annotations) {
+      {
+         "graphAnnotations" => [
+            {
+               "id" => "2UdH1QcZQaw",
+               "title" => "Deploy application",
+               "description" => "Deploy description",
+               "from" => from,
+               "to" => to,
+               "service" => service
+            },
+            {
+               "id" => "2UdH1QuiGgj",
+               "title" => "Release application",
+               "description" => "Release description",
+               "from" => from,
+               "to" => to,
+               "service" => service,
+               "roles" => [ "ExampleRole1", "ExampleRole2" ]
+            }
+         ]
+       }
+    }
+
+    let(:response_object) {
+      annotations
+    }
+
+    before do
+      allow(client).to receive(:http_client).and_return(test_client)
+    end
+
+    it "successfully get annotations" do
+      expect(client.get_graph_annotations(service, from, to).map(&:to_h)).to eq(annotations['graphAnnotations'])
+    end
+  end
+
+
+  describe '#update_graph_annotation' do
+    let(:stubbed_response) {
+      [
+        200,
+        {},
+        JSON.dump(response_object)
+      ]
+    }
+
+    let(:test_client) {
+      Faraday.new do |builder|
+        builder.adapter :test do |stubs|
+          stubs.put(api_path) { stubbed_response }
+        end
+      end
+    }
+
+    let(:api_path) { "/api/v0/graph-annotations/#{id}"}
+
+    let(:id) { "abcdefg" }
+    let(:service) { "ExampleServic" }
+    let(:from) { 1484020459 }
+    let(:to) { 1484020759 }
+    let(:annotation) {
+      {
+         "title" => "Deploy application",
+         "description" => "Deploy description",
+         "from" => from,
+         "to" => to,
+         "service" => service
+      }
+    }
+
+    let(:response_object) {
+      annotation.merge({ "id" => id })
+    }
+
+    before do
+      allow(client).to receive(:http_client).and_return(test_client)
+    end
+
+    it "successfully update annotation" do
+      expect(client.update_graph_annotation(id, annotation).to_h ).to eq(annotation.merge({"id" => id}))
+    end
+  end
+
+
+  describe '#delete_graph_annotation' do
+    let(:stubbed_response) {
+      [
+        200,
+        {},
+        JSON.dump(response_object)
+      ]
+    }
+
+    let(:test_client) {
+      Faraday.new do |builder|
+        builder.adapter :test do |stubs|
+          stubs.delete(api_path) { stubbed_response }
+        end
+      end
+    }
+
+    let(:api_path) { "/api/v0/graph-annotations/#{id}"}
+
+    let(:id) { "abcdefg" }
+    let(:service) { "ExampleServic" }
+    let(:from) { 1484020459 }
+    let(:to) { 1484020759 }
+    let(:annotation) {
+      {
+         "title" => "Deploy application",
+         "description" => "Deploy description",
+         "from" => from,
+         "to" => to,
+         "service" => service
+      }
+    }
+
+    let(:response_object) {
+      annotation.merge({ "id" => id })
+    }
+
+    before do
+      allow(client).to receive(:http_client).and_return(test_client)
+    end
+
+    it "successfully delete annotation" do
+      expect(client.delete_graph_annotation(id).to_h ).to eq(annotation.merge({"id" => id}))
+    end
+  end
+
+
+end

data/spec/mackerel/channel_spec.rb

@@ -0,0 +1,50 @@
+require 'mackerel'
+require 'mackerel/host'
+require 'json'
+
+describe Mackerel::Client do
+  let(:api_key) { 'xxxxxxxx' }
+  let(:client) { Mackerel::Client.new(:mackerel_api_key => api_key) }
+
+  describe '#get_channels' do
+
+    let(:api_path) { "/api/v0/channels" }
+    let(:stubbed_response) {
+      [
+        200,
+        {},
+        JSON.dump(response_object)
+      ]
+    }
+
+    let(:test_client) {
+      Faraday.new do |builder|
+        builder.adapter :test do |stubs|
+          stubs.get(api_path) { stubbed_response }
+        end
+      end
+    }
+
+    let(:channels) { 
+      {
+        "channels" => [
+          { "id" => "361DhijkGFS" , "name" => "Default", "type" => "email" },
+          { "id" => "361FijklHGT" , "name" => "alert_infomation", "type" => "slack" }
+        ]
+      }
+    }
+
+    let(:response_object) {
+      channels
+    }
+
+    before do
+      allow(client).to receive(:http_client).and_return(test_client)
+    end
+
+    it "successfully get channels" do
+      expect(client.get_channels().map(&:to_h)).to eq(channels['channels'])
+    end
+  end
+
+end