1 class AmfController < ApplicationController
5 # RAILS_DEFAULT_LOGGER.error("Args: #{args[0]}, #{args[1]}, #{args[2]}, #{args[3]}")
7 # ====================================================================
10 # ---- talk process AMF request
13 req=StringIO.new(request.raw_post) # Get POST data as request
14 req.read(2) # Skip version indicator and client ID
15 results={} # Results of each body
20 headers=getint(req) # Read number of headers
21 for i in (1..headers) # Read each header
22 name=getstring(req) # |
23 req.getc # | skip boolean
24 value=getvalue(req) # |
25 header["name"]=value # |
28 bodies=getint(req) # Read number of bodies
29 for i in (1..bodies) # Read each body
30 message=getstring(req) # | get message name
31 index=getstring(req) # | get index in response sequence
32 bytes=getlong(req) # | get total size in bytes
33 args=getvalue(req) # | get response (probably an array)
36 when 'getpresets'; results[index]=putdata(index,getpresets)
37 when 'whichways'; results[index]=putdata(index,whichways(args))
38 when 'getway'; results[index]=putdata(index,getway(args))
39 when 'putway'; results[index]=putdata(index,putway(args))
40 when 'deleteway'; results[index]=putdata(index,deleteway(args))
47 response.headers["Content-Type"]="application/x-amf"
48 a,b=results.length.divmod(256)
49 ans=0.chr+0.chr+0.chr+0.chr+a.chr+b.chr
58 # ====================================================================
62 # return presets,presetmenus and presetnames arrays
66 presetmenus={}; presetmenus['point']=[]; presetmenus['way']=[]
67 presetnames={}; presetnames['point']={}; presetnames['way']={}
71 File.open("config/potlatch/presets.txt") do |file|
72 file.each_line {|line|
74 if (t=~/(\w+)\/(\w+)/) then
77 presetmenus[presettype].push(presetcategory)
78 presetnames[presettype][presetcategory]=["(no preset)"]
79 elsif (t=~/^(.+):\s?(.+)$/) then
81 presetnames[presettype][presetcategory].push(pre)
83 kv.split(',').each {|a|
84 if (a=~/^(.+)=(.*)$/) then presets[pre][$1]=$2 end
89 [presets,presetmenus,presetnames]
92 # ----- whichways(left,bottom,right,top)
93 # return array of ways in current bounding box
94 # at present, instead of using correct (=more complex) SQL to find
95 # corner-crossing ways, it simply enlarges the bounding box by +/- 0.01
98 waylist=WaySegment.find_by_sql("SELECT DISTINCT current_way_segments.id AS wayid"+
99 " FROM current_way_segments,current_segments,current_nodes "+
100 " WHERE segment_id=current_segments.id "+
101 " AND current_segments.visible=1 "+
102 " AND node_a=current_nodes.id "+
103 " AND (latitude BETWEEN "+(args[1].to_f-0.01).to_s+" AND "+(args[3].to_f+0.01).to_s+") "+
104 " AND (longitude BETWEEN "+(args[0].to_f-0.01).to_s+" AND "+(args[2].to_f+0.01).to_s+")")
112 # ----- getway (objectname, way, baselong, basey, masterscale)
113 # returns objectname, array of co-ordinates, attributes,
114 # xmin,xmax,ymin,ymax
117 objname,wayid,baselong,basey,masterscale=args
121 xmin=999999; xmax=-999999
122 ymin=999999; ymax=-999999
124 readwayquery(wayid).each {|row|
125 xs1=long2coord(row['long1'].to_f,baselong,masterscale); ys1=lat2coord(row['lat1'].to_f,basey,masterscale)
126 xs2=long2coord(row['long2'].to_f,baselong,masterscale); ys2=lat2coord(row['lat2'].to_f,basey,masterscale)
127 if (row['id1'].to_i!=lastid)
128 points<<[xs1,ys1,row['id1'].to_i,0,tag2array(row['tags1']),0]
130 lastid=row['id2'].to_i
131 points<<[xs2,ys2,row['id2'].to_i,1,tag2array(row['tags2']),row['segment_id'].to_i]
132 xmin=[xmin,row['long1'].to_f,row['long2'].to_f].min
133 xmax=[xmax,row['long1'].to_f,row['long2'].to_f].max
134 ymin=[ymin,row['lat1'].to_f,row['lat2'].to_f].min
135 ymax=[ymax,row['lat1'].to_f,row['lat2'].to_f].max
139 attrlist=ActiveRecord::Base.connection.select_all "SELECT k,v FROM current_way_tags WHERE id=#{wayid}"
140 attrlist.each {|a| attributes[a['k']]=a['v'] }
142 [objname,points,attributes,xmin,xmax,ymin,ymax]
145 # ----- putway (user token, way, array of co-ordinates, array of attributes,
146 # baselong, basey, masterscale)
147 # returns current way ID, new way ID, hash of renumbered nodes,
148 # xmin,xmax,ymin,ymax
151 usertoken,originalway,points,attributes,baselong,basey,masterscale=args
152 uid=getuserid(usertoken); if !uid then return end
153 db_uqs='uniq'+uid.to_s+originalway.to_i.abs.to_s+Time.new.to_i.to_s # temp uniquesegments table name, typically 51 chars
154 db_now='@now'+uid.to_s+originalway.to_i.abs.to_s+Time.new.to_i.to_s # 'now' variable name, typically 51 chars
155 ActiveRecord::Base.connection.execute("SET #{db_now}=NOW()")
156 originalway=originalway.to_i
158 # -- 3. read original way into memory
160 xc={}; yc={}; tagc={}; seg={}
163 readwayquery(way).each { |row|
164 id1=row['id1'].to_i; xc[id1]=row['long1'].to_f; yc[id1]=row['lat1'].to_f; tagc[id1]=row['tags1']
165 id2=row['id2'].to_i; xc[id2]=row['long2'].to_f; yc[id2]=row['lat2'].to_f; tagc[id2]=row['tags2']
166 seg[row['segment_id'].to_i]=id1.to_s+'-'+id2.to_s
169 way=ActiveRecord::Base.connection.insert("INSERT INTO current_ways (user_id,timestamp,visible) VALUES (#{uid},#{db_now},1)")
172 # -- 4. get version by inserting new row into ways
174 version=ActiveRecord::Base.connection.insert("INSERT INTO ways (id,user_id,timestamp,visible) VALUES (#{way},#{uid},#{db_now},1)")
176 # -- 5. compare nodes and update xmin,xmax,ymin,ymax
178 xmin=999999; xmax=-999999
179 ymin=999999; ymax=-999999
184 points.each_index do |i|
185 xs=coord2long(points[i][0],masterscale,baselong)
186 ys=coord2lat(points[i][1],masterscale,basey)
187 xmin=[xs,xmin].min; xmax=[xs,xmax].max
188 ymin=[ys,ymin].min; ymax=[ys,ymax].max
189 node=points[i][2].to_i
190 tagstr=array2tag(points[i][4])
191 tagsql="'"+sqlescape(tagstr)+"'"
196 newnode=ActiveRecord::Base.connection.insert("INSERT INTO current_nodes ( latitude,longitude,timestamp,user_id,visible,tags) VALUES ( #{ys},#{xs},#{db_now},#{uid},1,#{tagsql})")
197 ActiveRecord::Base.connection.insert("INSERT INTO nodes (id,latitude,longitude,timestamp,user_id,visible,tags) VALUES (#{newnode},#{ys},#{xs},#{db_now},#{uid},1,#{tagsql})")
199 renumberednodes[node.to_s]=newnode.to_s
201 elsif xc.has_key?(node)
202 # old node from original way - update
203 if (xs!=xc[node] or (ys/0.0000001).round!=(yc[node]/0.0000001).round or tagstr!=tagc[node])
204 ActiveRecord::Base.connection.insert("INSERT INTO nodes (id,latitude,longitude,timestamp,user_id,visible,tags) VALUES (#{node},#{ys},#{xs},#{db_now},#{uid},1,#{tagsql})")
205 ActiveRecord::Base.connection.update("UPDATE current_nodes SET latitude=#{ys},longitude=#{xs},timestamp=#{db_now},user_id=#{uid},tags=#{tagsql},visible=1 WHERE id=#{node}")
207 if (nodelist!='') then nodelist+=',' end; nodelist+=node.to_s
210 # old node, created in another way and now added to this way
211 if (nodelist!='') then nodelist+=',' end; nodelist+=node.to_s
217 ActiveRecord::Base.connection.update("UPDATE current_nodes SET timestamp=#{db_now},user_id=#{uid},visible=1 WHERE id IN (#{nodelist})")
220 # -- 6.i compare segments
224 for i in (0..(points.length-2))
225 if (points[i+1][3].to_i==0) then next end
226 segid=points[i+1][5].to_i
227 from =points[i ][2].to_i
228 to =points[i+1][2].to_i
229 if seg.has_key?(segid)
230 if seg[segid]=="#{from}-#{to}" then
231 if (seglist!='') then seglist+=',' end; seglist+=segid.to_s
235 segid=ActiveRecord::Base.connection.insert("INSERT INTO current_segments ( node_a,node_b,timestamp,user_id,visible,tags) VALUES ( #{from},#{to},#{db_now},#{uid},1,'')")
236 ActiveRecord::Base.connection.insert("INSERT INTO segments (id,node_a,node_b,timestamp,user_id,visible,tags) VALUES (#{segid},#{from},#{to},#{db_now},#{uid},1,'')")
238 numberedsegments[(i+1).to_s]=segid.to_s
240 # numberedsegments.each{|a,b| RAILS_DEFAULT_LOGGER.error("Sending back: seg no. #{a} -> id #{b}") }
243 ActiveRecord::Base.connection.update("UPDATE current_segments SET timestamp=#{db_now},user_id=#{uid},visible=1 WHERE id IN (#{seglist})")
247 # -- 6.ii insert new way segments
249 createuniquesegments(way,db_uqs)
252 # ActiveRecord::Base.connection.select_values("SELECT segment_id FROM #{db_uqs}").each {|b| a+=b+',' }
253 # RAILS_DEFAULT_LOGGER.error("Unique segments are #{a}")
254 # a=ActiveRecord::Base.connection.select_value("SELECT #{db_now}")
255 # RAILS_DEFAULT_LOGGER.error("Timestamp of this edit is #{a}")
256 # RAILS_DEFAULT_LOGGER.error("Userid of this edit is #{uid}")
258 # delete nodes from uniquesegments (and not in modified way)
261 INSERT INTO nodes (id,latitude,longitude,timestamp,user_id,visible)
262 SELECT DISTINCT cn.id,cn.latitude,cn.longitude,#{db_now},#{uid},0
263 FROM current_nodes AS cn,
264 current_segments AS cs,
266 WHERE(cn.id=cs.node_a OR cn.id=cs.node_b)
267 AND cs.id=us.segment_id AND cs.visible=1
268 AND (cn.timestamp!=#{db_now} OR cn.user_id!=#{uid})
270 ActiveRecord::Base.connection.insert(sql)
273 UPDATE current_nodes AS cn,
274 current_segments AS cs,
276 SET cn.timestamp=#{db_now},cn.visible=0,cn.user_id=#{uid}
277 WHERE (cn.id=cs.node_a OR cn.id=cs.node_b)
278 AND cs.id=us.segment_id AND cs.visible=1
279 AND (cn.timestamp!=#{db_now} OR cn.user_id!=#{uid})
281 ActiveRecord::Base.connection.update(sql)
283 # delete segments from uniquesegments (and not in modified way)
286 INSERT INTO segments (id,node_a,node_b,timestamp,user_id,visible)
287 SELECT DISTINCT segment_id,node_a,node_b,#{db_now},#{uid},0
288 FROM current_segments AS cs, #{db_uqs} AS us
289 WHERE cs.id=us.segment_id AND cs.visible=1
290 AND (cs.timestamp!=#{db_now} OR cs.user_id!=#{uid})
292 ActiveRecord::Base.connection.insert(sql)
295 UPDATE current_segments AS cs, #{db_uqs} AS us
296 SET cs.timestamp=#{db_now},cs.visible=0,cs.user_id=#{uid}
297 WHERE cs.id=us.segment_id AND cs.visible=1
298 AND (cs.timestamp!=#{db_now} OR cs.user_id!=#{uid})
300 ActiveRecord::Base.connection.update(sql)
301 ActiveRecord::Base.connection.execute("DROP TABLE #{db_uqs}")
303 # insert new version of route into way_segments
308 for i in (0..(points.length-2))
309 if (points[i+1][3].to_i==0) then next end
310 if insertsql !='' then insertsql +=',' end
311 if currentsql!='' then currentsql+=',' end
312 insertsql +="(#{way},#{points[i+1][5]},#{version})"
313 currentsql+="(#{way},#{points[i+1][5]},#{sequence})"
317 ActiveRecord::Base.connection.execute("DELETE FROM current_way_segments WHERE id=#{way}");
318 ActiveRecord::Base.connection.insert("INSERT INTO way_segments (id,segment_id,version ) VALUES #{insertsql}");
319 ActiveRecord::Base.connection.insert("INSERT INTO current_way_segments (id,segment_id,sequence_id) VALUES #{currentsql}");
321 # -- 7. insert new way tags
325 attributes.each do |k,v|
326 if v=='' then next end
327 if v[0,6]=='(type ' then next end
328 if insertsql !='' then insertsql +=',' end
329 if currentsql!='' then currentsql+=',' end
330 insertsql +="(#{way},'"+sqlescape(k)+"','"+sqlescape(v)+"',version)"
331 currentsql+="(#{way},'"+sqlescape(k)+"','"+sqlescape(v)+"')"
334 ActiveRecord::Base.connection.execute("DELETE FROM current_way_tags WHERE id=#{way}")
335 if (insertsql !='') then ActiveRecord::Base.connection.insert("INSERT INTO way_tags (id,k,v,version) VALUES #{insertsql}" ) end
336 if (currentsql!='') then ActiveRecord::Base.connection.insert("INSERT INTO current_way_tags (id,k,v) VALUES #{currentsql}") end
338 [originalway,way,renumberednodes,numberedsegments,xmin,xmax,ymin,ymax]
341 # ----- deleteway (user token, way)
342 # returns way ID only
346 uid=getuserid(usertoken); if !uid then return end
349 db_uqs='uniq'+uid.to_s+way.to_i.abs.to_s+Time.new.to_i.to_s # temp uniquesegments table name, typically 51 chars
350 db_now='@now'+uid.to_s+way.to_i.abs.to_s+Time.new.to_i.to_s # 'now' variable name, typically 51 chars
351 ActiveRecord::Base.connection.execute("SET #{db_now}=NOW()")
352 createuniquesegments(way,db_uqs)
355 INSERT INTO nodes (id,latitude,longitude,timestamp,user_id,visible)
356 SELECT DISTINCT cn.id,cn.latitude,cn.longitude,#{db_now},#{uid},0
357 FROM current_nodes AS cn,
358 current_segments AS cs,
360 WHERE (cn.id=cs.node_a OR cn.id=cs.node_b)
361 AND cs.id=us.segment_id
363 ActiveRecord::Base.connection.insert(sql)
366 UPDATE current_nodes AS cn,
367 current_segments AS cs,
369 SET cn.timestamp=#{db_now},cn.visible=0,cn.user_id=#{uid}
370 WHERE (cn.id=cs.node_a OR cn.id=cs.node_b)
371 AND cs.id=us.segment_id
373 ActiveRecord::Base.connection.update(sql)
375 # - delete any otherwise unused segments
378 INSERT INTO segments (id,node_a,node_b,timestamp,user_id,visible)
379 SELECT DISTINCT segment_id,node_a,node_b,#{db_now},#{uid},0
380 FROM current_segments AS cs, #{db_uqs} AS us
381 WHERE cs.id=us.segment_id
383 ActiveRecord::Base.connection.insert(sql)
386 UPDATE current_segments AS cs, #{db_uqs} AS us
387 SET cs.timestamp=#{db_now},cs.visible=0,cs.user_id=#{uid}
388 WHERE cs.id=us.segment_id
390 ActiveRecord::Base.connection.update(sql)
391 ActiveRecord::Base.connection.execute("DROP TABLE #{db_uqs}")
395 ActiveRecord::Base.connection.insert("INSERT INTO ways (id,user_id,timestamp,visible) VALUES (#{way},#{uid},#{db_now},0)")
396 ActiveRecord::Base.connection.update("UPDATE current_ways SET user_id=#{uid},timestamp=#{db_now},visible=0 WHERE id=#{way}")
397 ActiveRecord::Base.connection.execute("DELETE FROM current_way_segments WHERE id=#{way}")
398 ActiveRecord::Base.connection.execute("DELETE FROM current_way_tags WHERE id=#{way}")
403 # ====================================================================
404 # Support functions for remote calls
407 ActiveRecord::Base.connection.select_all "SELECT n1.latitude AS lat1,n1.longitude AS long1,n1.id AS id1,n1.tags as tags1, "+
408 " n2.latitude AS lat2,n2.longitude AS long2,n2.id AS id2,n2.tags as tags2,segment_id "+
409 " FROM current_way_segments,current_segments,current_nodes AS n1,current_nodes AS n2 "+
410 " WHERE current_way_segments.id=#{id} "+
411 " AND segment_id=current_segments.id "+
412 " AND n1.id=node_a and n2.id=node_b "+
413 " ORDER BY sequence_id"
416 def createuniquesegments(way,uqs_name)
418 CREATE TEMPORARY TABLE #{uqs_name}
419 SELECT a.segment_id,COUNT(a.segment_id) AS ct
420 FROM current_way_segments AS a, current_way_segments AS b
421 WHERE a.segment_id=b.segment_id
423 GROUP BY a.segment_id
426 ActiveRecord::Base.connection.execute(sql)
431 a.gsub("'","''").gsub(92.chr,92.chr+92.chr)
436 a.gsub(';;;','#%').split(';').each do |b|
440 tags[k.gsub('#%','=')]=v.gsub('#%','=')
448 if v=='' then next end
449 if v[0,6]=='(type ' then next end
450 if str!='' then str+=';' end
451 str+=k.gsub(';',';;;').gsub('=','===')+'='+v.gsub(';',';;;').gsub('=','===')
457 token=sqlescape(token)
458 if (token=~/^(.+)\+(.+)$/) then
459 return ActiveRecord::Base.connection.select_value("SELECT id FROM users WHERE active=1 AND timeout>NOW() AND email='#{$1}' AND pass_crypt=MD5('#{$2}')")
461 return ActiveRecord::Base.connection.select_value("SELECT id FROM users WHERE active=1 AND timeout>NOW() AND token='#{token}'")
467 # ====================================================================
468 # AMF read subroutines
470 # ----- getint return two-byte integer
471 # ----- getlong return four-byte long
472 # ----- getstring return string with two-byte length
473 # ----- getdouble return eight-byte double-precision float
474 # ----- getobject return object/hash
475 # ----- getarray return numeric array
482 ((s.getc*256+s.getc)*256+s.getc)*256+s.getc
486 len=s.getc*256+s.getc
491 a=s.read(8).unpack('G') # G big-endian, E little-endian
506 while (key=getstring(s))
507 if (key=='') then break end
510 s.getc # skip the 9 'end of object' value
514 # ----- getvalue parse and get value
518 when 0; return getdouble(s) # number
519 when 1; return s.getc # boolean
520 when 2; return getstring(s) # string
521 when 3; return getobject(s) # object/hash
522 when 5; return nil # null
523 when 6; return nil # undefined
524 when 8; s.read(4) # mixedArray
525 return getobject(s) # |
526 when 10;return getarray(s) # array
527 else; return nil # error
531 # ====================================================================
532 # AMF write subroutines
534 # ----- putdata envelope data into AMF writeable form
535 # ----- encodevalue pack variables as AMF
538 d =encodestring(index+"/onResult")
539 d+=encodestring("null")
547 a=10.chr+encodelong(n.length)
555 a+=encodestring(k)+encodevalue(v)
559 2.chr+encodestring(n)
560 when 'Bignum','Fixnum','Float'
561 0.chr+encodedouble(n)
565 RAILS_DEFAULT_LOGGER.error("Unexpected Ruby type for AMF conversion: "+n.class.to_s)
569 # ----- encodestring encode string with two-byte length
570 # ----- encodedouble encode number as eight-byte double precision float
571 # ----- encodelong encode number as four-byte long
574 a,b=n.size.divmod(256)
586 # ====================================================================
587 # Co-ordinate conversion
589 def lat2coord(a,basey,masterscale)
590 -(lat2y(a)-basey)*masterscale+250
593 def long2coord(a,baselong,masterscale)
594 (a-baselong)*masterscale+350
598 180/Math::PI * Math.log(Math.tan(Math::PI/4+a*(Math::PI/180)/2))
601 def coord2lat(a,masterscale,basey)
602 y2lat((a-250)/-masterscale+basey)
605 def coord2long(a,masterscale,baselong)
606 (a-350)/masterscale+baselong
610 180/Math::PI * (2*Math.atan(Math.exp(a*Math::PI/180))-Math::PI/2)