NewSchemaElementsTask.java revision 2d3af58c2cee0a4a12390fae1eace3fdd15d9664
/*
* CDDL HEADER START
*
* The contents of this file are subject to the terms of the
* Common Development and Distribution License, Version 1.0 only
* (the "License"). You may not use this file except in compliance
* with the License.
*
* You can obtain a copy of the license at legal-notices/CDDLv1_0.txt
* or http://forgerock.org/license/CDDLv1.0.html.
* See the License for the specific language governing permissions
* and limitations under the License.
*
* When distributing Covered Code, include this CDDL HEADER in each
* file and include the License file at legal-notices/CDDLv1_0.txt.
* If applicable, add the following below this CDDL HEADER, with the
* fields enclosed by brackets "[]" replaced with your own identifying
* information:
* Portions Copyright [yyyy] [name of copyright owner]
*
* CDDL HEADER END
*
*
* Copyright 2009 Sun Microsystems, Inc.
* Portions Copyright 2013-2015 ForgeRock AS.
*/
package org.opends.guitools.controlpanel.task;
import static org.forgerock.util.Utils.*;
import static org.opends.messages.AdminToolMessages.*;
import static org.opends.server.types.CommonSchemaElements.*;
import java.io.File;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.LinkedHashMap;
import java.util.LinkedHashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
import javax.naming.NamingException;
import javax.naming.directory.BasicAttribute;
import javax.naming.directory.DirContext;
import javax.naming.directory.ModificationItem;
import javax.swing.SwingUtilities;
import org.forgerock.i18n.LocalizableMessage;
import org.forgerock.opendj.ldap.ModificationType;
import org.opends.guitools.controlpanel.datamodel.ControlPanelInfo;
import org.opends.guitools.controlpanel.ui.ColorAndFontConstants;
import org.opends.guitools.controlpanel.ui.ProgressDialog;
import org.opends.guitools.controlpanel.util.Utilities;
import org.opends.server.config.ConfigConstants;
import org.opends.server.core.DirectoryServer;
import org.opends.server.types.AttributeType;
import org.opends.server.types.Attributes;
import org.opends.server.types.CommonSchemaElements;
import org.opends.server.types.DirectoryException;
import org.opends.server.types.Entry;
import org.opends.server.types.ExistingFileBehavior;
import org.opends.server.types.LDIFExportConfig;
import org.opends.server.types.LDIFImportConfig;
import org.opends.server.types.Modification;
import org.opends.server.types.ObjectClass;
import org.opends.server.types.OpenDsException;
import org.opends.server.types.SchemaFileElement;
import org.opends.server.util.LDIFReader;
import org.opends.server.util.LDIFWriter;
import org.opends.server.util.ServerConstants;
/**
* An abstract class used to re-factor some code between the different tasks
* that create elements in the schema.
*/
public class NewSchemaElementsTask extends Task
{
final Set<ObjectClass> ocsToAdd = new LinkedHashSet<>();
final Set<AttributeType> attrsToAdd = new LinkedHashSet<>();
/**
* Constructor of the task.
*
* @param info
* the control panel information.
* @param dlg
* the progress dialog where the task progress will be displayed.
* @param ocsToAdd
* the object classes that must be created in order.
* @param attrsToAdd
* the attributes that must be created in order.
*/
public NewSchemaElementsTask(
ControlPanelInfo info, ProgressDialog dlg, Set<ObjectClass> ocsToAdd, Set<AttributeType> attrsToAdd)
{
super(info, dlg);
this.ocsToAdd.addAll(ocsToAdd);
this.attrsToAdd.addAll(attrsToAdd);
}
@Override
public Set<String> getBackends()
{
return Collections.emptySet();
}
@Override
public boolean canLaunch(Task taskToBeLaunched, Collection<LocalizableMessage> incompatibilityReasons)
{
if (state == State.RUNNING &&
(taskToBeLaunched.getType() == Task.Type.DELETE_SCHEMA_ELEMENT ||
taskToBeLaunched.getType() == Task.Type.MODIFY_SCHEMA_ELEMENT ||
taskToBeLaunched.getType() == Task.Type.NEW_SCHEMA_ELEMENT))
{
incompatibilityReasons.add(getIncompatibilityMessage(this, taskToBeLaunched));
return false;
}
return true;
}
@Override
public void runTask()
{
state = State.RUNNING;
lastException = null;
try
{
updateSchema();
state = State.FINISHED_SUCCESSFULLY;
}
catch (Throwable t)
{
lastException = t;
state = State.FINISHED_WITH_ERROR;
}
}
@Override
public Type getType()
{
return Type.NEW_SCHEMA_ELEMENT;
}
@Override
public LocalizableMessage getTaskDescription()
{
if (attrsToAdd.size() == 1 && ocsToAdd.isEmpty())
{
return INFO_CTRL_PANEL_NEW_ATTRIBUTE_TASK_DESCRIPTION.get(attrsToAdd.iterator().next().getNameOrOID());
}
else if (ocsToAdd.size() == 1 && attrsToAdd.isEmpty())
{
return INFO_CTRL_PANEL_NEW_OBJECTCLASS_TASK_DESCRIPTION.get(ocsToAdd.iterator().next().getNameOrOID());
}
else
{
final List<String> attrNames = getElementsNameOrOID(attrsToAdd);
final List<String> ocNames = getElementsNameOrOID(ocsToAdd);
if (ocNames.isEmpty())
{
return INFO_CTRL_PANEL_NEW_ATTRIBUTES_TASK_DESCRIPTION.get(joinAsString(", ", attrNames));
}
else if (attrNames.isEmpty())
{
return INFO_CTRL_PANEL_NEW_OBJECTCLASSES_TASK_DESCRIPTION.get(joinAsString(", ", ocNames));
}
else
{
return INFO_CTRL_PANEL_NEW_SCHEMA_ELEMENTS_TASK_DESCRIPTION.get(
joinAsString(", ", attrNames), joinAsString(", ", ocNames));
}
}
}
private <T extends CommonSchemaElements> List<String> getElementsNameOrOID(final Set<T> schemaElements)
{
final List<String> nameOrOIDs = new ArrayList<>();
for (CommonSchemaElements schemaElement : schemaElements)
{
nameOrOIDs.add(schemaElement.getNameOrOID());
}
return nameOrOIDs;
}
/**
* Update the schema.
*
* @throws OpenDsException
* if an error occurs.
*/
private void updateSchema() throws OpenDsException
{
if (isServerRunning())
{
updateSchemaOnline();
}
else
{
updateSchemaOffline();
}
}
@Override
protected String getCommandLinePath()
{
return null;
}
@Override
protected List<String> getCommandLineArguments()
{
return Collections.emptyList();
}
/**
* Add the schema elements one by one: we are not sure that the server will
* handle the adds sequentially if we only send one modification.
*
* @throws OpenDsException
*/
private void updateSchemaOnline() throws OpenDsException
{
for (AttributeType attr : attrsToAdd)
{
addAttributeOnline(attr);
appendNewLinesToProgress();
}
for (ObjectClass oc : ocsToAdd)
{
addObjectClassOnline(oc);
appendNewLinesToProgress();
}
}
private void appendNewLinesToProgress()
{
SwingUtilities.invokeLater(new Runnable()
{
@Override
public void run()
{
getProgressDialog().appendProgressHtml(Utilities.applyFont("<br><br>", ColorAndFontConstants.progressFont));
}
});
}
private void updateSchemaOffline() throws OpenDsException
{
// Group the changes in the same schema file.
final Map<String, List<AttributeType>> hmAttrs = copy(attrsToAdd);
final Map<String, List<ObjectClass>> hmOcs = copy(ocsToAdd);
final Set<String> allFileNames = new LinkedHashSet<>(hmAttrs.keySet());
allFileNames.addAll(hmOcs.keySet());
for (String fileName : allFileNames)
{
List<AttributeType> attrs = get(hmAttrs, fileName);
List<ObjectClass> ocs = get(hmOcs, fileName);
if ("".equals(fileName))
{
fileName = null;
}
updateSchemaOffline(fileName, attrs, ocs);
appendNewLinesToProgress();
}
}
private <T extends SchemaFileElement> List<T> get(Map<String, List<T>> hmElems, String fileName)
{
List<T> elems = hmElems.get(fileName);
if (elems != null)
{
return elems;
}
return Collections.emptyList();
}
private <T extends SchemaFileElement> Map<String, List<T>> copy(Set<T> elemsToAdd)
{
Map<String, List<T>> hmElems = new LinkedHashMap<>();
for (T elem : elemsToAdd)
{
String fileName = CommonSchemaElements.getSchemaFile(elem);
if (fileName == null)
{
fileName = "";
}
List<T> elems = hmElems.get(fileName);
if (elems == null)
{
elems = new ArrayList<>();
hmElems.put(fileName, elems);
}
elems.add(elem);
}
return hmElems;
}
private void addAttributeOnline(final AttributeType attribute) throws OpenDsException
{
addSchemaElementOnline(attribute, INFO_CTRL_PANEL_CREATING_ATTRIBUTE_PROGRESS.get(attribute.getNameOrOID()));
}
private void addObjectClassOnline(final ObjectClass objectClass) throws OpenDsException
{
addSchemaElementOnline(objectClass, INFO_CTRL_PANEL_CREATING_OBJECTCLASS_PROGRESS.get(objectClass.getNameOrOID()));
}
private void addSchemaElementOnline(final CommonSchemaElements schemaElement, final LocalizableMessage progressMsg)
throws OpenDsException
{
SwingUtilities.invokeLater(new Runnable()
{
@Override
public void run()
{
printEquivalentCommandLineToAddOnline(schemaElement);
getProgressDialog().appendProgressHtml(
Utilities.getProgressWithPoints(progressMsg, ColorAndFontConstants.progressFont));
}
});
try
{
final BasicAttribute attr = new BasicAttribute(getAttributeName(schemaElement));
attr.add(getValueOnline(schemaElement));
final ModificationItem mod = new ModificationItem(DirContext.ADD_ATTRIBUTE, attr);
getInfo().getDirContext().modifyAttributes(
ConfigConstants.DN_DEFAULT_SCHEMA_ROOT, new ModificationItem[] { mod });
}
catch (NamingException ne)
{
throw new OnlineUpdateException(ERR_CTRL_PANEL_ERROR_UPDATING_SCHEMA.get(ne), ne);
}
notifyConfigurationElementCreated(schemaElement);
SwingUtilities.invokeLater(new Runnable()
{
@Override
public void run()
{
getProgressDialog().appendProgressHtml(Utilities.getProgressDone(ColorAndFontConstants.progressFont));
}
});
}
private String getValueOnline(CommonSchemaElements element)
{
return element.toString();
}
private String getValueOffline(CommonSchemaElements element)
{
final Map<String, List<String>> props = element.getExtraProperties();
List<String> previousValues = props.get(ServerConstants.SCHEMA_PROPERTY_FILENAME);
setExtraProperty(element, ServerConstants.SCHEMA_PROPERTY_FILENAME, null);
String attributeWithoutFileDefinition = element.toString();
if (previousValues != null && !previousValues.isEmpty())
{
element.setExtraProperty(ServerConstants.SCHEMA_PROPERTY_FILENAME, new ArrayList<String>(previousValues));
}
return attributeWithoutFileDefinition;
}
private void printEquivalentCommandLineToAddOnline(CommonSchemaElements element)
{
List<String> args = new ArrayList<>();
args.add("-a");
args.addAll(getObfuscatedCommandLineArguments(getConnectionCommandLineArguments(true, true)));
args.add(getNoPropertiesFileArgument());
final String equivalentCmdLine = getEquivalentCommandLine(getCommandLinePath("ldapmodify"), args);
final String elementID = element.getNameOrOID();
final LocalizableMessage msg = element instanceof AttributeType ?
INFO_CTRL_PANEL_EQUIVALENT_CMD_TO_ADD_ATTRIBUTE_ONLINE.get(elementID)
: INFO_CTRL_PANEL_EQUIVALENT_CMD_TO_ADD_OBJECTCLASS_ONLINE.get(elementID);
final StringBuilder sb = new StringBuilder();
final String attName = getAttributeName(element);
sb.append(msg).append("<br><b>")
.append(equivalentCmdLine).append("<br>")
.append("dn: cn=schema<br>")
.append("changetype: modify<br>")
.append("add: ").append(attName).append("<br>")
.append(attName).append(": ").append(getValueOnline(element)).append("</b><br><br>");
getProgressDialog().appendProgressHtml(Utilities.applyFont(sb.toString(), ColorAndFontConstants.progressFont));
}
private String getAttributeName(CommonSchemaElements element)
{
return element instanceof AttributeType ? ConfigConstants.ATTR_ATTRIBUTE_TYPES : ConfigConstants.ATTR_OBJECTCLASSES;
}
private void updateSchemaOffline(
String file, final List<AttributeType> attributes, final List<ObjectClass> objectClasses) throws OpenDsException
{
final List<CommonSchemaElements> schemaElements = new ArrayList<CommonSchemaElements>(attributes);
schemaElements.addAll(objectClasses);
if (file == null)
{
file = ConfigConstants.FILE_USER_SCHEMA_ELEMENTS;
}
File f = new File(file);
if (!f.isAbsolute())
{
f = new File(DirectoryServer.getEnvironmentConfig().getSchemaDirectory(), file);
}
final String fileName = f.getAbsolutePath();
final boolean isSchemaFileDefined = isSchemaFileDefined(fileName);
SwingUtilities.invokeLater(new Runnable()
{
@Override
public void run()
{
final ProgressDialog progressDialog = getProgressDialog();
final String command = equivalentCommandToAddOffline(fileName, isSchemaFileDefined, schemaElements);
progressDialog.appendProgressHtml(Utilities.applyFont(command, ColorAndFontConstants.progressFont));
if (attributes.size() == 1 && objectClasses.isEmpty())
{
String attributeName = attributes.get(0).getNameOrOID();
progressDialog.appendProgressHtml(Utilities.getProgressWithPoints(
INFO_CTRL_PANEL_CREATING_ATTRIBUTE_PROGRESS.get(attributeName), ColorAndFontConstants.progressFont));
}
else if (objectClasses.size() == 1 && attributes.isEmpty())
{
String ocName = objectClasses.get(0).getNameOrOID();
progressDialog.appendProgressHtml(Utilities.getProgressWithPoints(
INFO_CTRL_PANEL_CREATING_OBJECTCLASS_PROGRESS.get(ocName), ColorAndFontConstants.progressFont));
}
else
{
progressDialog.appendProgressHtml(Utilities.getProgressWithPoints(
INFO_CTRL_PANEL_UPDATING_SCHEMA_FILE_PROGRESS.get(fileName), ColorAndFontConstants.progressFont));
}
}
});
if (isSchemaFileDefined)
{
updateSchemaFile(fileName, schemaElements);
}
else
{
updateSchemaUndefinedFile(fileName, schemaElements);
}
for (CommonSchemaElements schemaElement : schemaElements)
{
notifyConfigurationElementCreated(schemaElement);
}
SwingUtilities.invokeLater(new Runnable()
{
@Override
public void run()
{
getProgressDialog().appendProgressHtml(Utilities.getProgressDone(ColorAndFontConstants.progressFont));
}
});
}
private String equivalentCommandToAddOffline(
String schemaFile, boolean isSchemaFileDefined, List<CommonSchemaElements> schemaElements)
{
List<String> names = new ArrayList<>();
for (CommonSchemaElements schemaElement : schemaElements)
{
names.add(schemaElement.getNameOrOID());
}
final String namesString = joinAsString(", ", names);
final StringBuilder sb = new StringBuilder();
if (isSchemaFileDefined)
{
sb.append(INFO_CTRL_PANEL_EQUIVALENT_CMD_TO_ADD_SCHEMA_ELEMENT_OFFLINE.get(namesString, schemaFile))
.append("<br><b>");
}
else
{
sb.append(INFO_CTRL_PANEL_EQUIVALENT_CMD_TO_ADD_SCHEMA_ENTRY_OFFLINE.get(namesString, schemaFile))
.append("<br><b>");
for (String line : getSchemaEntryLines())
{
sb.append(line);
sb.append("<br>");
}
}
for (CommonSchemaElements schemaElement : schemaElements)
{
sb.append(getAttributeName(schemaElement)).append(": ").append(getValueOffline(schemaElement)).append("<br>");
}
sb.append("</b><br><br>");
return sb.toString();
}
/**
* Returns whether the file defined in the schema element exists or not.
*
* @param schemaFile
* the path to the schema file.
* @return <CODE>true</CODE> if the schema file is defined and
* <CODE>false</CODE> otherwise.
*/
private boolean isSchemaFileDefined(String schemaFile)
{
try (LDIFReader reader = new LDIFReader(new LDIFImportConfig(schemaFile)))
{
return reader.readEntry() != null;
}
catch (Throwable t)
{
return false;
}
}
/**
* Returns the list of LDIF lines that are enough to create the entry
* containing only the schema element associated with this task.
*
* @return the list of LDIF lines that are enough to create the entry
* containing only the schema element associated with this task.
*/
private List<String> getSchemaEntryLines()
{
List<String> lines = new ArrayList<>();
lines.add("dn: cn=schema");
lines.add("objectClass: top");
lines.add("objectClass: ldapSubentry");
lines.add("objectClass: subschema");
return lines;
}
/**
* Updates the contents of the schema file.
*
* @param schemaFile
* the schema file.
* @param isSchemaFileDefined
* whether the schema is defined or not.
* @param attributes
* the attributes to add.
* @param objectClasses
* the object classes to add.
* @throws OpenDsException
* if an error occurs updating the schema file.
*/
private void updateSchemaFile(String schemaFile, List<CommonSchemaElements> schemaElements)
throws OpenDsException
{
try (final LDIFExportConfig exportConfig = new LDIFExportConfig(schemaFile, ExistingFileBehavior.OVERWRITE))
{
try (final LDIFReader reader = new LDIFReader(new LDIFImportConfig(schemaFile)))
{
final Entry schemaEntry = reader.readEntry();
addElementsToEntry(schemaElements, schemaEntry);
try (final LDIFWriter writer = new LDIFWriter(exportConfig))
{
writer.writeEntry(schemaEntry);
exportConfig.getWriter().newLine();
}
}
catch (Throwable t)
{
throw new OfflineUpdateException(ERR_CTRL_PANEL_ERROR_UPDATING_SCHEMA.get(t), t);
}
}
}
private void addElementsToEntry(List<CommonSchemaElements> schemaElements, Entry schemaEntry)
throws DirectoryException
{
for (CommonSchemaElements schemaElement : schemaElements)
{
final Modification mod = new Modification(ModificationType.ADD,
Attributes.create(getAttributeName(schemaElement).toLowerCase(), getValueOffline(schemaElement)));
schemaEntry.applyModification(mod);
}
}
private void updateSchemaUndefinedFile(String schemaFile, List<CommonSchemaElements> schemaElements)
throws OfflineUpdateException
{
try (LDIFExportConfig exportConfig = new LDIFExportConfig(schemaFile, ExistingFileBehavior.FAIL))
{
List<String> lines = getSchemaEntryLines();
for (final CommonSchemaElements schemaElement : schemaElements)
{
lines.add(getAttributeName(schemaElement) + ": " + getValueOffline(schemaElement));
}
for (String line : lines)
{
final boolean wrapLines = exportConfig.getWrapColumn() > 1;
LDIFWriter.writeLDIFLine(
new StringBuilder(line), exportConfig.getWriter(), wrapLines, exportConfig.getWrapColumn());
}
exportConfig.getWriter().newLine();
}
catch (Throwable t)
{
throw new OfflineUpdateException(ERR_CTRL_PANEL_ERROR_UPDATING_SCHEMA.get(t), t);
}
}
}