/**
* 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.connection.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.jface.dialogs.MessageDialog;
import org.eclipse.swt.graphics.Image;
import org.eclipse.swt.dnd.Transfer;
import org.eclipse.ui.IWorkbenchActionConstants;
import org.eclipse.ui.IWorkbenchPage;
import org.eclipse.ui.IWorkbenchWindow;
import org.eclipse.ui.PlatformUI;
import org.safehaus.penrose.connection.ConnectionConfig;
import org.safehaus.penrose.connection.ConnectionClient;
import org.safehaus.penrose.connection.ConnectionManagerClient;
import org.safehaus.penrose.partition.PartitionClient;
import org.safehaus.penrose.partition.PartitionManagerClient;
import org.safehaus.penrose.client.PenroseClient;
import org.safehaus.penrose.studio.PenroseImage;
import org.safehaus.penrose.studio.PenroseStudio;
import org.safehaus.penrose.studio.dialog.ErrorDialog;
import org.safehaus.penrose.studio.connection.action.NewSourceAction;
import org.safehaus.penrose.studio.connection.editor.ConnectionEditorInput;
import org.safehaus.penrose.studio.connection.dnd.ConnectionTransfer;
import org.safehaus.penrose.studio.partition.tree.PartitionNode;
import org.safehaus.penrose.studio.partition.tree.PartitionsNode;
import org.safehaus.penrose.studio.plugin.Plugin;
import org.safehaus.penrose.studio.plugin.PluginManager;
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;
import java.util.ArrayList;
/**
* @author Endi S. Dewata
*/
public class ConnectionNode extends Node {
Logger log = Logger.getLogger(getClass());
private ServersView serversView;
private ServerNode serverNode;
private PartitionsNode partitionsNode;
private PartitionNode partitionNode;
private ConnectionsNode connectionsNode;
private String partitionName;
private String adapterName;
private String connectionName;
public ConnectionNode(String name, Image image, Object object, Node parent) {
super(name, image, object, parent);
connectionsNode = (ConnectionsNode)parent;
partitionNode = connectionsNode.getPartitionNode();
partitionsNode = partitionNode.getPartitionsNode();
serverNode = partitionsNode.getServerNode();
this.serversView = serverNode.getServersView();
}
public void showMenu(IMenuManager manager) {
manager.add(new Action("Open") {
public void run() {
try {
open();
} catch (Exception e) {
log.error(e.getMessage(), e);
ErrorDialog.open(e);
}
}
});
manager.add(new Separator(IWorkbenchActionConstants.MB_ADDITIONS));
manager.add(new NewSourceAction(this));
manager.add(new Separator(IWorkbenchActionConstants.MB_ADDITIONS));
manager.add(new Action("Copy") {
public void run() {
try {
copy();
} catch (Exception e) {
log.error(e.getMessage(), e);
ErrorDialog.open(e);
}
}
});
manager.add(new Action("Paste") {
public void run() {
try {
connectionsNode.paste();
} catch (Exception e) {
log.error(e.getMessage(), e);
ErrorDialog.open(e);
}
}
});
manager.add(new Action("Delete", PenroseStudio.getImageDescriptor(PenroseImage.DELETE_SMALL)) {
public void run() {
try {
remove();
} catch (Exception e) {
log.error(e.getMessage(), e);
ErrorDialog.open(e);
}
}
});
}
public void open() throws Exception {
PenroseStudio penroseStudio = PenroseStudio.getInstance();
PluginManager pluginManager = penroseStudio.getPluginManager();
Plugin plugin = pluginManager.getPlugin(adapterName);
ConnectionEditorInput ei = plugin.createConnectionEditorInput();
ei.setServer(serverNode.getServer());
ei.setPartitionName(partitionName);
ei.setConnectionName(connectionName);
String connectionEditorClass = plugin.getConnectionEditorClass();
log.debug("Opening "+connectionEditorClass);
IWorkbenchWindow window = PlatformUI.getWorkbench().getActiveWorkbenchWindow();
IWorkbenchPage page = window.getActivePage();
page.openEditor(ei, connectionEditorClass);
}
public void remove() throws Exception {
boolean confirm = MessageDialog.openQuestion(
serversView.getSite().getShell(),
"Confirmation", "Remove selected connections?");
if (!confirm) return;
Server server = serverNode.getServer();
PenroseClient client = server.getClient();
PartitionManagerClient partitionManagerClient = client.getPartitionManagerClient();
PartitionClient partitionClient = partitionManagerClient.getPartitionClient(partitionName);
ConnectionManagerClient connectionManagerClient = partitionClient.getConnectionManagerClient();
for (Node node : serversView.getSelectedNodes()) {
if (!(node instanceof ConnectionNode)) continue;
ConnectionNode connectionNode = (ConnectionNode)node;
connectionManagerClient.removeConnection(connectionNode.getConnectionName());
}
partitionClient.store();
ServersView serversView = ServersView.getInstance();
serversView.refresh(parent);
}
public void copy() throws Exception {
log.debug("Copying connections:");
Server server = serverNode.getServer();
PenroseClient client = server.getClient();
PartitionManagerClient partitionManagerClient = client.getPartitionManagerClient();
PartitionClient partitionClient = partitionManagerClient.getPartitionClient(partitionName);
ConnectionManagerClient connectionManagerClient = partitionClient.getConnectionManagerClient();
Collection<ConnectionConfig> list = new ArrayList<ConnectionConfig>();
for (Node node : serversView.getSelectedNodes()) {
if (!(node instanceof ConnectionNode)) continue;
ConnectionNode connectionNode = (ConnectionNode)node;
String connectionName = connectionNode.getConnectionName();
log.debug(" - "+connectionName);
ConnectionClient connectionClient = connectionManagerClient.getConnectionClient(connectionName);
ConnectionConfig connectionConfig = connectionClient.getConnectionConfig();
list.add(connectionConfig);
}
serversView.getSWTClipboard().setContents(
new Object[] { list.toArray(new ConnectionConfig[list.size()]) },
new Transfer[] { ConnectionTransfer.getInstance() }
);
}
public String getPartitionName() {
return partitionName;
}
public void setPartitionName(String partitionName) {
this.partitionName = partitionName;
}
public String getConnectionName() {
return connectionName;
}
public void setConnectionName(String connectionName) {
this.connectionName = connectionName;
}
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;
}
public ConnectionsNode getConnectionsNode() {
return connectionsNode;
}
public void setConnectionsNode(ConnectionsNode connectionsNode) {
this.connectionsNode = connectionsNode;
}
public String getAdapterName() {
return adapterName;
}
public void setAdapterName(String adapterName) {
this.adapterName = adapterName;
}
public boolean hasChildren() {
return false;
}
}