application_controller.rb in comable_backend-0.3.0

- old
+ new

@@ -1,7 +1,35 @@
 module Comable
   module Admin
     class ApplicationController < ActionController::Base
+      include Comable::ApplicationHelper
+
       layout 'comable/admin/application'
+
+      def current_ability
+        Comable::Ability.new(current_customer)
+      end
+
+      private
+
+      rescue_from CanCan::AccessDenied, with: :unauthorized
+
+      def unauthorized
+        if current_customer.signed_in?
+          flash[:alert] = Comable.t('admin.access_denied')
+          redirect_to after_access_denied_path
+        else
+          store_location
+          redirect_to comable.new_admin_customer_session_path
+        end
+      end
+
+      def after_access_denied_path
+        if current_customer.customer?
+          comable.root_path
+        else
+          comable.admin_root_path
+        end
+      end
     end
   end
 end