Hub export

9 files changed, 485 insertions, 0 deletions

diff --git a/app/exporters/chouette/hub/commercial_stop_area_exporter.rb b/app/exporters/chouette/hub/commercial_stop_area_exporter.rb
new file mode 100644
index 000000000..201fce3d2
--- /dev/null
+++ b/app/exporters/chouette/hub/commercial_stop_area_exporter.rb
@@ -0,0 +1,36 @@
+class Chouette::Hub::CommercialStopAreaExporter
+  include ERB::Util
+  attr_accessor :stop_area, :directory, :template
+  
+  def initialize(stop_area, directory)
+    @stop_area = stop_area
+    @directory = directory
+    @template = File.open('app/views/api/hub/stop_area/commercial_show.hub.erb' ){ |f| f.read }
+    @type = "ONNNNNNNNNNNNNNNNN"
+  end
+  
+  def render()
+    ERB.new(@template).result(binding)
+  end
+  
+  def hub_name
+    "/ARRET.TXT"
+  end
+  
+  def self.save( stop_areas, directory, hub_export)
+    stop_areas.each do |stop_area|
+      self.new( stop_area, directory).tap do |specific_exporter|
+        specific_exporter.save
+      end
+    end
+    hub_export.log_messages.create( :severity => "ok", :key => "EXPORT|COMMERCIAL_STOP_AREA_COUNT", :arguments => {"0" => stop_areas.size})
+  end
+  
+  def save
+    File.open(directory + hub_name , "a") do |f|
+      f.write("ARRET\n") if f.size == 0
+      f.write(render)
+    end if stop_area.present?
+  end
+end
+
diff --git a/app/exporters/chouette/hub/connection_link_exporter.rb b/app/exporters/chouette/hub/connection_link_exporter.rb
new file mode 100644
index 000000000..b3be19396
--- /dev/null
+++ b/app/exporters/chouette/hub/connection_link_exporter.rb
@@ -0,0 +1,35 @@
+class Chouette::Hub::Connection_LinkExporter
+  include ERB::Util
+  attr_accessor :connection_link, :directory, :template
+  
+  def initialize(connection_link, directory)
+    @connection_link = connection_link
+    @directory = directory
+    @template = File.open('app/views/api/hub/correspondance/show.hub.erb' ){ |f| f.read }
+  end
+  
+  def render()
+    ERB.new(@template).result(binding)
+  end
+  
+  def hub_name
+    "/CORRESPONDANCE.TXT"
+  end
+  
+  def self.save( connection_links, directory, hub_export)
+    connection_links.each do |connection_link|
+      self.new( connection_link, directory).tap do |specific_exporter|
+        specific_exporter.save
+      end
+    end
+    hub_export.log_messages.create( :severity => "ok", :key => "EXPORT|CONNECTION_LINK_COUNT", :arguments => {"0" => connection_links.size})
+  end
+  
+  def save
+    File.open(directory + hub_name , "a") do |f|
+      f.write("CORRESPONDANCE\n") if f.size == 0
+      f.write(render)
+    end if connection_link.present?
+  end
+end
+
diff --git a/app/exporters/chouette/hub/exporter.rb b/app/exporters/chouette/hub/exporter.rb
new file mode 100644
index 000000000..4b846fee0
--- /dev/null
+++ b/app/exporters/chouette/hub/exporter.rb
@@ -0,0 +1,185 @@
+class Chouette::Hub::Exporter
+  
+  attr_reader :referential
+  attr_reader :hub_export, :lines, :routes, :journey_patterns
+  attr_reader :time_tables, :vehicle_journeys
+  
+  def initialize(referential, hub_export)
+    @referential = referential
+    @hub_export = hub_export
+    @lines = nil
+    @routes = nil
+    @journey_patterns = nil
+    @time_tables = nil
+    @vehicle_journeys = nil
+  end
+  
+  def select_time_tables(start_date, end_date)
+    #TODO considere options[:o], options[:id]
+    all_time_tables = Chouette::TimeTable.all 
+    time_tables = []
+    s_date = Date.strptime(start_date, "%Y-%m-%d")
+    e_date = Date.strptime(end_date, "%Y-%m-%d")
+    while s_date <= e_date
+      (all_time_tables - time_tables).each { |time_table| time_tables << time_table if time_table.include_day?(s_date) }
+      s_date = s_date.next_day
+    end
+    return time_tables
+  end
+  
+  def select_lines(object, ids)
+    if object == "network"
+      ids.present? ? Chouette::Line.includes(:routes).where( :network_id => ids.split(",")).order(:name) :
+        Chouette::Line.joins(:network).includes(:routes).order(:name)
+    elsif object == "company"
+      ids.present? ? Chouette::Line.includes(:routes).where( :company_id => ids.split(",")).order(:name) :
+        Chouette::Line.joins(:company).includes(:routes).order(:name)
+    elsif object == "line" && ids.present?
+      Chouette::Line.includes(:routes).where( :id => ids.split(",")).order(:name)
+    else
+      Chouette::Line.includes(:routes).order(:name)
+    end
+  end
+  
+  def time_tables_exportable?
+    time_tables
+  end
+  
+  def routes_exportable?
+    routes # && routes.size < 150
+  end
+  
+  def lines_exportable?
+    lines # && lines.size < 150
+  end
+  
+  def journey_patterns_exportable?
+    journey_patterns # && journey_patterns.size < 150
+  end
+  
+  def vehicle_journeys_exportable?
+    vehicle_journeys
+  end
+  
+  def log_overflow_warning( target_class)
+    hub_export.log_messages.create( :severity => "warning", :key => "EXPORT_ERROR|EXCEPTION",
+                                    :arguments => {"0" => I18n.t( 'export_log_messages.messages.overflow',
+                                                                  :count => 150, :data => target_class.model_name.human.pluralize)})
+  end
+  
+  def export(zip_file_path, options = {})
+    begin
+      referential.switch
+      
+      FileUtils.rm(zip_file_path) if File.exists? zip_file_path
+      
+      hub_export.log_messages.create( :severity => "ok", :key => "EXPORT", :arguments => {"0" => "HUB"})
+      
+      Dir.mktmpdir(nil, "/tmp"){ |temp_dir|
+        
+        @time_tables = select_time_tables(options[:start_date], options[:end_date])
+        
+        @lines = select_lines( options[:o], options[:id] )
+        @routes = lines.map(&:routes).flatten.sort {|a,b| (a.name && b.name) ? a.name <=> b.name : a.id <=> b.id} if lines_exportable?
+        @journey_patterns = Chouette::JourneyPattern.where( :route_id => routes.map(&:id) ).order(:name) if routes_exportable?
+        
+        @vehicle_journeys = Chouette::VehicleJourney.where( :route_id => routes.map(&:id) ).order(:id) if routes_exportable?
+        
+        vjs = []
+        tts = []
+        @vehicle_journeys.each do |vj|
+          unless (vj.time_tables & @time_tables).empty?
+            vjs << vj
+            tts << (vj.time_tables & @time_tables)
+          end
+        end
+        @time_tables = tts.flatten.uniq
+        @vehicle_journeys = vjs.uniq
+        
+        vehicle_journey_at_stops = Chouette::VehicleJourneyAtStop.where( :vehicle_journey_id => @vehicle_journeys.map(&:id) ).order(:id) if vehicle_journeys_exportable?
+        
+        if time_tables_exportable?
+          Chouette::Hub::TimeTableExporter.save(@time_tables, temp_dir, hub_export)
+        else
+          log_overflow_warning(Chouette::TimeTable)
+        end
+        
+        if vehicle_journeys_exportable?
+          Chouette::Hub::VehicleJourneyExporter.save(@vehicle_journeys, temp_dir, hub_export)
+          Chouette::Hub::VehicleJourneyAtStopExporter.save(vehicle_journey_at_stops, temp_dir, hub_export)
+        else
+          log_overflow_warning(Chouette::VehicleJourney)
+        end
+        
+        stop_points = Chouette::StopPoint.where( :id => vehicle_journey_at_stops.map(&:stop_point_id)).order(:id)
+        physical_stop_areas =  Chouette::StopArea.where( :id => stop_points.map(&:stop_area_id)).order(:parent_id)
+        commercial_stop_areas = Chouette::StopArea.where( :id => physical_stop_areas.map(&:parent_id)).order(:id)
+        
+        Chouette::Hub::CommercialStopAreaExporter.save(commercial_stop_areas, temp_dir, hub_export)
+        Chouette::Hub::PhysicalStopAreaExporter.save(physical_stop_areas, temp_dir, hub_export)
+
+        connection_links = Chouette::ConnectionLink.where( "deprture_id IN (?) AND arrival_id IN (?)", (physical_stop_areas.map(&:id) + commercial_stop_areas.map(&:id)), (physical_stop_areas.map(&:id) + commercial_stop_areas.map(&:id)) )
+        
+        Chouette::Hub::ConnectionLinkExporter.save(connection_links, temp_dir, hub_export)
+        
+        if journey_patterns_exportable?
+          Chouette::Hub::JourneyPatternExporter.save(@journey_patterns, temp_dir, hub_export)
+        else
+          log_overflow_warning(Chouette::JourneyPattern) if routes_exportable?
+        end
+        
+        if lines_exportable?
+          Chouette::Hub::LineExporter.save(@lines, temp_dir, hub_export)
+        else
+          log_overflow_warning(Chouette::Line)
+        end
+        
+        if routes_exportable?
+          #Chouette::Hub::RouteExporter.save( routes, temp_dir, hub_export)
+        else
+          log_overflow_warning(Chouette::Route) if lines_exportable?
+        end
+
+        # if too many lines
+        # there may be too many stop_areas
+        if lines_exportable?
+          stop_areas = Chouette::StopArea.joins( :stop_points => [:route => :line]).where(:lines => {:id => lines.map(&:id)}).uniq.order(:name)
+          #Chouette::Hub::StopAreaExporter.save( stop_areas, temp_dir, hub_export, "Quay")
+
+          commercial_stop_areas = Chouette::StopArea.where( :id => stop_areas.map(&:parent_id).compact.uniq).order(:name)
+          #Chouette::Hub::StopAreaExporter.save( commercial_stop_areas, temp_dir, hub_export, "CommercialStopPoint")
+        end
+
+        if( options[:o] == "line" and not options[:id].present?) # Add all objects
+          stop_places = referential.stop_areas.stop_place.order(:name)
+          #Chouette::Hub::StopAreaExporter.save( stop_places, temp_dir, hub_export, "StopPlace")
+
+          itls = referential.stop_areas.itl.order(:name)
+          #Chouette::Hub::StopAreaExporter.save( itls, temp_dir, hub_export, "ITL")
+
+          connection_links = referential.connection_links.order(:name)
+          #Chouette::Hub::ConnectionLinkExporter.save( connection_links, temp_dir, hub_export)
+
+          access_links = referential.access_links.order(:name)
+          #Chouette::Hub::AccessLinkExporter.save(access_links, temp_dir, hub_export)
+
+          access_points = referential.access_points.order(:name)
+          #Chouette::Hub::AccessPointExporter.save(access_points, temp_dir, hub_export)
+
+        end
+
+        ::Zip::ZipFile.open(zip_file_path, ::Zip::ZipFile::CREATE) do |zipfile|
+          Dir[File.join(temp_dir, '*.TXT')].each do |f|
+            #Rails.logger.error("Adding File #{File.basename(f)}")
+            zipfile.add(File.basename(f), f)
+          end
+        end
+      }
+    ensure
+      # Always cleanup files
+      #FileUtils.remove_entry(temp_directory)
+    end
+  end
+
+end
+
diff --git a/app/exporters/chouette/hub/journey_pattern_exporter.rb b/app/exporters/chouette/hub/journey_pattern_exporter.rb
new file mode 100644
index 000000000..12c157408
--- /dev/null
+++ b/app/exporters/chouette/hub/journey_pattern_exporter.rb
@@ -0,0 +1,36 @@
+class Chouette::Hub::JourneyPatternExporter
+  include ERB::Util
+  attr_accessor :journey_pattern, :directory, :template
+  
+  def initialize(journey_pattern, directory)
+    @journey_pattern = journey_pattern
+    @directory = directory
+    @template = File.open('app/views/api/hub/chemin/show.hub.erb' ){ |f| f.read }
+    @type = "COM"
+  end
+  
+  def render()
+    ERB.new(@template).result(binding)
+  end
+  
+  def hub_name
+    "/CHEMIN.TXT"
+  end
+  
+  def self.save( journey_patterns, directory, hub_export)
+    journey_patterns.each do |journey_pattern|
+      self.new( journey_pattern, directory).tap do |specific_exporter|
+        specific_exporter.save
+      end
+    end
+    hub_export.log_messages.create( :severity => "ok", :key => "EXPORT|JOURNEY_PATTERN_COUNT", :arguments => {"0" => journey_patterns.size})
+  end
+  
+  def save
+    File.open(directory + hub_name , "a") do |f|
+      f.write("CHEMIN\n") if f.size == 0
+      f.write(render)
+    end if journey_pattern.present?
+  end
+end
+
diff --git a/app/exporters/chouette/hub/line_exporter.rb b/app/exporters/chouette/hub/line_exporter.rb
new file mode 100644
index 000000000..b9575876a
--- /dev/null
+++ b/app/exporters/chouette/hub/line_exporter.rb
@@ -0,0 +1,35 @@
+class Chouette::Hub::LineExporter
+  include ERB::Util
+  attr_accessor :line, :directory, :template
+  
+  def initialize(line, directory)
+    @line = line
+    @directory = directory
+    @template = File.open('app/views/api/hub/ligne/show.hub.erb' ){ |f| f.read }
+  end
+  
+  def render()
+    ERB.new(@template).result(binding)
+  end
+  
+  def hub_name
+    "/LIGNE.TXT"
+  end
+  
+  def self.save( lines, directory, hub_export)
+    lines.each do |line|
+      self.new( line, directory).tap do |specific_exporter|
+        specific_exporter.save
+      end
+    end
+    hub_export.log_messages.create( :severity => "ok", :key => "EXPORT|LINE_COUNT", :arguments => {"0" => lines.size})
+  end
+  
+  def save
+    File.open(directory + hub_name , "a") do |f|
+      f.write("LIGNE\n") if f.size == 0
+      f.write(render)
+    end if line.present?
+  end
+end
+
diff --git a/app/exporters/chouette/hub/physical_stop_area_exporter.rb b/app/exporters/chouette/hub/physical_stop_area_exporter.rb
new file mode 100644
index 000000000..4a2666321
--- /dev/null
+++ b/app/exporters/chouette/hub/physical_stop_area_exporter.rb
@@ -0,0 +1,37 @@
+class Chouette::Hub::PhysicalStopAreaExporter
+  include ERB::Util
+  attr_accessor :stop_area, :directory, :template
+  
+  def initialize(stop_area, directory)
+    @stop_area = stop_area
+    @directory = directory
+    @template = File.open('app/views/api/hub/stop_area/physical_show.hub.erb' ){ |f| f.read }
+    @type = "NNNNNNNNNNNNNNNNNN"
+    @stop_area.referential.projection_type = "27562"
+  end
+  
+  def render()
+    ERB.new(@template).result(binding)
+  end
+  
+  def hub_name
+    "/ARRET.TXT"
+  end
+  
+  def self.save( stop_areas, directory, hub_export)
+    stop_areas.each do |stop_area|
+      self.new( stop_area, directory).tap do |specific_exporter|
+        specific_exporter.save
+      end
+    end
+    hub_export.log_messages.create( :severity => "ok", :key => "EXPORT|PHYSICAL_STOP_AREA_COUNT", :arguments => {"0" => stop_areas.size})
+  end
+  
+  def save
+    File.open(directory + hub_name , "a") do |f|
+      f.write("ARRET\n") if f.size == 0
+      f.write(render)
+    end if stop_area.present?
+  end
+end
+
diff --git a/app/exporters/chouette/hub/time_table_exporter.rb b/app/exporters/chouette/hub/time_table_exporter.rb
new file mode 100644
index 000000000..1e91ead6c
--- /dev/null
+++ b/app/exporters/chouette/hub/time_table_exporter.rb
@@ -0,0 +1,46 @@
+class Chouette::Hub::TimeTableExporter
+  include ERB::Util
+  attr_accessor :time_table, :directory, :template, :start_date, :end_date
+  
+  def initialize(time_table, directory)
+    @time_table = time_table
+    @directory = directory
+    @template = File.open('app/views/api/hub/periode/show.hub.erb' ){ |f| f.read }
+    @calendar = ""
+    s_date = @time_table.start_date
+    e_date = @time_table.end_date
+    while s_date <= e_date
+      if time_table.include_day?(s_date)
+        @calendar += "1"
+      else
+        @calendar += "0"
+      end
+      s_date = s_date.next_day
+    end
+  end
+  
+  def render()
+    ERB.new(@template).result(binding)
+  end
+  
+  def hub_name
+    "/PERIODE.TXT"
+  end
+  
+  def self.save(time_tables, directory, hub_export)
+    time_tables.each do |time_table|
+      self.new(time_table, directory).tap do |specific_exporter|
+        specific_exporter.save
+      end
+    end
+    hub_export.log_messages.create( :severity => "ok", :key => "EXPORT|TIME_TABLE_COUNT", :arguments => {"0" => time_tables.size})
+  end
+
+  def save
+    File.open(directory + hub_name , "a") do |f|
+      f.write("PERIODE\n") if f.size == 0
+      f.write(render)
+    end if time_table.present?
+  end
+end
+
diff --git a/app/exporters/chouette/hub/vehicle_journey_at_stop_exporter.rb b/app/exporters/chouette/hub/vehicle_journey_at_stop_exporter.rb
new file mode 100644
index 000000000..e78048ecb
--- /dev/null
+++ b/app/exporters/chouette/hub/vehicle_journey_at_stop_exporter.rb
@@ -0,0 +1,39 @@
+class Chouette::Hub::VehicleJourneyAtStopExporter
+  include ERB::Util
+  attr_accessor :vehicle_journey_at_stop, :directory, :template
+  
+  def initialize(vehicle_journey_at_stop, directory)
+    @vehicle_journey_at_stop = vehicle_journey_at_stop
+    @directory = directory
+    @template = File.open('app/views/api/hub/horaire/show.hub.erb' ) { |f| f.read }
+    @arrival_time = @vehicle_journey_at_stop.arrival_time.sec + 60 * @vehicle_journey_at_stop.arrival_time.min + 60 * 60 * @vehicle_journey_at_stop.arrival_time.hour
+    @arrival_type = "A"
+    @departure_time = @vehicle_journey_at_stop.departure_time.sec + 60 *  @vehicle_journey_at_stop.departure_time.min + 60 * 60 *  @vehicle_journey_at_stop.departure_time.hour
+    @departure_type = "D"
+  end
+  
+  def render()
+    ERB.new(@template).result(binding)
+  end
+  
+  def hub_name
+    "/HORAIRE.TXT"
+  end
+  
+  def self.save( vehicle_journey_at_stops, directory, hub_export)
+    vehicle_journey_at_stops.each do |vehicle_journey_at_stop|
+      self.new( vehicle_journey_at_stop, directory).tap do |specific_exporter|
+        specific_exporter.save
+      end
+    end
+    hub_export.log_messages.create( :severity => "ok", :key => "EXPORT|VEHICLE_JOURNEY_AT_STOP_COUNT", :arguments => {"0" => vehicle_journey_at_stops.size})
+  end
+  
+  def save
+    File.open(directory + hub_name , "a") do |f|
+      f.write("HORAIRE\n") if f.size == 0
+      f.write(render)
+    end if vehicle_journey_at_stop.present?
+  end
+end
+
diff --git a/app/exporters/chouette/hub/vehicle_journey_exporter.rb b/app/exporters/chouette/hub/vehicle_journey_exporter.rb
new file mode 100644
index 000000000..ced67fdd8
--- /dev/null
+++ b/app/exporters/chouette/hub/vehicle_journey_exporter.rb
@@ -0,0 +1,36 @@
+class Chouette::Hub::VehicleJourneyExporter
+  include ERB::Util
+  attr_accessor :vehicle_journey, :directory, :template
+  
+  def initialize(vehicle_journey, directory)
+    @vehicle_journey = vehicle_journey
+    @directory = directory
+    @template = File.open('app/views/api/hub/course/show.hub.erb' ) { |f| f.read }
+  end
+  
+  def render()
+    ERB.new(@template).result(binding)
+  end
+  
+  def hub_name
+    "/COURSE.TXT"
+  end
+  
+  def self.save( vehicle_journeys, directory, hub_export)
+    vehicle_journeys.each do |vehicle_journey|
+      #Chouette::Hub::VehicleJourneyAtStopExporter.save(vehicle_journey.vehicle_journey_at_stops, directory, hub_export)
+      self.new( vehicle_journey, directory).tap do |specific_exporter|
+        specific_exporter.save
+      end
+    end
+    hub_export.log_messages.create( :severity => "ok", :key => "EXPORT|VEHICLE_JOURNEY_COUNT", :arguments => {"0" => vehicle_journeys.size})
+  end
+  
+  def save
+    File.open(directory + hub_name , "a") do |f|
+      f.write("COURSE\n") if f.size == 0
+      f.write(render)
+    end if vehicle_journey.present?
+  end
+end
+