app/controllers/search_controller.rb

   1 class SearchController < ApplicationController
   2   # Support searching for nodes, 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   skip_before_filter :verify_authenticity_token
   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
  16   def search_nodes
  17     do_search(false, true, false)
  18   end
  19
  20   def search_relations
  21     do_search(false, false, true)
  22   end
  23
  24   def do_search(do_ways, do_nodes, do_relations)
  25     type = params['type']
  26     value = params['value']
  27     unless type || value
  28       name = params['name']
  29       if name
  30         type = 'name'
  31         value = name
  32       end
  33     end
  34
  35     if do_nodes
  36       response.headers['Error'] = "Searching of nodes is currently unavailable"
  37       render :text => "", :status => :service_unavailable
  38       return false
  39     end
  40
  41     unless value
  42       response.headers['Error'] = "Searching for a key without value is currently unavailable"
  43       render :text => "", :status => :service_unavailable
  44       return false
  45     end
  46
  47     # Matching for node tags table
  48     if do_nodes
  49       nodes = Node.joins(:node_tags)
  50       nodes = nodes.where(:current_node_tags => { :k => type }) if type
  51       nodes = nodes.where(:current_node_tags => { :v => value }) if value
  52       nodes = nodes.limit(100)
  53     else
  54       nodes = []
  55     end
  56
  57     # Matching for way tags table
  58     if do_ways
  59       ways = Way.joins(:way_tags)
  60       ways = ways.where(:current_way_tags => { :k => type }) if type
  61       ways = ways.where(:current_way_tags => { :v => value }) if value
  62       ways = ways.limit(100)
  63     else
  64       ways = []
  65     end
  66
  67     # Matching for relation tags table
  68     if do_relations
  69       relations = Relation.joins(:relation_tags)
  70       relations = relations.where(:current_relation_tags => { :k => type }) if type
  71       relations = relations.where(:current_relation_tags => { :v => value }) if value
  72       relations = relations.limit(2000)
  73     else
  74       relations = []
  75     end
  76
  77     # Fetch any node needed for our ways (only have matching nodes so far)
  78     nodes += Node.find(ways.collect(&:nds).uniq)
  79
  80     # Print
  81     visible_nodes = {}
  82     changeset_cache = {}
  83     user_display_name_cache = {}
  84     doc = OSM::API.new.get_xml_doc
  85     nodes.each do |node|
  86       doc.root << node.to_xml_node(changeset_cache, user_display_name_cache)
  87       visible_nodes[node.id] = node
  88     end
  89
  90     ways.each do |way|
  91       doc.root << way.to_xml_node(visible_nodes, changeset_cache, user_display_name_cache)
  92     end
  93
  94     relations.each do |rel|
  95       doc.root << rel.to_xml_node(nil, changeset_cache, user_display_name_cache)
  96     end
  97
  98     render :text => doc.to_s, :content_type => "text/xml"
  99   end
 100 end