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