035dc993c636816f9b1c64d753ae0ab842edcf54
[utils] / security / src / main / java / org / wamblee / usermgt / InMemoryUserSet.java
1 /*
2  * Copyright 2005 the original author or authors.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 package org.wamblee.usermgt;
17
18 import org.wamblee.security.encryption.MessageDigester;
19
20 import java.util.Set;
21 import java.util.TreeSet;
22
23
24 /**
25  * In-memory user set.
26  *
27  * @author Erik Brakkee
28  */
29 public class InMemoryUserSet extends AbstractUserSet {
30     /**
31      * Users. All users in this set have their password validator and
32      * encoder set.
33      */
34     private Set<User> users;
35
36 /**
37      * Constructs an empty user set.
38      */
39     public InMemoryUserSet(NameValidator aPasswordValidator,
40         MessageDigester aPasswordEncoder) {
41         super(aPasswordValidator, aPasswordEncoder);
42         users = new TreeSet<User>();
43     }
44
45     /*
46      * (non-Javadoc)
47      *
48      * @see org.wamblee.usermgt.UserSet#userModified(org.wamblee.usermgt.User)
49      */
50     /**
51      * DOCUMENT ME!
52      *
53      * @param aUser DOCUMENT ME!
54      */
55     public void userModified(User aUser) {
56         users.remove(aUser);
57         setPasswordInfo(aUser);
58         users.add(aUser);
59     }
60
61     /*
62      * (non-Javadoc)
63      *
64      * @see org.wamblee.usermgt.UserSet#find(java.lang.String)
65      */
66     /**
67      * DOCUMENT ME!
68      *
69      * @param aName DOCUMENT ME!
70      *
71      * @return DOCUMENT ME!
72      */
73     public User find(String aName) {
74         for (User user : users) {
75             if (user.getName().equals(aName)) {
76                 return new User(user);
77             }
78         }
79
80         return null;
81     }
82
83     /*
84      * (non-Javadoc)
85      *
86      * @see org.wamblee.usermgt.UserSet#add(org.wamblee.usermgt.User)
87      */
88     /**
89      * DOCUMENT ME!
90      *
91      * @param aUser DOCUMENT ME!
92      *
93      * @return DOCUMENT ME!
94      */
95     public boolean add(User aUser) {
96         setPasswordInfo(aUser);
97
98         return users.add(aUser);
99     }
100
101     /*
102      * (non-Javadoc)
103      *
104      * @see org.wamblee.usermgt.UserSet#contains(org.wamblee.usermgt.User)
105      */
106     /**
107      * DOCUMENT ME!
108      *
109      * @param aUser DOCUMENT ME!
110      *
111      * @return DOCUMENT ME!
112      */
113     public boolean contains(User aUser) {
114         return users.contains(aUser);
115     }
116
117     /*
118      * (non-Javadoc)
119      *
120      * @see org.wamblee.usermgt.UserSet#remove(org.wamblee.usermgt.User)
121      */
122     /**
123      * DOCUMENT ME!
124      *
125      * @param aUser DOCUMENT ME!
126      *
127      * @return DOCUMENT ME!
128      */
129     public boolean remove(User aUser) {
130         return users.remove(aUser);
131     }
132
133     /*
134      * (non-Javadoc)
135      *
136      * @see org.wamblee.usermgt.UserSet#list()
137      */
138     /**
139      * DOCUMENT ME!
140      *
141      * @return DOCUMENT ME!
142      */
143     public Set<User> list() {
144         Set<User> list = new TreeSet<User>();
145
146         for (User user : users) {
147             list.add(new User(user));
148         }
149
150         return list;
151     }
152
153     /*
154      * (non-Javadoc)
155      *
156      * @see org.wamblee.usermgt.UserSet#list(org.wamblee.usermgt.Group)
157      */
158     /**
159      * DOCUMENT ME!
160      *
161      * @param aGroup DOCUMENT ME!
162      *
163      * @return DOCUMENT ME!
164      */
165     public Set<User> list(Group aGroup) {
166         Set<User> result = new TreeSet<User>();
167
168         for (User user : users) {
169             if (user.getGroups().contains(aGroup)) {
170                 result.add(new User(user));
171             }
172         }
173
174         return result;
175     }
176
177     /* (non-Javadoc)
178      * @see org.wamblee.usermgt.UserSet#size()
179      */
180     /**
181      * DOCUMENT ME!
182      *
183      * @return DOCUMENT ME!
184      */
185     public int size() {
186         return users.size();
187     }
188 }