flapjack-diner 1.0.0 → 1.2.0

data/lib/flapjack-diner/argument_validator.rb

@@ -1,6 +1,5 @@
 module Flapjack
   class ArgumentValidator
-
     attr_reader :query
 
     def initialize(query = {})
@@ -13,49 +12,56 @@ module Flapjack
       validations = args.delete(:as)
       validations = [validations] unless validations.is_a?(Array)
 
-      if elements = args[:query]
+      elements = args[:query]
+
+      unless elements.nil?
         elements = [elements] unless elements.is_a?(Array)
-        validations.each do |validation|
-          __send__(validation.to_s.downcase, *elements)
-        end
+        validations.each {|v|  __send__(v.to_s.downcase, *elements) }
       end
 
-      raise ArgumentError.new(@errors.join('; ')) unless @errors.empty?
+      raise(ArgumentError, @errors.join('; ')) unless @errors.empty?
     end
 
     private
 
+    def valid_time_str?(str)
+      Time.iso8601(str)
+      true
+    rescue ArgumentError
+      false
+    end
+
     def time(*elements)
       elements.each do |element|
-        if target = @query[element]
-          next if target.respond_to?(:iso8601) || (target.is_a?(String) &&
-            (begin; Time.iso8601(target); true; rescue ArgumentError; false; end))
-          @errors << "'#{target}' should be a time object or ISO 8601-formatted string."
-        end
+        target = @query[element]
+        next if target.nil? || target.respond_to?(:iso8601) ||
+          (target.is_a?(String) && valid_time_str?(target))
+        @errors << "'#{target}' should be a time object or ISO " \
+                   '8601-formatted string.'
       end
     end
 
     def boolean(*elements)
       elements.each do |element|
-        if target = @query[element]
-          next if [TrueClass, FalseClass].include?(target.class)
-          @errors << "'#{target}' should be 'true' or 'false'."
-        end
+        target = @query[element]
+        next if target.nil? || [TrueClass, FalseClass].include?(target.class)
+        @errors << "'#{target}' should be 'true' or 'false'."
       end
     end
 
     def array_of_strings(*elements)
       elements.each do |element|
-        if target = @query[element]
-          next if target.is_a?(Array) && target.all? {|t| t.is_a?(String)}
-          @errors << "'#{target}' should be an Array of Strings."
-        end
+        target = @query[element]
+        next if target.nil? || (target.is_a?(Array) &&
+          target.all? {|t| t.is_a?(String) })
+        @errors << "'#{target}' should be an Array of Strings."
       end
     end
 
     def required(*elements)
       elements.each do |element|
-        @errors << "'#{element}' is required." if @query[element].nil?
+        next unless @query[element].nil?
+        @errors << "'#{element}' is required."
       end
     end
 
@@ -64,10 +70,12 @@ module Flapjack
     end
 
     def method_missing(name, *args)
-      return super unless klass = classify_name(name)
+      klass = classify_name(name)
+      return super if klass.nil?
       elements = args
       elements.each do |element|
-        @errors << "'#{element}' is expected to be a #{klass}" unless @query[element].nil? || @query[element].is_a?(klass)
+        next if @query[element].nil? || @query[element].is_a?(klass)
+        @errors << "'#{element}' is expected to be a #{klass}"
       end
     end
 
@@ -75,6 +83,7 @@ module Flapjack
       class_name = name.to_s.split('_').map(&:capitalize).join
       Module.const_get(class_name)
     rescue NameError
+      nil
     end
   end
 end

data/lib/flapjack-diner/resources/checks.rb

@@ -0,0 +1,58 @@
+require 'httparty'
+require 'json'
+require 'uri'
+
+require 'flapjack-diner/version'
+require 'flapjack-diner/argument_validator'
+
+module Flapjack
+  module Diner
+    module Resources
+      module Checks
+        def create_checks(*args)
+          data = unwrap_create_data(*args)
+          validate_params(data) do
+            validate :query => :entity_id, :as => [:required, :string]
+            validate :query => :name,      :as => [:required, :string]
+            validate :query => :tags,      :as => :array_of_strings
+          end
+          perform_post('/checks', nil, :checks => data)
+        end
+
+        def checks(*ids)
+          perform_get('checks', '/checks', ids)
+        end
+
+        def update_checks(*args)
+          ids, params = unwrap_ids(*args), unwrap_params(*args)
+          raise "'update_checks' requires at least one check id " \
+                'parameter' if ids.nil? || ids.empty?
+          validate_params(params) do
+            validate :query => :enabled, :as => :boolean
+            validate :query => :tags,    :as => :array_of_strings
+          end
+          perform_patch("/checks/#{escaped_ids(ids)}", nil,
+                        update_checks_ops(params))
+        end
+
+        private
+
+        def update_checks_ops(params)
+          ops = params.each_with_object([]) do |(k, v), memo|
+            case k
+            when :enabled
+              memo << patch_replace('checks', k, v)
+            when :add_tag
+              memo << patch_add('checks', 'tags', v)
+            when :remove_tag
+              memo << patch_remove('checks', 'tags', v)
+            end
+          end
+          raise "'update_checks' did not find any valid update " \
+                'fields' if ops.empty?
+          ops
+        end
+      end
+    end
+  end
+end

data/lib/flapjack-diner/resources/contacts.rb

@@ -0,0 +1,70 @@
+require 'httparty'
+require 'json'
+require 'uri'
+
+require 'flapjack-diner/version'
+require 'flapjack-diner/argument_validator'
+
+module Flapjack
+  module Diner
+    module Resources
+      module Contacts
+        def create_contacts(*args)
+          data = unwrap_create_data(*args)
+          validate_params(data) do
+            validate :query => [:first_name, :last_name, :email],
+                     :as => [:required, :string]
+            validate :query => :timezone,   :as => :string
+            validate :query => :tags,       :as => :array_of_strings
+          end
+          perform_post('/contacts', nil, :contacts => data)
+        end
+
+        def contacts(*ids)
+          perform_get('contacts', '/contacts', ids)
+        end
+
+        def update_contacts(*args)
+          ids, params = unwrap_ids(*args), unwrap_params(*args)
+          raise "'update_contacts' requires at least one contact id " \
+                'parameter' if ids.nil? || ids.empty?
+          validate_params(params) do
+            validate :query => [:first_name, :last_name,
+                                :email, :timezone], :as => :string
+            validate :query => :tags,       :as => :array_of_strings
+          end
+          perform_patch("/contacts/#{escaped_ids(ids)}", nil,
+                        update_contacts_ops(params))
+        end
+
+        def delete_contacts(*ids)
+          raise "'delete_contacts' requires at least one contact id " \
+                'parameter' if ids.nil? || ids.empty?
+          perform_delete('/contacts', ids)
+        end
+
+        private
+
+        def update_contacts_ops(params)
+          ops = params.each_with_object([]) do |(k, v), memo|
+            case k
+            when :add_entity
+              memo << patch_add('contacts', 'entities', v)
+            when :remove_entity
+              memo << patch_remove('contacts', 'entities', v)
+            when :add_notification_rule
+              memo << patch_add('contacts', 'notification_rules', v)
+            when :remove_notification_rule
+              memo << patch_remove('contacts', 'notification_rules', v)
+            when :first_name, :last_name, :email, :timezone
+              memo << patch_replace('contacts', k, v)
+            end
+          end
+          raise "'update_contacts' did not find any valid update " \
+                'fields' if ops.empty?
+          ops
+        end
+      end
+    end
+  end
+end

data/lib/flapjack-diner/resources/entities.rb

@@ -0,0 +1,68 @@
+require 'httparty'
+require 'json'
+require 'uri'
+
+require 'flapjack-diner/version'
+require 'flapjack-diner/argument_validator'
+
+module Flapjack
+  module Diner
+    module Resources
+      module Entities
+        def create_entities(*args)
+          data = unwrap_create_data(*args)
+          validate_params(data) do
+            validate :query => :id,   :as => [:required, :string]
+            validate :query => :name, :as => :string
+            validate :query => :tags, :as => :array_of_strings
+          end
+          perform_post('/entities', nil, :entities => data)
+        end
+
+        def entities(*ids)
+          perform_get('entities', '/entities', ids)
+        end
+
+        def entities_matching(name_re)
+          raise "Must be a regexp: #{name_re.inspect}" unless
+            name_re.is_a?(Regexp)
+          entities.reject {|e| name_re.match(e[:name]).nil? }
+        end
+
+        def update_entities(*args)
+          ids, params = unwrap_ids(*args), unwrap_params(*args)
+          raise "'update_entities' requires at least one entity id " \
+                ' parameter' if ids.nil? || ids.empty?
+          validate_params(params) do
+            validate :query => :name, :as => :string
+            validate :query => :tags, :as => :array_of_strings
+          end
+          perform_patch("/entities/#{escaped_ids(ids)}", nil,
+                        update_entities_ops(params))
+        end
+
+        private
+
+        def update_entities_ops(params)
+          ops = params.each_with_object([]) do |(k, v), memo|
+            case k
+            when :name
+              memo << patch_replace('entities', k, v)
+            when :add_contact
+              memo << patch_add('entities', 'contacts', v)
+            when :remove_contact
+              memo << patch_remove('entities', 'contacts', v)
+            when :add_tag
+              memo << patch_add('entities', 'tags', v)
+            when :remove_tag
+              memo << patch_remove('entities', 'tags', v)
+           end
+          end
+          raise "'update_entities' did not find any valid update " \
+                'fields' if ops.empty?
+          ops
+        end
+      end
+    end
+  end
+end

data/lib/flapjack-diner/resources/maintenance_periods.rb

@@ -0,0 +1,82 @@
+require 'httparty'
+require 'json'
+require 'uri'
+
+require 'flapjack-diner/version'
+require 'flapjack-diner/argument_validator'
+
+module Flapjack
+  module Diner
+    module Resources
+      module MaintenancePeriods
+        %w(entities checks).each do |data_type|
+          define_method("create_scheduled_maintenances_#{data_type}") do |*args|
+            ids, data = unwrap_ids(*args), unwrap_create_data(*args)
+            if ids.nil? || ids.empty?
+              raise "'create_scheduled_maintenances_#{data_type}' requires " \
+                    "at least one #{data_type} id parameter"
+            end
+            validate_params(data) do
+              validate :query => :start_time, :as => [:required, :time]
+              validate :query => :duration,   :as => [:required, :integer]
+              validate :query => :summary,    :as => :string
+            end
+            perform_post("/scheduled_maintenances/#{data_type}", ids,
+                         :scheduled_maintenances => data)
+          end
+
+          define_method("create_unscheduled_maintenances_#{data_type}") do |*a|
+            ids, data = unwrap_ids(*a), unwrap_create_data(*a)
+            if ids.nil? || ids.empty?
+              raise "'create_unscheduled_maintenances_#{data_type}' requires " \
+                    "at least one #{data_type} id parameter"
+            end
+            validate_params(data) do
+              validate :query => :duration,   :as => [:required, :integer]
+              validate :query => :summary,    :as => :string
+            end
+            perform_post("/unscheduled_maintenances/#{data_type}", ids,
+                         :unscheduled_maintenances => data)
+          end
+
+          define_method("update_unscheduled_maintenances_#{data_type}") do |*a|
+            ids, params = unwrap_ids(*a), unwrap_params(*a)
+            if ids.nil? || ids.empty?
+              raise "'update_unscheduled_maintenances_#{data_type}' requires " \
+                    "at least one #{data_type} id parameter"
+            end
+            validate_params(params) do
+              validate :query => :end_time, :as => :time
+            end
+            ops = update_unscheduled_maintenances_ops(data_type, params)
+            perform_patch("/unscheduled_maintenances/#{data_type}", ids, ops)
+          end
+
+          define_method("delete_scheduled_maintenances_#{data_type}") do |*args|
+            ids, params = unwrap_ids(*args), unwrap_params(*args)
+            if ids.nil? || ids.empty?
+              raise "'delete_scheduled_maintenances_#{data_type}' requires " \
+                    "at least one #{data_type} id parameter"
+            end
+            validate_params(params) do
+              validate :query => :start_time, :as => [:required, :time]
+            end
+            perform_delete("/scheduled_maintenances/#{data_type}", ids, params)
+          end
+        end
+
+        private
+
+        def update_unscheduled_maintenances_ops(data_type, params)
+          ops = params.each_with_object([]) do |(k, v), memo|
+            next unless :end_time.eql?(k)
+            memo << patch_replace('unscheduled_maintenances', k, v)
+          end
+          raise "'update_unscheduled_maintenances_#{data_type}' did not " \
+                'find any valid update fields' if ops.empty?
+          ops
+        end
+      end
+    end
+  end
+end

data/lib/flapjack-diner/resources/media.rb

@@ -0,0 +1,61 @@
+require 'httparty'
+require 'json'
+require 'uri'
+
+require 'flapjack-diner/version'
+require 'flapjack-diner/argument_validator'
+
+module Flapjack
+  module Diner
+    module Resources
+      module Media
+        def create_contact_media(*args)
+          ids, data = unwrap_ids(*args), unwrap_create_data(*args)
+          raise "'create_contact_media' requires at least one contact id " \
+                'parameter' if ids.nil? || ids.empty?
+          validate_params(data) do
+            validate :query => [:type, :address], :as => [:required, :string]
+            validate :query => [:interval, :rollup_threshold],
+                     :as => [:required, :integer]
+          end
+          perform_post("/contacts/#{escaped_ids(ids)}/media", nil,
+                       :media => data)
+        end
+
+        def media(*ids)
+          perform_get('media', '/media', ids)
+        end
+
+        def update_media(*args)
+          ids, params = unwrap_ids(*args), unwrap_params(*args)
+          raise "'update_media' requires at least one media id " \
+                'parameter' if ids.nil? || ids.empty?
+          validate_params(params) do
+            validate :query => :address,                       :as => :string
+            validate :query => [:interval, :rollup_threshold], :as => :integer
+          end
+          perform_patch("/media/#{escaped_ids(ids)}", nil,
+                        update_media_ops(params))
+        end
+
+        def delete_media(*ids)
+          raise "'delete_media' requires at least one media id " \
+                'parameter' if ids.nil? || ids.empty?
+          perform_delete('/media', ids)
+        end
+
+        private
+
+        def update_media_ops(params)
+          ops = params.each_with_object([]) do |(k, v), memo|
+            next unless [:address, :interval, :rollup_threshold].include?(k)
+            memo << patch_replace('media', k, v)
+          end
+          raise "'update_media' did not find any valid update " \
+                'fields' if ops.empty?
+          ops
+        end
+      end
+    end
+  end
+end