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     res = XML::Document.new
  65     res.encoding = 'UTF-8'
  66     root = XML::Node.new 'osm'
  67     root['version'] = '0.6'
  68     root['creator'] = 'OpenStreetMap.org'
  69     res.root = root
  70
  71     root << XML::Node.new_comment(" Warning: this is a 0.6 result document, " +
  72       "not a normal OSM file. ")
  73
  74     Changeset.transaction do
  75       doc.find('//osm/create/node').each do |nd|
  76         elem = XML::Node.new 'node'
  77         node = Node.from_xml_node(nd, true)
  78         elem['old_id'] = nd['id']
  79         create_prim node_ids, node, nd
  80         elem['new_id'] = node.id.to_s
  81         elem['new_version'] = node.version.to_s
  82         root << elem
  83       end
  84       doc.find('//osm/create/way').each do |nd|
  85         elem = XML::Node.new 'way'
  86         way = Way.from_xml_node(nd, true)
  87         elem['old_id'] = nd['id']
  88         fix_way(way, node_ids)
  89         raise OSM::APIPreconditionFailedError.new if !way.preconditions_ok?
  90         create_prim way_ids, way, nd
  91         elem['new_id'] = way.id.to_s
  92         elem['new_version'] = way.version.to_s
  93         root << elem
  94       end
  95       doc.find('//osm/create/relation').each do |nd|
  96         elem = XML::Node.new 'relation'
  97         relation = Relation.from_xml_node(nd, true)
  98         elem['old_id'] = nd['id']
  99         fix_rel(relation, ids)
 100         raise OSM::APIPreconditionFailedError.new if !relation.preconditions_ok?
 101         create_prim rel_ids, relation, nd
 102         elem['new_id'] = relation.id.to_s
 103         elem['new_version'] = relation.version.to_s
 104         root << elem
 105       end
 106
 107       doc.find('//osm/modify/relation').each do |nd|
 108         elem = XML::Node.new 'relation'
 109         new_relation = Relation.from_xml_node(nd)
 110         relation = Relation.find(new_relation.id)
 111         relation.update_from new_relation, @user
 112         elem['old_id'] = elem['new_id'] = relation.id.to_s
 113         elem['new_version'] = relation.version.to_s
 114         root << elem
 115       end
 116       doc.find('//osm/modify/way').each do |nd|
 117         elem = XML::Node.new 'way'
 118         new_way = Way.from_xml_node(nd)
 119         way = Way.find(new_way.id)
 120         way.update_from new_way, @user
 121         elem['old_id'] = elem['new_id'] = way.id.to_s
 122         elem['new_version'] = way.version.to_s
 123         root << elem
 124       end
 125       doc.find('//osm/modify/node').each do |nd|
 126         elem = XML::Node.new 'node'
 127         new_node = Node.from_xml_node(nd)
 128         node = Node.find(new_node.id)
 129         node.update_from new_node, @user
 130         elem['old_id'] = elem['new_id'] = node.id.to_s
 131         elem['new_version'] = node.version.to_s
 132         root << elem
 133       end
 134
 135       doc.find('//osm/delete/relation').each do |nd|
 136         elem = XML::Node.new 'relation'
 137         relation = Relation.find(nd['id'])
 138         relation.delete_with_history(@user)
 139         elem['old_id'] = elem['new_id'] = relation.id.to_s
 140         elem['new_version'] = relation.version.to_s
 141         root << elem
 142       end
 143       doc.find('//osm/delete/way').each do |nd|
 144         elem = XML::Node.new 'way'
 145         way = Way.find(nd['id'])
 146         way.delete_with_relations_and_history(@user)
 147         elem['old_id'] = elem['new_id'] = way.id.to_s
 148         elem['new_version'] = way.version.to_s
 149         root << elem
 150       end
 151       doc.find('//osm/delete/node').each do |nd|
 152         elem = XML::Node.new 'node'
 153         new_node = Node.from_xml_node(nd)
 154         node = Node.find(nd['id'])
 155         node.delete_with_history(@user)
 156         elem['old_id'] = elem['new_id'] = node.id.to_s
 157         elem['new_version'] = node.version.to_s
 158         root << elem
 159       end
 160     end
 161
 162     render :text => res.to_s, :content_type => "text/xml"
 163   end
 164 end