1 # amf_controller is a semi-standalone API for Flash clients, particularly
2 # Potlatch. All interaction between Potlatch (as a .SWF application) and the
3 # OSM database takes place using this controller. Messages are
4 # encoded in the Actionscript Message Format (AMF).
6 # Helper functions are in /lib/potlatch.
8 # Author:: editions Systeme D / Richard Fairhurst 2004-2008
9 # Licence:: public domain.
11 # == General structure
13 # Apart from the amf_read and amf_write methods (which distribute the requests
14 # from the AMF message), each method generally takes arguments in the order
15 # they were sent by the Potlatch SWF. Do not assume typing has been preserved.
16 # Methods all return an array to the SWF.
20 # Any method that returns a status code (0 for ok) can also send:
21 # return(-1,"message") <-- just puts up a dialogue
22 # return(-2,"message") <-- also asks the user to e-mail me
24 # To write to the Rails log, use RAILS_DEFAULT_LOGGER.info("message").
26 class AmfController < ApplicationController
31 # Help methods for checking boundary sanity and area size
35 before_filter :check_write_availability
37 # Main AMF handlers: process the raw AMF string (using AMF library) and
38 # calls each action (private method) accordingly.
39 # ** FIXME: refactor to reduce duplication of code across read/write
42 req=StringIO.new(request.raw_post+0.chr)# Get POST data as request
43 # (cf http://www.ruby-forum.com/topic/122163)
44 req.read(2) # Skip version indicator and client ID
45 results={} # Results of each body
49 headers=AMF.getint(req) # Read number of headers
51 headers.times do # Read each header
52 name=AMF.getstring(req) # |
53 req.getc # | skip boolean
54 value=AMF.getvalue(req) # |
55 header["name"]=value # |
58 bodies=AMF.getint(req) # Read number of bodies
59 bodies.times do # Read each body
60 message=AMF.getstring(req) # | get message name
61 index=AMF.getstring(req) # | get index in response sequence
62 bytes=AMF.getlong(req) # | get total size in bytes
63 args=AMF.getvalue(req) # | get response (probably an array)
64 logger.info "Executing AMF #{message}:#{index}"
67 when 'getpresets'; results[index]=AMF.putdata(index,getpresets())
68 when 'whichways'; results[index]=AMF.putdata(index,whichways(*args))
69 when 'whichways_deleted'; results[index]=AMF.putdata(index,whichways_deleted(*args))
70 when 'getway'; results[index]=AMF.putdata(index,getway(args[0].to_i))
71 when 'getrelation'; results[index]=AMF.putdata(index,getrelation(args[0].to_i))
72 when 'getway_old'; results[index]=AMF.putdata(index,getway_old(args[0].to_i,args[1].to_i))
73 when 'getway_history'; results[index]=AMF.putdata(index,getway_history(args[0].to_i))
74 when 'getnode_history'; results[index]=AMF.putdata(index,getnode_history(args[0].to_i))
75 when 'findrelations'; results[index]=AMF.putdata(index,findrelations(*args))
76 when 'getpoi'; results[index]=AMF.putdata(index,getpoi(*args))
79 logger.info("encoding AMF results")
84 req=StringIO.new(request.raw_post+0.chr)
87 renumberednodes={} # Shared across repeated putways
88 renumberedways={} # Shared across repeated putways
90 headers=AMF.getint(req) # Read number of headers
91 headers.times do # Read each header
92 name=AMF.getstring(req) # |
93 req.getc # | skip boolean
94 value=AMF.getvalue(req) # |
95 header["name"]=value # |
98 bodies=AMF.getint(req) # Read number of bodies
99 bodies.times do # Read each body
100 message=AMF.getstring(req) # | get message name
101 index=AMF.getstring(req) # | get index in response sequence
102 bytes=AMF.getlong(req) # | get total size in bytes
103 args=AMF.getvalue(req) # | get response (probably an array)
106 when 'putway'; r=putway(renumberednodes,*args)
109 renumberedways[r[1]] = r[2]
111 results[index]=AMF.putdata(index,r)
112 when 'putrelation'; results[index]=AMF.putdata(index,putrelation(renumberednodes, renumberedways, *args))
113 when 'deleteway'; results[index]=AMF.putdata(index,deleteway(args[0],args[1].to_i))
114 when 'putpoi'; results[index]=AMF.putdata(index,putpoi(*args))
117 sendresponse(results)
122 # Return presets (default tags, localisation etc.):
123 # uses POTLATCH_PRESETS global, set up in OSM::Potlatch.
125 def getpresets() #:doc:
126 return POTLATCH_PRESETS
129 # Find all the ways, POI nodes (i.e. not part of ways), and relations
130 # in a given bounding box. Nodes are returned in full; ways and relations
133 def whichways(xmin, ymin, xmax, ymax) #:doc:
134 xmin -= 0.01; ymin -= 0.01
135 xmax += 0.01; ymax += 0.01
137 # check boundary is sane and area within defined
138 # see /config/application.yml
140 check_boundaries(xmin, ymin, xmax, ymax)
141 rescue Exception => err
142 # FIXME: report an error rather than just return an empty result
146 if POTLATCH_USE_SQL then
147 way_ids = sql_find_way_ids_in_area(xmin, ymin, xmax, ymax)
148 points = sql_find_pois_in_area(xmin, ymin, xmax, ymax)
149 relation_ids = sql_find_relations_in_area_and_ways(xmin, ymin, xmax, ymax, way_ids)
151 # find the way ids in an area
152 nodes_in_area = Node.find_by_area(ymin, xmin, ymax, xmax, :conditions => ["current_nodes.visible = ?", true], :include => :ways)
153 way_ids = nodes_in_area.collect { |node| node.way_ids }.flatten.uniq
155 # find the node ids in an area that aren't part of ways
156 nodes_not_used_in_area = nodes_in_area.select { |node| node.ways.empty? }
157 points = nodes_not_used_in_area.collect { |n| [n.id, n.lon, n.lat, n.tags_as_hash] }
159 # find the relations used by those nodes and ways
160 relations = Relation.find_for_nodes(nodes_in_area.collect { |n| n.id }, :conditions => {:visible => true}) +
161 Relation.find_for_ways(way_ids, :conditions => {:visible => true})
162 relation_ids = relations.collect { |relation| relation.id }.uniq
165 [way_ids, points, relation_ids]
168 # Find deleted ways in current bounding box (similar to whichways, but ways
169 # with a deleted node only - not POIs or relations).
171 def whichways_deleted(xmin, ymin, xmax, ymax) #:doc:
172 xmin -= 0.01; ymin -= 0.01
173 xmax += 0.01; ymax += 0.01
175 # check boundary is sane and area within defined
176 # see /config/application.yml
178 check_boundaries(xmin, ymin, xmax, ymax)
179 rescue Exception => err
180 # FIXME: report an error rather than just return an empty result
184 nodes_in_area = Node.find_by_area(ymin, xmin, ymax, xmax, :conditions => ["current_ways.visible = ?", false], :include => :ways_via_history)
185 way_ids = nodes_in_area.collect { |node| node.ways_via_history_ids }.flatten.uniq
190 # Get a way including nodes and tags.
191 # Returns 0 (success), a Potlatch-style array of points, and a hash of tags.
193 def getway(wayid) #:doc:
194 if POTLATCH_USE_SQL then
195 points = sql_get_nodes_in_way(wayid)
196 tags = sql_get_tags_in_way(wayid)
198 # Ideally we would do ":include => :nodes" here but if we do that
199 # then rails only seems to return the first copy of a node when a
200 # way includes a node more than once
202 way = Way.find(wayid)
203 rescue ActiveRecord::RecordNotFound
207 # check case where way has been deleted or doesn't exist
208 return [wayid,[],{}] if way.nil? or !way.visible
210 points = way.nodes.collect do |node|
211 nodetags=node.tags_as_hash
212 nodetags.delete('created_by')
213 [node.lon, node.lat, node.id, nodetags]
218 [wayid, points, tags]
221 # Get an old version of a way, and all constituent nodes.
223 # For undelete (version<0), always uses the most recent version of each node,
224 # even if it's moved. For revert (version >= 0), uses the node in existence
225 # at the time, generating a new id if it's still visible and has been moved/
228 def getway_old(id, version) #:doc:
230 old_way = OldWay.find(:first, :conditions => ['visible = ? AND id = ?', true, id], :order => 'version DESC')
231 points = old_way.get_nodes_undelete unless old_way.nil?
233 old_way = OldWay.find(:first, :conditions => ['id = ? AND version = ?', id, version])
234 points = old_way.get_nodes_revert unless old_way.nil?
238 return [0, id, [], {}, -1]
240 old_way.tags['history'] = "Retrieved from v#{old_way.version}"
241 return [0, id, points, old_way.tags, old_way.version]
245 # Find history of a way. Returns 'way', id, and
246 # an array of previous versions.
248 def getway_history(wayid) #:doc:
249 history = Way.find(wayid).old_ways.reverse.collect do |old_way|
250 user = old_way.user.data_public? ? old_way.user.display_name : 'anonymous'
251 uid = old_way.user.data_public? ? old_way.user.id : 0
252 [old_way.version, old_way.timestamp.strftime("%d %b %Y, %H:%M"), old_way.visible ? 1 : 0, user, uid]
255 ['way',wayid,history]
258 # Find history of a node. Returns 'node', id, and
259 # an array of previous versions.
261 def getnode_history(nodeid) #:doc:
262 history = Node.find(nodeid).old_nodes.reverse.collect do |old_node|
263 user = old_node.user.data_public? ? old_node.user.display_name : 'anonymous'
264 uid = old_node.user.data_public? ? old_node.user.id : 0
265 [old_node.timestamp.to_i, old_node.timestamp.strftime("%d %b %Y, %H:%M"), old_node.visible ? 1 : 0, user, uid]
268 ['node',nodeid,history]
271 # Get a relation with all tags and members.
275 # 2. list of members.
277 def getrelation(relid) #:doc:
279 rel = Relation.find(relid)
280 rescue ActiveRecord::RecordNotFound
281 return [relid, {}, []]
284 return [relid, {}, []] if rel.nil? or !rel.visible
286 [relid, rel.tags, rel.members]
289 # Find relations with specified name/id.
290 # Returns array of relations, each in same form as getrelation.
292 def findrelations(searchterm)
294 if searchterm.to_i>0 then
295 rel = Relation.find(searchterm.to_i)
296 if rel and rel.visible then
297 rels.push([rel.id, rel.tags, rel.members])
300 RelationTag.find(:all, :limit => 11, :conditions => ["match(v) against (?)", searchterm] ).each do |t|
301 if t.relation.visible then
302 rels.push([t.relation.id, t.relation.tags, t.relation.members])
312 # 1. original relation id (unchanged),
313 # 2. new relation id.
315 def putrelation(renumberednodes, renumberedways, usertoken, relid, tags, members, visible) #:doc:
316 uid = getuserid(usertoken)
317 if !uid then return -1,"You are not logged in, so the relation could not be saved." end
320 visible = (visible.to_i != 0)
322 # create a new relation, or find the existing one
326 rel = Relation.find(relid)
329 # check the members are all positive, and correctly type
334 mid = renumberednodes[mid] if m[0] == 'node'
335 mid = renumberedways[mid] if m[0] == 'way'
338 typedmembers << [m[0], mid, m[2]]
342 # assign new contents
343 rel.members = typedmembers
345 rel.visible = visible
348 # check it then save it
349 # BUG: the following is commented out because it always fails on my
350 # install. I think it's a Rails bug.
352 #if !rel.preconditions_ok?
353 # return -2, "Relation preconditions failed"
355 rel.save_with_history!
361 # Save a way to the database, including all nodes. Any nodes in the previous
362 # version and no longer used are deleted.
365 # 0. '0' (code for success),
366 # 1. original way id (unchanged),
368 # 3. hash of renumbered nodes (old id=>new id)
370 def putway(renumberednodes, usertoken, originalway, points, attributes) #:doc:
372 # -- Initialise and carry out checks
374 uid = getuserid(usertoken)
375 if !uid then return -1,"You are not logged in, so the way could not be saved." end
377 originalway = originalway.to_i
380 if a[2] == 0 or a[2].nil? then return -2,"Server error - node with id 0 found in way #{originalway}." end
381 if a[1] == 90 then return -2,"Server error - node with lat -90 found in way #{originalway}." end
384 if points.length < 2 then return -2,"Server error - way is only #{points.length} points long." end
386 # -- Get unique nodes
392 way = Way.find(originalway)
393 uniques = way.unshared_node_ids
396 # -- Compare nodes and save changes to any that have changed
406 if renumberednodes[id]
407 id = renumberednodes[id]
414 nodetags=node.tags_as_hash
415 nodetags.delete('created_by')
416 if !fpcomp(lat, node.lat) or !fpcomp(lon, node.lon) or
417 n[4] != nodetags or !node.visible?
426 node.tags = Tags.join(n[4])
428 node.save_with_history!
431 renumberednodes[id] = node.id
436 uniques = uniques - [id]
440 # -- Delete any unique nodes
443 deleteitemrelations(n, 'node')
448 node.save_with_history!
451 # -- Save revised way
453 way.tags = attributes
457 way.save_with_history!
459 [0, originalway, way.id, renumberednodes]
462 # Save POI to the database.
463 # Refuses save if the node has since become part of a way.
466 # 1. original node id (unchanged),
469 def putpoi(usertoken, id, lon, lat, tags, visible) #:doc:
470 uid = getuserid(usertoken)
471 if !uid then return -1,"You are not logged in, so the point could not be saved." end
474 visible = (visible.to_i == 1)
480 unless node.ways.empty? then return -1,"The point has since become part of a way, so you cannot save it as a POI." end
481 deleteitemrelations(id, 'node')
490 node.tags = Tags.join(tags)
491 node.visible = visible
492 node.save_with_history!
497 # Read POI from database
498 # (only called on revert: POIs are usually read by whichways).
500 # Returns array of id, long, lat, hash of tags.
502 def getpoi(id,timestamp) #:doc:
504 n = OldNode.find(id, :conditions=>['UNIX_TIMESTAMP(timestamp)=?',timestamp])
510 return [n.id, n.lon, n.lat, n.tags_as_hash]
512 return [nil, nil, nil, '']
516 # Delete way and all constituent nodes. Also removes from any relations.
517 # Returns 0 (success), unchanged way id.
519 def deleteway(usertoken, way_id) #:doc:
520 uid = getuserid(usertoken)
521 if !uid then return -1,"You are not logged in, so the way could not be deleted." end
523 # FIXME: would be good not to make two history entries when removing
524 # two nodes from the same relation
525 user = User.find(uid)
526 way = Way.find(way_id)
527 way.unshared_node_ids.each do |n|
528 deleteitemrelations(n, 'node')
530 deleteitemrelations(way_id, 'way')
532 way.delete_with_relations_and_nodes_and_history(user)
538 # ====================================================================
541 # Remove a node or way from all relations
543 def deleteitemrelations(objid, type) #:doc:
544 relations = RelationMember.find(:all,
545 :conditions => ['member_type = ? and member_id = ?', type, objid],
546 :include => :relation).collect { |rm| rm.relation }.uniq
548 relations.each do |rel|
549 rel.members.delete_if { |x| x[0] == type and x[1] == objid }
550 rel.save_with_history!
554 # Break out node tags into a hash
555 # (should become obsolete as of API 0.6)
557 def tagstring_to_hash(a) #:doc:
559 Tags.split(a) do |k, v|
566 # (could be removed if no-one uses the username+password form)
568 def getuserid(token) #:doc:
569 if (token =~ /^(.+)\+(.+)$/) then
570 user = User.authenticate(:username => $1, :password => $2)
572 user = User.authenticate(:token => token)
575 return user ? user.id : nil;
578 # Compare two floating-point numbers to within 0.0000001
580 def fpcomp(a,b) #:doc:
581 return ((a/0.0000001).round==(b/0.0000001).round)
586 def sendresponse(results)
587 a,b=results.length.divmod(256)
588 render :content_type => "application/x-amf", :text => proc { |response, output|
589 output.write 0.chr+0.chr+0.chr+0.chr+a.chr+b.chr
590 results.each do |k,v|
597 # ====================================================================
598 # Alternative SQL queries for getway/whichways
600 def sql_find_way_ids_in_area(xmin,ymin,xmax,ymax)
602 SELECT DISTINCT current_way_nodes.id AS wayid
603 FROM current_way_nodes
604 INNER JOIN current_nodes ON current_nodes.id=current_way_nodes.node_id
605 INNER JOIN current_ways ON current_ways.id =current_way_nodes.id
606 WHERE current_nodes.visible=TRUE
607 AND current_ways.visible=TRUE
608 AND #{OSM.sql_for_area(ymin, xmin, ymax, xmax, "current_nodes.")}
610 return ActiveRecord::Base.connection.select_all(sql).collect { |a| a['wayid'].to_i }
613 def sql_find_pois_in_area(xmin,ymin,xmax,ymax)
615 SELECT current_nodes.id,current_nodes.latitude*0.0000001 AS lat,current_nodes.longitude*0.0000001 AS lon,current_nodes.tags
617 LEFT OUTER JOIN current_way_nodes cwn ON cwn.node_id=current_nodes.id
618 WHERE current_nodes.visible=TRUE
620 AND #{OSM.sql_for_area(ymin, xmin, ymax, xmax, "current_nodes.")}
622 return ActiveRecord::Base.connection.select_all(sql).collect { |n| [n['id'].to_i,n['lon'].to_f,n['lat'].to_f,tagstring_to_hash(n['tags'])] }
625 def sql_find_relations_in_area_and_ways(xmin,ymin,xmax,ymax,way_ids)
626 # ** It would be more Potlatchy to get relations for nodes within ways
627 # during 'getway', not here
629 SELECT DISTINCT cr.id AS relid
630 FROM current_relations cr
631 INNER JOIN current_relation_members crm ON crm.id=cr.id
632 INNER JOIN current_nodes cn ON crm.member_id=cn.id AND crm.member_type='node'
633 WHERE #{OSM.sql_for_area(ymin, xmin, ymax, xmax, "cn.")}
635 unless way_ids.empty?
638 SELECT DISTINCT cr.id AS relid
639 FROM current_relations cr
640 INNER JOIN current_relation_members crm ON crm.id=cr.id
641 WHERE crm.member_type='way'
642 AND crm.member_id IN (#{way_ids.join(',')})
645 return ActiveRecord::Base.connection.select_all(sql).collect { |a| a['relid'].to_i }.uniq
648 def sql_get_nodes_in_way(wayid)
651 SELECT latitude*0.0000001 AS lat,longitude*0.0000001 AS lon,current_nodes.id,tags
652 FROM current_way_nodes,current_nodes
653 WHERE current_way_nodes.id=#{wayid.to_i}
654 AND current_way_nodes.node_id=current_nodes.id
655 AND current_nodes.visible=TRUE
658 ActiveRecord::Base.connection.select_all(sql).each do |row|
659 nodetags=tagstring_to_hash(row['tags'])
660 nodetags.delete('created_by')
661 points << [row['lon'].to_f,row['lat'].to_f,row['id'].to_i,nodetags]
666 def sql_get_tags_in_way(wayid)
668 ActiveRecord::Base.connection.select_all("SELECT k,v FROM current_way_tags WHERE id=#{wayid.to_i}").each do |row|
669 tags[row['k']]=row['v']
677 # indent-tabs-mode: t