ManageIQ/manageiq-api-common

module Insights
  module API
    module Common
      class EntitlementError < StandardError; end

      class Entitlement
        def initialize(identity)
          @identity = identity
        end

        %w[
          ansible
          hybrid_cloud
          insights
          migrations
          openshift
          smart_management
        ].each do |m|
          define_method("#{m}?") do
            find_entitlement_key(m)
          end
        end

        private

        attr_reader :identity

        def find_entitlement_key(key)
          result = identity.dig('entitlements', key.to_s)
          # TODO: Always force entitlements key
          return true unless result
          result['is_entitled']
        end
      end
    end
  end
end