Software Público Brasileiro

module Api
  class Entity < Grape::Entity

    def initialize(object, options = {})
      object = nil if object.is_a? Exception
      super object, options
    end

    def self.represent(objects, options = {})
      if options[:has_exception]
        data = super objects, options.merge(is_inner_data: true)
        if objects.is_a? Exception
          data.merge ok: false, error: {
            type: objects.class.name,
            message: objects.message
          }
        else
          data = data.serializable_hash if data.is_a? Entity
          data.merge ok: true, error: { type: 'Success', message: '' }
        end
      else
        super objects, options
      end
    end

  end
end