Refactoring of diagram element selection method to utils.

[simantics/district.git] / org.simantics.district.route.ui / src / org / simantics / district / route / ui / handlers / SelectRouteOnDiagram.java

diff --git a/org.simantics.district.route.ui/src/org/simantics/district/route/ui/handlers/SelectRouteOnDiagram.java b/org.simantics.district.route.ui/src/org/simantics/district/route/ui/handlers/SelectRouteOnDiagram.java
index 7568ea308d7b6e975cb3564ee28c6674092e5355..962c560a85cf28c064c1d1f68a6d570b548ffa20 100644 (file)
--- a/org.simantics.district.route.ui/src/org/simantics/district/route/ui/handlers/SelectRouteOnDiagram.java
+++ b/org.simantics.district.route.ui/src/org/simantics/district/route/ui/handlers/SelectRouteOnDiagram.java
@@ -1,26 +1,40 @@
 package org.simantics.district.route.ui.handlers;
 
+import java.util.Collections;
 import java.util.List;
 import java.util.concurrent.CompletableFuture;
 
 import javax.inject.Named;
 
+import org.eclipse.e4.core.di.annotations.CanExecute;
 import org.eclipse.e4.core.di.annotations.Execute;
 import org.eclipse.e4.ui.model.application.ui.basic.MPart;
 import org.eclipse.e4.ui.services.IServiceConstants;
 import org.eclipse.jface.viewers.ISelection;
+import org.eclipse.swt.widgets.Display;
 import org.eclipse.swt.widgets.Shell;
-import org.simantics.db.layer0.variable.Variable;
+import org.simantics.db.Resource;
+import org.simantics.db.exception.DatabaseException;
+import org.simantics.district.network.ui.DistrictNetworkUIUtil;
 import org.simantics.district.route.Route;
 import org.simantics.district.route.RouteJob;
 import org.simantics.district.route.RouterConfiguration;
 import org.simantics.utils.ui.ISelectionUtils;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
 
 /**
  * @author Tuukka Lehtonen
  */
 public class SelectRouteOnDiagram {
 
+    private static final Logger LOGGER = LoggerFactory.getLogger(SelectRouteOnDiagram.class);
+
+    @CanExecute
+    public boolean canExecute(@Named(IServiceConstants.ACTIVE_SELECTION) ISelection selection) {
+        return ISelectionUtils.filterSingleSelection(selection, Route.class) != null;
+    }
+
     @Execute
     public void execute(@Named(IServiceConstants.ACTIVE_SHELL) Shell activeShell,
             @Named(IServiceConstants.ACTIVE_PART) MPart part,
@@ -30,9 +44,21 @@ public class SelectRouteOnDiagram {
         if (route == null)
             return;
 
-        CompletableFuture<List<Variable>> result = new CompletableFuture<>();
+        Display display = activeShell.getDisplay();
+        CompletableFuture<List<Resource>> result = new CompletableFuture<>();
+        result
+        .exceptionally(e -> Collections.emptyList())
+        .thenAccept(dnElements -> {
+            if (!dnElements.isEmpty()) {
+                try {
+                    DistrictNetworkUIUtil.openDNDiagramWithSelection(display, dnElements);
+                } catch (DatabaseException e) {
+                    LOGGER.error("Failed to open district network diagram with selection {}", dnElements); //$NON-NLS-1$
+                }
+            }
+        });
+
         RouterConfiguration config = new RouterConfiguration();
         new RouteJob(config, route, result).schedule();
     }
-
-}
\ No newline at end of file
+}