app/controllers/changeset_controller.rb

   1 # The ChangesetController is the RESTful interface to Changeset objects
   2
   3 class ChangesetController < ApplicationController
   4   require 'xml/libxml'
   5
   6   before_filter :authorize, :only => [:create, :update, :delete, :upload]
   7   before_filter :check_write_availability, :only => [:create, :update, :delete, :upload]
   8   before_filter :check_read_availability, :except => [:create, :update, :delete, :upload]
   9   after_filter :compress_output
  10
  11   # Create a changeset from XML.
  12   def create
  13     if request.put?
  14       cs = Changeset.from_xml(request.raw_post, true)
  15
  16       if cs
  17         cs.user_id = @user.id
  18         cs.save_with_tags!
  19         render :text => cs.id.to_s, :content_type => "text/plain"
  20       else
  21         render :nothing => true, :status => :bad_request
  22       end
  23     else
  24       render :nothing => true, :status => :method_not_allowed
  25     end
  26   end
  27
  28   def create_prim(ids, prim, nd)
  29     prim.version = 0
  30     prim.user_id = @user.id
  31     prim.visible = true
  32     prim.save_with_history!
  33
  34     ids[nd['id'].to_i] = prim.id
  35   end
  36
  37   def fix_way(w, node_ids)
  38     w.nds = w.instance_eval { @nds }.
  39       map { |nd| node_ids[nd] || nd }
  40     return w
  41   end
  42
  43   def fix_rel(r, ids)
  44     r.members = r.instance_eval { @members }.
  45       map { |memb| [memb[0], ids[memb[0]][memb[1].to_i] || memb[1], memb[2]] }
  46     return r
  47   end
  48
  49   def upload
  50     if not request.put?
  51       render :nothing => true, :status => :method_not_allowed
  52       return
  53     end
  54
  55     # FIXME: this should really be done without loading the whole XML file
  56     # into memory.
  57     p = XML::Parser.new
  58     p.string  = request.raw_post
  59     doc = p.parse
  60
  61     node_ids, way_ids, rel_ids = {}, {}, {}
  62     ids = {"node"=>node_ids, "way"=>way_ids, "relation"=>rel_ids}
  63
  64     Changeset.transaction do
  65       doc.find('//osm/create/node').each do |nd|
  66         create_prim node_ids, Node.from_xml_node(nd, true), nd
  67       end
  68       doc.find('//osm/create/way').each do |nd|
  69         way = Way.from_xml_node(nd, true)
  70         fix_way(way, node_ids)
  71         raise OSM::APIPreconditionFailedError.new if !way.preconditions_ok?
  72         create_prim way_ids, way, nd
  73       end
  74       doc.find('//osm/create/relation').each do |nd|
  75         relation = Relation.from_xml_node(nd, true)
  76         fix_rel(relation, ids)
  77         raise OSM::APIPreconditionFailedError.new if !relation.preconditions_ok?
  78         create_prim rel_ids, relation, nd
  79       end
  80
  81       doc.find('//osm/modify/node').each do |nd|
  82         unless NodeController.new.update_internal nil, Node.from_xml_node(nd)
  83           raise OSM::APIPreconditionFailedError.new
  84         end
  85       end
  86       doc.find('//osm/modify/way').each do |nd|
  87         unless WayController.update_internal nil, fix_way(Way.from_xml_node(nd), node_ids)
  88           raise OSM::APIPreconditionFailedError.new
  89         end
  90       end
  91       doc.find('//osm/modify/relation').each do |nd|
  92         unless RelationController.update_internal nil, fix_rel(Relation.from_xml_node(nd), ids)
  93           raise OSM::APIPreconditionFailedError.new
  94         end
  95       end
  96
  97       doc.find('//osm/delete/node').each do |nd|
  98         unless NodeController.delete_internal nil, Node.from_xml_node(n)
  99           raise OSM::APIPreconditionFailedError.new
 100         end
 101       end
 102       doc.find('//osm/delete/way').each do |nd|
 103         Way.from_xml_node(nd).delete_with_relations_and_history(@user)
 104       end
 105       doc.find('//osm/delete/relation').each do |nd|
 106         unless RelationController.delete_internal nil, fix_rel(Relation.from_xml_node(nd), ids)
 107           raise OSM::APIPreconditionFailedError.new
 108         end
 109       end
 110     end
 111
 112     render :text => "Ok, Fine. Upload worked without errors.\n", :status => 200
 113   end
 114 end