123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343 |
- package com.rthoni.intellij.codefromds.ui.dialogs;
-
- import com.intellij.database.model.DasObject;
- import com.intellij.database.psi.DbDataSource;
- import com.intellij.openapi.fileChooser.FileChooserDescriptorFactory;
- import com.intellij.openapi.project.Project;
- import com.intellij.openapi.ui.DialogWrapper;
- import com.intellij.openapi.ui.TextFieldWithBrowseButton;
- import com.intellij.openapi.ui.ValidationInfo;
- import com.intellij.ui.ColoredListCellRenderer;
- import com.intellij.ui.JBColor;
- import com.intellij.ui.components.JBList;
- import com.rthoni.intellij.codefromds.business.Helper;
- import com.rthoni.intellij.codefromds.dbo.options.*;
- import org.jetbrains.annotations.NotNull;
- import org.jetbrains.annotations.Nullable;
-
- import javax.swing.*;
- import javax.swing.event.DocumentEvent;
- import javax.swing.event.DocumentListener;
- import javax.swing.event.ListSelectionListener;
- import java.awt.event.ActionListener;
- import java.io.File;
- import java.util.List;
- import java.util.Vector;
- import java.util.function.Consumer;
- import java.util.stream.Collectors;
-
- /**
- * Created by robin on 11/14/16.
- */
- public class GenerateDialog extends DialogWrapper {
-
- private JPanel _panel;
- private JBList _listDatasources;
- private JBList _listTables;
- private JBList _listColumns;
- private TextFieldWithBrowseButton _textModels;
- private TextFieldWithBrowseButton _textDataSource;
- private TextFieldWithBrowseButton _textDataSourceTemplate;
- private TextFieldWithBrowseButton _textModelsTemplate;
- private TextFieldWithBrowseButton _textConfigPath;
- private JTextField _textFilesExtension;
- private JLabel _lblModelsPath;
- private JLabel _lblDataSourcePath;
- private JLabel _lblDataSourceTemplatePath;
- private JLabel _lblModelsTemplatePath;
- private TextFieldWithBrowseButton _textCastFile;
- private JLabel _lblCastFile;
- private JBList _listStoredProcedure;
-
- private GenerateOptions _options;
-
- private Project _project;
-
- public GenerateDialog(Project project) {
- super(project);
- _project = project;
- setTitle("Code FROM data source");
- setOptions(null);
- init();
- }
-
- public GenerateOptions getOptions() {
- return _options;
- }
-
- public void setOptions(GenerateOptions options) {
- _options = options;
- if (_options != null) {
- showSource(_options.getSelection());
- _textModels.setText(Helper.getAbsolutePath(_project, _options.getModelsRelativePath()));
- _textDataSource.setText(Helper.getAbsolutePath(_project, _options.getDataSourceRelativePath()));
- _textDataSourceTemplate.setText(Helper.getAbsolutePath(_project, _options.getDataSourceTemplateRelativePath()));
- _textModelsTemplate.setText(Helper.getAbsolutePath(_project, _options.getModelsTemplateRelativePath()));
- _textFilesExtension.setText(_options.getFilesExtension());
- _textCastFile.setText(Helper.getAbsolutePath(_project, _options.getCastFileRelativePath()));
- _textConfigPath.setText(_options.getConfigAbsolutePath());
- }
- else {
- showSource(null);
- _textModels.setText("");
- _textDataSource.setText("");
- _textDataSourceTemplate.setText("");
- _textModelsTemplate.setText("");
- _textFilesExtension.setText("");
- _textCastFile.setText("");
- _textConfigPath.setText("");
- }
- }
-
- @Nullable
- @Override
- protected ValidationInfo doValidate() {
- ValidationInfo info = null;
- File modelDir = _options == null ? null : new File(Helper.getAbsolutePath(_project, _options.getModelsRelativePath()));
- File dataSourcePath = _options == null ? null : new File(Helper.getAbsolutePath(_project, _options.getDataSourceRelativePath()));
- File dataSourceDir = _options == null ? null : new File(dataSourcePath.getParent());
- File dataSourceTemplatePath = _options == null ? null : new File(Helper.getAbsolutePath(_project, _options.getDataSourceTemplateRelativePath()));
- File modelsTemplatePath = _options == null ? null : new File(Helper.getAbsolutePath(_project, _options.getModelsTemplateRelativePath()));
- File configPath = _options == null ? null : new File(_options.getConfigAbsolutePath());
- File configDir = _options == null ? null : new File(configPath.getParent());
- String extension = _options == null ? null : _options.getFilesExtension();
- File castFilePath = _options == null ? null : new File(Helper.getAbsolutePath(_project, _options.getCastFileRelativePath()));
- if (_options == null) {
- info = new ValidationInfo("No Data Source Selected", _listDatasources);
- }
- else if (!modelDir.exists() || !modelDir.isDirectory()) {
- info = new ValidationInfo("Models folder does not exists", _textModels.getTextField());
- }
- else if (!dataSourceDir.exists() || !dataSourceDir.isDirectory()) {
- info = new ValidationInfo("Data source folder does not exists", _textDataSource.getTextField());
- }
- else if (!dataSourceTemplatePath.exists() || !dataSourceTemplatePath.isFile()) {
- info = new ValidationInfo("Data source template file does not exists", _textDataSourceTemplate.getTextField());
- }
- else if (!modelsTemplatePath.exists() || !modelsTemplatePath.isFile()) {
- info = new ValidationInfo("Models template file does not exists", _textModelsTemplate.getTextField());
- }
- else if (!configDir.exists() || !configDir.isDirectory()) {
- info = new ValidationInfo("Configuration file parent folder does not exists", _textConfigPath.getTextField());
- }
- else if (extension == null || extension.isEmpty()) {
- info = new ValidationInfo("Files extension is required", _textFilesExtension);
- }
- else if (extension.startsWith(".")) {
- info = new ValidationInfo("Files extension must not include dot (.)", _textFilesExtension);
- }
- else if (!castFilePath.exists() || !castFilePath.isFile()) {
- info = new ValidationInfo("Cast file does not exists", _textCastFile.getTextField());
- }
-
- return info;
- }
-
- @Nullable
- @Override
- protected JComponent createCenterPanel() {
-
- List<DbDataSource> dataSources = Helper.getDataSources();
-
- _listDatasources.setListData(dataSources.stream().map(DasObject::getName).toArray(String[]::new));
- _listDatasources.addListSelectionListener(e -> {
- if (!e.getValueIsAdjusting()) {
- int index = _listDatasources.getSelectedIndex();
- if (index == -1) {
- setOptions(null);
- }
- else {
- setOptions(new GenerateOptions(dataSources.get(index)));
- }
- }
- });
- _listTables.addListSelectionListener(e -> {
- if (!e.getValueIsAdjusting()) {
- int index = _listTables.getSelectedIndex();
- if (index == -1) {
- showTable(null);
- }
- else {
- showTable(_options.getSelection().getTables().get(index));
- }
- }
- });
-
- _listStoredProcedure.addListSelectionListener(e -> {
- if (!e.getValueIsAdjusting()) {
- updateSpSelection();
- }
- });
-
- setupTextField(_textModels, null, true, "Models");
- setupTextField(_textDataSource, null, false, "Data Source");
- setupTextField(_textDataSourceTemplate, null, false, "Data Source Template");
- setupTextField(_textModelsTemplate, null, false, "Models Template");
- setupTextField(_textConfigPath, null, false, "Configuration");
- setupTextField(_textCastFile, null, false, "Types Cast");
-
- setupTextFieldListener(_textModels.getTextField(), s -> {
- _options.setModelsRelativePath(Helper.getRelativePath(_project, s));
- _lblModelsPath.setText("$ProjectRoot/" + _options.getModelsRelativePath());
- });
-
- setupTextFieldListener(_textDataSource.getTextField(), s -> {
- _options.setDataSourceRelativePath(Helper.getRelativePath(_project, s));
- _lblDataSourcePath.setText("$ProjectRoot/" + _options.getDataSourceRelativePath());
- });
- setupTextFieldListener(_textDataSourceTemplate.getTextField(), s -> {
- _options.setDataSourceTemplateRelativePath(Helper.getRelativePath(_project, s));
- _lblDataSourceTemplatePath.setText("$ProjectRoot/" + _options.getDataSourceTemplateRelativePath());
- });
- setupTextFieldListener(_textModelsTemplate.getTextField(), s -> {
- _options.setModelsTemplateRelativePath(Helper.getRelativePath(_project, s));
- _lblModelsTemplatePath.setText("$ProjectRoot/" + _options.getModelsTemplateRelativePath());
- });
- setupTextFieldListener(_textFilesExtension, s -> _options.setFilesExtension(s));
- setupTextFieldListener(_textConfigPath.getTextField(), s -> _options.setConfigAbsolutePath(s));
- setupTextFieldListener(_textCastFile.getTextField(), s -> {
- _options.setCastFileRelativePath(Helper.getRelativePath(_project, s));
- _lblCastFile.setText("$ProjectRoot/" + _options.getCastFileRelativePath());
- });
-
- _listTables.setCellRenderer(new ColoredListCellRenderer() {
- @Override
- protected void customizeCellRenderer(@NotNull JList jList, Object o, int i, boolean b, boolean b1) {
- TableSelection tableSelection = _options.getSelection().getTables().get(i);
- if (tableSelection.hasAll()) {
- setBackground(JBColor.GREEN);
- }
- else if (tableSelection.hasNone()) {
- setBackground(JBColor.RED);
- }
- else {
- setBackground(JBColor.ORANGE);
- }
- append(tableSelection.getTable().getName());
- }
- });
-
- _listStoredProcedure.setCellRenderer(new ColoredListCellRenderer() {
- @Override
- protected void customizeCellRenderer(@NotNull JList jList, Object o, int i, boolean b, boolean b1) {
- StoredProcedureSelection spSelection = _options.getSelection().getStoredProcedures().get(i);
- if (spSelection.isSelected()) {
- setBackground(JBColor.GREEN);
- }
- else {
- setBackground(JBColor.RED);
- }
- append(spSelection.getStoredProcedure().getText());
- }
- });
-
- if (dataSources.size() > 0) {
- _listDatasources.setSelectedIndices(new int[]{0});
- }
-
- return _panel;
- }
-
- private void setupTextFieldListener(JTextField field, Consumer<String> consumer)
- {
- field.getDocument().addDocumentListener(new DocumentListener() {
- @Override
- public void insertUpdate(DocumentEvent e) {
- changed();
- }
-
- @Override
- public void removeUpdate(DocumentEvent e) {
- changed();
- }
-
- @Override
- public void changedUpdate(DocumentEvent e) {
- changed();
- }
-
- public void changed() {
- consumer.accept(field.getText());
- }
- });
- }
-
- private void setupTextField(TextFieldWithBrowseButton field, Project project, boolean dirsOnly, String title)
- {
- for (ActionListener l : field.getButton().getActionListeners()) {
- field.getButton().removeActionListener(l);
- }
- field.addBrowseFolderListener("Choose " + title + " " + (dirsOnly ? "Folder" : "File"), "Choose " + (dirsOnly ? "Folder" : "File"), project,
- dirsOnly ? FileChooserDescriptorFactory.createSingleFolderDescriptor() : FileChooserDescriptorFactory.createSingleFileDescriptor());
- }
-
- private void showSource(DataSourceSelection source)
- {
- if (source != null) {
- List<String> tables = source.getTables().stream().map(t -> t.getTable().getName()).collect(Collectors.toList());
- _listTables.setListData(tables.toArray(new String[tables.size()]));
- if (tables.size() > 0) {
- showTable(source.getTables().get(0));
- }
- else {
- showTable(null);
- }
- List<String> sps = source.getStoredProcedures().stream().map(t -> t.getStoredProcedure().getText()).collect(Collectors.toList());
- _listStoredProcedure.setListData(sps.toArray(new String[sps.size()]));
- }
- else {
- _listTables.setListData(new String[]{});
- _listStoredProcedure.setListData(new String[]{});
- showTable(null);
- }
- }
-
- private int[] getSelectedIndices(final TableSelection table)
- {
- List<ColumnSelection> columns = table.getColumns();
- List<Integer> indices = new Vector<>();
- for (int i = 0; i < columns.size(); ++i) {
- if (columns.get(i).isSelected()) {
- indices.add(i);
- }
- }
- return indices.stream().mapToInt(Integer::intValue).toArray();
- }
-
- private void updateColumnSelection(final TableSelection table)
- {
- List<ColumnSelection> columns = table.getColumns();
- for (int i = 0; i < columns.size(); ++i) {
- columns.get(i).setSelected(_listColumns.isSelectedIndex(i));
- }
- }
-
- private void updateSpSelection()
- {
- List<StoredProcedureSelection> sps = _options.getSelection().getStoredProcedures();
- for (int i = 0; i < sps.size(); ++i) {
- sps.get(i).setSelected(_listStoredProcedure.isSelectedIndex(i));
- }
- }
-
- private void showTable(TableSelection table)
- {
- for (ListSelectionListener e : _listColumns.getListSelectionListeners()) {
- _listColumns.removeListSelectionListener(e);
- }
- if (table != null) {
- _listColumns.setListData(table.getColumns().stream().map(c -> c.getColumn().getName()).toArray(String[]::new));
- int[] indices = getSelectedIndices(table);
- _listColumns.setSelectedIndices(indices);
- _listColumns.addListSelectionListener(e -> {
- if (!e.getValueIsAdjusting()) {
- _listTables.updateUI();
- updateColumnSelection(table);
- }
- });
- }
- else {
- _listColumns.setListData(new String[]{});
- }
- }
- }
|