danielweinmann/unlock

class ApplicationController < ActionController::Base

  include Pundit
  protect_from_forgery with: :exception
  before_action :configure_permitted_parameters, if: :devise_controller?
  helper_method :namespace, :add_headline_and_article!, :suppress_headline_and_article!, :add_headline_and_article?, :add_logo!, :suppress_logo!, :add_logo?, :locale_country

  after_action :verify_authorized, unless: -> {devise_controller? || self.class == HighVoltage::PagesController}
  after_action :verify_policy_scoped, unless: -> {devise_controller? || self.class == HighVoltage::PagesController}
  rescue_from Pundit::NotAuthorizedError, with: :user_not_authorized
  rescue_from I18n::InvalidLocale, with: :user_not_authorized

  # This will send the current_user to the view and instantiate UserDecorator for it
  before_action :current_user

  before_filter :set_locale

  force_ssl if: :in_production?

  responders :flash, :location

  def in_production?
    Rails.env.production?
  end
    
  def add_headline_and_article?
    !@suppress_headline_and_article
  end

  def add_headline_and_article!
    @suppress_headline_and_article = false
  end

  def suppress_headline_and_article!
    @suppress_headline_and_article = true
  end

  def add_logo?
    !@suppress_logo
  end

  def add_logo!
    @suppress_logo = false
  end

  def suppress_logo!
    @suppress_logo = true
  end

  def namespace
    names = self.class.to_s.split('::')
    return "null" if names.length < 2
    names[0..(names.length-2)].map(&:downcase).join('_')
  end

  def configure_permitted_parameters
    devise_parameter_sanitizer.for(:account_update) << :name
    devise_parameter_sanitizer.for(:account_update) << :image
  end

  def user_not_authorized(exception)
    flash[:alert] = t('flash.not_authorized')
    redirect_to(request.referrer || root_path)
  end

  def set_locale
    if params[:locale]
      # Enforces locale with country
      raise I18n::InvalidLocale, params[:locale].to_sym unless locale_country(params[:locale], allow_default_locales?)
      I18n.locale = params[:locale]
      current_user.try(:set_locale, params[:locale])
    elsif request.method == "GET"
      new_locale = current_user.try(:locale) || I18n.default_locale
      redirect_to url_for(params.merge(locale: new_locale, only_path: true))
    end
  end

  def locale_country(locale, allow_default_locales = false)
    country = locale.to_s.split('-')
    if allow_default_locales
      country = ["en", "GB"] if country == ["en"]
      country.size > 1 ? country[1].downcase : country[0]
    else
      return unless country && country.size > 1 && country[1] != '419'
      country[1].downcase
    end
  end

  private
  
  def allow_default_locales?
    false
  end

end