eclipse/plugins/org.argeo.slc.client.ui/src/org/argeo/slc/client/ui/views/ExecutionModulesView.java

   1 package org.argeo.slc.client.ui.views;
   2
   3 import java.io.ByteArrayOutputStream;
   4 import java.io.IOException;
   5 import java.util.ArrayList;
   6 import java.util.List;
   7 import java.util.Properties;
   8 import java.util.UUID;
   9
  10 import org.apache.commons.io.IOUtils;
  11 import org.argeo.slc.SlcException;
  12 import org.argeo.slc.client.oxm.OxmInterface;
  13 import org.argeo.slc.client.ui.ClientUiPlugin;
  14 import org.argeo.slc.client.ui.controllers.ProcessController;
  15 import org.argeo.slc.client.ui.providers.ExecutionModulesContentProvider;
  16 import org.argeo.slc.client.ui.providers.ExecutionModulesContentProvider.FlowNode;
  17 import org.argeo.slc.execution.ExecutionModuleDescriptor;
  18 import org.argeo.slc.process.RealizedFlow;
  19 import org.argeo.slc.process.SlcExecution;
  20 import org.eclipse.jface.viewers.DoubleClickEvent;
  21 import org.eclipse.jface.viewers.IContentProvider;
  22 import org.eclipse.jface.viewers.IDoubleClickListener;
  23 import org.eclipse.jface.viewers.IStructuredSelection;
  24 import org.eclipse.jface.viewers.ITableLabelProvider;
  25 import org.eclipse.jface.viewers.LabelProvider;
  26 import org.eclipse.jface.viewers.TreeViewer;
  27 import org.eclipse.swt.SWT;
  28 import org.eclipse.swt.dnd.DND;
  29 import org.eclipse.swt.dnd.DragSourceEvent;
  30 import org.eclipse.swt.dnd.DragSourceListener;
  31 import org.eclipse.swt.dnd.TextTransfer;
  32 import org.eclipse.swt.dnd.Transfer;
  33 import org.eclipse.swt.graphics.Image;
  34 import org.eclipse.swt.widgets.Composite;
  35 import org.eclipse.ui.ISharedImages;
  36 import org.eclipse.ui.PlatformUI;
  37 import org.eclipse.ui.part.ViewPart;
  38
  39 public class ExecutionModulesView extends ViewPart {
  40         // private final static Log log = LogFactory
  41         // .getLog(ExecutionModulesView.class);
  42
  43         public static final String ID = "org.argeo.slc.client.ui.executionModulesView";
  44
  45         private TreeViewer viewer;
  46
  47         // Ioc
  48         private IContentProvider contentProvider;
  49         private OxmInterface oxmBean;
  50         private ProcessController processController;
  51
  52         public void createPartControl(Composite parent) {
  53                 viewer = new TreeViewer(parent, SWT.MULTI | SWT.H_SCROLL | SWT.V_SCROLL);
  54                 viewer.setContentProvider(contentProvider);
  55                 viewer.setLabelProvider(new ViewLabelProvider());
  56                 viewer.setInput(getViewSite());
  57                 viewer.addDoubleClickListener(new ViewDoubleClickListener());
  58                 int operations = DND.DROP_COPY | DND.DROP_MOVE;
  59                 Transfer[] tt = new Transfer[] { TextTransfer.getInstance() };
  60                 viewer.addDragSupport(operations, tt, new ViewDragListener());
  61         }
  62
  63         public void setFocus() {
  64                 viewer.getControl().setFocus();
  65         }
  66
  67         public TreeViewer getViewer() {
  68                 return viewer;
  69         }
  70
  71         class ViewLabelProvider extends LabelProvider implements
  72                         ITableLabelProvider {
  73                 public String getColumnText(Object obj, int index) {
  74                         if (obj instanceof ExecutionModulesContentProvider.ExecutionModuleNode) {
  75                                 ExecutionModuleDescriptor emd = ((ExecutionModulesContentProvider.ExecutionModuleNode) obj)
  76                                                 .getDescriptor();
  77                                 if (emd.getLabel() != null)
  78                                         return emd.getLabel();
  79                                 else
  80                                         return getText(emd);
  81                         } else
  82                                 return getText(obj);
  83                 }
  84
  85                 public Image getColumnImage(Object obj, int index) {
  86                         return getImage(obj);
  87                 }
  88
  89                 public Image getImage(Object obj) {
  90                         if (obj instanceof ExecutionModulesContentProvider.AgentNode)
  91                                 return ClientUiPlugin.getDefault().getImageRegistry()
  92                                                 .get("agent");
  93                         else if (obj instanceof ExecutionModulesContentProvider.ExecutionModuleNode)
  94                                 return ClientUiPlugin.getDefault().getImageRegistry()
  95                                                 .get("executionModule");
  96                         else if (obj instanceof ExecutionModulesContentProvider.FolderNode)
  97                                 return ClientUiPlugin.getDefault().getImageRegistry()
  98                                                 .get("folder");
  99                         else if (obj instanceof ExecutionModulesContentProvider.FlowNode)
 100                                 return ClientUiPlugin.getDefault().getImageRegistry()
 101                                                 .get("flow");
 102                         else
 103                                 return PlatformUI.getWorkbench().getSharedImages()
 104                                                 .getImage(ISharedImages.IMG_OBJ_ELEMENT);
 105                 }
 106         }
 107
 108         class ViewDoubleClickListener implements IDoubleClickListener {
 109                 public void doubleClick(DoubleClickEvent evt) {
 110                         Object obj = ((IStructuredSelection) evt.getSelection())
 111                                         .getFirstElement();
 112                         if (obj instanceof ExecutionModulesContentProvider.FlowNode) {
 113                                 ExecutionModulesContentProvider.FlowNode fn = (ExecutionModulesContentProvider.FlowNode) obj;
 114
 115                                 List<RealizedFlow> realizedFlows = new ArrayList<RealizedFlow>();
 116                                 RealizedFlow realizedFlow = new RealizedFlow();
 117                                 realizedFlow.setModuleName(fn.getExecutionModuleNode()
 118                                                 .getDescriptor().getName());
 119                                 realizedFlow.setModuleVersion(fn.getExecutionModuleNode()
 120                                                 .getDescriptor().getVersion());
 121                                 realizedFlow.setFlowDescriptor(fn.getExecutionModuleNode()
 122                                                 .getFlowDescriptors().get(fn.getFlowName()));
 123                                 realizedFlows.add(realizedFlow);
 124
 125                                 SlcExecution slcExecution = new SlcExecution();
 126                                 slcExecution.setUuid(UUID.randomUUID().toString());
 127                                 slcExecution.setRealizedFlows(realizedFlows);
 128                                 slcExecution.setHost(fn.getExecutionModuleNode().getAgentNode()
 129                                                 .getAgent().toString());
 130                                 processController.execute(fn.getExecutionModuleNode()
 131                                                 .getAgentNode().getAgent(), slcExecution);
 132                         }
 133                 }
 134
 135         }
 136
 137         class ViewDragListener implements DragSourceListener {
 138
 139                 public void dragStart(DragSourceEvent event) {
 140                         System.out.println("Start Drag");
 141                 }
 142
 143                 public void dragSetData(DragSourceEvent event) {
 144                         System.out.println("dragSetData: " + event);
 145                         // System.out.println("dataType: " + event.dataType);
 146
 147                         IStructuredSelection selection = (IStructuredSelection) viewer
 148                                         .getSelection();
 149                         if (selection.getFirstElement() instanceof ExecutionModulesContentProvider.FlowNode) {
 150
 151                                 if (TextTransfer.getInstance().isSupportedType(event.dataType)) {
 152                                         ExecutionModulesContentProvider.FlowNode flowNode = (ExecutionModulesContentProvider.FlowNode) selection
 153                                                         .getFirstElement();
 154
 155                                         Properties props = new Properties();
 156                                         flowNodeAsProperties(props, flowNode);
 157                                         props.setProperty("agentId", flowNode
 158                                                         .getExecutionModuleNode().getAgentNode().getAgent()
 159                                                         .getAgentUuid());
 160                                         props.setProperty("host", flowNode.getExecutionModuleNode().getAgentNode()
 161                                                         .getAgent().toString());
 162
 163                                         ByteArrayOutputStream out = new ByteArrayOutputStream();
 164                                         try {
 165                                                 props.store(out, "");
 166                                                 event.data = new String(out.toByteArray());
 167                                         } catch (IOException e) {
 168                                                 throw new SlcException(
 169                                                                 "Cannot transform realized flow", e);
 170                                         } finally {
 171                                                 IOUtils.closeQuietly(out);
 172                                         }
 173                                 }
 174                         }
 175                 }
 176
 177                 public void dragFinished(DragSourceEvent event) {
 178                         System.out.println("Finished Drag");
 179                 }
 180
 181                 protected void flowNodeAsProperties(Properties props, FlowNode fn) {
 182
 183                         RealizedFlow realizedFlow = new RealizedFlow();
 184                         realizedFlow.setModuleName(fn.getExecutionModuleNode()
 185                                         .getDescriptor().getName());
 186                         realizedFlow.setModuleVersion(fn.getExecutionModuleNode()
 187                                         .getDescriptor().getVersion());
 188                         realizedFlow.setFlowDescriptor(fn.getExecutionFlowDescriptor());
 189
 190                         // As we want to have the effective ExecutionSpec and not a
 191                         // reference; we store it at the RealizeFlow level : thus the
 192                         // marshaller will store the object and not only a reference.
 193                         realizedFlow.setExecutionSpec(fn.getExecutionFlowDescriptor()
 194                                         .getExecutionSpec());
 195
 196                         props.setProperty("RealizedFlowAsXml",
 197                                         oxmBean.marshal(realizedFlow));
 198                         System.out
 199                                         .println(oxmBean.marshal(fn.getExecutionFlowDescriptor()));
 200
 201                 }
 202
 203         }
 204
 205         // IoC
 206         public void setContentProvider(IContentProvider contentProvider) {
 207                 this.contentProvider = contentProvider;
 208         }
 209
 210         public void setProcessController(ProcessController processController) {
 211                 this.processController = processController;
 212         }
 213
 214         public void setOxmBean(OxmInterface oxmBean) {
 215                 this.oxmBean = oxmBean;
 216         }
 217
 218 }