updates from hacking day
[rails.git] / app / controllers / api_controller.rb
1 class ApiController < ApplicationController
2
3   before_filter :authorize
4   after_filter :compress_output
5
6   helper :user
7   model :user
8
9   def authorize_web
10     @current_user = User.find_by_token(session[:token])
11   end
12   
13   # The maximum area you're allowed to request, in square degrees
14   MAX_REQUEST_AREA = 0.25
15
16   def map
17     response.headers["Content-Type"] = 'text/xml'
18     # Figure out the bbox
19     bbox = params['bbox']
20     unless bbox and bbox.count(',') == 3
21        report_error("The parameter bbox is required, and must be of the form min_lon,min_lat,max_lon,max_lat")
22       return
23     end
24
25     bbox = bbox.split(',')
26
27     min_lon = bbox[0].to_f
28     min_lat = bbox[1].to_f
29     max_lon = bbox[2].to_f
30     max_lat = bbox[3].to_f
31
32     # check the bbox is sane
33     unless min_lon <= max_lon
34        report_error("The minimum longitude must be less than the maximum longitude, but it wasn't")
35       return
36     end
37     unless min_lat <= max_lat
38        report_error("The minimum latitude must be less than the maximum latitude, but it wasn't")
39       return
40     end
41         unless min_lon >= -180 && min_lat >= -90 && max_lon <= 180 && max_lat <= 90
42        report_error("The latitudes must be between -90 and 90, and longitudes between -180 and 180")
43       return
44         end
45
46     # check the bbox isn't too large
47     requested_area = (max_lat-min_lat)*(max_lon-min_lon)
48     if requested_area > MAX_REQUEST_AREA
49        report_error("The maximum bbox size is " + MAX_REQUEST_AREA.to_s + ", and your request was too large. Either request a smaller area, or use planet.osm")
50       return
51     end
52
53     # get all the nodes
54     nodes = Node.find(:all, :conditions => ['latitude > ? AND longitude > ? AND latitude < ? AND longitude < ? AND visible = 1', min_lat, min_lon, max_lat, max_lon])
55
56     node_ids = nodes.collect {|node| node.id }
57
58     if node_ids.length > 50_000
59        report_error("You requested too many nodes (limit is 50,000). Either request a smaller area, or use planet.osm")
60     end
61
62     # grab the segments
63     segments = Array.new
64     if node_ids.length > 0
65       node_ids_sql = "(#{node_ids.join(',')})"
66       # get the referenced segments
67       segments = Segment.find_by_sql "select * from current_segments where visible = 1 and (node_a in #{node_ids_sql} or node_b in #{node_ids_sql})"
68     end
69     # see if we have any missing nodes
70     segments_nodes = segments.collect {|segment| segment.node_a }
71     segments_nodes += segments.collect {|segment| segment.node_b }
72
73     segments_nodes.uniq!
74
75     missing_nodes = segments_nodes - node_ids
76
77     # get missing nodes if there are any
78     nodes += Node.find(missing_nodes) if missing_nodes.length > 0
79
80     doc = OSM::API.new.get_xml_doc
81
82     # get ways
83     # find which ways are needed
84     segment_ids = segments.collect {|segment| segment.id }
85     ways = Array.new
86     if segment_ids.length > 0
87       way_segments = WaySegment.find_all_by_segment_id(segment_ids)
88       way_ids = way_segments.collect {|way_segment| way_segment.id }
89       ways = Way.find(way_ids) # NB: doesn't pick up segments, tags from db until accessed via way.way_segments etc.
90
91      # seg_ids = way_segments.collect {|way_segment| way_segment.segment_id }
92       
93       list_of_way_segs = ways.collect {|way| way.way_segments}
94       list_of_way_segs.flatten!
95
96       list_of_way_segments = list_of_way_segs.collect { |way_seg| way_seg.segment_id }
97
98     end
99
100     segments_to_fetch = list_of_way_segments.uniq - segment_ids
101
102     if segments_to_fetch.length > 0
103       segments += Segment.find(segments_to_fetch)
104     end
105     
106     # get more nodes
107     #
108
109     segments_nodes = segments.collect {|segment| segment.node_a }
110     segments_nodes += segments.collect {|segment| segment.node_b }
111
112     node_ids_a = nodes.collect {|node| node.id }
113
114     nodes_to_get = segments_nodes - node_ids_a
115     nodes += Node.find(nodes_to_get) if nodes_to_get.length > 0
116
117     visible_nodes = {}
118     user_display_name_cache = {}
119
120     nodes.each do |node|
121       if node.visible?
122         doc.root << node.to_xml_node(user_display_name_cache)
123         visible_nodes[node.id] = node
124       end
125     end
126
127     visible_segments = {}
128
129     segments.each do |segment|
130       if visible_nodes[segment.node_a] and visible_nodes[segment.node_b] and segment.visible?
131         doc.root << segment.to_xml_node(user_display_name_cache) 
132         visible_segments[segment.id] = segment
133       end
134     end
135
136     ways.each do |way|
137       doc.root << way.to_xml_node(visible_segments, user_display_name_cache) if way.visible?
138     end 
139
140     render :text => doc.to_s
141
142   end
143 end