# 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

    p = XML::Reader.new request.raw_post

    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
      while p.read == 1
	break if p.node_type == 15 # end element
	next unless p.node_type == 1 # element

	case p.name
	when 'create':
	  while p.read == 1
	    break if p.node_type == 15 # end element
	    next unless p.node_type == 1 # element

	    case p.name
	    when 'node':
	      elem = XML::Node.new 'node'
	      node = Node.from_xml_node(p.expand, true)
	      elem['old_id'] = p.expand['id']
	      create_prim node_ids, node, p.expand
	      elem['new_id'] = node.id.to_s
	      elem['new_version'] = node.version.to_s
	      root << elem
	    when 'way':
	      elem = XML::Node.new 'way'
	      way = Way.from_xml_node(p.expand, true)
	      elem['old_id'] = p.expand['id']
	      fix_way(way, node_ids)
	      raise OSM::APIPreconditionFailedError.new if !way.preconditions_ok?
	      create_prim way_ids, way, p.expand
	      elem['new_id'] = way.id.to_s
	      elem['new_version'] = way.version.to_s
	      root << elem
	    when 'relation':
	      elem = XML::Node.new 'relation'
	      relation = Relation.from_xml_node(p.expand, true)
	      elem['old_id'] = p.expand['id']
	      fix_rel(relation, ids)
	      raise OSM::APIPreconditionFailedError.new if !relation.preconditions_ok?
	      create_prim rel_ids, relation, p.expand
	      elem['new_id'] = relation.id.to_s
	      elem['new_version'] = relation.version.to_s
	      root << elem
	    end
	  end
	when 'modify':
	  while p.read == 1
	    break if p.node_type == 15 # end element
	    next unless p.node_type == 1 # element

	    case p.name
	    when 'node':
	      elem = XML::Node.new 'node'
	      new_node = Node.from_xml_node(p.expand)
	      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
	    when 'way':
	      elem = XML::Node.new 'way'
	      new_way = Way.from_xml_node(p.expand)
	      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
	    when 'relation':
	      elem = XML::Node.new 'relation'
	      new_relation = Relation.from_xml_node(p.expand)
	      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
	  end
	when 'delete':
	  while p.read == 1
	    break if p.node_type == 15 # end element
	    next unless p.node_type == 1 # element

	    case p.name
	    when 'node':
	      elem = XML::Node.new 'node'
	      node = Node.find(p.expand['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
	    when 'way':
	      elem = XML::Node.new 'way'
	      way = Way.find(p.expand['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
	    when 'relation':
	      elem = XML::Node.new 'relation'
	      relation = Relation.find(p.expand['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
	  end
	end
      end
    end

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

  rescue OSM::APIError => ex
    render ex.render_opts
  end
end