app/controllers/cas/application_controller.rb in cas-cms-0.1.0 vs app/controllers/cas/application_controller.rb in cas-cms-0.1.1

@@ -1,17 +1,27 @@
 module Cas
   class ApplicationController < ActionController::Base
     protect_from_forgery with: :exception
     before_action :authenticate_user!
     before_action :set_current_user
+    before_action :set_user_sites
+    before_action :set_domain
 
     private
 
     def set_current_user
       @current_user = current_user
     end
 
+    def set_user_sites
+      @user_sites = @current_user.sites if @current_user.present?
+    end
+
     def set_domain
       @domain ||= (ENV["DOMAIN"] || request.domain)
+
+      if @domain.blank? || (ENV["DOMAIN"].blank? && @domain == "localhost")
+        @domain = ::Cas::Site.first!.domains.first
+      end
     end
   end
 end