diff --git a/app/controllers/users_controller.rb b/app/controllers/users_controller.rb index 4b5af22..8c2c444 100644 --- a/app/controllers/users_controller.rb +++ b/app/controllers/users_controller.rb @@ -8,7 +8,13 @@ class UsersController < ApplicationController if params[:role].downcase == "staff" @users = User.all.select {|u| u.role >= Role.get(:mod) } else - @users = User.where(role: Role.get(params[:role])) + if role = Role.get(params[:role]) + @users = User.where(role: role) + else + flash[:alert] = "role '#{params[:role]}' does not exist!" + redirect_to users_path + return + end end else @users = User.where.not(id: User.first.id) #Remove first user diff --git a/app/views/users/index.html.erb b/app/views/users/index.html.erb index 33fed1d..e445b8d 100644 --- a/app/views/users/index.html.erb +++ b/app/views/users/index.html.erb @@ -1,9 +1,11 @@ <% if params[:role] %> -
User count: <%= @users.length %>
+