Only allow users to read their own messages.

[rails.git] / app / controllers / old_segment_controller.rb

diff --git a/app/controllers/old_segment_controller.rb b/app/controllers/old_segment_controller.rb
index a35a622f975d846575c63e80012652d8d287e849..2d48fb5b9489547e6d8b423d4221990a20f81461 100644 (file)
--- a/app/controllers/old_segment_controller.rb
+++ b/app/controllers/old_segment_controller.rb
@@ -1,20 +1,24 @@
 class OldSegmentController < ApplicationController
+  require 'xml/libxml'
+
+  session :off
+  after_filter :compress_output
 
   def history
-    response.headers["Content-Type"] = 'application/xml'
-    segment = Segment.find(params[:id])
+    begin
+      segment = Segment.find(params[:id])
 
-    unless segment
-      render :nothing => true, :staus => 404
-      return
-    end
+      doc = OSM::API.new.get_xml_doc
 
-    doc = OSM::API.new.get_xml_doc
+      segment.old_segments.each do |old_segment|
+        doc.root << old_segment.to_xml_node
+      end
 
-    segment.old_segments.each do |old_segment|
-      doc.root << old_segment.to_xml_node
+     render :text => doc.to_s, :content_type => "text/xml"
+    rescue ActiveRecord::RecordNotFound
+      render :nothing => true, :status => :not_found
+    rescue
+      render :nothing => true, :status => :internal_server_error
     end
-
-    render :text => doc.to_s
   end
 end