ochko/prudge

# -*- coding: utf-8 -*-
class ApplicationController < ActionController::Base
  helper :all
  protect_from_forgery

  helper_method :current_user_session, :current_user, :current_user?, :admin?, :judge?

  before_filter :set_locale

  def current_ability
    @current_ability ||=
      if current_user.nil?
        BaseAbility
      elsif current_user.admin?
        AdminAbility
      elsif current_user.judge?
        JudgeAbility
      else
        CoderAbility
      end.
      new(current_user)
  end

  rescue_from CanCan::AccessDenied do |exception|
    respond_to do |format|
      format.js { render :partial => 'access_denied', :object => exception }
      format.html { redirect_to root_url, :alert => exception.message }
    end
  end

  private

  def set_locale
    if params[:locale].present?
      session[:locale] = params[:locale]
    elsif session[:locale].blank?
      session[:locale] = http_accept_language
    end
    I18n.locale = session[:locale] || I18n.default_locale
  end

  def http_accept_language
    if lang = request.env['HTTP_ACCEPT_LANGUAGE']
      lang.scan(/^[a-z]{2}/).first
    end
  end

  def message_for(name, options = {})
    name ||= "#{controller_name}.#{action_name}"
    I18n.translate name, options.merge(:scope => :message)
  end

  def flash_notice(name=nil, options = {})
    flash[:notice] = message_for(name, options)
  end

  def flash_error(name=nil, options = {})
    flash[:error] = message_for(name, options)
  end

  def admin?
    current_user && current_user.admin
  end

  def judge?
    current_user && current_user.judge
  end

  def current_user_session
    return @current_user_session if defined?(@current_user_session)
    @current_user_session = UserSession.find
  end

  def current_user
    return @current_user if defined?(@current_user)
    @current_user = current_user_session && current_user_session.user
  end

  def current_user?(user)
    return false unless current_user
    current_user.id == user.id
  end

  def require_user
    unless current_user
      respond_to do |format|
        format.html do
          store_location
          flash_notice :login_required
          redirect_to login_url
        end
        format.js { render :text => message_for(:login_required)}
      end
      return false
    end
  end

  def require_no_user
    if current_user
      store_location
      flash_notice = :logout_required
      redirect_to root_url
      return false
    end
  end

  def store_location
    if request.format == "html"
      session[:return_to] = request.url
    end
  end

  def redirect_back_or_default(default)
    redirect_to(session[:return_to] || default)
    session[:return_to] = nil
  end
end