Merge pull request #10176 from dbeaver/postgre-tasks#9911

#9911 Add null check

plugins/org.jkiss.dbeaver.ext.postgresql.ui/src/org/jkiss/dbeaver/ext/postgresql/tools/PostgreRestoreWizardPageSettings.java

@@ -84,7 +84,7 @@ class PostgreRestoreWizardPageSettings extends PostgreToolWizardPageSettings<Pos
         inputFileText = new TextWithOpenFile(inputGroup, PostgreMessages.wizard_restore_page_setting_label_choose_backup_file, new String[] {"*.backup","*.sql","*"});
         inputFileText.setLayoutData(new GridData(GridData.FILL_HORIZONTAL));
         inputFileText.getTextControl().addListener(SWT.Modify, updateListener);
-        //inputFileText.setText(wizard.getSettings().getInputFile());
+        inputFileText.setText(wizard.getSettings().getInputFile());
 
         createExtraArgsInput(inputGroup);
 

plugins/org.jkiss.dbeaver.ext.postgresql/src/org/jkiss/dbeaver/ext/postgresql/tasks/PostgreDatabaseRestoreSettings.java

@@ -24,6 +24,7 @@ import org.jkiss.dbeaver.model.DBUtils;
 import org.jkiss.dbeaver.model.preferences.DBPPreferenceMap;
 import org.jkiss.dbeaver.model.preferences.DBPPreferenceStore;
 import org.jkiss.dbeaver.model.runtime.DBRRunnableContext;
+import org.jkiss.dbeaver.model.struct.DBSObject;
 import org.jkiss.utils.CommonUtils;
 
 import java.lang.reflect.InvocationTargetException;
@@ -99,8 +100,19 @@ public class PostgreDatabaseRestoreSettings extends PostgreBackupRestoreSettings
                 } catch (InterruptedException e) {
                     // Ignore
                 }
+            } else {
+                for (DBSObject object : getDatabaseObjects()) {
+                    if (object instanceof PostgreDatabase) {
+                        restoreInfo = new PostgreDatabaseRestoreInfo((PostgreDatabase) object);
+                        break;
+                    }
+                }
             }
         }
+
+        if (restoreInfo == null) {
+            throw new DBException("Cannot find database for restoring");
+        }
     }
 
     @Override

plugins/org.jkiss.dbeaver.ext.postgresql/src/org/jkiss/dbeaver/ext/postgresql/tasks/PostgreScriptExecuteSettings.java

@@ -67,8 +67,19 @@ public class PostgreScriptExecuteSettings extends AbstractScriptExecuteSettings<
                 } catch (InterruptedException e) {
                     // Ignore
                 }
+            } else {
+                for (DBSObject object : getDatabaseObjects()) {
+                    if (object instanceof PostgreDatabase) {
+                        database = (PostgreDatabase) object;
+                        break;
+                    }
+                }
             }
         }
+
+        if (database == null) {
+            throw new DBException("Cannot find database for script execution");
+        }
     }
 
     @Override