PumpkinHouseWeb/Account/RoleProvider.cs

   1 using System;
   2 using System.Collections.Generic;
   3 using System.Linq;
   4 using System.Web;
   5 using System.Web.Security;
   6 using PumpkinHouseDatabase;
   7
   8 namespace PumpkinHouse
   9 {
  10     public class RoleProvider : SqlRoleProvider
  11     {
  12         public override bool IsUserInRole(string username, string roleName)
  13         {
  14             if (roleName == "admin")
  15             {
  16                 using (DataUtils utils = new DataUtils(false))
  17                 {
  18                     DB_User user = utils.FindUserByNameOrEmail(username);
  19                     if (user == null)
  20                     {
  21                         return false;
  22                     }
  23                     return user.Is_Admin == 1;
  24                 }
  25             }
  26             if (roleName == "locked")
  27             {
  28                 using (DataUtils utils = new DataUtils(false))
  29                 {
  30                     DB_User user = utils.FindUserByNameOrEmail(username);
  31                     if (user == null)
  32                     {
  33                         return false;
  34                     }
  35                     if (user.Is_Locked == 1)
  36                     {
  37                         HttpContext.Current.Session["username"] = user.Username;
  38                         HttpContext.Current.Session["lockedReason"] = user.Locked_Reason;
  39                         return true;
  40                     };
  41                     return false;
  42                 }
  43             }
  44             return true;
  45         }
  46
  47         public override string[] GetRolesForUser(string username)
  48         {
  49             using (DataUtils utils = new DataUtils(false))
  50             {
  51                 DB_User user = utils.FindUserByNameOrEmail(username);
  52                 if (user.Is_Locked == 1)
  53                 {
  54                     return new string[] { "locked" };
  55                 }
  56                 if (user.Is_Admin == 1)
  57                 {
  58                     return new string[] { "admin" };
  59                 }
  60                 return new string[] { };
  61             }
  62         }
  63
  64         public override string[] GetAllRoles()
  65         {
  66             return new string[] { "admin", "locked" };
  67         }
  68
  69     }
  70 }