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 # Note that this requires a patched version of composite_primary_keys 1.1.0
21 # (see http://groups.google.com/group/compositekeys/t/a00e7562b677e193)
22 # if you are to run with POTLATCH_USE_SQL=false .
26 # Any method that returns a status code (0 for ok) can also send:
27 # return(-1,"message") <-- just puts up a dialogue
28 # return(-2,"message") <-- also asks the user to e-mail me
30 # To write to the Rails log, use RAILS_DEFAULT_LOGGER.info("message").
32 class AmfController < ApplicationController
37 # Help methods for checking boundary sanity and area size
41 before_filter :check_write_availability
43 # Main AMF handlers: process the raw AMF string (using AMF library) and
44 # calls each action (private method) accordingly.
45 # ** FIXME: refactor to reduce duplication of code across read/write
48 req=StringIO.new(request.raw_post+0.chr)# Get POST data as request
49 # (cf http://www.ruby-forum.com/topic/122163)
50 req.read(2) # Skip version indicator and client ID
51 results={} # Results of each body
55 headers=AMF.getint(req) # Read number of headers
57 headers.times do # Read each header
58 name=AMF.getstring(req) # |
59 req.getc # | skip boolean
60 value=AMF.getvalue(req) # |
61 header["name"]=value # |
64 bodies=AMF.getint(req) # Read number of bodies
65 bodies.times do # Read each body
66 message=AMF.getstring(req) # | get message name
67 index=AMF.getstring(req) # | get index in response sequence
68 bytes=AMF.getlong(req) # | get total size in bytes
69 args=AMF.getvalue(req) # | get response (probably an array)
70 logger.info "Executing AMF #{message}:#{index}"
73 when 'getpresets'; results[index]=AMF.putdata(index,getpresets())
74 when 'whichways'; results[index]=AMF.putdata(index,whichways(*args))
75 when 'whichways_deleted'; results[index]=AMF.putdata(index,whichways_deleted(*args))
76 when 'getway'; results[index]=AMF.putdata(index,getway(args[0].to_i))
77 when 'getrelation'; results[index]=AMF.putdata(index,getrelation(args[0].to_i))
78 when 'getway_old'; results[index]=AMF.putdata(index,getway_old(args[0].to_i,args[1].to_i))
79 when 'getway_history'; results[index]=AMF.putdata(index,getway_history(args[0].to_i))
80 when 'getnode_history'; results[index]=AMF.putdata(index,getnode_history(args[0].to_i))
81 when 'findgpx'; results[index]=AMF.putdata(index,findgpx(*args))
82 when 'findrelations'; results[index]=AMF.putdata(index,findrelations(*args))
83 when 'getpoi'; results[index]=AMF.putdata(index,getpoi(*args))
86 logger.info("encoding AMF results")
91 req=StringIO.new(request.raw_post+0.chr)
94 renumberednodes={} # Shared across repeated putways
95 renumberedways={} # Shared across repeated putways
97 headers=AMF.getint(req) # Read number of headers
98 headers.times do # Read each header
99 name=AMF.getstring(req) # |
100 req.getc # | skip boolean
101 value=AMF.getvalue(req) # |
102 header["name"]=value # |
105 bodies=AMF.getint(req) # Read number of bodies
106 bodies.times do # Read each body
107 message=AMF.getstring(req) # | get message name
108 index=AMF.getstring(req) # | get index in response sequence
109 bytes=AMF.getlong(req) # | get total size in bytes
110 args=AMF.getvalue(req) # | get response (probably an array)
113 when 'putway'; r=putway(renumberednodes,*args)
115 if r[1] != r[2] then renumberedways[r[1]] = r[2] end
116 results[index]=AMF.putdata(index,r)
117 when 'putrelation'; results[index]=AMF.putdata(index,putrelation(renumberednodes, renumberedways, *args))
118 when 'deleteway'; results[index]=AMF.putdata(index,deleteway(*args))
119 when 'putpoi'; r=putpoi(*args)
120 if r[1] != r[2] then renumberednodes[r[1]] = r[2] end
121 results[index]=AMF.putdata(index,r)
122 when 'startchangeset'; results[index]=AMF.putdata(index,startchangeset(*args))
125 sendresponse(results)
130 # Start new changeset
132 def startchangeset(usertoken, cstags, closeid, closecomment)
133 uid = getuserid(usertoken)
134 if !uid then return -1,"You are not logged in, so Potlatch can't write any changes to the database." end
136 # close previous changeset and add comment
138 cs = Changeset.find(closeid)
139 cs.set_closed_time_now
140 if closecomment.empty?
143 cs.tags['comment']=closecomment
148 # open a new changeset
152 # Don't like the next two lines. These need to be abstracted to the model more/better
153 cs.created_at = Time.now
154 cs.closed_at = Time.new + Changeset::IDLE_TIMEOUT
159 # Return presets (default tags, localisation etc.):
160 # uses POTLATCH_PRESETS global, set up in OSM::Potlatch.
162 def getpresets() #:doc:
163 return POTLATCH_PRESETS
166 # Find all the ways, POI nodes (i.e. not part of ways), and relations
167 # in a given bounding box. Nodes are returned in full; ways and relations
170 def whichways(xmin, ymin, xmax, ymax) #:doc:
171 xmin -= 0.01; ymin -= 0.01
172 xmax += 0.01; ymax += 0.01
174 # check boundary is sane and area within defined
175 # see /config/application.yml
177 check_boundaries(xmin, ymin, xmax, ymax)
178 rescue Exception => err
179 return [-2,"Sorry - I can't get the map for that area."]
182 if POTLATCH_USE_SQL then
183 ways = sql_find_ways_in_area(xmin, ymin, xmax, ymax)
184 points = sql_find_pois_in_area(xmin, ymin, xmax, ymax)
185 relations = sql_find_relations_in_area_and_ways(xmin, ymin, xmax, ymax, ways.collect {|x| x[0]})
187 # find the way ids in an area
188 nodes_in_area = Node.find_by_area(ymin, xmin, ymax, xmax, :conditions => ["current_nodes.visible = ?", true], :include => :ways)
189 ways = nodes_in_area.collect { |node|
190 node.ways.collect { |w| [w.id,w.version] }.flatten
194 # find the node ids in an area that aren't part of ways
195 nodes_not_used_in_area = nodes_in_area.select { |node| node.ways.empty? }
196 points = nodes_not_used_in_area.collect { |n| [n.id, n.lon, n.lat, n.tags] }
198 # find the relations used by those nodes and ways
199 relations = Relation.find_for_nodes(nodes_in_area.collect { |n| n.id }, :conditions => {:visible => true}) +
200 Relation.find_for_ways(ways.collect { |w| w[0] }, :conditions => {:visible => true})
201 relations = relations.collect { |relation| [relation.id,relation.version] }.uniq
204 [0,ways, points, relations]
207 # Find deleted ways in current bounding box (similar to whichways, but ways
208 # with a deleted node only - not POIs or relations).
210 def whichways_deleted(xmin, ymin, xmax, ymax) #:doc:
211 xmin -= 0.01; ymin -= 0.01
212 xmax += 0.01; ymax += 0.01
214 # check boundary is sane and area within defined
215 # see /config/application.yml
217 check_boundaries(xmin, ymin, xmax, ymax)
218 rescue Exception => err
219 return [-2,"Sorry - I can't get the map for that area."]
222 nodes_in_area = Node.find_by_area(ymin, xmin, ymax, xmax, :conditions => ["current_ways.visible = ?", false], :include => :ways_via_history)
223 way_ids = nodes_in_area.collect { |node| node.ways_via_history_ids }.flatten.uniq
228 # Get a way including nodes and tags.
229 # Returns the way id, a Potlatch-style array of points, a hash of tags, and the version number.
231 def getway(wayid) #:doc:
232 if POTLATCH_USE_SQL then
233 points = sql_get_nodes_in_way(wayid)
234 tags = sql_get_tags_in_way(wayid)
235 version = sql_get_way_version(wayid)
237 # Ideally we would do ":include => :nodes" here but if we do that
238 # then rails only seems to return the first copy of a node when a
239 # way includes a node more than once
241 way = Way.find(wayid)
242 rescue ActiveRecord::RecordNotFound
246 # check case where way has been deleted or doesn't exist
247 return [wayid,[],{}] if way.nil? or !way.visible
249 points = way.nodes.collect do |node|
251 nodetags.delete('created_by')
252 [node.lon, node.lat, node.id, nodetags]
255 version = way.version
258 [wayid, points, tags, version]
261 # Get an old version of a way, and all constituent nodes.
263 # For undelete (version<0), always uses the most recent version of each node,
264 # even if it's moved. For revert (version >= 0), uses the node in existence
265 # at the time, generating a new id if it's still visible and has been moved/
271 # 2. array of points,
274 # 5. is this the current, visible version? (boolean)
276 def getway_old(id, version) #:doc:
278 old_way = OldWay.find(:first, :conditions => ['visible = ? AND id = ?', true, id], :order => 'version DESC')
279 points = old_way.get_nodes_undelete unless old_way.nil?
281 old_way = OldWay.find(:first, :conditions => ['id = ? AND version = ?', id, version])
282 points = old_way.get_nodes_revert unless old_way.nil?
286 return [-1, id, [], {}, -1,0]
289 old_way.tags['history'] = "Retrieved from v#{old_way.version}"
290 return [0, id, points, old_way.tags, old_way.version, (curway.version==old_way.version and curway.visible)]
294 # Find history of a way. Returns 'way', id, and
295 # an array of previous versions.
297 def getway_history(wayid) #:doc:
299 history = Way.find(wayid).old_ways.reverse.collect do |old_way|
300 user_object = old_way.changeset.user
301 user = user_object.data_public? ? user_object.display_name : 'anonymous'
302 uid = user_object.data_public? ? user_object.id : 0
303 [old_way.version, old_way.timestamp.strftime("%d %b %Y, %H:%M"), old_way.visible ? 1 : 0, user, uid]
306 return ['way',wayid,history]
307 rescue ActiveRecord::RecordNotFound
308 return ['way', wayid, []]
312 # Find history of a node. Returns 'node', id, and
313 # an array of previous versions.
315 def getnode_history(nodeid) #:doc:
317 history = Node.find(nodeid).old_nodes.reverse.collect do |old_node|
318 user_object = old_node.changeset.user
319 user = user_object.data_public? ? user_object.display_name : 'anonymous'
320 uid = user_object.data_public? ? user_object.id : 0
321 [old_node.version, old_node.timestamp.strftime("%d %b %Y, %H:%M"), old_node.visible ? 1 : 0, user, uid]
324 return ['node',nodeid,history]
325 rescue ActiveRecord::RecordNotFound
326 return ['node', nodeid, []]
330 # Find GPS traces with specified name/id.
331 # Returns array listing GPXs, each one comprising id, name and description.
333 def findgpx(searchterm, usertoken)
334 uid = getuserid(usertoken)
335 if !uid then return -1,"You must be logged in to search for GPX traces." end
338 if searchterm.to_i>0 then
339 gpx = Trace.find(searchterm.to_i, :conditions => ["visible=? AND (public=? OR user_id=?)",true,true,uid] )
341 gpxs.push([gpx.id, gpx.name, gpx.description])
344 Trace.find(:all, :limit => 21, :conditions => ["visible=? AND (public=? OR user_id=?) AND MATCH(name) AGAINST (?)",true,true,uid,searchterm] ).each do |gpx|
345 gpxs.push([gpx.id, gpx.name, gpx.description])
351 # Get a relation with all tags and members.
355 # 2. list of members,
358 def getrelation(relid) #:doc:
360 rel = Relation.find(relid)
361 rescue ActiveRecord::RecordNotFound
362 return [relid, {}, []]
365 return [relid, {}, [], nil] if rel.nil? or !rel.visible
366 [relid, rel.tags, rel.members, rel.version]
369 # Find relations with specified name/id.
370 # Returns array of relations, each in same form as getrelation.
372 def findrelations(searchterm)
374 if searchterm.to_i>0 then
375 rel = Relation.find(searchterm.to_i)
376 if rel and rel.visible then
377 rels.push([rel.id, rel.tags, rel.members])
380 RelationTag.find(:all, :limit => 11, :conditions => ["match(v) against (?)", searchterm] ).each do |t|
381 if t.relation.visible then
382 rels.push([t.relation.id, t.relation.tags, t.relation.members])
392 # 1. original relation id (unchanged),
393 # 2. new relation id.
395 def putrelation(renumberednodes, renumberedways, usertoken, changeset, relid, tags, members, visible) #:doc:
396 uid = getuserid(usertoken)
397 if !uid then return -1,"You are not logged in, so the relation could not be saved." end
400 visible = (visible.to_i != 0)
402 # create a new relation, or find the existing one
407 rel = Relation.find(relid)
410 # check the members are all positive, and correctly type
415 mid = renumberednodes[mid] if m[0] == 'node'
416 mid = renumberedways[mid] if m[0] == 'way'
419 typedmembers << [m[0], mid, m[2]]
423 # assign new contents
424 rel.members = typedmembers
426 rel.visible = visible
427 rel.changeset_id = changeset
429 # check it then save it
430 # BUG: the following is commented out because it always fails on my
431 # install. I think it's a Rails bug.
433 #if !rel.preconditions_ok?
434 # return -2, "Relation preconditions failed"
436 rel.save_with_history!
442 # Save a way to the database, including all nodes. Any nodes in the previous
443 # version and no longer used are deleted.
446 # 0. '0' (code for success),
447 # 1. original way id (unchanged),
449 # 3. hash of renumbered nodes (old id=>new id),
452 def putway(renumberednodes, usertoken, changeset, originalway, points, attributes) #:doc:
454 # -- Initialise and carry out checks
456 uid = getuserid(usertoken)
457 if !uid then return -1,"You are not logged in, so the way could not be saved." end
459 originalway = originalway.to_i
462 if a[2] == 0 or a[2].nil? then return -2,"Server error - node with id 0 found in way #{originalway}." end
463 if a[1] == 90 then return -2,"Server error - node with lat -90 found in way #{originalway}." end
466 if points.length < 2 then return -2,"Server error - way is only #{points.length} points long." end
468 # -- Get unique nodes
472 way.version = 0 # otherwise +=1 breaks
475 way = Way.find(originalway)
476 uniques = way.unshared_node_ids
479 # -- Compare nodes and save changes to any that have changed
489 if renumberednodes[id]
490 id = renumberednodes[id]
494 node.version = 0 # otherwise +=1 breaks
499 nodetags.delete('created_by')
500 if !fpcomp(lat, node.lat) or !fpcomp(lon, node.lon) or
501 n[4] != nodetags or !node.visible?
507 node.changeset_id = changeset
512 node.save_with_history!
515 renumberednodes[id] = node.id
520 uniques = uniques - [id]
524 # -- Delete any unique nodes
527 deleteitemrelations(n, 'node')
530 node.changeset_id = changeset
532 node.save_with_history!
535 # -- Save revised way
537 if way.tags!=attributes or way.nds!=nodes or !way.visible?
538 way.tags = attributes
540 way.changeset_id = changeset
542 way.save_with_history!
545 [0, originalway, way.id, renumberednodes, way.version]
548 # Save POI to the database.
549 # Refuses save if the node has since become part of a way.
552 # 1. original node id (unchanged),
556 def putpoi(usertoken, changeset, id, lon, lat, tags, visible) #:doc:
557 uid = getuserid(usertoken)
558 if !uid then return -1,"You are not logged in, so the point could not be saved." end
561 visible = (visible.to_i == 1)
567 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
568 deleteitemrelations(id, 'node')
575 node.changeset_id = changeset
579 node.visible = visible
580 node.save_with_history!
582 [0, id, node.id, node.version]
585 # Read POI from database
586 # (only called on revert: POIs are usually read by whichways).
588 # Returns array of id, long, lat, hash of tags, version.
590 def getpoi(id,version) #:doc:
592 n = OldNode.find(id, :conditions=>['version=?',version])
598 return [n.id, n.lon, n.lat, n.tags, n.version]
600 return [nil, nil, nil, {}, nil]
604 # Delete way and all constituent nodes. Also removes from any relations.
605 # Returns 0 (success), unchanged way id.
607 def deleteway(usertoken, changeset_id, way_id) #:doc:
608 if !getuserid(usertoken) then return -1,"You are not logged in, so the way could not be deleted." end
612 # FIXME: would be good not to make two history entries when removing
613 # two nodes from the same relation
614 way = Way.find(way_id)
615 way.unshared_node_ids.each do |n|
616 deleteitemrelations(n, 'node')
618 deleteitemrelations(way_id, 'way')
620 way.delete_with_relations_and_nodes_and_history(changeset_id.to_i)
626 # ====================================================================
629 # Remove a node or way from all relations
631 def deleteitemrelations(objid, type) #:doc:
632 relations = RelationMember.find(:all,
633 :conditions => ['member_type = ? and member_id = ?', type, objid],
634 :include => :relation).collect { |rm| rm.relation }.uniq
636 relations.each do |rel|
637 rel.members.delete_if { |x| x[0] == type and x[1] == objid }
638 rel.save_with_history!
642 # Break out node tags into a hash
643 # (should become obsolete as of API 0.6)
645 def tagstring_to_hash(a) #:doc:
647 Tags.split(a) do |k, v|
654 # (can also be of form user:pass)
656 def getuserid(token) #:doc:
657 if (token =~ /^(.+)\:(.+)$/) then
658 user = User.authenticate(:username => $1, :password => $2)
660 user = User.authenticate(:token => token)
663 return user ? user.id : nil;
666 # Compare two floating-point numbers to within 0.0000001
668 def fpcomp(a,b) #:doc:
669 return ((a/0.0000001).round==(b/0.0000001).round)
674 def sendresponse(results)
675 a,b=results.length.divmod(256)
676 render :content_type => "application/x-amf", :text => proc { |response, output|
677 # ** move amf writing loop into here -
678 # basically we read the messages in first (into an array of some sort),
679 # then iterate through that array within here, and do all the AMF writing
680 output.write 0.chr+0.chr+0.chr+0.chr+a.chr+b.chr
681 results.each do |k,v|
688 # ====================================================================
689 # Alternative SQL queries for getway/whichways
691 def sql_find_ways_in_area(xmin,ymin,xmax,ymax)
693 SELECT DISTINCT current_ways.id AS wayid,current_ways.version AS version
694 FROM current_way_nodes
695 INNER JOIN current_nodes ON current_nodes.id=current_way_nodes.node_id
696 INNER JOIN current_ways ON current_ways.id =current_way_nodes.id
697 WHERE current_nodes.visible=TRUE
698 AND current_ways.visible=TRUE
699 AND #{OSM.sql_for_area(ymin, xmin, ymax, xmax, "current_nodes.")}
701 return ActiveRecord::Base.connection.select_all(sql).collect { |a| [a['wayid'].to_i,a['version'].to_i] }
704 def sql_find_pois_in_area(xmin,ymin,xmax,ymax)
707 SELECT current_nodes.id,current_nodes.latitude*0.0000001 AS lat,current_nodes.longitude*0.0000001 AS lon,current_nodes.version
709 LEFT OUTER JOIN current_way_nodes cwn ON cwn.node_id=current_nodes.id
710 WHERE current_nodes.visible=TRUE
712 AND #{OSM.sql_for_area(ymin, xmin, ymax, xmax, "current_nodes.")}
714 ActiveRecord::Base.connection.select_all(sql).each do |row|
716 ActiveRecord::Base.connection.select_all("SELECT k,v FROM current_node_tags WHERE id=#{row['id']}").each do |n|
717 poitags[n['k']]=n['v']
719 pois << [row['id'].to_i, row['lon'].to_f, row['lat'].to_f, poitags, row['version'].to_i]
724 def sql_find_relations_in_area_and_ways(xmin,ymin,xmax,ymax,way_ids)
725 # ** It would be more Potlatchy to get relations for nodes within ways
726 # during 'getway', not here
728 SELECT DISTINCT cr.id AS relid,cr.version AS version
729 FROM current_relations cr
730 INNER JOIN current_relation_members crm ON crm.id=cr.id
731 INNER JOIN current_nodes cn ON crm.member_id=cn.id AND crm.member_type='node'
732 WHERE #{OSM.sql_for_area(ymin, xmin, ymax, xmax, "cn.")}
734 unless way_ids.empty?
737 SELECT DISTINCT cr.id AS relid,cr.version AS version
738 FROM current_relations cr
739 INNER JOIN current_relation_members crm ON crm.id=cr.id
740 WHERE crm.member_type='way'
741 AND crm.member_id IN (#{way_ids.join(',')})
744 return ActiveRecord::Base.connection.select_all(sql).collect { |a| [a['relid'].to_i,a['version'].to_i] }
747 def sql_get_nodes_in_way(wayid)
750 SELECT latitude*0.0000001 AS lat,longitude*0.0000001 AS lon,current_nodes.id
751 FROM current_way_nodes,current_nodes
752 WHERE current_way_nodes.id=#{wayid.to_i}
753 AND current_way_nodes.node_id=current_nodes.id
754 AND current_nodes.visible=TRUE
757 ActiveRecord::Base.connection.select_all(sql).each do |row|
759 ActiveRecord::Base.connection.select_all("SELECT k,v FROM current_node_tags WHERE id=#{row['id']}").each do |n|
760 nodetags[n['k']]=n['v']
762 nodetags.delete('created_by')
763 points << [row['lon'].to_f,row['lat'].to_f,row['id'].to_i,nodetags]
768 def sql_get_tags_in_way(wayid)
770 ActiveRecord::Base.connection.select_all("SELECT k,v FROM current_way_tags WHERE id=#{wayid.to_i}").each do |row|
771 tags[row['k']]=row['v']
776 def sql_get_way_version(wayid)
777 ActiveRecord::Base.connection.select_one("SELECT version FROM current_ways WHERE id=#{wayid.to_i}")