Bladeren bron

Add UserDetailsMapFactoryBean

Fixes gh-4804
Rob Winch 7 jaren geleden
bovenliggende
commit
f2ccc53549

+ 86 - 0
config/src/main/java/org/springframework/security/config/core/userdetails/UserDetailsMapFactoryBean.java

@@ -0,0 +1,86 @@
+/*
+ * Copyright 2002-2017 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.springframework.security.config.core.userdetails;
+
+import org.springframework.beans.factory.FactoryBean;
+import org.springframework.lang.Nullable;
+import org.springframework.security.core.userdetails.User;
+import org.springframework.security.core.userdetails.UserDetails;
+import org.springframework.security.core.userdetails.memory.UserAttribute;
+import org.springframework.security.core.userdetails.memory.UserAttributeEditor;
+import org.springframework.util.Assert;
+
+import java.util.ArrayList;
+import java.util.Collection;
+import java.util.Map;
+
+/**
+ * Creates a {@code Collection<UserDetails>} from a @{code Map} in the format of
+ * <p>
+ * <code>
+ * username=password[,enabled|disabled],roles...
+ * </code>
+ * <p>
+ * The enabled and disabled properties are optional with enabled being the default. For example:
+ * <p>
+ * <code>
+ * user=password,ROLE_USER
+ * admin=secret,ROLE_USER,ROLE_ADMIN
+ * disabled_user=does_not_matter,disabled,ROLE_USER
+ * </code>
+ *
+ * @author Rob Winch
+ * @since 5.0
+ */
+public class UserDetailsMapFactoryBean implements FactoryBean<Collection<UserDetails>> {
+	private final Map<String, String> userProperties;
+
+	public UserDetailsMapFactoryBean(Map<String, String> userProperties) {
+		Assert.notNull(userProperties, "userProperties cannot be null");
+		this.userProperties = userProperties;
+	}
+
+	@Nullable
+	@Override
+	public Collection<UserDetails> getObject() throws Exception {
+		Collection<UserDetails> users = new ArrayList<>(this.userProperties.size());
+
+		UserAttributeEditor editor = new UserAttributeEditor();
+		for (Map.Entry<String, String> entry : this.userProperties.entrySet()) {
+			String name = entry.getKey();
+			String property = entry.getValue();
+			editor.setAsText(property);
+			UserAttribute attr = (UserAttribute) editor.getValue();
+			if (attr == null) {
+				throw new IllegalStateException("The entry with username '" + name
+					+ "' and value '" + property + "' could not be converted to a UserDetails.");
+			}
+			UserDetails user = User.withUsername(name)
+				.password(attr.getPassword())
+				.disabled(!attr.isEnabled())
+				.authorities(attr.getAuthorities())
+				.build();
+			users.add(user);
+		} return users;
+	}
+
+	@Nullable
+	@Override
+	public Class<?> getObjectType() {
+		return Collection.class;
+	}
+}

+ 2 - 26
config/src/main/java/org/springframework/security/config/core/userdetails/UserDetailsResourceFactoryBean.java

@@ -21,17 +21,13 @@ import org.springframework.context.ResourceLoaderAware;
 import org.springframework.core.io.DefaultResourceLoader;
 import org.springframework.core.io.Resource;
 import org.springframework.core.io.ResourceLoader;
-import org.springframework.security.core.userdetails.User;
 import org.springframework.security.core.userdetails.UserDetails;
-import org.springframework.security.core.userdetails.memory.UserAttribute;
-import org.springframework.security.core.userdetails.memory.UserAttributeEditor;
 import org.springframework.security.util.InMemoryResource;
 import org.springframework.util.Assert;
 
 import java.io.InputStream;
-import java.util.ArrayList;
 import java.util.Collection;
-import java.util.Enumeration;
+import java.util.Map;
 import java.util.Properties;
 
 /**
@@ -72,27 +68,7 @@ public class UserDetailsResourceFactoryBean implements ResourceLoaderAware, Fact
 		try(InputStream in = resource.getInputStream()){
 			userProperties.load(in);
 		}
-
-		Collection<UserDetails> users = new ArrayList<>(userProperties.size());
-		Enumeration<?> names = userProperties.propertyNames();
-		UserAttributeEditor editor = new UserAttributeEditor();
-
-		while (names.hasMoreElements()) {
-			String name = (String) names.nextElement();
-			String property = userProperties.getProperty(name);
-			editor.setAsText(property);
-			UserAttribute attr = (UserAttribute) editor.getValue();
-			if(attr == null) {
-				throw new IllegalStateException("The entry with username '" + name + "' and value '" + property + "' could not be converted to a UserDetails.");
-			}
-			UserDetails user = User.withUsername(name)
-				.password(attr.getPassword())
-				.disabled(!attr.isEnabled())
-				.authorities(attr.getAuthorities())
-				.build();
-			users.add(user);
-		}
-		return users;
+		return new UserDetailsMapFactoryBean((Map) userProperties).getObject();
 	}
 
 	@Override