app/controllers/api/search_controller.rb

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