]> git.openstreetmap.org Git - rails.git/blobdiff - app/controllers/changeset_controller.rb
Changing if not to unless to make code more readable.
[rails.git] / app / controllers / changeset_controller.rb
index 19d1591a0405d1ce731969db5a4e0ca4b302862f..8950126b4d86d77fff189a023b71ce85d7a89ed3 100644 (file)
@@ -35,80 +35,135 @@ class ChangesetController < ApplicationController
   end
 
   def fix_way(w, node_ids)
-    w.nds.each { |nd|
-      new_id = node_ids[nd.node_id]
-      nd.node_id = new_id unless new_id.nil?
-    }
+    w.nds = w.instance_eval { @nds }.
+      map { |nd| node_ids[nd] || nd }
+    return w
   end
 
   def fix_rel(r, ids)
-    r.members.each { |memb|
-      new_id = ids[memb.member_type][memb.member_id]
-      nd.member_id = new_id unless new_id.nil?
-    }
+    r.members = r.instance_eval { @members }.
+      map { |memb| [memb[0], ids[memb[0]][memb[1].to_i] || memb[1], memb[2]] }
+    return r
+  end
+  
+  def read
+    begin
+      changeset = Changeset.find(params[:id])
+      render :text => changeset.to_xml.to_s, :content_type => "text/xml"
+    rescue ActiveRecord::RecordNotFound
+      render :nothing => true, :status => :not_found
+    end
+  end
+  
+  def close 
+    begin
+      unless request.put?
+        render :nothing => true, :status => :method_not_allowed
+        return
+      end
+      changeset = Changeset.find(params[:id])
+      changeset.open = false
+      changeset.save
+      render :nothing => true
+    rescue ActiveRecord::RecordNotFound
+      render :nothing => true, :status => :not_found
+    end
   end
 
   def upload
-    if not request.put?
+    unless request.put?
       render :nothing => true, :status => :method_not_allowed
       return
     end
 
-    # FIXME: this should really be done without loading the whole XML file
-    # into memory.
-    p = XML::Parser.new
-    p.string  = request.raw_post
-    doc = p.parse
+    p = XML::Reader.new request.raw_post
 
     node_ids, way_ids, rel_ids = {}, {}, {}
     ids = {"node"=>node_ids, "way"=>way_ids, "relation"=>rel_ids}
 
+    models = {"node"=>Node, "way"=>Way, "relation"=>Relation}
+
+    res = XML::Document.new
+    res.encoding = 'UTF-8'
+    root = XML::Node.new 'osm'
+    root['version'] = '0.6'
+    root['creator'] = 'OpenStreetMap.org'
+    res.root = root
+
+    root << XML::Node.new_comment(" Warning: this is a 0.6 result document, " +
+      "not a normal OSM file. ")
+
     Changeset.transaction do
-      doc.find('//osm/create/node').each do |nd|
-       create_prim node_ids, Node.from_xml_node(nd, true), nd
-      end
-      doc.find('//osm/create/way').each do |nd|
-       way = Way.from_xml_node(nd, true)
-       raise OSM::APIPreconditionFailedError.new if !way.preconditions_ok?
-       create_prim way_ids, fix_way(way, node_ids), nd
-      end
-      doc.find('//osm/create/relation').each do |nd|
-       relation = Relation.from_xml_node(nd, true)
-       raise OSM::APIPreconditionFailedError.new if !way.preconditions_ok?
-       create_prim relation_ids, fix_rel(relation, ids), nd
-      end
+      while p.read == 1
+       break if p.node_type == 15 # end element
+       next unless p.node_type == 1 # element
 
-      doc.find('//osm/modify/node').each do |nd|
-       unless NodeController.update_internal nil, Node.from_xml_node(nd)
-         raise OSM::APIPreconditionFailedError.new
-       end
-      end
-      doc.find('//osm/modify/way').each do |nd|
-       unless WayController.update_internal nil, fix_way(Way.from_xml_node(nd), node_ids)
-         raise OSM::APIPreconditionFailedError.new
-       end
-      end
-      doc.find('//osm/modify/relation').each do |nd|
-       unless RelationController.update_internal nil, fix_rel(Relation.from_xml_node(nd), ids)
-         raise OSM::APIPreconditionFailedError.new
-       end
-      end
+       case p.name
+       when 'create':
+         while p.read == 1
+           break if p.node_type == 15 # end element
+           next unless p.node_type == 1 # element
 
-      doc.find('//osm/delete/node').each do |nd|
-       unless NodeController.delete_internal nil, Node.from_xml_node(n)
-         raise OSM::APIPreconditionFailedError.new
-       end
-      end
-      doc.find('//osm/delete/way').each do |nd|
-       Way.from_xml_node(nd).delete_with_relations_and_history(@user)
-      end
-      doc.find('//osm/delete/relation').each do |nd|
-       unless RelationController.delete_internal nil, fix_rel(Relation.from_xml_node(nd), ids)
-         raise OSM::APIPreconditionFailedError.new
+           model = models[p.name]
+           next if model.nil?
+
+           elem = XML::Node.new p.name
+           nd = p.expand; p.next
+           osm = model.from_xml_node(nd, true)
+           elem['old_id'] = nd['id']
+
+           case nd.name
+           when 'way':
+             fix_way(osm, node_ids)
+             raise OSM::APIPreconditionFailedError.new if !osm.preconditions_ok?
+           when 'relation':
+             fix_rel(osm, ids)
+             raise OSM::APIPreconditionFailedError.new if !osm.preconditions_ok?
+           end
+
+           create_prim ids[nd.name], osm, nd
+           elem['new_id'] = osm.id.to_s
+           elem['new_version'] = osm.version.to_s
+           root << elem
+         end
+       when 'modify':
+         while p.read == 1
+           break if p.node_type == 15 # end element
+           next unless p.node_type == 1 # element
+
+           model = models[p.name]
+           next if model.nil?
+
+           elem = XML::Node.new p.name
+           new_osm = model.from_xml_node(p.expand); p.next
+           osm = model.find(new_osm.id)
+           osm.update_from new_osm, @user
+           elem['old_id'] = elem['new_id'] = osm.id.to_s
+           elem['new_version'] = osm.version.to_s
+           root << elem
+         end
+       when 'delete':
+         while p.read == 1
+           break if p.node_type == 15 # end element
+           next unless p.node_type == 1 # element
+
+           model = models[p.name]
+           next if model.nil?
+
+           elem = XML::Node.new p.name
+           osm = model.find(p.expand['id']); p.next
+           osm.delete_with_history(@user)
+           elem['old_id'] = elem['new_id'] = osm.id.to_s
+           elem['new_version'] = osm.version.to_s
+           root << elem
+         end
        end
       end
     end
 
-    render :text => "Ok, Fine. Upload worked without errors.\n", :status => 200
+    render :text => res.to_s, :content_type => "text/xml"
+
+  rescue OSM::APIError => ex
+    render ex.render_opts
   end
 end