From c837da6a702be74daa551794933036c8268b300c Mon Sep 17 00:00:00 2001
From: Evgeny Fradkin <Eugene.Fradkin@gmail.com>
Date: Fri, 2 Mar 2018 17:30:21 +0300
Subject: [PATCH] #1599 Mock Data menu item is moved from the Tools menu to the
 upper level.

---
 plugins/org.jkiss.dbeaver.core/plugin.xml     |  2 +
 .../org.jkiss.dbeaver.ext.mockdata/plugin.xml | 37 ++++++++-----
 .../mockdata/handlers/MockDataHandler.java    | 54 +++++++++++++++++++
 3 files changed, 81 insertions(+), 12 deletions(-)
 create mode 100644 plugins/org.jkiss.dbeaver.ext.mockdata/src/org/jkiss/dbeaver/ext/mockdata/handlers/MockDataHandler.java

diff --git a/plugins/org.jkiss.dbeaver.core/plugin.xml b/plugins/org.jkiss.dbeaver.core/plugin.xml
index 1fe8c0c33f..c135e49e2f 100644
--- a/plugins/org.jkiss.dbeaver.core/plugin.xml
+++ b/plugins/org.jkiss.dbeaver.core/plugin.xml
@@ -2149,6 +2149,8 @@
                 <visibleWhen checkEnabled="true"/>
             </command>
 
+            <separator name="generate" visible="false"/>
+
             <menu label="%menu.database.sql.generate" icon="icons/misc/sql.png">
                 <visibleWhen>
                     <with variable="activePart">
diff --git a/plugins/org.jkiss.dbeaver.ext.mockdata/plugin.xml b/plugins/org.jkiss.dbeaver.ext.mockdata/plugin.xml
index 1204dd6cd6..21180d4c1d 100644
--- a/plugins/org.jkiss.dbeaver.ext.mockdata/plugin.xml
+++ b/plugins/org.jkiss.dbeaver.ext.mockdata/plugin.xml
@@ -5,18 +5,31 @@
 
     <extension-point id="org.jkiss.dbeaver.mockGenerator" name="Mock data generator" schema="schema/org.jkiss.dbeaver.mockGenerator.exsd"/>
 
-    <extension point="org.jkiss.dbeaver.tools">
-        <tools>
-            <tool
-                class="org.jkiss.dbeaver.ext.mockdata.MockDataGenerateTool"
-                description="Mock Data"
-                id="org.jkiss.dbeaver.ext.mockdata.MockDataGenerateTool"
-                label="Mock Data"
-                singleton="false">
-
-                <objectType name="org.jkiss.dbeaver.model.struct.DBSDataManipulator" forceCheck="true"/>
-            </tool>
-        </tools>
+    <extension point="org.eclipse.ui.handlers">
+        <handler commandId="org.jkiss.dbeaver.ext.mockdata.forSelection" class="org.jkiss.dbeaver.ext.mockdata.handlers.MockDataHandler">
+            <enabledWhen>
+                <with variable="selection">
+                    <count value="+"/>
+                    <iterate operator="and">
+                        <adapt type="org.jkiss.dbeaver.model.struct.DBSDataContainer"/>
+                    </iterate>
+                </with>
+            </enabledWhen>
+        </handler>
+    </extension>
+
+    <extension point="org.eclipse.ui.commands">
+        <command id="org.jkiss.dbeaver.ext.mockdata.forSelection" name="Generate Mock Data" description="Mock Data Generator"/>
+    </extension>
+
+    <extension point="org.eclipse.ui.menus">
+        <menuContribution allPopups="false" locationURI="popup:org.eclipse.ui.popup.any?before=generate">
+            <command
+                    commandId="org.jkiss.dbeaver.ext.mockdata.forSelection"
+                    label="Generate Mock Data">
+                <visibleWhen checkEnabled="true"/>
+            </command>
+        </menuContribution>
     </extension>
 
     <extension point="org.jkiss.dbeaver.mockGenerator">
diff --git a/plugins/org.jkiss.dbeaver.ext.mockdata/src/org/jkiss/dbeaver/ext/mockdata/handlers/MockDataHandler.java b/plugins/org.jkiss.dbeaver.ext.mockdata/src/org/jkiss/dbeaver/ext/mockdata/handlers/MockDataHandler.java
new file mode 100644
index 0000000000..927df177fd
--- /dev/null
+++ b/plugins/org.jkiss.dbeaver.ext.mockdata/src/org/jkiss/dbeaver/ext/mockdata/handlers/MockDataHandler.java
@@ -0,0 +1,54 @@
+/*
+ * DBeaver - Universal Database Manager
+ * Copyright (C) 2010-2017 Serge Rider (serge@jkiss.org)
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.jkiss.dbeaver.ext.mockdata.handlers;
+
+import org.eclipse.core.commands.AbstractHandler;
+import org.eclipse.core.commands.ExecutionEvent;
+import org.eclipse.core.commands.ExecutionException;
+import org.eclipse.ui.handlers.HandlerUtil;
+import org.jkiss.dbeaver.DBException;
+import org.jkiss.dbeaver.Log;
+import org.jkiss.dbeaver.ext.mockdata.MockDataGenerateTool;
+import org.jkiss.dbeaver.model.struct.DBSObject;
+import org.jkiss.dbeaver.ui.navigator.NavigatorUtils;
+
+import java.util.List;
+
+public class MockDataHandler extends AbstractHandler {
+
+    private static final Log log = Log.getLog(MockDataHandler.class);
+
+    public MockDataHandler() {
+    }
+
+    @Override
+    public Object execute(ExecutionEvent event) throws ExecutionException {
+
+        List<DBSObject> selectedObjects = NavigatorUtils.getSelectedObjects(
+                HandlerUtil.getCurrentSelection(event));
+
+        MockDataGenerateTool mockDataGenerator = new MockDataGenerateTool();
+        try {
+            mockDataGenerator.execute(
+                    HandlerUtil.getActiveWorkbenchWindow(event), null, selectedObjects);
+        } catch (DBException e) {
+            log.error("Error launching the Mock Data Generator", e);
+        }
+
+        return null;
+    }
+}
-- 
GitLab