org.eclipse.linuxtools.lttng2.ui/src/org/eclipse/linuxtools/internal/lttng2/ui/views/control/handlers/NewConnectionHandler.java

   1 /**********************************************************************
   2  * Copyright (c) 2012 Ericsson
   3  *
   4  * All rights reserved. This program and the accompanying materials are
   5  * made available under the terms of the Eclipse Public License v1.0 which
   6  * accompanies this distribution, and is available at
   7  * http://www.eclipse.org/legal/epl-v10.html
   8  *
   9  * Contributors:
  10  *   Bernd Hufmann - Initial API and implementation
  11  **********************************************************************/
  12 package org.eclipse.linuxtools.internal.lttng2.ui.views.control.handlers;
  13
  14 import org.eclipse.core.commands.ExecutionEvent;
  15 import org.eclipse.core.commands.ExecutionException;
  16 import org.eclipse.jface.dialogs.MessageDialog;
  17 import org.eclipse.jface.window.Window;
  18 import org.eclipse.linuxtools.internal.lttng2.ui.views.control.ControlView;
  19 import org.eclipse.linuxtools.internal.lttng2.ui.views.control.Messages;
  20 import org.eclipse.linuxtools.internal.lttng2.ui.views.control.dialogs.INewConnectionDialog;
  21 import org.eclipse.linuxtools.internal.lttng2.ui.views.control.dialogs.TraceControlDialogFactory;
  22 import org.eclipse.linuxtools.internal.lttng2.ui.views.control.model.ITraceControlComponent;
  23 import org.eclipse.linuxtools.internal.lttng2.ui.views.control.model.impl.TargetNodeComponent;
  24 import org.eclipse.rse.core.IRSESystemType;
  25 import org.eclipse.rse.core.RSECorePlugin;
  26 import org.eclipse.rse.core.model.IHost;
  27 import org.eclipse.rse.core.model.ISystemRegistry;
  28 import org.eclipse.ui.IWorkbenchPage;
  29 import org.eclipse.ui.IWorkbenchPart;
  30 import org.eclipse.ui.IWorkbenchWindow;
  31 import org.eclipse.ui.PlatformUI;
  32
  33 /**
  34  * <b><u>NewConnectionHandler</u></b>
  35  * <p>
  36  * Command handler for creation new connection for trace control.
  37  * </p>
  38  */
  39 public class NewConnectionHandler extends BaseControlViewHandler {
  40
  41     // ------------------------------------------------------------------------
  42     // Constants
  43     // ------------------------------------------------------------------------
  44     /**
  45      * The trace control system type defined for LTTng version 2.0 and later.
  46      */
  47     public final static String TRACE_CONTROL_SYSTEM_TYPE = "org.eclipse.linuxtools.internal.lttng2.ui.control.systemType"; //$NON-NLS-1$
  48
  49     // ------------------------------------------------------------------------
  50     // Attributes
  51     // ------------------------------------------------------------------------
  52     /**
  53      * The parent trace control component the new node will be added to.
  54      */
  55     private ITraceControlComponent fRoot = null;
  56
  57     /*
  58      * (non-Javadoc)
  59      * @see org.eclipse.core.commands.AbstractHandler#execute(org.eclipse.core.commands.ExecutionEvent)
  60      */
  61     @Override
  62     public Object execute(ExecutionEvent event) throws ExecutionException {
  63         assert (fRoot != null);
  64
  65         IWorkbenchWindow window = PlatformUI.getWorkbench().getActiveWorkbenchWindow();
  66         if (window == null) {
  67             return false;
  68         }
  69
  70         ISystemRegistry registry = RSECorePlugin.getTheSystemRegistry();
  71
  72         // get system type definition for LTTng 2.x connection
  73         IRSESystemType sysType = RSECorePlugin.getTheCoreRegistry().getSystemTypeById(TRACE_CONTROL_SYSTEM_TYPE);
  74
  75         // get all hosts for this system type
  76         IHost[] hosts = registry.getHostsBySystemType(sysType);
  77
  78         // Open dialog box for the node name and address
  79         final INewConnectionDialog dialog = TraceControlDialogFactory.getInstance().getNewConnectionDialog();
  80         dialog.setTraceControlParent(fRoot);
  81         dialog.setHosts(hosts);
  82
  83         if (dialog.open() != Window.OK) {
  84             return null;
  85         }
  86
  87         String hostName = dialog.getConnectionName();
  88         String hostAddress = dialog.getHostName();
  89
  90         // get the singleton RSE registry
  91         IHost host = null;
  92
  93         for (int i = 0; i < hosts.length; i++) {
  94             if (hosts[i].getAliasName().equals(hostName)) {
  95                 host = hosts[i];
  96                 break;
  97             }
  98         }
  99
 100         if (host == null) {
 101             // if there's no host then we will create it
 102             try {
 103                 // create the host object as an SSH Only connection
 104                 host = registry.createHost(
 105                         sysType,       //System Type Name
 106                         hostName,      //Connection name
 107                         hostAddress,   //IP Address
 108                         "Connection to Host"); //description //$NON-NLS-1$
 109             }
 110             catch (Exception e) {
 111                 MessageDialog.openError(window.getShell(),
 112                         Messages.TraceControl_EclipseCommandFailure,
 113                         Messages.TraceControl_NewNodeCreationFailure + " (" + hostName + ", " + hostAddress + ")" + ":\n" + e.toString());  //$NON-NLS-1$ //$NON-NLS-2$ //$NON-NLS-3$ //$NON-NLS-4$
 114                 return null;
 115             }
 116         }
 117
 118         if (host != null) {
 119             // successful creation of host
 120             TargetNodeComponent node = null;
 121             if (!fRoot.containsChild(hostName)) {
 122                 node = new TargetNodeComponent(hostName, fRoot, host);
 123                 fRoot.addChild(node);
 124             }
 125             else {
 126                 node = (TargetNodeComponent)fRoot.getChild(hostName);
 127             }
 128
 129             node.connect();
 130         }
 131         return null;
 132     }
 133
 134
 135     /*
 136      * (non-Javadoc)
 137      * @see org.eclipse.core.commands.AbstractHandler#isEnabled()
 138      */
 139     @Override
 140     public boolean isEnabled() {
 141
 142         // Get workbench page for the Control View
 143         IWorkbenchPage page = getWorkbenchPage();
 144         if (page == null) {
 145             return false;
 146         }
 147
 148         fRoot = null;
 149
 150         // no need to verify part because it has been already done in getWorkbenchPage()
 151         IWorkbenchPart part = page.getActivePart();
 152         fRoot = ((ControlView) part).getTraceControlRoot();
 153
 154         return (fRoot != null);
 155     }
 156 }