Module: Dooly::Attachment::ClassMethods

Defined in:
lib/dooly/attachment.rb

Instance Method Summary collapse

Instance Method Details

#make_attachable_as_json(klass = nil) ⇒ Object



11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
# File 'lib/dooly/attachment.rb', line 11

def make_attachable_as_json(klass = nil)
  klass = Dooly::Attachment::Base if klass.nil?
  unless klass <= Dooly::Attachment::Base
    raise "Attachment class must inherit Dooly::Attachment::Base: #{klass}"
  end
  self.attachment_klass = klass
  
  if self.instance_methods.include?(:as_json)
    self.class_eval do 
      alias :as_json_exclude_attachment :as_json
      define_method(:as_json) do |options = {}, &block|
        js = if block
          as_json_exclude_attachment(options, &block)
        else
          as_json_exclude_attachment(options)
        end
        
        js.class <= Hash ? js.merge(attachment.as_json) : js
      end
    end
  else
    self.class_eval do 
      define_method(:as_json) do |options = {}|
        attachment.as_json
      end
    end
  end
end