complete vehicle translation. Mantis 26839

1 files changed, 39 insertions, 15 deletions

diff --git a/app/models/vehicle_translation.rb b/app/models/vehicle_translation.rb
index 0457da0fa..639981d63 100644
--- a/app/models/vehicle_translation.rb
+++ b/app/models/vehicle_translation.rb
@@ -4,12 +4,14 @@ class VehicleTranslation
   extend ActiveModel::Naming
 
   attr_accessor :vehicle_journey_id, :count, :duration
-  attr_accessor :first_stop_arrival_time, :first_stop_departure_time
+  attr_accessor :first_stop_time, :departure_or_arrival
+  attr_accessor :delta
 
-  validates_presence_of :count, :duration
+  validates_presence_of :count, :duration, :first_stop_time, :departure_or_arrival
+  validates_inclusion_of :departure_or_arrival, :in =>  %w( departure arrival)
   validates_numericality_of :count, greater_than: 0
   validates_numericality_of :duration, greater_than: 0
-  validate :starting_time_provided
+  validate :firts_stop_time_format
   validate :vehicle_has_stop_times
 
   def initialize(attributes = {})
@@ -18,15 +20,17 @@ class VehicleTranslation
     end
   end
 
-  def starting_time_provided
-    if ( first_stop_arrival_time.blank? && first_stop_departure_time.blank?)
-      errors.add :first_stop_arrival_time, I18n.t('activemodel.errors.models.vehicle_translation.missing_start_time')
-      errors.add :first_stop_departure_time, I18n.t('activemodel.errors.models.vehicle_translation.missing_start_time')
-      #errors.add( :first_stop_departure_time => "un horaire de départ ou d'arrivée au premier arrêt doit être renseigné")
-    elsif first_stop_departure_time.blank?
-      errors.add :first_stop_arrival_time, I18n.t('activemodel.errors.models.vehicle_translation.unreadable_time') unless Time.parse( self.first_stop_arrival_time) rescue false
-    elsif first_stop_arrival_time.blank?
-      errors.add :first_stop_departure_time, I18n.t('activemodel.errors.models.vehicle_translation.unreadable_time') unless Time.parse( self.first_stop_departure_time) rescue false
+  def firts_stop_time_format
+    if first_stop_time.blank?
+      errors.add :first_stop_time, I18n.t('activemodel.errors.models.vehicle_translation.unreadable_time') unless self.class.time_format_text?( self.first_stop_time)
+    end
+  end
+
+  def self.time_format_text?( text)
+    begin
+      Time.parse text
+    rescue
+      false
     end
   end
 
@@ -40,23 +44,43 @@ class VehicleTranslation
     false
   end
 
+  def move_time_in_current_time_zone( time )
+    Time.parse( "#{time.hour}:#{time.min}")
+  end
+  def evaluate_delta( actual_time)
+    Time.parse( first_stop_time) - move_time_in_current_time_zone( actual_time)
+  end
+  def vjas_time_attribute
+    "#{departure_or_arrival}_time"
+  end
+  def first_vjas_time
+    vehicle_journey.vehicle_journey_at_stops.first.send( vjas_time_attribute)
+  end
+  def first_delta
+    evaluate_delta( first_vjas_time)
+  end
+
   def translate
     vehicle = vehicle_journey
     copied_attributes = vehicle_journey.attributes
     copied_attributes.delete( "id")
     copied_attributes.delete( "objectid")
 
+    # time shift for current duplicated vehicle
+    delta = first_delta
+
     1.upto( count.to_i) do |index|
       translated = Chouette::VehicleJourney.create( copied_attributes)
       translated.time_tables = vehicle.time_tables
+
       vehicle.vehicle_journey_at_stops.each do |vjas|
         vjas_attributes = vjas.attributes.merge( "vehicle_journey_id" => translated.id)
-        vjas_attributes.merge! "departure_time" => ( vjas_attributes[ "departure_time"] + (index * duration.to_i.minutes) ),
-          "arrival_time" => ( vjas_attributes[ "arrival_time"] + (index * duration.to_i.minutes) )
+        vjas_attributes.merge! "departure_time" => ( vjas_attributes[ "departure_time"] + delta),
+                               "arrival_time" => ( vjas_attributes[ "arrival_time"] + delta)
 
         Chouette::VehicleJourneyAtStop.create( vjas_attributes)
       end
-
+      delta += duration.to_i.minutes
     end
   end