app/controllers/storytime/application_controller.rb in storytime-1.0.5 vs app/controllers/storytime/application_controller.rb in storytime-1.0.6

@@ -6,10 +6,16 @@
   include Pundit
   rescue_from Pundit::NotAuthorizedError, with: :user_not_authorized
   
   helper :all
 
+  if Storytime.user_class_symbol != :user
+    helper_method :authenticate_user!
+    helper_method :current_user
+    helper_method :user_signed_in?
+  end
+
   def setup
     url = if Storytime.user_class.count == 0
       main_app.new_user_registration_url
     elsif current_user.nil?
       main_app.new_user_session_url
@@ -18,9 +24,23 @@
     else
       url_for([:dashboard, Storytime::BlogPost])
     end
 
     redirect_to url
+  end
+
+  if Storytime.user_class_symbol != :user
+    def authenticate_user!
+      send("authenticate_#{Storytime.user_class.to_s.downcase}!".to_sym)
+    end
+
+    def current_user
+      send("current_#{Storytime.user_class.to_s.downcase}".to_sym)
+    end
+
+    def user_signed_in?
+      send("#{Storytime.user_class.to_s.downcase}_signed_in?".to_sym)
+    end
   end
 
 private
   def ensure_site
     redirect_to new_dashboard_site_url unless devise_controller? || @site = Storytime::Site.first