1 /**********************************************************************
2 * Copyright (c) 2012 Ericsson
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
10 * Bernd Hufmann - Initial API and implementation
11 **********************************************************************/
12 package org
.eclipse
.linuxtools
.internal
.lttng2
.ui
.views
.control
.handlers
;
14 import java
.util
.Iterator
;
15 import java
.util
.List
;
17 import org
.eclipse
.core
.commands
.ExecutionEvent
;
18 import org
.eclipse
.core
.commands
.ExecutionException
;
19 import org
.eclipse
.core
.resources
.IFolder
;
20 import org
.eclipse
.core
.resources
.IProject
;
21 import org
.eclipse
.core
.runtime
.CoreException
;
22 import org
.eclipse
.core
.runtime
.IProgressMonitor
;
23 import org
.eclipse
.core
.runtime
.IStatus
;
24 import org
.eclipse
.core
.runtime
.NullProgressMonitor
;
25 import org
.eclipse
.core
.runtime
.Status
;
26 import org
.eclipse
.core
.runtime
.jobs
.Job
;
27 import org
.eclipse
.jface
.viewers
.ISelection
;
28 import org
.eclipse
.jface
.viewers
.StructuredSelection
;
29 import org
.eclipse
.jface
.window
.Window
;
30 import org
.eclipse
.linuxtools
.internal
.lttng2
.core
.control
.model
.TraceSessionState
;
31 import org
.eclipse
.linuxtools
.internal
.lttng2
.ui
.Activator
;
32 import org
.eclipse
.linuxtools
.internal
.lttng2
.ui
.views
.control
.ControlView
;
33 import org
.eclipse
.linuxtools
.internal
.lttng2
.ui
.views
.control
.dialogs
.IImportDialog
;
34 import org
.eclipse
.linuxtools
.internal
.lttng2
.ui
.views
.control
.dialogs
.ImportFileInfo
;
35 import org
.eclipse
.linuxtools
.internal
.lttng2
.ui
.views
.control
.dialogs
.TraceControlDialogFactory
;
36 import org
.eclipse
.linuxtools
.internal
.lttng2
.ui
.views
.control
.messages
.Messages
;
37 import org
.eclipse
.linuxtools
.internal
.lttng2
.ui
.views
.control
.model
.impl
.TraceSessionComponent
;
38 import org
.eclipse
.linuxtools
.tmf
.ui
.project
.model
.TmfTraceFolder
;
39 import org
.eclipse
.rse
.services
.clientserver
.messages
.SystemMessageException
;
40 import org
.eclipse
.rse
.services
.files
.IFileService
;
41 import org
.eclipse
.rse
.subsystems
.files
.core
.subsystems
.IRemoteFile
;
42 import org
.eclipse
.rse
.subsystems
.files
.core
.subsystems
.IRemoteFileSubSystem
;
43 import org
.eclipse
.ui
.IWorkbenchPage
;
44 import org
.eclipse
.ui
.IWorkbenchWindow
;
45 import org
.eclipse
.ui
.PlatformUI
;
49 * Command handler implementation to import traces from a (remote) session to a tracing project.
52 * @author Bernd Hufmann
54 public class ImportHandler
extends BaseControlViewHandler
{
56 // ------------------------------------------------------------------------
58 // ------------------------------------------------------------------------
61 * The command parameter
63 protected CommandParameter fParam
;
65 // ------------------------------------------------------------------------
67 // ------------------------------------------------------------------------
69 // ------------------------------------------------------------------------
71 // ------------------------------------------------------------------------
75 * @see org.eclipse.core.commands.AbstractHandler#execute(org.eclipse.core.commands.ExecutionEvent)
78 public Object
execute(ExecutionEvent event
) throws ExecutionException
{
80 IWorkbenchWindow window
= PlatformUI
.getWorkbench().getActiveWorkbenchWindow();
88 final CommandParameter param
= fParam
.clone();
90 final IImportDialog dialog
= TraceControlDialogFactory
.getInstance().getImportDialog();
91 dialog
.setSession(param
.getSession());
93 if ((dialog
.open() != Window
.OK
) || param
.getSession().isStreamedTrace()) {
97 Job job
= new Job(Messages
.TraceControl_ImportJob
) {
99 protected IStatus
run(IProgressMonitor monitor
) {
101 List
<ImportFileInfo
> traces
= dialog
.getTracePathes();
102 IProject project
= dialog
.getProject();
104 for (Iterator
<ImportFileInfo
> iterator
= traces
.iterator(); iterator
.hasNext();) {
105 ImportFileInfo remoteFile
= iterator
.next();
106 downloadTrace(remoteFile
, project
);
109 } catch (ExecutionException e
) {
110 return new Status(IStatus
.ERROR
, Activator
.PLUGIN_ID
, Messages
.TraceControl_ImportFailure
, e
);
112 return Status
.OK_STATUS
;
125 * @see org.eclipse.core.commands.AbstractHandler#isEnabled()
128 public boolean isEnabled() {
129 // Get workbench page for the Control View
130 IWorkbenchPage page
= getWorkbenchPage();
135 // Check if one or more session are selected
136 ISelection selection
= page
.getSelection(ControlView
.ID
);
137 TraceSessionComponent session
= null;
138 if (selection
instanceof StructuredSelection
) {
139 StructuredSelection structered
= ((StructuredSelection
) selection
);
140 for (Iterator
<?
> iterator
= structered
.iterator(); iterator
.hasNext();) {
141 Object element
= iterator
.next();
142 if (element
instanceof TraceSessionComponent
) {
143 // Add only TraceSessionComponents that are inactive and not destroyed
144 TraceSessionComponent tmpSession
= (TraceSessionComponent
) element
;
145 if ((tmpSession
.getSessionState() == TraceSessionState
.INACTIVE
) && (!tmpSession
.isDestroyed())) {
146 session
= tmpSession
;
151 boolean isEnabled
= session
!= null;
157 fParam
= new CommandParameter(session
);
165 // ------------------------------------------------------------------------
167 // ------------------------------------------------------------------------
169 * Downloads a trace from the remote host to the given project.
172 * - trace information of trace to import
174 * - project to import to
175 * @throws ExecutionException
177 private static void downloadTrace(ImportFileInfo trace
, IProject project
)
178 throws ExecutionException
{
180 IRemoteFileSubSystem fsss
= trace
.getImportFile().getParentRemoteFileSubSystem();
182 IFolder traceFolder
= project
.getFolder(TmfTraceFolder
.TRACE_FOLDER_NAME
);
183 if (!traceFolder
.exists()) {
184 throw new ExecutionException(Messages
.TraceControl_ImportDialogInvalidTracingProject
+ " (" + TmfTraceFolder
.TRACE_FOLDER_NAME
+ ")"); //$NON-NLS-1$//$NON-NLS-2$
187 String traceName
= trace
.getLocalTraceName();
188 IFolder folder
= traceFolder
.getFolder(traceName
);
189 if (folder
.exists()) {
190 if(!trace
.isOverwrite()) {
191 throw new ExecutionException(Messages
.TraceControl_ImportDialogTraceAlreadyExistError
+ ": " + traceName
); //$NON-NLS-1$
194 folder
.create(true, true, null);
197 IRemoteFile
[] sources
= fsss
.list(trace
.getImportFile(), IFileService
.FILE_TYPE_FILES
, new NullProgressMonitor());
199 String
[] destinations
= new String
[sources
.length
];
200 String
[] encodings
= new String
[sources
.length
];
201 for (int i
= 0; i
< sources
.length
; i
++) {
202 destinations
[i
] = folder
.getLocation().addTrailingSeparator().append(sources
[i
].getName()).toString();
206 fsss
.downloadMultiple(sources
, destinations
, encodings
, new NullProgressMonitor());
208 } catch (SystemMessageException e
) {
209 throw new ExecutionException(e
.toString(), e
);
210 } catch (CoreException e
) {
211 throw new ExecutionException(e
.toString(), e
);