]> git.openstreetmap.org Git - rails.git/blobdiff - lib/daemons/gpx_import.rb
apply patch from TomH for #477 and dont require http auth on GET to the API.
[rails.git] / lib / daemons / gpx_import.rb
index 015a2791648451ae0555ad8c1a2de53a9a2ae517..acf2153001bb065ddf4a04e879f1ed375cb11711 100755 (executable)
@@ -1,7 +1,7 @@
 #!/usr/bin/env ruby
 
 #You might want to change this
-ENV["RAILS_ENV"] ||= "development"
+#ENV["RAILS_ENV"] ||= "development"
 
 require File.dirname(__FILE__) + "/../../config/environment"
 
@@ -10,8 +10,10 @@ Signal.trap("TERM") do
   $running = false
 end
 
+logger = ActiveRecord::Base.logger
+
 while($running) do
-  
+
   ActiveRecord::Base.logger.info("GPX Import daemon wake @ #{Time.now}.")
 
   traces = Trace.find(:all, :conditions => ['inserted = ?', false])
@@ -20,18 +22,93 @@ while($running) do
     traces.each do |trace|
       begin
 
-        ActiveRecord::Base.logger.info("GPX Import importing #{trace.name} from #{trace.user.email}")
+        logger.info("GPX Import importing #{trace.name} (#{trace.id}) from #{trace.user.email}")
+
+        # TODO *nix specific, could do to work on windows... would be functionally inferior though - check for '.gz'
+        filetype = `file -b /home/osm/gpx/#{trace.id}.gpx`.chomp
+        gzipped = filetype =~ /^gzip/
+        zipped = filetype =~ /^Zip/
+
+        if gzipped
+          logger.info("gzipped")
+          filename = "/tmp/#{rand}"
+          system("gunzip -c /home/osm/gpx/#{trace.id}.gpx > #{filename}")
+        elsif zipped
+          logger.info("zipped")
+          filename = "/tmp/#{rand}"
+          system("unzip -p /home/osm/gpx/#{trace.id}.gpx > #{filename}")
+        else
+          logger.info("not gzipped")
+          filename = "/home/osm/gpx/#{trace.id}.gpx"
+        end
+
+        gpx = OSM::GPXImporter.new(filename)
+
+        f_lat = 0
+        f_lon = 0
+        first = true
+
+        gpx.points do |point|
+          if first
+            f_lat = point['latitude']
+            f_lon = point['longitude']
+          end
+
+          tp = Tracepoint.new
+          tp.lat = point['latitude'].to_f
+          tp.lng = point['longitude'].to_f
+          tp.altitude = point['altitude'].to_f
+          tp.user_id = trace.user.id
+          tp.gpx_id = trace.id
+          tp.trackid = point['segment'].to_i
+          tp.save!
+        end
 
-        #  gpx = OSM::GPXImporter.new('/tmp/2.gpx')
-        #  gpx.points do |point|
-        #    puts point['latitude']
-        #  end
-        
-        Notifier::deliver_gpx_success(trace)
-      rescue
-        Notifier::deliver_gpx_failure(trace)
+        if gpx.actual_points > 0
+          max_lat = Tracepoint.maximum('latitude', :conditions => ['gpx_id = ?', trace.id])
+          min_lat = Tracepoint.minimum('latitude', :conditions => ['gpx_id = ?', trace.id])
+          max_lon = Tracepoint.maximum('longitude', :conditions => ['gpx_id = ?', trace.id])
+          min_lon = Tracepoint.minimum('longitude', :conditions => ['gpx_id = ?', trace.id])
+
+          max_lat = max_lat.to_f / 1000000
+          min_lat = min_lat.to_f / 1000000
+          max_lon = max_lon.to_f / 1000000
+          min_lon = min_lon.to_f / 1000000
+
+          trace.latitude = f_lat
+          trace.longitude = f_lon
+          trace.large_picture = gpx.get_picture(min_lat, min_lon, max_lat, max_lon, gpx.actual_points)
+          trace.icon_picture = gpx.get_icon(min_lat, min_lon, max_lat, max_lon)
+          trace.size = gpx.actual_points
+          trace.inserted = true
+          trace.save
+
+          if gzipped || zipped
+            FileUtils.rm_f(filename)
+          end
+
+          logger.info "done trace #{trace.id}"
+          Notifier::deliver_gpx_success(trace, gpx.possible_points)
+        else
+          FileUtils.rm_f("/home/osm/gpx/#{trace.id}.gpx", filename)
+          trace.destroy
+          Notifier::deliver_gpx_failure(trace, '0 points parsed ok. Do they all have lat,lng,alt,timestamp?')
+        end
+
+      rescue Exception => ex
+        logger.info ex
+        ex.backtrace.each {|l| logger.info l }
+          FileUtils.rm_f("/home/osm/gpx/#{trace.id}.gpx", filename)
+          trace.destroy
+          Notifier::deliver_gpx_failure(trace, ex.to_s + ex.backtrace.join("\n") )
       end
     end
   end
+
+  Trace.find(:all, :conditions => ['inserted = ?', false]).each do |trace|
+     FileUtils.rm_f("/home/osm/gpx/#{trace.id}.gpx")
+     trace.destroy
+  end
+  exit
   sleep 15.minutes
 end