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/relation').each do |nd|
  82         new_relation = Relation.from_xml_node(nd)
  83         Relation.find(new_relation.id).update_from new_relation, @user
  84       end
  85       doc.find('//osm/modify/way').each do |nd|
  86         new_way = Way.from_xml_node(nd)
  87         Way.find(new_way.id).update_from new_way, @user
  88       end
  89       doc.find('//osm/modify/node').each do |nd|
  90         new_node = Node.from_xml_node(nd)
  91         Node.find(new_node.id).update_from new_node, @user
  92       end
  93
  94       doc.find('//osm/delete/relation').each do |nd|
  95         Relation.find(nd['id']).delete_with_history(@user)
  96       end
  97       doc.find('//osm/delete/way').each do |nd|
  98         Way.find(nd['id']).delete_with_relations_and_history(@user)
  99       end
 100       doc.find('//osm/delete/node').each do |nd|
 101         Node.find(nd['id']).delete_with_history(@user)
 102       end
 103     end
 104
 105     render :text => "Ok, Fine. Upload worked without errors.\n", :status => 200
 106   end
 107 end