openstreetmap-website/app/controllers/changeset_controller.rb

# The ChangesetController is the RESTful interface to Changeset objects

class ChangesetController < ApplicationController
  require 'xml/libxml'

  before_filter :authorize, :only => [:create, :update, :delete, :upload]
  before_filter :check_write_availability, :only => [:create, :update, :delete, :upload]
  before_filter :check_read_availability, :except => [:create, :update, :delete, :upload]
  after_filter :compress_output

  # Create a changeset from XML.
  def create
    if request.put?
      cs = Changeset.from_xml(request.raw_post, true)

      if cs
        cs.user_id = @user.id
        cs.save_with_tags!
        render :text => cs.id.to_s, :content_type => "text/plain"
      else
        render :nothing => true, :status => :bad_request
      end
    else
      render :nothing => true, :status => :method_not_allowed
    end
  end

  def create_prim(ids, prim, nd)
    prim.version = 0
    prim.user_id = @user.id
    prim.visible = true
    prim.save_with_history!

    ids[nd['id'].to_i] = prim.id
  end

  def fix_way(w, node_ids)
    w.nds = w.instance_eval { @nds }.
      map { |nd| node_ids[nd] || nd }
    return w
  end

  def fix_rel(r, ids)
    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
      if not 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?
      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

    node_ids, way_ids, rel_ids = {}, {}, {}
    ids = {"node"=>node_ids, "way"=>way_ids, "relation"=>rel_ids}

    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|
	elem = XML::Node.new 'node'
	node = Node.from_xml_node(nd, true)
	elem['old_id'] = nd['id']
	create_prim node_ids, node, nd
	elem['new_id'] = node.id.to_s
	elem['new_version'] = node.version.to_s
	root << elem
      end
      doc.find('//osm/create/way').each do |nd|
	elem = XML::Node.new 'way'
	way = Way.from_xml_node(nd, true)
	elem['old_id'] = nd['id']
	fix_way(way, node_ids)
	raise OSM::APIPreconditionFailedError.new if !way.preconditions_ok?
	create_prim way_ids, way, nd
	elem['new_id'] = way.id.to_s
	elem['new_version'] = way.version.to_s
	root << elem
      end
      doc.find('//osm/create/relation').each do |nd|
	elem = XML::Node.new 'relation'
	relation = Relation.from_xml_node(nd, true)
	elem['old_id'] = nd['id']
	fix_rel(relation, ids)
	raise OSM::APIPreconditionFailedError.new if !relation.preconditions_ok?
	create_prim rel_ids, relation, nd
	elem['new_id'] = relation.id.to_s
	elem['new_version'] = relation.version.to_s
	root << elem
      end

      doc.find('//osm/modify/relation').each do |nd|
	elem = XML::Node.new 'relation'
	new_relation = Relation.from_xml_node(nd)
	relation = Relation.find(new_relation.id)
	relation.update_from new_relation, @user
	elem['old_id'] = elem['new_id'] = relation.id.to_s
	elem['new_version'] = relation.version.to_s
	root << elem
      end
      doc.find('//osm/modify/way').each do |nd|
	elem = XML::Node.new 'way'
	new_way = Way.from_xml_node(nd)
	way = Way.find(new_way.id)
	way.update_from new_way, @user
	elem['old_id'] = elem['new_id'] = way.id.to_s
	elem['new_version'] = way.version.to_s
	root << elem
      end
      doc.find('//osm/modify/node').each do |nd|
	elem = XML::Node.new 'node'
	new_node = Node.from_xml_node(nd)
	node = Node.find(new_node.id)
	node.update_from new_node, @user
	elem['old_id'] = elem['new_id'] = node.id.to_s
	elem['new_version'] = node.version.to_s
	root << elem
      end

      doc.find('//osm/delete/relation').each do |nd|
	elem = XML::Node.new 'relation'
	relation = Relation.find(nd['id'])
	relation.delete_with_history(@user)
	elem['old_id'] = elem['new_id'] = relation.id.to_s
	elem['new_version'] = relation.version.to_s
	root << elem
      end
      doc.find('//osm/delete/way').each do |nd|
	elem = XML::Node.new 'way'
	way = Way.find(nd['id'])
	way.delete_with_history(@user)
	elem['old_id'] = elem['new_id'] = way.id.to_s
	elem['new_version'] = way.version.to_s
	root << elem
      end
      doc.find('//osm/delete/node').each do |nd|
	elem = XML::Node.new 'node'
	new_node = Node.from_xml_node(nd)
	node = Node.find(nd['id'])
	node.delete_with_history(@user)
	elem['old_id'] = elem['new_id'] = node.id.to_s
	elem['new_version'] = node.version.to_s
	root << elem
      end
    end

    render :text => res.to_s, :content_type => "text/xml"
  end
end