上次我們實現了登錄的頁面,現在我們區別管理員和普通用戶,根據用戶的不同身份,轉入到不同的頁面
在rails中提供了filter來截獲訪問某個action的請求,可以用filter在action被調用前或調用後添加我們自己的處理。在這裡,我們在admin的controller中的action被調用前添加攔截器。如果是管理員,就進入管理頁面,如果是普通用戶,就進入登錄頁面。
1.在application.rb中給ApplicationController類添加authorize方法,代碼如下:
def authorize unless session[:user_id] flash[:notice] = "Please log in" redirect_to(:controller => "login", :action => "login") end end
然後在admin.rb文件裡AdminController類開始的地方添加一行代碼:
class AdminController < ApplicationController before_filter :authorize ……
這樣,我們就添加了一個攔截器,現在,AdminController中所有的action被調用前都會先調用authorize方法。
我們還要修改login_controller.rb文件,給LoginController裡添加一句代碼:
class LoginController < ApplicationController before_filter :authorize, :except => :login ……
2.現在,我們如果直接訪問http://localhost:3003/admin/ship,將會直接定位到login頁面,並且提示先請登錄。如圖:
如果之前已經登錄過,因為session還存在,所以不會看到效果,只要把浏覽器關掉才重新定位admin/ship頁面就可以了。
3.下面我們先完成一個用戶列表的功能。代碼很簡單:
在login_controller.rb文件中,修改list_users方法:
def list_users @all_users = User.find(:all) end
修改views\login目錄下的list_users.rhtml文件:
<% @page_title = "User List" -%> <table> <% for user in @all_users -%> <tr> <td><%= user.name %></td> <td><%= link_to("(delete)", :action => :delete_user, :id => user.id) %></td> </tr> <% end -%> </table>
完成後,如果你已經登錄的話,直接打開http://localhost:3003/login/list_users頁面,會看到一個簡單的用戶列表頁面,如圖:
4.在用戶列表頁面上可以看到有delete鏈接,當然,我們不能把所有的管理員都刪除掉,至少要保留一個,所以我們添加一些代碼,來確保某個管理員不會被刪除。在users.rb文件裡添加代碼:
before_destroy :dont_destroy_dahuzizyd def dont_destroy_dahuzizyd raise "Can't destroy dahuzizyd" if self.name == 'dahuzizyd' end
並且修改login_controller.rb文件中的delete_user方法,如下:
def delete_user id = params[:id] if id && user = User.find(id) begin user.destroy flash[:notice] = "User #{user.name} deleted" rescue flash[:notice] = "Can't delete that user" end end redirect_to(:action => :list_users) end
這樣,在用戶列表頁面上,當刪除名為dahuzizyd的用戶時會提示不能刪除,如圖:
5.接下來,我們來修改views\layout目錄下的admin.rhtml文件 ,這樣我們就可以在管理員登錄以後顯示不同的功能鏈接,代碼如下:
<html> <head> <title>ADMINISTER Pragprog Books Online Store</title> <%= stylesheet_link_tag "scaffold", "depot", "admin", :media => "all" %> </head> <body> <div id="banner"> <%= @page_title || "Administer Bookshelf" %> </div> <div id="columns"> <div id="side"> <% if session[:user_id] -%> <%= link_to("Products", :controller => "admin", :action => "list") %><br /> <%= link_to("Shipping", :controller => "admin", :action => "ship") %><br /> <hr/> <%= link_to("Add user", :controller => "login", :action => "add_user") %><br /> <%= link_to("List users", :controller => "login", :action => "list_users") %><br /> <hr/> <%= link_to("Log out", :controller => "login", :action => "logout") %> <% end -%> </div> <div id="main"> <% if flash[:notice] -%> <div id="notice"><%= flash[:notice] %></div> <% end -%> <%= @content_for_layout %> </div> </div> </body> </html>
完成後,我們可以點擊Add user鏈接,出現下面的頁面:
6.好了,現在在頁面左側的鏈接裡有一個logout,我們還沒有實現它的功能,現在就開始,修改login_controller.rb文件中的log_out方法,代碼如下:
def logout session[:user_id] = nil flash[:notice] = "Logged out" redirect_to(:action => "login") end
OK,現在我們點擊logout按鈕,會重新定位到login頁面,如圖:
7.最後我們再整理下代碼,在store_controller.rb文件中,@cart = find_cart這句代碼在除了empty_cart以外的所有方法中都是第一句,所以,我們現在修改方法find_cart方法:
def find_cart @cart = (session[:cart] ||= Cart.new) end
並且,給store_controller.rb添加一個攔截器:
before_filter :find_cart, :except => :index
這樣我們就可以在store_controller.rb中所有的@cart = find_cart代碼都注釋掉了。