/**
* Copyright 2009 Red Hat, Inc.
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* This program 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 General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
*/
package org.safehaus.penrose.studio.directory.wizard;
import org.apache.log4j.Logger;
import org.eclipse.jface.wizard.IWizardPage;
import org.eclipse.jface.wizard.Wizard;
import org.safehaus.penrose.directory.*;
import org.safehaus.penrose.ldap.DN;
import org.safehaus.penrose.ldap.DNBuilder;
import org.safehaus.penrose.ldap.RDNBuilder;
import org.safehaus.penrose.partition.PartitionClient;
import org.safehaus.penrose.partition.PartitionManagerClient;
import org.safehaus.penrose.client.PenroseClient;
import org.safehaus.penrose.mapping.Relationship;
import org.safehaus.penrose.studio.server.Server;
import org.safehaus.penrose.studio.attribute.wizard.AttributesWizardPage;
import org.safehaus.penrose.studio.dialog.ErrorDialog;
import java.util.Collection;
/**
* @author Endi S. Dewata
*/
public class DynamicEntryWizard extends Wizard {
Logger log = Logger.getLogger(getClass());
private Server server;
private String partitionName;
private DN parentDn;
private EntryConfig entryConfig;
public EntrySourceWizardPage sourcesPage;
public RelationshipWizardPage relationsPage;
public ObjectClassWizardPage objectClassesPage;
public AttributesWizardPage attributesPage;
public DynamicEntryWizard() {
setWindowTitle("Adding dynamic entry");
}
public void addPages() {
sourcesPage = new EntrySourceWizardPage();
sourcesPage.setDescription("Add data sources to be mapped into this entry.");
sourcesPage.setServer(server);
sourcesPage.setPartitionName(partitionName);
addPage(sourcesPage);
relationsPage = new RelationshipWizardPage(server, partitionName);
addPage(relationsPage);
objectClassesPage = new ObjectClassWizardPage();
objectClassesPage.setServer(server);
addPage(objectClassesPage);
attributesPage = new AttributesWizardPage();
attributesPage.setServer(server);
attributesPage.setPartitionName(partitionName);
attributesPage.setDefaultType(AttributesWizardPage.VARIABLE);
addPage(attributesPage);
}
public boolean canFinish() {
if (!sourcesPage.isPageComplete()) return false;
if (!relationsPage.isPageComplete()) return false;
if (!objectClassesPage.isPageComplete()) return false;
if (!attributesPage.isPageComplete()) return false;
return true;
}
public IWizardPage getNextPage(IWizardPage page) {
if (sourcesPage == page) {
Collection<EntrySourceConfig> sourceConfigs = sourcesPage.getEntrySourceConfigs();
relationsPage.setSourceMappings(sourceConfigs);
attributesPage.setEntrySourceConfigs(sourceConfigs);
if (sourceConfigs.size() < 2) return objectClassesPage;
} else if (objectClassesPage == page) {
Collection<String> objectClasses = objectClassesPage.getSelectedObjectClasses();
attributesPage.setObjectClasses(objectClasses);
}
return super.getNextPage(page);
}
public boolean performFinish() {
try {
entryConfig.setEntryClass("org.safehaus.penrose.directory.DynamicEntry");
entryConfig.addSourceConfigs(sourcesPage.getEntrySourceConfigs());
Collection<Relationship> relationships = relationsPage.getRelationships();
for (Relationship relationship : relationships) {
String lfield = relationship.getLeftField();
EntrySourceConfig lsource = entryConfig.getSourceConfig(relationship.getLeftSource());
int lindex = entryConfig.getSourceConfigIndex(lsource);
String rfield = relationship.getRightField();
EntrySourceConfig rsource = entryConfig.getSourceConfig(relationship.getRightSource());
int rindex = entryConfig.getSourceConfigIndex(rsource);
if (lindex < rindex) { // rhs is dependent on lhs
rsource.addFieldConfig(new EntryFieldConfig(rfield, EntryFieldConfig.VARIABLE, relationship.getLhs()));
} else {
lsource.addFieldConfig(new EntryFieldConfig(lfield, EntryFieldConfig.VARIABLE, relationship.getRhs()));
}
}
entryConfig.addObjectClasses(objectClassesPage.getSelectedObjectClasses());
log.debug("Attribute mappings:");
Collection<EntryAttributeConfig> attributeMappings = attributesPage.getAttributeConfigs();
entryConfig.addAttributeConfigs(attributeMappings);
RDNBuilder rb = new RDNBuilder();
for (EntryAttributeConfig attributeMapping : attributeMappings) {
if (!attributeMapping.isRdn()) continue;
rb.set(attributeMapping.getName(), "...");
}
DNBuilder db = new DNBuilder();
db.append(rb.toRdn());
db.append(parentDn);
entryConfig.setDn(db.toDn());
log.debug("Reverse mappings:");
for (EntryAttributeConfig attributeMapping : entryConfig.getAttributeConfigs()) {
String name = attributeMapping.getName();
String variable = attributeMapping.getVariable();
if (variable == null) {
log.debug("Attribute " + name + " can't be reverse mapped.");
continue;
}
int j = variable.indexOf(".");
String sourceName = variable.substring(0, j);
String fieldName = variable.substring(j + 1);
EntrySourceConfig sourceMapping = entryConfig.getSourceConfig(sourceName);
Collection fieldMappings = sourceMapping.getFieldConfigs(fieldName);
if (fieldMappings != null && !fieldMappings.isEmpty()) {
log.debug("Attribute " + name + " has been reverse mapped.");
continue;
}
log.debug(" - " + sourceName + "." + fieldName + " <= " + name);
EntryFieldConfig fieldMapping = new EntryFieldConfig(fieldName, EntryFieldConfig.VARIABLE, attributeMapping.getName());
sourceMapping.addFieldConfig(fieldMapping);
}
PenroseClient client = server.getClient();
PartitionManagerClient partitionManagerClient = client.getPartitionManagerClient();
PartitionClient partitionClient = partitionManagerClient.getPartitionClient(partitionName);
DirectoryClient directoryClient = partitionClient.getDirectoryClient();
directoryClient.createEntry(entryConfig);
partitionClient.store();
return true;
} catch (Exception e) {
log.error(e.getMessage(), e);
ErrorDialog.open(e);
return false;
}
}
public boolean needsPreviousAndNextButtons() {
return true;
}
public DN getParentDn() {
return parentDn;
}
public void setParentDn(DN parentDn) {
this.parentDn = parentDn;
}
public EntryConfig getEntryConfig() {
return entryConfig;
}
public void setEntryConfig(EntryConfig entryConfig) {
this.entryConfig = entryConfig;
}
public String getPartitionName() {
return partitionName;
}
public void setPartitionName(String partitionName) {
this.partitionName = partitionName;
}
public Server getServer() {
return server;
}
public void setServer(Server server) {
this.server = server;
}
}