Fix more parameter sanitisation issues and add tests

[rails.git] / app / controllers / user_blocks_controller.rb

diff --git a/app/controllers/user_blocks_controller.rb b/app/controllers/user_blocks_controller.rb
index 7d3830c251791c029b737fb5a2885da4b73bc418..ea5cdab10353987042340320a840b2fa1488f69e 100644 (file)
--- a/app/controllers/user_blocks_controller.rb
+++ b/app/controllers/user_blocks_controller.rb
@@ -1,17 +1,18 @@
 class UserBlocksController < ApplicationController
 class UserBlocksController < ApplicationController

-  layout 'site'
-
-  before_filter :authorize_web
-  before_filter :set_locale
-  before_filter :require_user, :only => [:new, :create, :edit, :update, :revoke]
-  before_filter :require_moderator, :only => [:create, :update, :revoke]
-  before_filter :lookup_this_user, :only => [:new, :create, :blocks_on, :blocks_by]
-  before_filter :lookup_user_block, :only => [:show, :edit, :update, :revoke]
-  before_filter :require_valid_params, :only => [:create, :update]
-  before_filter :check_database_readable
-  before_filter :check_database_writable, :only => [:create, :update, :revoke]
+  layout "site"
+
+  before_action :authorize_web
+  before_action :set_locale
+  before_action :require_user, :only => [:new, :create, :edit, :update, :revoke]
+  before_action :require_moderator, :only => [:new, :create, :edit, :update, :revoke]
+  before_action :lookup_this_user, :only => [:new, :create, :blocks_on, :blocks_by]
+  before_action :lookup_user_block, :only => [:show, :edit, :update, :revoke]
+  before_action :require_valid_params, :only => [:create, :update]
+  before_action :check_database_readable
+  before_action :check_database_writable, :only => [:create, :update, :revoke]

 
   def index
 
   def index

+    @params = params.permit

     @user_blocks_pages, @user_blocks = paginate(:user_blocks,
                                                 :include => [:user, :creator, :revoker],
                                                 :order => "user_blocks.ends_at DESC",
     @user_blocks_pages, @user_blocks = paginate(:user_blocks,
                                                 :include => [:user, :creator, :revoker],
                                                 :order => "user_blocks.ends_at DESC",


@@ -19,7 +20,7 @@ class UserBlocksController < ApplicationController
   end
 
   def show
   end
 
   def show

-    if @user and @user.id == @user_block.user_id
+    if @user && @user.id == @user_block.user_id

       @user_block.needs_view = false
       @user_block.save!
     end
       @user_block.needs_view = false
       @user_block.save!
     end


@@ -34,44 +35,43 @@ class UserBlocksController < ApplicationController
   end
 
   def create
   end
 
   def create

-    unless @valid_params 
-      redirect_to :action => "new"
-      return
-    end
-
-    @user_block = UserBlock.new(:user_id => @this_user.id,
-                                :creator_id => @user.id,
-                                :reason => params[:user_block][:reason],
-                                :ends_at => Time.now.getutc() + @block_period.hours,
-                                :needs_view => params[:user_block][:needs_view])
-    
-    if @user_block.save
-      flash[:notice] = t('user_block.create.flash', :name => @this_user.display_name)
-      redirect_to @user_block
+    if @valid_params
+      @user_block = UserBlock.new(
+        :user_id => @this_user.id,
+        :creator_id => @user.id,
+        :reason => params[:user_block][:reason],
+        :ends_at => Time.now.getutc + @block_period.hours,
+        :needs_view => params[:user_block][:needs_view]
+      )
+
+      if @user_block.save
+        flash[:notice] = t("user_block.create.flash", :name => @this_user.display_name)
+        redirect_to @user_block
+      else
+        render :action => "new"
+      end

     else
     else

-      render :action => "new"
+      redirect_to new_user_block_path(:display_name => params[:display_name])

     end
   end
 
     end
   end
 

-  def update  
-    unless @valid_params 
-      redirect_to :action => "edit"
-      return
-    end
-
-    if @user_block.creator_id != @user.id
-      flash[:notice] = t('user_block.update.only_creator_can_edit')
-      redirect_to :action => "edit"
-      return
-    end
-      
-    if @user_block.update_attributes({ :ends_at => Time.now.getutc() + @block_period.hours,
-                                       :reason => params[:user_block][:reason],
-                                       :needs_view => params[:user_block][:needs_view] })
-      flash[:notice] = t('user_block.update.success')
-      redirect_to(@user_block)
+  def update
+    if @valid_params
+      if @user_block.creator_id != @user.id
+        flash[:error] = t("user_block.update.only_creator_can_edit")
+        redirect_to :action => "edit"
+      elsif @user_block.update_attributes(
+        :ends_at => Time.now.getutc + @block_period.hours,
+        :reason => params[:user_block][:reason],
+        :needs_view => params[:user_block][:needs_view]
+      )
+        flash[:notice] = t("user_block.update.success")
+        redirect_to(@user_block)
+      else
+        render :action => "edit"
+      end

     else
     else

-      render :action => "edit"
+      redirect_to edit_user_block_path(:id => params[:id])

     end
   end
 
     end
   end
 


@@ -80,7 +80,7 @@ class UserBlocksController < ApplicationController
   def revoke
     if params[:confirm]
       if @user_block.revoke! @user
   def revoke
     if params[:confirm]
       if @user_block.revoke! @user

-        flash[:notice] = t'user_block.revoke.flash'
+        flash[:notice] = t "user_block.revoke.flash"

         redirect_to(@user_block)
       end
     end
         redirect_to(@user_block)
       end
     end


@@ -89,9 +89,10 @@ class UserBlocksController < ApplicationController
   ##
   # shows a list of all the blocks on the given user
   def blocks_on
   ##
   # shows a list of all the blocks on the given user
   def blocks_on

+    @params = params.permit(:display_name)

     @user_blocks_pages, @user_blocks = paginate(:user_blocks,
                                                 :include => [:user, :creator, :revoker],
     @user_blocks_pages, @user_blocks = paginate(:user_blocks,
                                                 :include => [:user, :creator, :revoker],

-                                                :conditions => {:user_id => @this_user.id},
+                                                :conditions => { :user_id => @this_user.id },

                                                 :order => "user_blocks.ends_at DESC",
                                                 :per_page => 20)
   end
                                                 :order => "user_blocks.ends_at DESC",
                                                 :per_page => 20)
   end


@@ -99,35 +100,22 @@ class UserBlocksController < ApplicationController
   ##
   # shows a list of all the blocks by the given user.
   def blocks_by
   ##
   # shows a list of all the blocks by the given user.
   def blocks_by

+    @params = params.permit(:display_name)

     @user_blocks_pages, @user_blocks = paginate(:user_blocks,
                                                 :include => [:user, :creator, :revoker],
     @user_blocks_pages, @user_blocks = paginate(:user_blocks,
                                                 :include => [:user, :creator, :revoker],

-                                                :conditions => {:creator_id => @this_user.id},
+                                                :conditions => { :creator_id => @this_user.id },

                                                 :order => "user_blocks.ends_at DESC",
                                                 :per_page => 20)
   end
 
   private
                                                 :order => "user_blocks.ends_at DESC",
                                                 :per_page => 20)
   end
 
   private

-  ##
-  # require that the user is a moderator, or fill out a helpful error message
-  # and return them to the login screen where they might be able to login as
-  # a moderator.
-  def require_moderator
-    unless @user.moderator?
-      flash[:notice] = t('user_block.filter.not_a_moderator')
-      redirect_to :action => 'index'
-    end
-  end
-
-  ##
-  # ensure that there is a "this_user" instance variable
-  def lookup_this_user
-    @this_user = User.find_by_display_name(params[:display_name])
-  end

 
   ##
   # ensure that there is a "user_block" instance variable
   def lookup_user_block
     @user_block = UserBlock.find(params[:id])
 
   ##
   # ensure that there is a "user_block" instance variable
   def lookup_user_block
     @user_block = UserBlock.find(params[:id])

+  rescue ActiveRecord::RecordNotFound
+    render :action => "not_found", :status => :not_found

   end
 
   ##
   end
 
   ##


@@ -140,14 +128,13 @@ class UserBlocksController < ApplicationController
     @valid_params = false
 
     if !UserBlock::PERIODS.include?(@block_period)
     @valid_params = false
 
     if !UserBlock::PERIODS.include?(@block_period)

-      flash[:notice] = t('user_block.filter.block_period')
-      
-    elsif @user_block and !@user_block.active?
-      flash[:notice] = t('user_block.filter.block_expired')
-      
+      flash[:error] = t("user_block.filter.block_period")
+
+    elsif @user_block && !@user_block.active?
+      flash[:error] = t("user_block.filter.block_expired")
+

     else
       @valid_params = true
     end
   end
     else
       @valid_params = true
     end
   end

-

 end
 end