summaryrefslogtreecommitdiffstats
path: root/projecttemplates/WebFormsRelyingParty/Code/DataRoleProvider.cs
blob: 8117e4b895fd04a236051973e49df088150f83b0 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
//-----------------------------------------------------------------------
// <copyright file="DataRoleProvider.cs" company="Andrew Arnott">
//     Copyright (c) Andrew Arnott. All rights reserved.
// </copyright>
//-----------------------------------------------------------------------

namespace WebFormsRelyingParty.Code {
	using System;
	using System.Collections.Generic;
	using System.Linq;
	using System.Web;
	using System.Web.Security;

	public class DataRoleProvider : RoleProvider {
		public override string ApplicationName {
			get { throw new NotImplementedException(); }
			set { throw new NotImplementedException(); }
		}

		public override void AddUsersToRoles(string[] usernames, string[] roleNames) {
			var users = from token in Global.DataContext.AuthenticationToken
						where usernames.Contains(token.ClaimedIdentifier)
						select token.User;
			var roles = from role in Global.DataContext.Role
						where roleNames.Contains(role.Name, StringComparer.OrdinalIgnoreCase)
						select role;
			foreach (User user in users) {
				foreach (Role role in roles) {
					user.Roles.Add(role);
				}
			}
		}

		public override void RemoveUsersFromRoles(string[] usernames, string[] roleNames) {
			var users = from token in Global.DataContext.AuthenticationToken
						where usernames.Contains(token.ClaimedIdentifier)
						select token.User;
			var roles = from role in Global.DataContext.Role
						where roleNames.Contains(role.Name, StringComparer.OrdinalIgnoreCase)
						select role;
			foreach (User user in users) {
				foreach (Role role in roles) {
					user.Roles.Remove(role);
				}
			}
		}

		public override void CreateRole(string roleName) {
			Global.DataContext.AddToRole(new Role { Name = roleName });
		}

		/// <summary>
		/// Removes a role from the data source for the configured applicationName.
		/// </summary>
		/// <param name="roleName">The name of the role to delete.</param>
		/// <param name="throwOnPopulatedRole">If true, throw an exception if <paramref name="roleName"/> has one or more members and do not delete <paramref name="roleName"/>.</param>
		/// <returns>
		/// true if the role was successfully deleted; otherwise, false.
		/// </returns>
		public override bool DeleteRole(string roleName, bool throwOnPopulatedRole) {
			Role role = Global.DataContext.Role.SingleOrDefault(r => r.Name == roleName);
			if (role == null) {
				return false;
			}

			if (throwOnPopulatedRole && role.Users.Count > 0) {
				throw new InvalidOperationException();
			}

			Global.DataContext.DeleteObject(roleName);
			return true;
		}

		/// <summary>
		/// Gets an array of user names in a role where the user name contains the specified user name to match.
		/// </summary>
		/// <param name="roleName">The role to search in.</param>
		/// <param name="usernameToMatch">The user name to search for.</param>
		/// <returns>
		/// A string array containing the names of all the users where the user name matches <paramref name="usernameToMatch"/> and the user is a member of the specified role.
		/// </returns>
		public override string[] FindUsersInRole(string roleName, string usernameToMatch) {
			return (from role in Global.DataContext.Role
					where role.Name == roleName
					from user in role.Users
					from authTokens in user.AuthenticationTokens
					where authTokens.ClaimedIdentifier == usernameToMatch
					select authTokens.ClaimedIdentifier).ToArray();
		}

		public override string[] GetAllRoles() {
			return Global.DataContext.Role.Select(role => role.Name).ToArray();
		}

		public override string[] GetRolesForUser(string username) {
			return (from authToken in Global.DataContext.AuthenticationToken
					where authToken.ClaimedIdentifier == username
					from role in authToken.User.Roles
					select role.Name).ToArray();
		}

		public override string[] GetUsersInRole(string roleName) {
			return (from role in Global.DataContext.Role
					where string.Equals(role.Name, roleName, StringComparison.OrdinalIgnoreCase)
					from user in role.Users
					from token in user.AuthenticationTokens
					select token.ClaimedIdentifier).ToArray();
		}

		public override bool IsUserInRole(string username, string roleName) {
			Role role = Global.DataContext.Role.SingleOrDefault(r => string.Equals(r.Name, roleName, StringComparison.OrdinalIgnoreCase));
			if (role != null) {
				return role.Users.Any(user => user.AuthenticationTokens.Any(token => token.ClaimedIdentifier == username));
			}

			return false;
		}

		public override bool RoleExists(string roleName) {
			return Global.DataContext.Role.Any(role => string.Equals(role.Name, roleName, StringComparison.OrdinalIgnoreCase));
		}
	}
}