extended_attachments.rb in couchrest_model-1.0.0.beta8

- old
+ new

@@ -5,15 +5,19 @@
       # Add a file attachment to the current document. Expects
       # :file and :name to be included in the arguments.
       def create_attachment(args={})
         raise ArgumentError unless args[:file] && args[:name]
         return if has_attachment?(args[:name])
-        self['_attachments'] ||= {}
         set_attachment_attr(args)
       rescue ArgumentError => e
         raise ArgumentError, 'You must specify :file and :name'
       end
+      
+      # return all attachments
+      def attachments
+        self['_attachments'] ||= {}
+      end
 
       # reads the data from an attachment
       def read_attachment(attachment_name)
         database.fetch_attachment(self, attachment_name)
       end
@@ -28,17 +32,17 @@
         raise ArgumentError, 'You must specify :file and :name'
       end
 
       # deletes a file attachment from the current doc
       def delete_attachment(attachment_name)
-        return unless self['_attachments']
-        self['_attachments'].delete attachment_name
+        return unless attachments
+        attachments.delete attachment_name
       end
 
       # returns true if attachment_name exists
       def has_attachment?(attachment_name)
-        !!(self['_attachments'] && self['_attachments'][attachment_name] && !self['_attachments'][attachment_name].empty?)
+        !!(attachments && attachments[attachment_name] && !attachments[attachment_name].empty?)
       end
 
       # returns URL to fetch the attachment from
       def attachment_url(attachment_name)
         return unless has_attachment?(attachment_name)
@@ -60,10 +64,10 @@
         end
 
         def set_attachment_attr(args)
           content_type = args[:content_type] ? args[:content_type] : get_mime_type(args[:file].path)
           content_type ||= (get_mime_type(args[:name]) || 'text/plain')
-          self['_attachments'][args[:name]] = {
+          attachments[args[:name]] = {
             'content_type' => content_type,
             'data'         => args[:file].read
           }
         end
       