X-Git-Url: https://git.openstreetmap.org/rails.git/blobdiff_plain/2e44f9ccf3ac8c54c407d74422247b5a721b269c..337bda6a90b8d37a533f334ad50f3893c55b8e57:/app/controllers/user_preference_controller.rb

diff --git a/app/controllers/user_preference_controller.rb b/app/controllers/user_preference_controller.rb
index 7f841b72b..bf9bab213 100644
--- a/app/controllers/user_preference_controller.rb
+++ b/app/controllers/user_preference_controller.rb
@@ -1,68 +1,87 @@
+# Update and read user preferences, which are arbitrayr key/val pairs
 class UserPreferenceController < ApplicationController
-  before_filter :authorize
-
+  skip_before_action :verify_authenticity_token
+  before_action :authorize
+  before_action :require_allow_read_prefs, :only => [:read_one, :read]
+  before_action :require_allow_write_prefs, :except => [:read_one, :read]
+  around_action :api_call_handle_error
+
+  ##
+  # return all the preferences as an XML document
   def read
-
     doc = OSM::API.new.get_xml_doc
 
     prefs = @user.preferences
 
-    el1 = XML::Node.new 'preferences'
+    el1 = XML::Node.new "preferences"
 
     prefs.each do |pref|
-      el1 <<  pref.to_xml_node
+      el1 << pref.to_xml_node
     end
 
     doc.root << el1
     render :text => doc.to_s, :content_type => "text/xml"
+  end
 
+  ##
+  # return the value for a single preference
+  def read_one
+    pref = UserPreference.find([@user.id, params[:preference_key]])
+
+    render :text => pref.v.to_s, :content_type => "text/plain"
   end
 
+  # update the entire set of preferences
   def update
-    begin
-      p = XML::Parser.new
-      p.string = request.raw_post
-      doc = p.parse
-
-      prefs = []
-
-      keyhash = {}
-
-      doc.find('//preferences/preference').each do |pt|
-        pref = UserPreference.new
-        
-        unless keyhash[pt['k']].nil? # already have that key
-          render :text => 'OH NOES! CAN HAS UNIQUE KEYS?', :status => 406
-          return
-        end
-        
-        keyhash[pt['k']] = 1
-
-        pref.k = pt['k']
-        pref.v = pt['v']
-        pref.user_id = @user.id
-        prefs << pref
-      end
+    old_preferences = @user.preferences.each_with_object({}) do |preference, preferences|
+      preferences[preference.k] = preference
+    end
 
-      if prefs.size > 150
-        render :text => 'Too many preferences', :status => 413
-        return
-      end
+    new_preferences = {}
 
-      # kill the existing ones
-      UserPreference.delete_all(['user_id = ?', @user.id])
+    doc = XML::Parser.string(request.raw_post, :options => XML::Parser::Options::NOERROR).parse
 
-      # save the new ones
-      prefs.each do |pref|
-        pref.save!
+    doc.find("//preferences/preference").each do |pt|
+      if preference = old_preferences.delete(pt["k"])
+        preference.v = pt["v"]
+      elsif new_preferences.include?(pt["k"])
+        raise OSM::APIDuplicatePreferenceError.new(pt["k"])
+      else
+        preference = @user.preferences.build(:k => pt["k"], :v => pt["v"])
       end
 
-    rescue Exception => ex
-      render :text => 'OH NOES! FAIL!: ' + ex.to_s, :status => 500
-      return
+      new_preferences[preference.k] = preference
+    end
+
+    old_preferences.each_value(&:delete)
+
+    new_preferences.each_value(&:save!)
+
+    render :text => "", :content_type => "text/plain"
+  end
+
+  ##
+  # update the value of a single preference
+  def update_one
+    begin
+      pref = UserPreference.find([@user.id, params[:preference_key]])
+    rescue ActiveRecord::RecordNotFound
+      pref = UserPreference.new
+      pref.user = @user
+      pref.k = params[:preference_key]
     end
 
-    render :nothing => true
+    pref.v = request.raw_post.chomp
+    pref.save!
+
+    render :text => "", :content_type => "text/plain"
   end
 
+  ##
+  # delete a single preference
+  def delete_one
+    UserPreference.find([@user.id, params[:preference_key]]).delete
+
+    render :text => "", :content_type => "text/plain"
+  end
 end