app/controllers/search_controller.rb

   1 class SearchController < ApplicationController
   2   # Support searching for nodes, segments, ways, or all
   3   # Can search by tag k, v, or both (type->k,value->v)
   4   # Can search by name (k=name,v=....)
   5
   6   after_filter :compress_output
   7
   8   def search_all
   9     do_search(true,true,true)
  10   end
  11
  12   def search_ways
  13     do_search(true,false,false)
  14   end
  15   def search_segments
  16     do_search(false,true,false)
  17   end
  18   def search_nodes
  19     do_search(false,false,true)
  20   end
  21
  22
  23   def do_search(do_ways,do_segments,do_nodes)
  24     type = params['type']
  25     value = params['value']
  26     unless type or value
  27       name = params['name']
  28       if name
  29         type = 'name'
  30         value = name
  31       end
  32     end
  33
  34     way_ids = Array.new
  35     ways = Array.new
  36     segments = Array.new
  37     nodes = Array.new
  38
  39     # Matching for tags table
  40     cond_tbl = Array.new
  41     sql = '1=1'
  42     if type
  43       sql += ' AND k=?'
  44       cond_tbl += [type]
  45     end
  46     if value
  47       sql += ' AND v=?'
  48       cond_tbl += [value]
  49     end
  50     cond_tbl = [sql] + cond_tbl
  51
  52     # Matching for tags column
  53     if type and value
  54       cond_tags = ['tags LIKE ? OR tags LIKE ? OR tags LIKE ? OR tags LIKE ?',
  55       ''+type+'='+value+'',
  56       ''+type+'='+value+';%',
  57       '%;'+type+'='+value+';%',
  58       '%;'+type+'='+value+'' ]
  59     elsif type
  60       cond_tags = ['tags LIKE ? OR tags LIKE ?',
  61       ''+type+'=%',
  62       '%;'+type+'=%' ]
  63     elsif value
  64       cond_tags = ['tags LIKE ? OR tags LIKE ?',
  65       '%='+value+';%',
  66       '%='+value+'' ]
  67     else
  68       cond_tags = ['1=1']
  69     end
  70
  71
  72     # First up, look for the ids of the ways we want
  73     if do_ways
  74       ways_tmp = WayTag.find(:all, :conditions => cond_tbl)
  75       way_ids = ways_tmp.collect {|way| way.id }
  76     end
  77
  78     # Now, segments matching
  79     if do_segments
  80       segs = Segment.find(:all, :conditions => cond_tags)
  81     end
  82
  83     # Now, nodes
  84     if do_nodes
  85       nodes = Node.find(:all, :conditions => cond_tags)
  86     end
  87
  88     # Get the remaining objects:
  89     # Fetch the ways (until now only had their ids)
  90     ways = Way.find(way_ids)
  91
  92     # Fetch any segments needed for our ways (only have matching segments so far)
  93     seg_ids = Array.new
  94     ways.each do |way|
  95       seg_ids += way.segs
  96     end
  97     segments += Segment.find(seg_ids)
  98
  99     # Fetch any nodes needed for our segments (only have matching nodes so far)
 100     node_ids = Array.new
 101     segments.each do |seg|
 102       node_ids += seg.node_a
 103       node_ids += seg.node_b
 104     end
 105     nodes += Node.find(node_ids)
 106
 107
 108     # Print
 109     doc = OSM::API.new.get_xml_doc
 110     nodes.each do |node|
 111       doc.root << node.to_xml_node()
 112     end
 113
 114     segments.each do |segment|
 115       doc.root << segment.to_xml_node()
 116     end
 117
 118     ways.each do |way|
 119       doc.root << way.to_xml_node()
 120     end
 121
 122     render :text => doc.to_s, :content_type => "text/xml"
 123   end
 124 end