]> git.openstreetmap.org Git - rails.git/blob - app/models/trace.rb
Merge remote-tracking branch 'upstream/pull/3332'
[rails.git] / app / models / trace.rb
1 # == Schema Information
2 #
3 # Table name: gpx_files
4 #
5 #  id          :bigint(8)        not null, primary key
6 #  user_id     :bigint(8)        not null
7 #  visible     :boolean          default(TRUE), not null
8 #  name        :string           default(""), not null
9 #  size        :bigint(8)
10 #  latitude    :float
11 #  longitude   :float
12 #  timestamp   :datetime         not null
13 #  description :string           default(""), not null
14 #  inserted    :boolean          not null
15 #  visibility  :enum             default("public"), not null
16 #
17 # Indexes
18 #
19 #  gpx_files_timestamp_idx           (timestamp)
20 #  gpx_files_user_id_idx             (user_id)
21 #  gpx_files_visible_visibility_idx  (visible,visibility)
22 #
23 # Foreign Keys
24 #
25 #  gpx_files_user_id_fkey  (user_id => users.id)
26 #
27
28 class Trace < ApplicationRecord
29   require "open3"
30
31   self.table_name = "gpx_files"
32
33   belongs_to :user, :counter_cache => true
34   has_many :tags, :class_name => "Tracetag", :foreign_key => "gpx_id", :dependent => :delete_all
35   has_many :points, :class_name => "Tracepoint", :foreign_key => "gpx_id", :dependent => :delete_all
36
37   scope :visible, -> { where(:visible => true) }
38   scope :visible_to, ->(u) { visible.where("visibility IN ('public', 'identifiable') OR user_id = ?", u) }
39   scope :visible_to_all, -> { where(:visibility => %w[public identifiable]) }
40   scope :tagged, ->(t) { joins(:tags).where(:gpx_file_tags => { :tag => t }) }
41
42   validates :user, :presence => true, :associated => true
43   validates :name, :presence => true, :length => 1..255, :characters => true
44   validates :description, :presence => { :on => :create }, :length => 1..255, :characters => true
45   validates :timestamp, :presence => true
46   validates :visibility, :inclusion => %w[private public trackable identifiable]
47
48   after_destroy :remove_files
49
50   def tagstring
51     tags.collect(&:tag).join(", ")
52   end
53
54   def tagstring=(s)
55     self.tags = if s.include? ","
56                   s.split(/\s*,\s*/).grep_v(/^\s*$/).collect do |tag|
57                     tt = Tracetag.new
58                     tt.tag = tag
59                     tt
60                   end
61                 else
62                   # do as before for backwards compatibility:
63                   s.split.collect do |tag|
64                     tt = Tracetag.new
65                     tt.tag = tag
66                     tt
67                   end
68                 end
69   end
70
71   def public?
72     visibility == "public" || visibility == "identifiable"
73   end
74
75   def trackable?
76     visibility == "trackable" || visibility == "identifiable"
77   end
78
79   def identifiable?
80     visibility == "identifiable"
81   end
82
83   def large_picture=(data)
84     f = File.new(large_picture_name, "wb")
85     f.syswrite(data)
86     f.close
87   end
88
89   def icon_picture=(data)
90     f = File.new(icon_picture_name, "wb")
91     f.syswrite(data)
92     f.close
93   end
94
95   def large_picture
96     f = File.new(large_picture_name, "rb")
97     data = f.sysread(File.size(f.path))
98     f.close
99     data
100   end
101
102   def icon_picture
103     f = File.new(icon_picture_name, "rb")
104     data = f.sysread(File.size(f.path))
105     f.close
106     data
107   end
108
109   def large_picture_name
110     "#{Settings.gpx_image_dir}/#{id}.gif"
111   end
112
113   def icon_picture_name
114     "#{Settings.gpx_image_dir}/#{id}_icon.gif"
115   end
116
117   def trace_name
118     "#{Settings.gpx_trace_dir}/#{id}.gpx"
119   end
120
121   def mime_type
122     filetype = Open3.capture2("/usr/bin/file", "-Lbz", trace_name).first.chomp
123     gzipped = filetype.include?("gzip compressed")
124     bzipped = filetype.include?("bzip2 compressed")
125     zipped = filetype.include?("Zip archive")
126     tarred = filetype.include?("tar archive")
127
128     if gzipped
129       "application/x-gzip"
130     elsif bzipped
131       "application/x-bzip2"
132     elsif zipped
133       "application/x-zip"
134     elsif tarred
135       "application/x-tar"
136     else
137       "application/gpx+xml"
138     end
139   end
140
141   def extension_name
142     filetype = Open3.capture2("/usr/bin/file", "-Lbz", trace_name).first.chomp
143     gzipped = filetype.include?("gzip compressed")
144     bzipped = filetype.include?("bzip2 compressed")
145     zipped = filetype.include?("Zip archive")
146     tarred = filetype.include?("tar archive")
147
148     if tarred && gzipped
149       ".tar.gz"
150     elsif tarred && bzipped
151       ".tar.bz2"
152     elsif tarred
153       ".tar"
154     elsif gzipped
155       ".gpx.gz"
156     elsif bzipped
157       ".gpx.bz2"
158     elsif zipped
159       ".zip"
160     else
161       ".gpx"
162     end
163   end
164
165   def update_from_xml(xml, create: false)
166     p = XML::Parser.string(xml, :options => XML::Parser::Options::NOERROR)
167     doc = p.parse
168     pt = doc.find_first("//osm/gpx_file")
169
170     if pt
171       update_from_xml_node(pt, :create => create)
172     else
173       raise OSM::APIBadXMLError.new("trace", xml, "XML doesn't contain an osm/gpx_file element.")
174     end
175   rescue LibXML::XML::Error, ArgumentError => e
176     raise OSM::APIBadXMLError.new("trace", xml, e.message)
177   end
178
179   def update_from_xml_node(pt, create: false)
180     raise OSM::APIBadXMLError.new("trace", pt, "visibility missing") if pt["visibility"].nil?
181
182     self.visibility = pt["visibility"]
183
184     unless create
185       raise OSM::APIBadXMLError.new("trace", pt, "ID is required when updating.") if pt["id"].nil?
186
187       id = pt["id"].to_i
188       # .to_i will return 0 if there is no number that can be parsed.
189       # We want to make sure that there is no id with zero anyway
190       raise OSM::APIBadUserInput, "ID of trace cannot be zero when updating." if id.zero?
191       raise OSM::APIBadUserInput, "The id in the url (#{self.id}) is not the same as provided in the xml (#{id})" unless self.id == id
192     end
193
194     # We don't care about the time, as it is explicitly set on create/update/delete
195     # We don't care about the visibility as it is implicit based on the action
196     # and set manually before the actual delete
197     self.visible = true
198
199     description = pt.find("description").first
200     raise OSM::APIBadXMLError.new("trace", pt, "description missing") if description.nil?
201
202     self.description = description.content
203
204     self.tags = pt.find("tag").collect do |tag|
205       Tracetag.new(:tag => tag.content)
206     end
207   end
208
209   def xml_file
210     filetype = Open3.capture2("/usr/bin/file", "-Lbz", trace_name).first.chomp
211     gzipped = filetype.include?("gzip compressed")
212     bzipped = filetype.include?("bzip2 compressed")
213     zipped = filetype.include?("Zip archive")
214     tarred = filetype.include?("tar archive")
215
216     if gzipped || bzipped || zipped || tarred
217       file = Tempfile.new("trace.#{id}")
218
219       if tarred && gzipped
220         system("tar", "-zxOf", trace_name, :out => file.path)
221       elsif tarred && bzipped
222         system("tar", "-jxOf", trace_name, :out => file.path)
223       elsif tarred
224         system("tar", "-xOf", trace_name, :out => file.path)
225       elsif gzipped
226         system("gunzip", "-c", trace_name, :out => file.path)
227       elsif bzipped
228         system("bunzip2", "-c", trace_name, :out => file.path)
229       elsif zipped
230         system("unzip", "-p", trace_name, "-x", "__MACOSX/*", :out => file.path, :err => "/dev/null")
231       end
232
233       file.unlink
234     else
235       file = File.open(trace_name)
236     end
237
238     file
239   end
240
241   def import
242     logger.info("GPX Import importing #{name} (#{id}) from #{user.email}")
243
244     gpx = GPX::File.new(trace_name)
245
246     f_lat = 0
247     f_lon = 0
248     first = true
249
250     # If there are any existing points for this trace then delete them
251     Tracepoint.where(:gpx_id => id).delete_all
252
253     gpx.points.each_slice(1_000) do |points|
254       # Gather the trace points together for a bulk import
255       tracepoints = []
256
257       points.each do |point|
258         if first
259           f_lat = point.latitude
260           f_lon = point.longitude
261           first = false
262         end
263
264         tp = Tracepoint.new
265         tp.lat = point.latitude
266         tp.lon = point.longitude
267         tp.altitude = point.altitude
268         tp.timestamp = point.timestamp
269         tp.gpx_id = id
270         tp.trackid = point.segment
271         tracepoints << tp
272       end
273
274       # Run the before_save and before_create callbacks, and then import them in bulk with activerecord-import
275       tracepoints.each do |tp|
276         tp.run_callbacks(:save) { false }
277         tp.run_callbacks(:create) { false }
278       end
279
280       Tracepoint.import!(tracepoints)
281     end
282
283     if gpx.actual_points.positive?
284       max_lat = Tracepoint.where(:gpx_id => id).maximum(:latitude)
285       min_lat = Tracepoint.where(:gpx_id => id).minimum(:latitude)
286       max_lon = Tracepoint.where(:gpx_id => id).maximum(:longitude)
287       min_lon = Tracepoint.where(:gpx_id => id).minimum(:longitude)
288
289       max_lat = max_lat.to_f / 10000000
290       min_lat = min_lat.to_f / 10000000
291       max_lon = max_lon.to_f / 10000000
292       min_lon = min_lon.to_f / 10000000
293
294       self.latitude = f_lat
295       self.longitude = f_lon
296       self.large_picture = gpx.picture(min_lat, min_lon, max_lat, max_lon, gpx.actual_points)
297       self.icon_picture = gpx.icon(min_lat, min_lon, max_lat, max_lon)
298       self.size = gpx.actual_points
299       self.inserted = true
300       save!
301     end
302
303     logger.info "done trace #{id}"
304
305     gpx
306   end
307
308   private
309
310   def remove_files
311     FileUtils.rm_f(trace_name)
312     FileUtils.rm_f(icon_picture_name)
313     FileUtils.rm_f(large_picture_name)
314   end
315 end