login_system.rb
上传用户:netsea168
上传日期:2022-07-22
资源大小:4652k
文件大小:3k
源码类别:

Ajax

开发平台:

Others

  1. module LoginSystem
  2.   protected
  3.   
  4.     def logged_in?
  5.       current_user != :false
  6.     end
  7.     def current_user
  8.       @current_user ||= (login_from_session || login_from_basic_auth || login_from_cookie || :false)
  9.     end
  10.     def current_user=(new_user)
  11.       session[:user] = (new_user.nil? || new_user.is_a?(Symbol)) ? nil : new_user.id
  12.       @current_user = new_user
  13.     end
  14.     
  15.     # If the current actions are in our access rule will be verifyed
  16.     def allowed?
  17.       return AccessControl.allowed_controllers(current_user.profile.label, current_user.profile.modules).include?(params[:controller])
  18.     end
  19.     
  20.     def authorized?
  21.       logged_in? && allowed?
  22.     end
  23.     
  24.     def login_required
  25.       authorized? || access_denied
  26.     end
  27.     def access_denied
  28.       respond_to do |accepts|
  29.         accepts.html do
  30.           #store_location
  31.           session[:return_to] = request.request_uri
  32.           if User.find(:first)
  33.             redirect_to :controller => "accounts/login", :action => :index
  34.           else
  35.             redirect_to :controller => "accounts/login", :action => :new
  36.           end
  37.         end
  38.         accepts.xml do
  39.           headers["Status"]           = "Unauthorized"
  40.           headers["WWW-Authenticate"] = %(Basic realm="Web Password")
  41.           render :text => "Could't authenticate you", :status => '401 Unauthorized'
  42.         end
  43.       end
  44.       false
  45.     end  
  46.     def store_location
  47.       session[:return_to] = request.request_uri
  48.     end
  49.     def redirect_back_or_default(default)
  50.       redirect_to(session[:return_to] || default)
  51.       session[:return_to] = nil
  52.     end
  53.     def self.included(base)
  54.       base.send :helper_method, :current_user, :logged_in?
  55.     end
  56.     def login_from_session
  57.       self.current_user = User.find_by_id(session[:user]) if session[:user]
  58.     end
  59.     def login_from_basic_auth
  60.       email, passwd = get_auth_data
  61.       self.current_user = User.authenticate(email, passwd) if email && passwd
  62.     end
  63.     # Called from #current_user.  Finaly, attempt to login by an expiring token in the cookie.
  64.     def login_from_cookie      
  65.       user = cookies[:auth_token] && User.find_by_remember_token(cookies[:auth_token])
  66.       if user && user.remember_token?
  67.         user.remember_me
  68.         cookies[:auth_token] = { :value => user.remember_token, :expires => user.remember_token_expires_at }
  69.         self.current_user = user
  70.       end
  71.     end
  72.   private
  73.     @@http_auth_headers = %w(X-HTTP_AUTHORIZATION HTTP_AUTHORIZATION Authorization)
  74.     # gets BASIC auth info
  75.     def get_auth_data
  76.       auth_key  = @@http_auth_headers.detect { |h| request.env.has_key?(h) }
  77.       auth_data = request.env[auth_key].to_s.split unless auth_key.blank?
  78.       return auth_data && auth_data[0] == 'Basic' ? Base64.decode64(auth_data[1]).split(':')[0..1] : [nil, nil] 
  79.     end
  80. end