app/models/bodega/order.rb in bodega-0.2.0 vs app/models/bodega/order.rb in bodega-0.3.0

@@ -1,28 +1,43 @@
 module Bodega
   class Order < ActiveRecord::Base
-    extend Bodega::Monetize
-
+    before_save :set_total
     before_create :set_identifier
 
     belongs_to :customer, polymorphic: true
     has_many :order_products, class_name: 'Bodega::OrderProduct', dependent: :destroy
     has_many :products, through: :order_products
 
-    monetize :subtotal
-    monetize :tax
-    monetize :total
+    monetize :tax_cents
+    monetize :total_cents
 
+    def finalize!(payment_method)
+      self.class.transaction do
+        self.save!
+        begin
+          self.payment_id = payment_method.complete!
+          self.save!
+        rescue Exception => e
+          raise ActiveRecord::Rollback
+          raise e.inspect
+        end
+      end
+    end
+
     def subtotal
-      order_products.inject(0) {|sum, order_product| sum += order_product.subtotal }
+      @subtotal ||= order_products.inject(Money.new(0)) {|sum, order_product| sum += order_product.subtotal }
     end
 
+    def set_total
+      self.total = subtotal + tax
+    end
+
     def to_param
       identifier
     end
 
     protected
     def set_identifier
-      self.identifier = "#{Time.now.to_i}--#{rand(12)}"
+      self.identifier = self.class.count.succ.to_s(36)
     end
   end
 end