X-Git-Url: https://git.openstreetmap.org/rails.git/blobdiff_plain/6fa8647855b81a2f518c4995969113038187759c..9752466c0342700ebef5a7741e9105f44957a82a:/app/controllers/amf_controller.rb

diff --git a/app/controllers/amf_controller.rb b/app/controllers/amf_controller.rb
index 28e2e4b2b..febc4db68 100644
--- a/app/controllers/amf_controller.rb
+++ b/app/controllers/amf_controller.rb
@@ -10,10 +10,10 @@
 #
 # == General structure
 #
-# Apart from the talk method (which distributes the requests from the
-# AMF message), each method generally takes arguments in the order they were 
-# sent by the Potlatch SWF. Do not assume typing has been preserved. Methods 
-# all return an array to the SWF.
+# Apart from the amf_read and amf_write methods (which distribute the requests
+# from the AMF message), each method generally takes arguments in the order 
+# they were sent by the Potlatch SWF. Do not assume typing has been preserved. 
+# Methods all return an array to the SWF.
 # 
 # == Debugging
 # 
@@ -22,11 +22,6 @@
 #	return(-2,"message")		<-- also asks the user to e-mail me
 # 
 # To write to the Rails log, use RAILS_DEFAULT_LOGGER.info("message").
-#
-# == To do
-# 
-# - Check authentication
-# - Check the right things are being written to the database!
 
 class AmfController < ApplicationController
   require 'stringio'
@@ -36,16 +31,15 @@ class AmfController < ApplicationController
   session :off
   before_filter :check_write_availability
 
-  # Main AMF handler: processes the raw AMF string (using AMF library) and
+  # Main AMF handlers: process the raw AMF string (using AMF library) and
   # calls each action (private method) accordingly.
+  # ** FIXME: refactor to reduce duplication of code across read/write
   
-  def talk
-	req=StringIO.new(request.raw_post+0.chr)	# Get POST data as request
-												# (cf http://www.ruby-forum.com/topic/122163)
-	req.read(2)									# Skip version indicator and client ID
-	results={}									# Results of each body
-	renumberednodes={}							# Shared across repeated putways
-	renumberedways={}							# Shared across repeated putways
+  def amf_read
+	req=StringIO.new(request.raw_post+0.chr)# Get POST data as request
+											# (cf http://www.ruby-forum.com/topic/122163)
+	req.read(2)								# Skip version indicator and client ID
+	results={}								# Results of each body
 
 	# Parse request
 
@@ -73,6 +67,37 @@ class AmfController < ApplicationController
 		when 'getrelation';			results[index]=AMF.putdata(index,getrelation(args[0].to_i))
 		when 'getway_old';			results[index]=AMF.putdata(index,getway_old(args[0].to_i,args[1].to_i))
 		when 'getway_history';		results[index]=AMF.putdata(index,getway_history(args[0].to_i))
+		when 'getnode_history';		results[index]=AMF.putdata(index,getnode_history(args[0].to_i))
+		when 'findrelations';		results[index]=AMF.putdata(index,findrelations(*args))
+		when 'getpoi';				results[index]=AMF.putdata(index,getpoi(*args))
+	  end
+	end
+    sendresponse(results)
+  end
+
+  def amf_write
+	req=StringIO.new(request.raw_post+0.chr)
+	req.read(2)
+	results={}
+	renumberednodes={}						# Shared across repeated putways
+	renumberedways={}						# Shared across repeated putways
+
+	headers=AMF.getint(req)					# Read number of headers
+	headers.times do						# Read each header
+	  name=AMF.getstring(req)				#  |
+	  req.getc				   				#  | skip boolean
+	  value=AMF.getvalue(req)				#  |
+	  header["name"]=value					#  |
+	end
+
+	bodies=AMF.getint(req)					# Read number of bodies
+	bodies.times do							# Read each body
+	  message=AMF.getstring(req)			#  | get message name
+	  index=AMF.getstring(req)				#  | get index in response sequence
+	  bytes=AMF.getlong(req)				#  | get total size in bytes
+	  args=AMF.getvalue(req)				#  | get response (probably an array)
+
+	  case message
 		when 'putway';				r=putway(renumberednodes,*args)
 									renumberednodes=r[3]
 									if r[1] != r[2]
@@ -82,19 +107,9 @@ class AmfController < ApplicationController
 		when 'putrelation';			results[index]=AMF.putdata(index,putrelation(renumberednodes, renumberedways, *args))
 		when 'deleteway';			results[index]=AMF.putdata(index,deleteway(args[0],args[1].to_i))
 		when 'putpoi';				results[index]=AMF.putdata(index,putpoi(*args))
-		when 'getpoi';				results[index]=AMF.putdata(index,getpoi(args[0].to_i))
 	  end
 	end
-
-	# Write out response
-
-	a,b=results.length.divmod(256)
-	render :content_type => "application/x-amf", :text => proc { |response, output| 
-	  output.write 0.chr+0.chr+0.chr+0.chr+a.chr+b.chr
-	  results.each do |k,v|
-		output.write(v)
-	  end
-	}
+    sendresponse(results)
   end
 
   private
@@ -110,14 +125,15 @@ class AmfController < ApplicationController
   # in a given bounding box. Nodes are returned in full; ways and relations 
   # are IDs only. 
 
-  def whichways(xmin,ymin,xmax,ymax) #:doc:
-	xmin-=0.01; ymin-=0.01
-	xmax+=0.01; ymax+=0.01
+  def whichways(xmin, ymin, xmax, ymax) #:doc:
+	enlarge = [(xmax-xmin)/8,0.01].min
+	xmin -= enlarge; ymin -= enlarge
+	xmax += enlarge; ymax += enlarge
 
 	if POTLATCH_USE_SQL then
-	  way_ids=sql_find_way_ids_in_area(xmin,ymin,xmax,ymax)
-	  points=sql_find_pois_in_area(xmin,ymin,xmax,ymax)
-	  relation_ids=sql_find_relations_in_area_and_ways(xmin,ymin,xmax,ymax,way_ids)
+	  way_ids = sql_find_way_ids_in_area(xmin, ymin, xmax, ymax)
+	  points = sql_find_pois_in_area(xmin, ymin, xmax, ymax)
+	  relation_ids = sql_find_relations_in_area_and_ways(xmin, ymin, xmax, ymax, way_ids)
 	else
 	  # find the way ids in an area
 	  nodes_in_area = Node.find_by_area(ymin, xmin, ymax, xmax, :conditions => "current_nodes.visible = 1", :include => :ways)
@@ -128,23 +144,24 @@ class AmfController < ApplicationController
 	  points = nodes_not_used_in_area.collect { |n| [n.id, n.lon, n.lat, n.tags_as_hash] }
 
 	  # find the relations used by those nodes and ways
-	  relations = nodes_in_area.collect { |node| node.containing_relations.visible }.flatten +
-	              way_ids.collect { |id| Way.find(id).containing_relations.visible }.flatten
+	  relations = Relation.find_for_nodes(nodes_in_area.collect { |n| n.id }, :conditions => "visible = 1") +
+                  Relation.find_for_ways(way_ids, :conditions => "visible = 1")
 	  relation_ids = relations.collect { |relation| relation.id }.uniq
 	end
 
-	[way_ids,points,relation_ids]
+	[way_ids, points, relation_ids]
   end
 
   # Find deleted ways in current bounding box (similar to whichways, but ways
   # with a deleted node only - not POIs or relations).
 
-  def whichways_deleted(xmin,ymin,xmax,ymax) #:doc:
-	xmin-=0.01; ymin-=0.01
-	xmax+=0.01; ymax+=0.01
+  def whichways_deleted(xmin, ymin, xmax, ymax) #:doc:
+	xmin -= 0.01; ymin -= 0.01
+	xmax += 0.01; ymax += 0.01
 
-	nodes_in_area = Node.find_by_area(ymin, xmin, ymax,xmax, :conditions => "current_nodes.visible=0 AND current_ways.visible=0", :include => :ways_via_history )
+	nodes_in_area = Node.find_by_area(ymin, xmin, ymax, xmax, :conditions => "current_nodes.visible = 0 AND current_ways.visible = 0", :include => :ways_via_history)
 	way_ids = nodes_in_area.collect { |node| node.ways_via_history_ids }.flatten.uniq
+
 	[way_ids]
   end
 
@@ -153,20 +170,22 @@ class AmfController < ApplicationController
 
   def getway(wayid) #:doc:
 	if POTLATCH_USE_SQL then
-	  points=sql_get_nodes_in_way(wayid)
-	  tags=sql_get_tags_in_way(wayid)
+	  points = sql_get_nodes_in_way(wayid)
+	  tags = sql_get_tags_in_way(wayid)
 	else
 	  # Ideally we would do ":include => :nodes" here but if we do that
 	  # then rails only seems to return the first copy of a node when a
 	  # way includes a node more than once
-	  points = []
 	  way = Way.find(wayid)
-	  way.nodes.each do |node|
-	    points << [node.lon, node.lat, node.id, nil, node.tags_as_hash]
+	  points = way.nodes.collect do |node|
+		nodetags=node.tags_as_hash
+		nodetags.delete('created_by')
+		[node.lon, node.lat, node.id, nodetags]
 	  end
-	  tags=way.tags
+	  tags = way.tags
 	end
-	[wayid,points,tags]
+
+	[wayid, points, tags]
   end
 
   # Get an old version of a way, and all constituent nodes.
@@ -176,28 +195,44 @@ class AmfController < ApplicationController
   # at the time, generating a new id if it's still visible and has been moved/
   # retagged.
 
-  def getway_old(id,version) #:doc:
-	if version<0
-	  old_way = OldWay.find(:first, :conditions => ['visible=1 AND id=?', id], :order => 'version DESC')
+  def getway_old(id, version) #:doc:
+	if version < 0
+	  old_way = OldWay.find(:first, :conditions => ['visible = 1 AND id = ?', id], :order => 'version DESC')
 	  points = old_way.get_nodes_undelete
 	else
-	  old_way = OldWay.find(:first, :conditions => ['id=? AND version=?', id, version])
+	  old_way = OldWay.find(:first, :conditions => ['id = ? AND version = ?', id, version])
 	  points = old_way.get_nodes_revert
 	end
-	old_way.tags['history']="Retrieved from v#{old_way.version}"
-	[0,id,points,old_way.tags,old_way.version]
+
+	old_way.tags['history'] = "Retrieved from v#{old_way.version}"
+
+	[0, id, points, old_way.tags, old_way.version]
   end
   
-  # Find history of a way. Returns an array of previous versions.
+  # Find history of a way. Returns 'way', id, and 
+  # an array of previous versions.
 
   def getway_history(wayid) #:doc:
-	history=[]
-	way=Way.find(wayid)
-	way.old_ways.each do |old_way|
-	  if old_way.user.data_public then user=old_way.user.display_name else user='anonymous' end
-	  history<<[old_way.version,old_way.timestamp.strftime("%d %b %Y, %H:%M"),old_way.visible ? 1 : 0,user]
+	history = Way.find(wayid).old_ways.reverse.collect do |old_way|
+	  user = old_way.user.data_public? ? old_way.user.display_name : 'anonymous'
+	  uid  = old_way.user.data_public? ? old_way.user.id : 0
+	  [old_way.version, old_way.timestamp.strftime("%d %b %Y, %H:%M"), old_way.visible ? 1 : 0, user, uid]
+	end
+
+	['way',wayid,history]
+  end
+
+  # Find history of a node. Returns 'node', id, and 
+  # an array of previous versions.
+
+  def getnode_history(nodeid) #:doc:
+	history = Node.find(nodeid).old_nodes.reverse.collect do |old_node|
+	  user = old_node.user.data_public? ? old_node.user.display_name : 'anonymous'
+	  uid  = old_node.user.data_public? ? old_node.user.id : 0
+	  [old_node.timestamp.to_i, old_node.timestamp.strftime("%d %b %Y, %H:%M"), old_node.visible ? 1 : 0, user, uid]
 	end
-	[history]
+
+	['node',nodeid,history]
   end
 
   # Get a relation with all tags and members.
@@ -208,7 +243,28 @@ class AmfController < ApplicationController
   
   def getrelation(relid) #:doc:
 	rel = Relation.find(relid)
-	[relid,rel.tags,rel.members]
+
+	[relid, rel.tags, rel.members]
+  end
+
+  # Find relations with specified name/id.
+  # Returns array of relations, each in same form as getrelation.
+  
+  def findrelations(searchterm)
+	rels = []
+	if searchterm.to_i>0 then
+	  rel = Relation.find(searchterm.to_i)
+	  if rel and rel.visible then
+	    rels.push([rel.id, rel.tags, rel.members])
+	  end
+	else
+	  RelationTag.find(:all, :limit => 11, :conditions => ["match(v) against (?)", searchterm] ).each do |t|
+		if t.relation.visible then
+	      rels.push([t.relation.id, t.relation.tags, t.relation.members])
+	    end
+	  end
+	end
+	rels
   end
 
   # Save a relation.
@@ -217,9 +273,9 @@ class AmfController < ApplicationController
   # 1. original relation id (unchanged),
   # 2. new relation id.
 
-  def putrelation(renumberednodes, renumberedways, usertoken,relid,tags,members,visible) #:doc:
-	uid=getuserid(usertoken)
-	if !uid then return -1,"You are not logged in, so the point could not be saved." end
+  def putrelation(renumberednodes, renumberedways, usertoken, relid, tags, members, visible) #:doc:
+	uid = getuserid(usertoken)
+	if !uid then return -1,"You are not logged in, so the relation could not be saved." end
 
 	relid = relid.to_i
 	visible = visible.to_i
@@ -238,11 +294,10 @@ class AmfController < ApplicationController
 	  if mid < 0
 		mid = renumberednodes[mid] if m[0] == 'node'
 		mid = renumberedways[mid] if m[0] == 'way'
-		if mid < 0
-		  return -2, "Negative ID unresolved"
-		end
 	  end
-	  typedmembers << [m[0], mid, m[2]]
+      if mid
+	    typedmembers << [m[0], mid, m[2]]
+	  end
 	end
 
 	# assign new contents
@@ -261,7 +316,7 @@ class AmfController < ApplicationController
 	  rel.save_with_history!
 	#end
 
-	[0,relid,rel.id]
+	[0, relid, rel.id]
   end
 
   # Save a way to the database, including all nodes. Any nodes in the previous
@@ -273,87 +328,96 @@ class AmfController < ApplicationController
   # 2. new way id,
   # 3. hash of renumbered nodes (old id=>new id)
 
-  def putway(renumberednodes,usertoken,originalway,points,attributes) #:doc:
+  def putway(renumberednodes, usertoken, originalway, points, attributes) #:doc:
 
 	# -- Initialise and carry out checks
 	
-	uid=getuserid(usertoken)
+	uid = getuserid(usertoken)
 	if !uid then return -1,"You are not logged in, so the way could not be saved." end
-	originalway=originalway.to_i
+
+	originalway = originalway.to_i
 
 	points.each do |a|
-	  if a[2]==0 or a[2].nil? then return -2,"Server error - node with id 0 found in way #{originalway}." end
-	  if a[1]==90 then return -2,"Server error - node with lat -90 found in way #{originalway}." end
+	  if a[2] == 0 or a[2].nil? then return -2,"Server error - node with id 0 found in way #{originalway}." end
+	  if a[1] == 90 then return -2,"Server error - node with lat -90 found in way #{originalway}." end
 	end
 
-	if points.length<2 then return -2,"Server error - way is only #{points.length} points long." end
+	if points.length < 2 then return -2,"Server error - way is only #{points.length} points long." end
 
 	# -- Get unique nodes
 
-	if originalway<0
-	  way=Way.new
-	  uniques=[]
+	if originalway < 0
+	  way = Way.new
+	  uniques = []
 	else
-	  way=Way.find(originalway)
-	  uniques=way.unshared_node_ids
+	  way = Way.find(originalway)
+	  uniques = way.unshared_node_ids
 	end
 
 	# -- Compare nodes and save changes to any that have changed
 
-	nodes=[]
+	nodes = []
 
 	points.each do |n|
-	  lon=n[0].to_f
-	  lat=n[1].to_f
-	  id =n[2].to_i
-	  savenode=false
+	  lon = n[0].to_f
+	  lat = n[1].to_f
+	  id = n[2].to_i
+	  savenode = false
+
 	  if renumberednodes[id]
-	    id=renumberednodes[id]
-	  elsif id<0
+	    id = renumberednodes[id]
+	  elsif id < 0
 		# Create new node
-		node=Node.new
-		savenode=true
+		node = Node.new
+		savenode = true
 	  else
-		node=Node.find(id)
-		if (!fpcomp(lat,node.lat) or !fpcomp(lon,node.lon) or \
-			Tags.join(n[4])!=node.tags or node.visible==0)
-		  savenode=true
+		node = Node.find(id)
+		nodetags=node.tags_as_hash
+		nodetags.delete('created_by')
+		if !fpcomp(lat, node.lat) or !fpcomp(lon, node.lon) or
+		   n[4] != nodetags or !node.visible?
+		  savenode = true
 		end
 	  end
+
 	  if savenode
-		node.user_id=uid
-	    node.lat=lat; node.lon=lon
-	    node.tags=Tags.join(n[4])
-	    node.visible=true
+		node.user_id = uid
+	    node.lat = lat
+        node.lon = lon
+	    node.tags = Tags.join(n[4])
+	    node.visible = true
 	    node.save_with_history!
-		if id!=node.id
-		  renumberednodes[id]=node.id
-		  id=node.id
+
+		if id != node.id
+		  renumberednodes[id] = node.id
+		  id = node.id
 	    end
 	  end
-	  uniques=uniques-[id]
+
+	  uniques = uniques - [id]
 	  nodes.push(id)
 	end
 
+	# -- Save revised way
+
+	way.tags = attributes
+	way.nds = nodes
+	way.user_id = uid
+	way.visible = true
+	way.save_with_history!
+
 	# -- Delete any unique nodes
 	
 	uniques.each do |n|
-	  deleteitemrelations(n,'node')
-	  node=Node.find(n)
-	  node.user_id=uid
-	  node.visible=false
+	  deleteitemrelations(n, 'node')
+
+	  node = Node.find(n)
+	  node.user_id = uid
+	  node.visible = false
 	  node.save_with_history!
 	end
 
-	# -- Save revised way
-
-	way.tags=attributes
-	way.nds=nodes
-	way.user_id=uid
-	way.visible=true
-	way.save_with_history!
-
-	[0,originalway,way.id,renumberednodes]
+	[0, originalway, way.id, renumberednodes]
   end
 
   # Save POI to the database.
@@ -363,31 +427,32 @@ class AmfController < ApplicationController
   # 1. original node id (unchanged),
   # 2. new node id.
 
-  def putpoi(usertoken,id,lon,lat,tags,visible) #:doc:
-	uid=getuserid(usertoken)
+  def putpoi(usertoken, id, lon, lat, tags, visible) #:doc:
+	uid = getuserid(usertoken)
 	if !uid then return -1,"You are not logged in, so the point could not be saved." end
 
-	id=id.to_i
-	visible=(visible.to_i==1)
+	id = id.to_i
+	visible = (visible.to_i == 1)
+
+	if id > 0 then
+	  node = Node.find(id)
 
-	if (id>0) then
-	  node=Node.find(id)
 	  if !visible then
 	    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
-	    deleteitemrelations(id,'node')
+	    deleteitemrelations(id, 'node')
 	  end
 	else
-	  node=Node.new
+	  node = Node.new
 	end
 
 	node.user_id = uid
-	node.latitude = (lat*10000000).round
-	node.longitude = (lon*10000000).round
+	node.lat = lat
+	node.lon = lon
 	node.tags = Tags.join(tags)
-	node.visible=visible
+	node.visible = visible
 	node.save_with_history!
-	newid=node.id
-	[0,id,newid]
+
+	[0, id, node.id]
   end
 
   # Read POI from database
@@ -395,20 +460,25 @@ class AmfController < ApplicationController
   #
   # Returns array of id, long, lat, hash of tags.
 
-  def getpoi(id) #:doc:
-	n = Node.find(id)
+  def getpoi(id,timestamp) #:doc:
+	if timestamp>0 then
+	  n = OldNode.find(id, :conditions=>['UNIX_TIMESTAMP(timestamp)=?',timestamp])
+	else
+	  n = Node.find(id)
+	end
+
 	if n
 	  return [n.id, n.lon, n.lat, n.tags_as_hash]
 	else
-	  return [nil,nil,nil,'']
+	  return [nil, nil, nil, '']
 	end
   end
 
   # Delete way and all constituent nodes. Also removes from any relations.
   # Returns 0 (success), unchanged way id.
 
-  def deleteway(usertoken,way_id) #:doc:
-	uid=getuserid(usertoken)
+  def deleteway(usertoken, way_id) #:doc:
+	uid = getuserid(usertoken)
 	if !uid then return -1,"You are not logged in, so the way could not be deleted." end
 
 	# FIXME: would be good not to make two history entries when removing
@@ -416,11 +486,13 @@ class AmfController < ApplicationController
 	user = User.find(uid)
 	way = Way.find(way_id)
 	way.unshared_node_ids.each do |n|
-	  deleteitemrelations(n,'node')
+	  deleteitemrelations(n, 'node')
 	end
+	deleteitemrelations(way_id, 'way')
 
 	way.delete_with_relations_and_nodes_and_history(user)  
-	return [0,way_id]
+
+	[0, way_id]
   end
 
 
@@ -429,11 +501,13 @@ class AmfController < ApplicationController
 
   # Remove a node or way from all relations
 
-  def deleteitemrelations(objid,type) #:doc:
-	relationids = RelationMember.find(:all, :conditions => ['member_type=? and member_id=?', type, objid]).collect { |ws| ws.id }.uniq
-	relationids.each do |relid|
-	  rel=Relation.find(relid)
-	  rel.members.delete_if {|x| x[0]==type and x[1]==objid}
+  def deleteitemrelations(objid, type) #:doc:
+	relations = RelationMember.find(:all, 
+									:conditions => ['member_type = ? and member_id = ?', type, objid], 
+									:include => :relation).collect { |rm| rm.relation }.uniq
+
+	relations.each do |rel|
+	  rel.members.delete_if { |x| x[0] == type and x[1] == objid }
 	  rel.save_with_history!
 	end
   end
@@ -468,6 +542,18 @@ class AmfController < ApplicationController
 	return ((a/0.0000001).round==(b/0.0000001).round)
   end
 
+  # Send AMF response
+  
+  def sendresponse(results)
+	a,b=results.length.divmod(256)
+	render :content_type => "application/x-amf", :text => proc { |response, output| 
+	  output.write 0.chr+0.chr+0.chr+0.chr+a.chr+b.chr
+	  results.each do |k,v|
+		output.write(v)
+	  end
+	}
+  end
+
 
   # ====================================================================
   # Alternative SQL queries for getway/whichways
@@ -531,7 +617,9 @@ class AmfController < ApplicationController
 	  ORDER BY sequence_id
 	  EOF
 	ActiveRecord::Base.connection.select_all(sql).each do |row|
-	  points << [row['lon'].to_f,row['lat'].to_f,row['id'].to_i,nil,tagstring_to_hash(row['tags'])]
+	  nodetags=tagstring_to_hash(row['tags'])
+	  nodetags.delete('created_by')
+	  points << [row['lon'].to_f,row['lat'].to_f,row['id'].to_i,nodetags]
 	end
 	points
   end
@@ -545,3 +633,8 @@ class AmfController < ApplicationController
   end
 
 end
+
+# Local Variables:
+# indent-tabs-mode: t
+# tab-width: 4
+# End: