/*
* JBoss, Home of Professional Open Source.
* Copyright 2011, Red Hat, Inc., and individual contributors
* as indicated by the @author tags. See the copyright.txt file in the
* distribution for a full listing of individual contributors.
*
* This is free software; you can redistribute it and/or modify it
* under the terms of the GNU Lesser General Public License as
* published by the Free Software Foundation; either version 2.1 of
* the License, or (at your option) any later version.
*
* This software is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this software; if not, write to the Free
* Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
* 02110-1301 USA, or see the FSF site: http://www.fsf.org.
*/
package org.jboss.as.domain.management.security.adduser;
import org.jboss.as.domain.management.security.PropertiesFileLoader;
import org.jboss.as.domain.management.security.UserPropertiesFileLoader;
import org.jboss.msc.service.StartException;
import org.wildfly.security.sasl.util.UsernamePasswordHashUtil;
import java.io.File;
import java.io.IOException;
import java.security.NoSuchAlgorithmException;
import java.util.Properties;
import static org.jboss.as.domain.management.security.adduser.AddUser.NEW_LINE;
public abstract class UpdatePropertiesHandler {
private ConsoleWrapper theConsole;
public UpdatePropertiesHandler(ConsoleWrapper theConsole) {
this.theConsole = theConsole;
}
/**
* Implement the persistence handler for storing the group properties.
*/
void persist(final String key, final String value, final boolean enableDisableMode, final boolean disable, final File file) throws IOException, StartException {
persist(key, value, enableDisableMode, disable, file, null);
}
/**
* Implement the persistence handler for storing the user properties.
*/
void persist(final String key, final String value, final boolean enableDisableMode, final boolean disable, final File file, final String realm) throws IOException, StartException {
final PropertiesFileLoader propertiesHandler = realm == null ? new PropertiesFileLoader(file.getAbsolutePath(), null) :
new UserPropertiesFileLoader(file.getAbsolutePath(), null);
try {
propertiesHandler.start(null);
if (realm != null) {
((UserPropertiesFileLoader) propertiesHandler).setRealmName(realm);
}
Properties prob = propertiesHandler.getProperties();
if (value != null) {
prob.setProperty(key, value);
}
if (enableDisableMode) {
prob.setProperty(key + "!disable", String.valueOf(disable));
}
propertiesHandler.persistProperties();
} finally {
propertiesHandler.stop(null);
}
}
/**
* Customize message for update or add users
* @param fileName - the filename of the updated property file;
* @return the console message that should be present to the user
*/
abstract String consoleUserMessage(String fileName);
/**
* Customize message for update or add groups
* @param fileName - the filename of the updated property file;
* @return the console message that should be present to the user
*/
abstract String consoleGroupsMessage(String fileName);
/**
* Customize error message for update or add users groups / properties
* @param fileName - the filename of the updated property file;
* @return the console message that should be present to the user
*/
abstract String errorMessage(String fileName, Throwable e);
State update(StateValues stateValues) {
final String userName = stateValues.getUserName();
final boolean enableDisableMode = stateValues.getOptions().isEnableDisableMode();
final boolean disable = stateValues.getOptions().isDisable();
final String groups = stateValues.getGroups();
final String password;
if (stateValues.getPassword() != null) {
try {
password = new UsernamePasswordHashUtil().generateHashedHexURP(
stateValues.getUserName(),
stateValues.getRealm(),
stateValues.getPassword().toCharArray());
} catch (NoSuchAlgorithmException e) {
return new ErrorState(theConsole, e.getMessage(), null, stateValues);
}
} else {
password = null;
}
// Persist username=password
for (File current : stateValues.getUserFiles()) {
try {
persist(userName, password, enableDisableMode, disable, current, stateValues.getRealm());
if (stateValues.isSilent() == false) {
theConsole.printf(consoleUserMessage(current.getCanonicalPath()));
theConsole.printf(NEW_LINE);
}
} catch (Exception e) {
return new ErrorState(theConsole, errorMessage(current.getAbsolutePath(), e), null, stateValues);
}
}
// Persist username=groups
if (stateValues.groupPropertiesFound() && (groups != null || enableDisableMode)) {
for (final File current : stateValues.getGroupFiles()) {
try {
persist(userName, groups, enableDisableMode, disable, current);
if (stateValues.isSilent() == false) {
theConsole.printf(consoleGroupsMessage(current.getCanonicalPath()));
theConsole.printf(NEW_LINE);
}
} catch (Exception e) {
return new ErrorState(theConsole, errorMessage(current.getAbsolutePath(), e), null, stateValues);
}
}
}
return null;
}
}