From 025ee8d7b4e955afaa00a1707606013ee4764d55 Mon Sep 17 00:00:00 2001
From: serge-rider <serge@jkiss.org>
Date: Sat, 14 Apr 2018 23:19:57 +0300
Subject: [PATCH] #3304 Data transfer - extra output file template variables

Former-commit-id: 270da14fe331fb91007788fee4e3fde8b073dd1f
---
 .../stream/StreamConsumerPageOutput.java      |  7 ++++
 .../stream/StreamTransferConsumer.java        | 34 ++++++++++++-------
 .../src/org/jkiss/dbeaver/model/DBUtils.java  | 16 +++++++++
 3 files changed, 45 insertions(+), 12 deletions(-)

diff --git a/plugins/org.jkiss.dbeaver.core/src/org/jkiss/dbeaver/tools/transfer/stream/StreamConsumerPageOutput.java b/plugins/org.jkiss.dbeaver.core/src/org/jkiss/dbeaver/tools/transfer/stream/StreamConsumerPageOutput.java
index e140103f8e..94718445fb 100644
--- a/plugins/org.jkiss.dbeaver.core/src/org/jkiss/dbeaver/tools/transfer/stream/StreamConsumerPageOutput.java
+++ b/plugins/org.jkiss.dbeaver.core/src/org/jkiss/dbeaver/tools/transfer/stream/StreamConsumerPageOutput.java
@@ -88,6 +88,9 @@ public class StreamConsumerPageOutput extends ActiveWizardPage<DataTransferWizar
             GridData gd = new GridData(GridData.FILL_HORIZONTAL);
             gd.horizontalSpan = 4;
             UIUtils.setContentProposalToolTip(fileNameText, "Output file name pattern",
+                StreamTransferConsumer.VARIABLE_DATASOURCE,
+                StreamTransferConsumer.VARIABLE_CATALOG,
+                StreamTransferConsumer.VARIABLE_SCHEMA,
                 StreamTransferConsumer.VARIABLE_TABLE,
                 StreamTransferConsumer.VARIABLE_TIMESTAMP,
                 StreamTransferConsumer.VARIABLE_DATE,
@@ -101,6 +104,9 @@ public class StreamConsumerPageOutput extends ActiveWizardPage<DataTransferWizar
                 fileNameText,
                 new TextContentAdapter(),
                 new SimpleContentProposalProvider(new String[] {
+                    GeneralUtils.variablePattern(StreamTransferConsumer.VARIABLE_DATASOURCE),
+                    GeneralUtils.variablePattern(StreamTransferConsumer.VARIABLE_CATALOG),
+                    GeneralUtils.variablePattern(StreamTransferConsumer.VARIABLE_SCHEMA),
                     GeneralUtils.variablePattern(StreamTransferConsumer.VARIABLE_TABLE),
                     GeneralUtils.variablePattern(StreamTransferConsumer.VARIABLE_TIMESTAMP),
                     GeneralUtils.variablePattern(StreamTransferConsumer.VARIABLE_DATE),
@@ -194,6 +200,7 @@ public class StreamConsumerPageOutput extends ActiveWizardPage<DataTransferWizar
         boolean isBinary = getWizard().getSettings().getProcessor().isBinaryFormat();
         boolean clipboard = !isBinary && clipboardCheck.getSelection();
 
+        clipboardCheck.setEnabled(!isBinary);
         directoryText.setEnabled(!clipboard);
         fileNameText.setEnabled(!clipboard);
         compressCheckbox.setEnabled(!clipboard);
diff --git a/plugins/org.jkiss.dbeaver.core/src/org/jkiss/dbeaver/tools/transfer/stream/StreamTransferConsumer.java b/plugins/org.jkiss.dbeaver.core/src/org/jkiss/dbeaver/tools/transfer/stream/StreamTransferConsumer.java
index 26d1a494bb..bc0ca0df9a 100644
--- a/plugins/org.jkiss.dbeaver.core/src/org/jkiss/dbeaver/tools/transfer/stream/StreamTransferConsumer.java
+++ b/plugins/org.jkiss.dbeaver.core/src/org/jkiss/dbeaver/tools/transfer/stream/StreamTransferConsumer.java
@@ -40,6 +40,8 @@ import org.jkiss.dbeaver.model.runtime.DBRProgressMonitor;
 import org.jkiss.dbeaver.model.runtime.DBRShellCommand;
 import org.jkiss.dbeaver.model.sql.SQLDataSource;
 import org.jkiss.dbeaver.model.struct.DBSObject;
+import org.jkiss.dbeaver.model.struct.rdb.DBSCatalog;
+import org.jkiss.dbeaver.model.struct.rdb.DBSSchema;
 import org.jkiss.dbeaver.runtime.ui.DBUserInterface;
 import org.jkiss.dbeaver.tools.transfer.IDataTransferConsumer;
 import org.jkiss.dbeaver.ui.UIUtils;
@@ -66,6 +68,9 @@ public class StreamTransferConsumer implements IDataTransferConsumer<StreamConsu
 
     private static final String LOB_DIRECTORY_NAME = "files"; //$NON-NLS-1$
 
+    public static final String VARIABLE_DATASOURCE = "datasource";
+    public static final String VARIABLE_CATALOG = "catalog";
+    public static final String VARIABLE_SCHEMA = "schema";
     public static final String VARIABLE_TABLE = "table";
     public static final String VARIABLE_TIMESTAMP = "timestamp";
     public static final String VARIABLE_DATE = "date";
@@ -350,8 +355,6 @@ public class StreamTransferConsumer implements IDataTransferConsumer<StreamConsu
     private void executeFinishCommand() {
         String commandLine = translatePattern(
             settings.getFinishProcessCommand(),
-            DBUtils.getObjectOwnerProject(sourceObject),
-            stripObjectName(sourceObject.getName()),
             outputFile);
         DBRShellCommand command = new DBRShellCommand(commandLine);
         DBRProcessDescriptor processDescriptor = new DBRProcessDescriptor(command);
@@ -373,8 +376,6 @@ public class StreamTransferConsumer implements IDataTransferConsumer<StreamConsu
         Object extension = processorProperties.get(StreamConsumerSettings.PROP_FILE_EXTENSION);
         String fileName = translatePattern(
             settings.getOutputFilePattern(),
-            DBUtils.getObjectOwnerProject(sourceObject),
-            stripObjectName(sourceObject.getName()),
             null);
         if (extension != null) {
             return fileName + "." + extension;
@@ -396,27 +397,36 @@ public class StreamTransferConsumer implements IDataTransferConsumer<StreamConsu
         return new File(dir, fileName);
     }
 
-    private String translatePattern(String pattern, final IProject project, final String tableName, final File targetFile)
+    private String translatePattern(String pattern, final File targetFile)
     {
-        pattern = GeneralUtils.replaceVariables(pattern, name -> {
+        return GeneralUtils.replaceVariables(pattern, name -> {
             switch (name) {
+                case VARIABLE_DATASOURCE: {
+                    return stripObjectName(sourceObject.getDataSource().getContainer().getName());
+                }
+                case VARIABLE_CATALOG: {
+                    DBSCatalog catalog = DBUtils.getParentOfType(DBSCatalog.class, sourceObject);
+                    return catalog == null ? "" : stripObjectName(catalog.getName());
+                }
+                case VARIABLE_SCHEMA: {
+                    DBSSchema schema = DBUtils.getParentOfType(DBSSchema.class, sourceObject);
+                    return schema == null ? "" : stripObjectName(schema.getName());
+                }
                 case VARIABLE_TABLE:
-                    return tableName;
+                    return stripObjectName(sourceObject.getName());
                 case VARIABLE_TIMESTAMP:
                     return RuntimeUtils.getCurrentTimeStamp();
                 case VARIABLE_DATE:
                     return RuntimeUtils.getCurrentDate();
-                case VARIABLE_PROJECT:
+                case VARIABLE_PROJECT: {
+                    IProject project = DBUtils.getObjectOwnerProject(sourceObject);
                     return project == null ? "" : project.getName();
+                }
                 case VARIABLE_FILE:
                     return targetFile == null ? "" : targetFile.getAbsolutePath();
             }
             return null;
         });
-        // Replace legacy patterns (without dollar prefix)
-        return pattern
-            .replace("{table}", tableName)
-            .replace("{timestamp}", RuntimeUtils.getCurrentTimeStamp());
     }
 
     private static String stripObjectName(String name)
diff --git a/plugins/org.jkiss.dbeaver.model/src/org/jkiss/dbeaver/model/DBUtils.java b/plugins/org.jkiss.dbeaver.model/src/org/jkiss/dbeaver/model/DBUtils.java
index 43a7c2256b..ecbe9ed67c 100644
--- a/plugins/org.jkiss.dbeaver.model/src/org/jkiss/dbeaver/model/DBUtils.java
+++ b/plugins/org.jkiss.dbeaver.model/src/org/jkiss/dbeaver/model/DBUtils.java
@@ -460,6 +460,22 @@ public final class DBUtils {
         return adapter == object ? null : adapter;
     }
 
+    @Nullable
+    public static <T> T getParentOfType(@NotNull Class<T> type, DBSObject object)
+    {
+        if (object == null) {
+            return null;
+        }
+        for (DBSObject parent = object.getParentObject(); parent != null; parent = parent.getParentObject()) {
+            if (type.isInstance(parent)) {
+                return type.cast(parent);
+            } else if (parent instanceof DBPDataSource || parent instanceof DBPDataSourceContainer) {
+                break;
+            }
+        }
+        return null;
+    }
+
     /**
      * Search for virtual entity descriptor
      * @param object object
-- 
GitLab