invoice.rb in erp_invoicing-3.0.1

- old
+ new
@@ -1,16 +1,109 @@
 class Invoice < ActiveRecord::Base
+  acts_as_document
   
+  belongs_to  :billing_account
   belongs_to	:invoice_type
-	has_many 	  :invoice_items
-	has_many 	  :invoice_party_roles
-	has_many	  :parties, :through => :invoice_party_roles
-	
-	#This line of code connects the invoice to a polymorphic payment application.
-	#The effect of this is to allow payments to be "applied_to" invoices
-  has_many    :payment_applications, :as => :payment_applied_to
+  belongs_to  :invoice_payment_strategy_type
+  has_many    :invoice_payment_term_sets, :dependent => :destroy
+  has_many    :payment_applications, :as => :payment_applied_to, :dependent => :destroy do
+    def successful
+      all.select{|item| item.financial_txn.has_captured_payment?}
+    end
+    def pending
+      all.select{|item| item.is_pending?}
+    end
+  end
+  has_many 	  :invoice_items, :dependent => :destroy do
+    def by_date
+      order('created_at')
+    end
 
+    def unpaid
+      select{|item| item.balance > 0 }
+    end
+  end
+  has_many 	  :invoice_party_roles, :dependent => :destroy
+	has_many	  :parties, :through => :invoice_party_roles
+  
   alias :items :invoice_items
   alias :type :invoice_type
   alias :party_roles :invoice_party_roles
+  alias :payment_strategy :invoice_payment_strategy_type
+
+  def has_payments?(status)
+    selected_payment_applications = self.get_payment_applications(status)
+    !(selected_payment_applications.nil? or selected_payment_applications.empty?)
+  end
+
+  def get_payment_applications(status=:all)
+    selected_payment_applications = case status.to_sym
+    when :pending
+      self.payment_applications.pending
+    when :successful
+      self.payment_applications.successful
+    when :all
+      self.payment_applications
+    end
+
+    unless self.items.empty?
+      selected_payment_applications = (selected_payment_applications | self.items.collect{|item| item.get_payment_applications(status)}).flatten! unless (self.items.collect{|item| item.get_payment_applications(status)}.empty?)
+    end
+    
+    selected_payment_applications
+  end
+
+  def balance
+    (self.payment_due - self.total_payments)
+  end
+
+  def payment_due
+    self.items.all.sum(&:total_amount)
+  end
+
+  def total_payments
+    self.get_payment_applications(:successful).sum{|item| item.money.amount}
+  end
+
+  def transactions
+    transactions = []
+
+    self.items.each do |item|
+      transactions << {
+        :date => item.created_at,
+        :description => item.item_description,
+        :quantity => item.quantity,
+        :amount => item.amount
+      }
+    end
+
+    self.get_payment_applications(:successful).each do |item|
+      transactions << {
+        :date => item.financial_txn.payments.last.created_at,
+        :description => item.financial_txn.description,
+        :quantity => 1,
+        :amount => (0 - item.financial_txn.money.amount)
+      }
+    end
+
+    transactions.sort_by{|item| [item[:date]]}
+  end
+
+  def add_party_with_role_type(party, role_type)
+    self.invoice_party_roles << InvoicePartyRole.create(:party => party, :role_type => convert_role_type(role_type))
+    self.save
+  end
+
+  def find_parties_by_role_type(role_type)
+    self.invoice_party_roles.where('role_type_id = ?', convert_role_type(role_type).id).all.collect(&:party)
+  end
+
+  private
+
+  def convert_role_type(role_type)
+    role_type = RoleType.iid(role_type) if role_type.is_a? String
+    raise "Role type does not exist" if role_type.nil?
+
+    role_type
+  end
 	
 end