org.eclipse.linuxtools.tmf.ui/src/org/eclipse/linuxtools/internal/tmf/ui/project/handlers/SelectTraceTypeHandler.java

   1 /*******************************************************************************
   2  * Copyright (c) 2011, 2013 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  *   Francois Chouinard - Initial API and implementation
  11  *   Patrick Tasse - Fix propagation to experiment traces
  12  *******************************************************************************/
  13
  14 package org.eclipse.linuxtools.internal.tmf.ui.project.handlers;
  15
  16 import java.util.ArrayList;
  17 import java.util.Iterator;
  18 import java.util.List;
  19
  20 import org.eclipse.core.commands.AbstractHandler;
  21 import org.eclipse.core.commands.ExecutionEvent;
  22 import org.eclipse.core.commands.ExecutionException;
  23 import org.eclipse.core.resources.IProject;
  24 import org.eclipse.core.resources.IResource;
  25 import org.eclipse.core.runtime.CoreException;
  26 import org.eclipse.core.runtime.IStatus;
  27 import org.eclipse.core.runtime.MultiStatus;
  28 import org.eclipse.core.runtime.Status;
  29 import org.eclipse.jface.dialogs.ErrorDialog;
  30 import org.eclipse.jface.viewers.ISelection;
  31 import org.eclipse.jface.viewers.ISelectionProvider;
  32 import org.eclipse.jface.viewers.TreeSelection;
  33 import org.eclipse.linuxtools.internal.tmf.ui.Activator;
  34 import org.eclipse.linuxtools.tmf.core.TmfCommonConstants;
  35 import org.eclipse.linuxtools.tmf.core.trace.ITmfTrace;
  36 import org.eclipse.linuxtools.tmf.ui.project.model.ITmfProjectModelElement;
  37 import org.eclipse.linuxtools.tmf.ui.project.model.TmfExperimentFolder;
  38 import org.eclipse.linuxtools.tmf.ui.project.model.TmfTraceElement;
  39 import org.eclipse.linuxtools.tmf.ui.project.model.TmfTraceFolder;
  40 import org.eclipse.swt.widgets.Shell;
  41 import org.eclipse.ui.IWorkbenchPage;
  42 import org.eclipse.ui.IWorkbenchPart;
  43 import org.eclipse.ui.IWorkbenchWindow;
  44 import org.eclipse.ui.PlatformUI;
  45
  46 /**
  47  * <b><u>SetTraceTypeHandler</u></b>
  48  * <p>
  49  */
  50 public class SelectTraceTypeHandler extends AbstractHandler {
  51
  52     // ------------------------------------------------------------------------
  53     // Constants
  54     // ------------------------------------------------------------------------
  55
  56     private static final String BUNDLE_PARAMETER = "org.eclipse.linuxtools.tmf.ui.commandparameter.select_trace_type.bundle"; //$NON-NLS-1$
  57     private static final String TYPE_PARAMETER = "org.eclipse.linuxtools.tmf.ui.commandparameter.select_trace_type.type"; //$NON-NLS-1$
  58     private static final String ICON_PARAMETER = "org.eclipse.linuxtools.tmf.ui.commandparameter.select_trace_type.icon"; //$NON-NLS-1$
  59
  60     // ------------------------------------------------------------------------
  61     // Attributes
  62     // ------------------------------------------------------------------------
  63
  64     private TreeSelection fSelection = null;
  65
  66     // ------------------------------------------------------------------------
  67     // Validation
  68     // ------------------------------------------------------------------------
  69
  70     @Override
  71     public boolean isEnabled() {
  72
  73         // Check if we are closing down
  74         IWorkbenchWindow window = PlatformUI.getWorkbench().getActiveWorkbenchWindow();
  75         if (window == null) {
  76             return false;
  77         }
  78
  79         // Get the selection
  80         IWorkbenchPage page = PlatformUI.getWorkbench().getActiveWorkbenchWindow().getActivePage();
  81         IWorkbenchPart part = page.getActivePart();
  82         if (part == null) {
  83             return false;
  84         }
  85         ISelectionProvider selectionProvider = part.getSite().getSelectionProvider();
  86         if (selectionProvider == null) {
  87             return false;
  88         }
  89         ISelection selection = selectionProvider.getSelection();
  90
  91         // Make sure selection contains only traces
  92         fSelection = null;
  93         if (selection instanceof TreeSelection) {
  94             fSelection = (TreeSelection) selection;
  95             Iterator<Object> iterator = fSelection.iterator();
  96             while (iterator.hasNext()) {
  97                 Object element = iterator.next();
  98                 if (!(element instanceof TmfTraceElement)) {
  99                     return false;
 100                 }
 101             }
 102         }
 103
 104         // If we get here, either nothing is selected or everything is a trace
 105         return !selection.isEmpty();
 106     }
 107
 108     // ------------------------------------------------------------------------
 109     // Execution
 110     // ------------------------------------------------------------------------
 111
 112     @Override
 113     public Object execute(ExecutionEvent event) throws ExecutionException {
 114
 115         // Check if we are closing down
 116         IWorkbenchWindow window = PlatformUI.getWorkbench().getActiveWorkbenchWindow();
 117         if (window == null) {
 118             return null;
 119         }
 120         List<IStatus> statuses = new ArrayList<IStatus>();
 121         boolean ok = true;
 122         for (Object element : fSelection.toList()) {
 123             TmfTraceElement trace = (TmfTraceElement) element;
 124             trace = trace.getElementUnderTraceFolder();
 125             IResource resource = trace.getResource();
 126             if (resource != null) {
 127                 try {
 128                     // Set the properties for this resource
 129                     String bundleName = event.getParameter(BUNDLE_PARAMETER);
 130                     String traceType = event.getParameter(TYPE_PARAMETER);
 131                     String iconUrl = event.getParameter(ICON_PARAMETER);
 132                     IStatus status = propagateProperties(trace, bundleName, traceType, iconUrl);
 133                     ok &= status.isOK();
 134                     if (!status.isOK()) {
 135                         statuses.add(status);
 136                     }
 137                 } catch (CoreException e) {
 138                     Activator.getDefault().logError(Messages.SelectTraceTypeHandler_ErrorSelectingTrace + trace.getName(), e);
 139                 }
 140             }
 141         }
 142         ((ITmfProjectModelElement) fSelection.getFirstElement()).getProject().refresh();
 143
 144         if (!ok) {
 145             final Shell shell = window.getShell();
 146             MultiStatus info = new MultiStatus(Activator.PLUGIN_ID, 1, Messages.SelectTraceTypeHandler_TraceFailedValidation, null);
 147             if (statuses.size() > 1)
 148             {
 149                 info = new MultiStatus(Activator.PLUGIN_ID, 1, Messages.SelectTraceTypeHandler_TracesFailedValidation, null);
 150             }
 151             for (IStatus status : statuses) {
 152                 info.add(status);
 153             }
 154             ErrorDialog.openError(shell, Messages.SelectTraceTypeHandler_Title, Messages.SelectTraceTypeHandler_InvalidTraceType, info);
 155         }
 156         return null;
 157     }
 158
 159     private static IStatus propagateProperties(TmfTraceElement trace,
 160             String bundleName, String traceType, String iconUrl)
 161             throws CoreException {
 162
 163         IResource svResource = trace.getResource();
 164         String svBundleName = svResource.getPersistentProperty(TmfCommonConstants.TRACEBUNDLE);
 165         String svTraceType = svResource.getPersistentProperty(TmfCommonConstants.TRACETYPE);
 166         String svIconUrl = svResource.getPersistentProperty(TmfCommonConstants.TRACEICON);
 167
 168         setProperties(trace.getResource(), bundleName, traceType, iconUrl);
 169         trace.refreshTraceType();
 170         final IStatus validateTraceType = validateTraceType(trace);
 171         if (!validateTraceType.isOK()) {
 172             setProperties(trace.getResource(), svBundleName, svTraceType, svIconUrl);
 173             trace.refreshTraceType();
 174             return validateTraceType;
 175         }
 176
 177         trace.refreshTraceType();
 178
 179         if (trace.getParent() instanceof TmfTraceFolder) {
 180             TmfExperimentFolder experimentFolder = trace.getProject().getExperimentsFolder();
 181             for (final ITmfProjectModelElement experiment : experimentFolder.getChildren()) {
 182                 for (final ITmfProjectModelElement child : experiment.getChildren()) {
 183                     if (child instanceof TmfTraceElement) {
 184                         TmfTraceElement linkedTrace = (TmfTraceElement) child;
 185                         if (linkedTrace.getName().equals(trace.getName())) {
 186                             IResource resource = linkedTrace.getResource();
 187                             setProperties(resource, bundleName, traceType, iconUrl);
 188                             linkedTrace.refreshTraceType();
 189                         }
 190                     }
 191                 }
 192             }
 193         }
 194
 195         return Status.OK_STATUS;
 196     }
 197
 198     private static void setProperties(IResource resource, String bundleName,
 199             String traceType, String iconUrl) throws CoreException {
 200         resource.setPersistentProperty(TmfCommonConstants.TRACEBUNDLE, bundleName);
 201         resource.setPersistentProperty(TmfCommonConstants.TRACETYPE, traceType);
 202         resource.setPersistentProperty(TmfCommonConstants.TRACEICON, iconUrl);
 203     }
 204
 205     private static IStatus validateTraceType(TmfTraceElement trace) {
 206         IProject project = trace.getProject().getResource();
 207         ITmfTrace tmfTrace = null;
 208         IStatus validate = null;
 209         try {
 210             tmfTrace = trace.instantiateTrace();
 211             if (tmfTrace != null) {
 212                 validate = tmfTrace.validate(project, trace.getLocation().getPath());
 213             }
 214             else{
 215                 validate =  new Status(IStatus.ERROR, trace.getName(), "File does not exist : " + trace.getLocation().getPath()); //$NON-NLS-1$
 216             }
 217         } finally {
 218             if (tmfTrace != null) {
 219                 tmfTrace.dispose();
 220             }
 221         }
 222         if (validate == null) {
 223             validate = new Status(IStatus.ERROR, "unknown", "unknown"); //$NON-NLS-1$ //$NON-NLS-2$
 224         }
 225         return validate;
 226     }
 227
 228 }