2 files changed, 126 insertions, 0 deletions
diff --git a/app/services/referential_consolidated.rb b/app/services/referential_consolidated.rb
new file mode 100644
index 000000000..465eab405
--- /dev/null
+++ b/app/services/referential_consolidated.rb
@@ -0,0 +1,124 @@
+class ReferentialConsolidated
+  attr_reader :params
+
+  def initialize vehicle_journeys, params
+    @vehicle_journeys = vehicle_journeys
+    @params = params
+  end
+
+  def paginated_lines
+    @paginated_lines ||= begin
+      line_ids = @vehicle_journeys.joins(route: :line).pluck('lines.id')
+      lines = Chouette::Line.where(id: line_ids).order(:name)
+      lines.paginate page: params[:page], per_page: params[:per_page] || 10
+    end
+  end
+
+  def lines
+    @lines ||= paginated_lines.to_a.map {|l| Line.new(self, l, @vehicle_journeys, params) }
+  end
+
+  def _should_highlight?
+    return false unless params[:q].present?
+    keys = params[:q].keys - ["stop_areas"]
+    params[:q].values_at(*keys).each do |value|
+      if value.is_a?(Hash)
+        return true if value.values.any?(&:present?)
+      elsif value.is_a?(Array)
+        return true if value.any?(&:present?)
+      else
+        if value.present?
+          return true
+        end
+      end
+    end
+    false
+  end
+
+  def should_highlight?
+    if @should_highlight.nil?
+      @should_highlight = _should_highlight?
+    end
+    @should_highlight
+  end
+
+  class Base
+    extend Forwardable
+    attr_reader :params
+    attr_reader :parent
+    attr_reader :ar_model
+
+    def initialize parent, ar_model, vehicle_journeys, params
+      @parent = parent
+      @ar_model = ar_model
+      @all_vehicle_journeys = vehicle_journeys
+      @params = params
+    end
+
+    def should_highlight?
+      parent.should_highlight?
+    end
+  end
+
+  class Line < Base
+    delegate name: :ar_model
+    delegate id: :ar_model
+
+    def routes
+      @routes ||= begin
+        ar_model.routes.order(:name).map {|r| Route.new(self, r, @all_vehicle_journeys, params) }
+      end
+    end
+  end
+
+  class Route < Base
+    def_delegators :ar_model, :name, :id, :time_tables, :purchase_windows, :stop_area_ids
+
+    def vehicle_journeys
+      @vehicle_journeys ||= begin
+        ar_model.vehicle_journeys.map {|vj| VehicleJourney.new(self, vj, @all_vehicle_journeys, params) }
+      end
+    end
+
+    def highlighted_journeys
+      @all_vehicle_journeys.joins(:journey_pattern).where(route_id: self.id)
+    end
+
+    def highlighted_count
+      highlighted_journeys.count
+    end
+
+    def highlighted?
+      matching_stop_areas = params[:q] && params[:q]["stop_areas"] && (params[:q]["stop_areas"].values & self.stop_area_ids.map(&:to_s)).present?
+      (should_highlight? || matching_stop_areas) && highlighted_journeys.exists?
+    end
+
+    def stop_points
+      @stop_points ||= ar_model.stop_points.map {|sp| StopPoint.new(self, sp, @all_vehicle_journeys, params) }
+    end
+  end
+
+  class VehicleJourney < Base
+    def_delegators :ar_model, :id, :published_journey_name, :journey_pattern, :time_tables, :purchase_windows, :vehicle_journey_at_stops, :time_table_ids, :purchase_window_ids, :route
+
+    def highlighted?
+      should_highlight? && @all_vehicle_journeys.where(id: self.id).exists?
+    end
+
+    def has_purchase_window? purchase_window
+      purchase_window_ids.include?(purchase_window.id)
+    end
+
+    def has_time_table? time_table
+      time_table_ids.include?(time_table.id)
+    end
+  end
+
+  class StopPoint < Base
+    def_delegators :ar_model, :id, :arrival_time, :departure_time, :name, :stop_area, :stop_area_id
+
+    def highlighted?
+      params[:q] && params[:q]["stop_areas"] && params[:q]["stop_areas"].values.any?{|v| v.to_s == stop_area_id.to_s}
+    end
+  end
+end
diff --git a/app/services/route_way_cost_calculator.rb b/app/services/route_way_cost_calculator.rb
index d41a2e59a..ca47a6772 100644
--- a/app/services/route_way_cost_calculator.rb
+++ b/app/services/route_way_cost_calculator.rb
@@ -8,5 +8,7 @@ class RouteWayCostCalculator
     way_costs = TomTom.matrix(way_costs)
     way_costs = WayCostCollectionJSONSerializer.dump(way_costs)
     @route.update(costs: way_costs)
+  rescue TomTom::Errors::MatrixRemoteError => e
+    Rails.logger.error "TomTom::Matrix server error: #{e}"
   end
 end