/**
* 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.source.tree;
import org.apache.log4j.Logger;
import org.eclipse.jface.action.Action;
import org.eclipse.jface.action.IMenuManager;
import org.eclipse.jface.action.Separator;
import org.eclipse.swt.graphics.Image;
import org.eclipse.ui.IWorkbenchActionConstants;
import org.safehaus.penrose.partition.PartitionClient;
import org.safehaus.penrose.partition.PartitionManagerClient;
import org.safehaus.penrose.client.PenroseClient;
import org.safehaus.penrose.source.SourceClient;
import org.safehaus.penrose.source.SourceConfig;
import org.safehaus.penrose.source.SourceManagerClient;
import org.safehaus.penrose.studio.PenroseImage;
import org.safehaus.penrose.studio.PenroseStudio;
import org.safehaus.penrose.studio.dialog.ErrorDialog;
import org.safehaus.penrose.studio.action.RefreshAction;
import org.safehaus.penrose.studio.source.dnd.SourceTransfer;
import org.safehaus.penrose.studio.nis.source.action.NewNISSourceAction;
import org.safehaus.penrose.studio.ldap.source.action.NewLDAPSourceAction;
import org.safehaus.penrose.studio.jdbc.source.action.NewJDBCSourceAction;
import org.safehaus.penrose.studio.partition.tree.PartitionNode;
import org.safehaus.penrose.studio.partition.tree.PartitionsNode;
import org.safehaus.penrose.studio.server.Server;
import org.safehaus.penrose.studio.server.tree.ServerNode;
import org.safehaus.penrose.studio.server.ServersView;
import org.safehaus.penrose.studio.tree.Node;
import java.util.Collection;
/**
* @author Endi S. Dewata
*/
public class SourcesNode extends Node {
Logger log = Logger.getLogger(getClass());
protected ServersView serversView;
protected ServerNode serverNode;
protected PartitionsNode partitionsNode;
protected PartitionNode partitionNode;
private String partitionName;
public SourcesNode(String name, Image image, Object object, Node parent) {
super(name, image, object, parent);
partitionNode = (PartitionNode)parent;
partitionsNode = partitionNode.getPartitionsNode();
serverNode = partitionsNode.getServerNode();
serversView = serverNode.getServersView();
}
public void update() throws Exception {
Server server = serverNode.getServer();
PenroseClient client = server.getClient();
PartitionManagerClient partitionManagerClient = client.getPartitionManagerClient();
PartitionClient partitionClient = partitionManagerClient.getPartitionClient(partitionName);
SourceManagerClient sourceManagerClient = partitionClient.getSourceManagerClient();
for (String sourceName : sourceManagerClient.getSourceNames()) {
SourceClient sourceClient = sourceManagerClient.getSourceClient(sourceName);
SourceNode sourceNode = new SourceNode(
sourceName,
PenroseStudio.getImage(PenroseImage.SOURCE),
sourceName,
this
);
sourceNode.setPartitionName(partitionName);
sourceNode.setAdapterName(sourceClient.getAdapterName());
sourceNode.setSourceName(sourceName);
addChild(sourceNode);
}
}
public void expand() throws Exception {
if (children == null) update();
}
public void refresh() throws Exception {
removeChildren();
update();
}
public void showMenu(IMenuManager manager) {
manager.add(new NewJDBCSourceAction(this));
manager.add(new NewLDAPSourceAction(this));
manager.add(new NewNISSourceAction(this));
manager.add(new Separator(IWorkbenchActionConstants.MB_ADDITIONS));
manager.add(new Action("Paste") {
public void run() {
try {
paste();
} catch (Exception e) {
log.error(e.getMessage(), e);
ErrorDialog.open(e);
}
}
});
manager.add(new Separator(IWorkbenchActionConstants.MB_ADDITIONS));
manager.add(new RefreshAction(this));
}
public void paste() throws Exception {
log.debug("Pasting sources:");
SourceConfig[] sourceConfigs = (SourceConfig[]) serversView.getSWTClipboard().getContents(SourceTransfer.getInstance());
if (sourceConfigs == null) return;
Server server = serverNode.getServer();
PenroseClient client = server.getClient();
PartitionManagerClient partitionManagerClient = client.getPartitionManagerClient();
PartitionClient partitionClient = partitionManagerClient.getPartitionClient(partitionName);
SourceManagerClient sourceManagerClient = partitionClient.getSourceManagerClient();
Collection<String> names = sourceManagerClient.getSourceNames();
for (SourceConfig sourceConfig : sourceConfigs) {
String name = sourceConfig.getName();
int counter = 1;
String newName = name;
while (names.contains(newName)) {
counter++;
newName = name+"_"+counter;
}
log.debug(" - "+name+" -> "+newName);
sourceConfig.setName(newName);
sourceManagerClient.createSource(sourceConfig);
}
partitionClient.store();
ServersView serversView = ServersView.getInstance();
serversView.refresh(this);
}
public String getPartitionName() {
return partitionName;
}
public void setPartitionName(String partitionName) {
this.partitionName = partitionName;
}
public ServersView getServersView() {
return serversView;
}
public void setServersView(ServersView serversView) {
this.serversView = serversView;
}
public ServerNode getServerNode() {
return serverNode;
}
public void setServerNode(ServerNode serverNode) {
this.serverNode = serverNode;
}
public PartitionsNode getPartitionsNode() {
return partitionsNode;
}
public void setPartitionsNode(PartitionsNode partitionsNode) {
this.partitionsNode = partitionsNode;
}
public PartitionNode getPartitionNode() {
return partitionNode;
}
public void setPartitionNode(PartitionNode partitionNode) {
this.partitionNode = partitionNode;
}
}