quotation.rb in dorsale-2.1.8

- old
+ new

@@ -1,10 +1,12 @@
 module Dorsale
   module BillingMachine
     class Quotation < ActiveRecord::Base
       self.table_name = "dorsale_billing_machine_quotations"
 
+      STATES = %w(pending accepted refused canceled)
+
       belongs_to :customer, polymorphic: true
       belongs_to :id_card
       belongs_to :payment_term
       has_many :lines,
         :inverse_of => :quotation,
@@ -20,20 +22,22 @@
 
       polymorphic_id_for :customer
 
       validates :id_card, presence: true
       validates :date,    presence: true
+      validates :state,   presence: true, inclusion: {in: STATES}
 
       # simple_form
       validates :id_card_id, presence: true
 
       def initialize(*args)
         super
         self.date                = Date.today     if date.nil?
         self.expires_at          = date + 1.month if expires_at.nil?
         self.vat_rate            = 20             if vat_rate.nil?
         self.commercial_discount = 0              if commercial_discount.nil?
+        self.state               = STATES.first   if state.nil?
       end
 
       before_create :assign_unique_index
       before_create :assign_tracking_id
 
@@ -59,9 +63,65 @@
 
       def pdf
         pdf = ::Dorsale::BillingMachine::QuotationPdf.new(self)
         pdf.build
         pdf
+      end
+
+      def create_copy!
+        new_quotation = self.dup
+
+        self.lines.each do |line|
+          new_quotation.lines << line.dup
+        end
+
+        new_quotation.unique_index = nil
+        new_quotation.created_at   = nil
+        new_quotation.updated_at   = nil
+        new_quotation.date         = Date.today
+        new_quotation.state        = Quotation::STATES.first
+
+        new_quotation.save!
+
+        self.attachments.each do |attachment|
+          new_attachment            = attachment.dup
+          new_attachment.attachable = new_quotation
+          new_attachment.file       = File.open(attachment.file.path)
+          new_attachment.save!
+        end
+
+        new_quotation
+      end
+
+      def create_invoice!
+        new_invoice = Dorsale::BillingMachine::Invoice.new
+
+        self.attributes.each do |k, v|
+          next if k.to_s == "id"
+          next if k.to_s.match /index|tracking|_at/
+
+          if new_invoice.respond_to?("#{k}=")
+            new_invoice.public_send("#{k}=", v)
+          end
+        end
+
+        self.lines.each do |line|
+          new_line = new_invoice.lines.new
+          line.attributes.each do |k, v|
+            next if k.to_s == "id"
+            next if k.to_s.match /index|tracking|_at/
+
+          if new_line.respond_to?("#{k}=")
+            new_line.public_send("#{k}=", v)
+          end
+          end
+        end
+
+        new_invoice.date = Date.today
+
+        new_invoice.save!
+
+        new_invoice
       end
 
     end # Quotation
   end # BillingMachine
 end # Dorsale