Use TreeNodeModelApi everywhere instead of DataSourceApi (#28)

* Use TreeNodeModelApi everywhere
* Filter out datasources

Author: Vorago

ui/jetbrains/src/main/java/com/neueda/jetbrains/plugin/graphdb/jetbrains/ui/datasource/interactions/DataSourceInteractions.java

@@ -11,6 +11,8 @@
 import com.neueda.jetbrains.plugin.graphdb.jetbrains.component.datasource.state.DataSourceApi;
 import com.neueda.jetbrains.plugin.graphdb.jetbrains.ui.datasource.DataSourcesView;
 import com.neueda.jetbrains.plugin.graphdb.jetbrains.ui.datasource.interactions.neo4j.bolt.Neo4jBoltDataSourceDialog;
+import com.neueda.jetbrains.plugin.graphdb.jetbrains.ui.datasource.tree.Neo4jTreeNodeType;
+import com.neueda.jetbrains.plugin.graphdb.jetbrains.ui.datasource.tree.TreeNodeModelApi;
 import com.neueda.jetbrains.plugin.graphdb.jetbrains.util.FileUtil;
 import com.neueda.jetbrains.plugin.graphdb.jetbrains.util.Notifier;
 
@@ -57,10 +59,10 @@ private void initAddAction() {
     private void initRemoveAction() {
         decorator.setRemoveAction(anActionButton -> {
             DefaultMutableTreeNode[] selectedNodes = dataSourceTree.getSelectedNodes(DefaultMutableTreeNode.class,
-                    (node) -> node.getUserObject() instanceof DataSourceApi);
+                    this::isDataSource);
 
             List<DataSourceApi> dataSourcesForRemoval = Arrays.stream(selectedNodes)
-                    .map((node) -> (DataSourceApi) node.getUserObject())
+                    .map(this::getDataSourceApi)
                     .collect(Collectors.toList());
 
             if (dataSourcesForRemoval.size() > 0) {
@@ -69,7 +71,7 @@ private void initRemoveAction() {
         });
         decorator.setRemoveActionUpdater(e -> {
             DefaultMutableTreeNode[] selectedNodes = dataSourceTree.getSelectedNodes(DefaultMutableTreeNode.class,
-                    (node) -> node.getUserObject() instanceof DataSourceApi);
+                    this::isDataSource);
 
             return selectedNodes.length > 0;
         });
@@ -78,18 +80,18 @@ private void initRemoveAction() {
     private void initEditAction() {
         decorator.setEditActionUpdater(e -> {
             DefaultMutableTreeNode[] selectedNodes = dataSourceTree.getSelectedNodes(DefaultMutableTreeNode.class,
-                    (node) -> node.getUserObject() instanceof DataSourceApi);
+                    this::isDataSource);
 
             return selectedNodes.length == 1;
         });
         decorator.setEditAction(anActionButton -> {
             PatchedDefaultMutableTreeNode[] selectedNodes = dataSourceTree.getSelectedNodes(PatchedDefaultMutableTreeNode.class,
-                    (node) -> node.getUserObject() instanceof DataSourceApi);
+                    (node) -> node.getUserObject() instanceof TreeNodeModelApi);
 
             if (selectedNodes.length == 1) {
                 PatchedDefaultMutableTreeNode treeNode = selectedNodes[0];
 
-                DataSourceApi dataSourceToEdit = (DataSourceApi) treeNode.getUserObject();
+                DataSourceApi dataSourceToEdit = getDataSourceApi(treeNode);
 
                 DataSourceDialog dialog = null;
                 if (dataSourceToEdit.getDataSourceType().equals(DataSourceType.NEO4J_BOLT)) {
@@ -112,13 +114,13 @@ public void mouseClicked(MouseEvent e) {
                 int clickCount = e.getClickCount();
                 if (clickCount == 2) {
                     DefaultMutableTreeNode[] selectedNodes = dataSourceTree.getSelectedNodes(DefaultMutableTreeNode.class,
-                            (node) -> node.getUserObject() instanceof DataSourceApi);
+                            DataSourceInteractions.this::isDataSource);
 
                     if (selectedNodes.length != 1) {
                         return;
                     }
 
-                    DataSourceApi dataSource = (DataSourceApi) selectedNodes[0].getUserObject();
+                    DataSourceApi dataSource = getDataSourceApi(selectedNodes[0]);
                     Analytics.event(dataSource, "openEditor");
 
                     try {
@@ -131,4 +133,13 @@ public void mouseClicked(MouseEvent e) {
         };
         dataSourceTree.addMouseListener(mouseAdapter);
     }
+
+    private boolean isDataSource(DefaultMutableTreeNode node) {
+        return node.getUserObject() instanceof TreeNodeModelApi
+                && ((TreeNodeModelApi) node.getUserObject()).getType() == Neo4jTreeNodeType.DATASOURCE;
+    }
+
+    private DataSourceApi getDataSourceApi(DefaultMutableTreeNode node) {
+        return ((TreeNodeModelApi) node.getUserObject()).getDataSourceApi();
+    }
 }