Noosfero/noosfero

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