127 lines
2.8 KiB
Ruby
127 lines
2.8 KiB
Ruby
class SearchController < ApplicationController
|
|
# Support searching for nodes, segments, ways, or all
|
|
# Can search by tag k, v, or both (type->k,value->v)
|
|
# Can search by name (k=name,v=....)
|
|
|
|
before_filter :authorize
|
|
after_filter :compress_output
|
|
|
|
def search_all
|
|
do_search(true,true,true)
|
|
end
|
|
|
|
def search_ways
|
|
do_search(true,false,false)
|
|
end
|
|
def search_segments
|
|
do_search(false,true,false)
|
|
end
|
|
def search_nodes
|
|
do_search(false,false,true)
|
|
end
|
|
|
|
|
|
def do_search(do_ways,do_segments,do_nodes)
|
|
response.headers["Content-Type"] = 'text/xml'
|
|
|
|
type = params['type']
|
|
value = params['value']
|
|
unless type or value
|
|
name = params['name']
|
|
if name
|
|
type = 'name'
|
|
value = name
|
|
end
|
|
end
|
|
|
|
way_ids = Array.new
|
|
ways = Array.new
|
|
segments = Array.new
|
|
nodes = Array.new
|
|
|
|
# Matching for tags table
|
|
cond_tbl = Array.new
|
|
sql = '1=1'
|
|
if type
|
|
sql += ' AND k=?'
|
|
cond_tbl += [type]
|
|
end
|
|
if value
|
|
sql += ' AND v=?'
|
|
cond_tbl += [value]
|
|
end
|
|
cond_tbl = [sql] + cond_tbl
|
|
|
|
# Matching for tags column
|
|
if type and value
|
|
cond_tags = ['tags LIKE ? OR tags LIKE ? OR tags LIKE ? OR tags LIKE ?',
|
|
''+type+'='+value+'',
|
|
''+type+'='+value+';%',
|
|
'%;'+type+'='+value+';%',
|
|
'%;'+type+'='+value+'' ]
|
|
elsif type
|
|
cond_tags = ['tags LIKE ? OR tags LIKE ?',
|
|
''+type+'=%',
|
|
'%;'+type+'=%' ]
|
|
elsif value
|
|
cond_tags = ['tags LIKE ? OR tags LIKE ?',
|
|
'%='+value+';%',
|
|
'%='+value+'' ]
|
|
else
|
|
cond_tags = ['1=1']
|
|
end
|
|
|
|
|
|
# First up, look for the ids of the ways we want
|
|
if do_ways
|
|
ways_tmp = WayTag.find(:all, :conditions => cond_tbl)
|
|
way_ids = ways_tmp.collect {|way| way.id }
|
|
end
|
|
|
|
# Now, segments matching
|
|
if do_segments
|
|
segs = Segment.find(:all, :conditions => cond_tags)
|
|
end
|
|
|
|
# Now, nodes
|
|
if do_nodes
|
|
nodes = Node.find(:all, :conditions => cond_tags)
|
|
end
|
|
|
|
# Get the remaining objects:
|
|
# Fetch the ways (until now only had their ids)
|
|
ways = Way.find(way_ids)
|
|
|
|
# Fetch any segments needed for our ways (only have matching segments so far)
|
|
seg_ids = Array.new
|
|
ways.each do |way|
|
|
seg_ids += way.segments
|
|
end
|
|
segments += Segment.find(seg_ids)
|
|
|
|
# Fetch any nodes needed for our segments (only have matching nodes so far)
|
|
node_ids = Array.new
|
|
segments.each do |seg|
|
|
node_ids += seg.node_a
|
|
node_ids += seg.node_b
|
|
end
|
|
nodes += Node.find(node_ids)
|
|
|
|
|
|
# Print
|
|
doc = OSM::API.get_xml_doc
|
|
nodes.each do |node|
|
|
doc.root << node.to_xml_node()
|
|
end
|
|
|
|
segments.each do |segment|
|
|
doc.root << segment.to_xml_node()
|
|
end
|
|
|
|
ways.each do |way|
|
|
doc.root << way.to_xml_node()
|
|
end
|
|
|
|
render :text => doc.to_s
|
|
end
|
|
end
|