lib/media_magick/model.rb in media_magick-0.0.1 vs lib/media_magick/model.rb in media_magick-0.1.0

@@ -6,18 +6,32 @@
   module Model
     extend ActiveSupport::Concern
 
     module ClassMethods
       def attachs_many(name, options = {}, &block)
+        warn "[DEPRECATION] `attachs_many` is deprecated.  Please use `attaches_many` instead."
+        attaches_many(name, options, &block)
+      end
+
+      #
+      # TODO 
+      # * refactor these methods to remove duplication 
+      #
+      def attaches_many(name, options = {}, &block)
         klass = Class.new do
           include Mongoid::Document
           extend CarrierWave::Mount
 
           field :priority, type: Integer, default: 0
           default_scope asc(:priority)
 
-          embedded_in :attachmentable, polymorphic: true
+          if options[:relation] == :referenced
+            belongs_to :attachmentable, polymorphic: true
+          else
+            embedded_in :attachmentable, polymorphic: true
+          end
+
           mount_uploader name.to_s.singularize, (options[:uploader] || AttachmentUploader)
 
           self.const_set "TYPE", options[:type] || :image
           self.const_set "ATTACHMENT", name.to_s.singularize
 
@@ -28,12 +42,55 @@
             self.send(self.class::ATTACHMENT).send(method)
           end
         end
 
         name_camelcase = name.to_s.camelcase
-        Object.const_set "#{self.name}#{name_camelcase}", klass
+        Object.const_set "#{self}#{name_camelcase}", klass
 
-        embeds_many(name, :as => :attachmentable, class_name: "#{self}#{name_camelcase}")
+        if options[:relation] == :referenced
+          klass.collection_name = "#{self.name}_#{name.to_s.camelcase}".parameterize
+
+          has_many(name, :as => :attachmentable, class_name: "#{self}#{name_camelcase}")
+
+          field "#{name.to_s.singularize}_ids", type: Array
+
+          before_create do
+            ids = self.send("#{name.to_s.singularize}_ids") || []
+
+            ids.each do |id|
+              "#{self.class}#{name_camelcase}".constantize.find(id).update_attributes(attachmentable: self)
+            end
+          end
+        else
+          embeds_many(name, :as => :attachmentable, class_name: "#{self}#{name_camelcase}")
+        end
+      end
+
+      def attaches_one(name, options = {}, &block)
+        klass = Class.new do
+          include Mongoid::Document
+          extend CarrierWave::Mount
+
+          embedded_in name
+          mount_uploader name.to_s.singularize, (options[:uploader] || AttachmentUploader)
+
+          accepts_nested_attributes_for name.to_s.singularize
+
+          self.const_set "TYPE", options[:type] || :image
+          self.const_set "ATTACHMENT", name.to_s.singularize
+
+          class_eval(&block) if block_given?
+
+          def method_missing(method, args = nil)
+            return self.send(self.class::ATTACHMENT).file.filename if method == :filename
+            self.send(self.class::ATTACHMENT).send(method)
+          end
+        end
+
+        name_camelcase = name.to_s.camelcase
+        Object.const_set "#{self}#{name_camelcase}", klass
+
+        embeds_one name, class_name: "#{self}#{name_camelcase}", cascade_callbacks: true
       end
     end
   end
 end