init

.gitignore

@@ -0,0 +1,57 @@
+# Created by .ignore support plugin (hsz.mobi)
+### Java template
+*.class
+
+# Mobile Tools for Java (J2ME)
+.mtj.tmp/
+
+# Package Files #
+*.jar
+*.war
+*.ear
+
+# virtual machine crash logs, see http://www.java.com/en/download/help/error_hotspot.xml
+hs_err_pid*
+
+# Covers JetBrains IDEs: IntelliJ, RubyMine, PhpStorm, AppCode, PyCharm, CLion, Android Studio and Webstorm
+# Reference: https://intellij-support.jetbrains.com/hc/en-us/articles/206544839
+
+# User-specific stuff:
+.idea/workspace.xml
+.idea/tasks.xml
+
+# Sensitive or high-churn files:
+.idea/dataSources/
+.idea/dataSources.ids
+.idea/dataSources.xml
+.idea/dataSources.local.xml
+.idea/sqlDataSources.xml
+.idea/dynamic.xml
+.idea/uiDesigner.xml
+
+# Gradle:
+.idea/gradle.xml
+.idea/libraries
+
+# Mongo Explorer plugin:
+.idea/mongoSettings.xml
+
+## File-based project format:
+*.iws
+
+## Plugin-specific files:
+
+# IntelliJ
+/out/
+
+# mpeltonen/sbt-idea plugin
+.idea_modules/
+
+# JIRA plugin
+atlassian-ide-plugin.xml
+
+# Crashlytics plugin (for Android Studio and IntelliJ)
+com_crashlytics_export_strings.xml
+crashlytics.properties
+crashlytics-build.properties
+fabric.properties

.idea/compiler.xml

@@ -0,0 +1,22 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="CompilerConfiguration">
+    <resourceExtensions />
+    <wildcardResourcePatterns>
+      <entry name="!?*.java" />
+      <entry name="!?*.form" />
+      <entry name="!?*.class" />
+      <entry name="!?*.groovy" />
+      <entry name="!?*.scala" />
+      <entry name="!?*.flex" />
+      <entry name="!?*.kt" />
+      <entry name="!?*.clj" />
+      <entry name="!?*.aj" />
+    </wildcardResourcePatterns>
+    <annotationProcessing>
+      <profile default="true" name="Default" enabled="false">
+        <processorPath useClasspath="true" />
+      </profile>
+    </annotationProcessing>
+  </component>
+</project>

.idea/copyright/profiles_settings.xml

@@ -0,0 +1,3 @@
+<component name="CopyrightManager">
+  <settings default="" />
+</component>

.idea/misc.xml

@@ -0,0 +1,26 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="EntryPointsManager">
+    <entry_points version="2.0" />
+  </component>
+  <component name="MavenImportPreferences">
+    <option name="generalSettings">
+      <MavenGeneralSettings>
+        <option name="mavenHome" value="Bundled (Maven 3)" />
+      </MavenGeneralSettings>
+    </option>
+  </component>
+  <component name="ProjectLevelVcsManager" settingsEditedManually="false">
+    <OptionsSetting value="true" id="Add" />
+    <OptionsSetting value="true" id="Remove" />
+    <OptionsSetting value="true" id="Checkout" />
+    <OptionsSetting value="true" id="Update" />
+    <OptionsSetting value="true" id="Status" />
+    <OptionsSetting value="true" id="Edit" />
+    <ConfirmationsSetting value="0" id="Add" />
+    <ConfirmationsSetting value="0" id="Remove" />
+  </component>
+  <component name="ProjectRootManager" version="2" languageLevel="JDK_1_8" default="true" assert-keyword="true" jdk-15="true" project-jdk-name="IntelliJ IDEA RS-163.7608" project-jdk-type="IDEA JDK">
+    <output url="file://$PROJECT_DIR$/out" />
+  </component>
+</project>

.idea/modules.xml

@@ -0,0 +1,8 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="ProjectModuleManager">
+    <modules>
+      <module fileurl="file://$PROJECT_DIR$/code-from-ds.iml" filepath="$PROJECT_DIR$/code-from-ds.iml" />
+    </modules>
+  </component>
+</project>

code-from-ds.iml

@@ -0,0 +1,13 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<module type="PLUGIN_MODULE" version="4">
+  <component name="DevKit.ModuleBuildProperties" url="file://$MODULE_DIR$/resources/META-INF/plugin.xml" />
+  <component name="NewModuleRootManager" inherit-compiler-output="true">
+    <exclude-output />
+    <content url="file://$MODULE_DIR$">
+      <sourceFolder url="file://$MODULE_DIR$/src" isTestSource="false" />
+      <sourceFolder url="file://$MODULE_DIR$/resources" type="java-resource" />
+    </content>
+    <orderEntry type="inheritedJdk" />
+    <orderEntry type="sourceFolder" forTests="false" />
+  </component>
+</module>

resources/META-INF/plugin.xml

@@ -0,0 +1,37 @@
+<idea-plugin version="2">
+  <id>com.rthoni.intellij.codefromds</id>
+  <name>Code FROM ds</name>
+  <version>1.0</version>
+  <vendor email="robin@rthoni.com" url="https://www.rthoni.com">Robin Thoni</vendor>
+
+  <description><![CDATA[
+      Generate code from data sources
+    ]]></description>
+
+  <change-notes><![CDATA[
+      First version
+    ]]>
+  </change-notes>
+
+  <!-- please see http://www.jetbrains.org/intellij/sdk/docs/basics/getting_started/build_number_ranges.html for description -->
+  <idea-version since-build="141.0"/>
+
+  <!-- please see http://www.jetbrains.org/intellij/sdk/docs/basics/getting_started/plugin_compatibility.html
+       on how to target different products -->
+  <depends>com.intellij.modules.lang</depends>
+  <depends>com.intellij.database</depends>
+
+  <extensions defaultExtensionNs="com.intellij">
+    <!-- Add your extensions here -->
+  </extensions>
+
+  <actions>
+    <!-- Add your actions here -->
+    <action id="codefromds-generate" class="com.rthoni.intellij.codefromds.ui.actions.GenerateAction"
+            text="Generate code from data source" description="Generate models from an intellij data source">
+      <add-to-group group-id="ToolsMenu" anchor="last"/>
+      <keyboard-shortcut keymap="$default" first-keystroke="ctrl alt G"/>
+    </action>
+  </actions>
+
+</idea-plugin>

src/com/rthoni/intellij/codefromds/dbo/ColumnSelection.java

@@ -0,0 +1,28 @@
+package com.rthoni.intellij.codefromds.dbo;
+
+import com.intellij.database.model.DasColumn;
+
+/**
+ * Created by robin on 11/15/16.
+ */
+public class ColumnSelection {
+    private DasColumn _column;
+
+    private boolean _selected;
+
+    public DasColumn getColumn() {
+        return _column;
+    }
+
+    public void setColumn(DasColumn column) {
+        _column = column;
+    }
+
+    public boolean isSelected() {
+        return _selected;
+    }
+
+    public void setSelected(boolean selected) {
+        _selected = selected;
+    }
+}

src/com/rthoni/intellij/codefromds/dbo/DataSourceSelection.java

@@ -0,0 +1,30 @@
+package com.rthoni.intellij.codefromds.dbo;
+
+import com.intellij.database.psi.DbDataSource;
+
+import java.util.List;
+
+/**
+ * Created by robin on 11/15/16.
+ */
+public class DataSourceSelection {
+    private DbDataSource _source;
+
+    private List<TableSelection> _tables;
+
+    public DbDataSource getSource() {
+        return _source;
+    }
+
+    public void setSource(DbDataSource source) {
+        _source = source;
+    }
+
+    public List<TableSelection> getTables() {
+        return _tables;
+    }
+
+    public void setTables(List<TableSelection> tables) {
+        _tables = tables;
+    }
+}

src/com/rthoni/intellij/codefromds/dbo/GenerateOptions.java

@@ -0,0 +1,26 @@
+package com.rthoni.intellij.codefromds.dbo;
+
+/**
+ * Created by robin on 11/15/16.
+ */
+public class GenerateOptions {
+    private DataSourceSelection _selection;
+
+    private String _modelsFolder;
+
+    public DataSourceSelection getSelection() {
+        return _selection;
+    }
+
+    public void setSelection(DataSourceSelection selection) {
+        _selection = selection;
+    }
+
+    public String getModelsFolder() {
+        return _modelsFolder;
+    }
+
+    public void setModelsFolder(String modelsFolder) {
+        _modelsFolder = modelsFolder;
+    }
+}

src/com/rthoni/intellij/codefromds/dbo/TableSelection.java

@@ -0,0 +1,41 @@
+package com.rthoni.intellij.codefromds.dbo;
+
+import com.intellij.database.model.DasTable;
+
+import java.util.List;
+
+/**
+ * Created by robin on 11/15/16.
+ */
+public class TableSelection {
+
+    private DasTable _table;
+
+    private List<ColumnSelection> _columns;
+
+    public DasTable getTable() {
+        return _table;
+    }
+
+    public void setTable(DasTable table) {
+        _table = table;
+    }
+
+    public List<ColumnSelection> getColumns() {
+        return _columns;
+    }
+
+    public void setColumns(List<ColumnSelection> columns) {
+        _columns = columns;
+    }
+
+    public boolean hasNone()
+    {
+        return _columns != null && _columns.stream().filter(ColumnSelection::isSelected).count() == 0;
+    }
+
+    public boolean hasAll()
+    {
+        return _columns != null && _columns.stream().filter(ColumnSelection::isSelected).count() == _columns.size();
+    }
+}

src/com/rthoni/intellij/codefromds/ui/actions/GenerateAction.java

@@ -0,0 +1,28 @@
+package com.rthoni.intellij.codefromds.ui.actions;
+
+import com.intellij.openapi.actionSystem.AnAction;
+import com.intellij.openapi.actionSystem.AnActionEvent;
+import com.intellij.openapi.project.Project;
+import com.intellij.openapi.project.ProjectManager;
+import com.intellij.openapi.ui.DialogWrapper;
+import com.rthoni.intellij.codefromds.dbo.GenerateOptions;
+import com.rthoni.intellij.codefromds.ui.dialogs.GenerateDialog;
+
+import javax.swing.*;
+
+/**
+ * Created by robin on 11/14/16.
+ */
+public class GenerateAction extends AnAction {
+
+    @Override
+    public void actionPerformed(AnActionEvent e) {
+        GenerateDialog dlg = new GenerateDialog(null);
+        dlg.show();
+        int res = dlg.getExitCode();
+        if (res == DialogWrapper.OK_EXIT_CODE) {
+            GenerateOptions options = dlg.getOptions();
+            options.getSelection();
+        }
+    }
+}

src/com/rthoni/intellij/codefromds/ui/dialogs/GenerateDialog.java

@@ -0,0 +1,230 @@
+package com.rthoni.intellij.codefromds.ui.dialogs;
+
+import com.intellij.database.model.DasObject;
+import com.intellij.database.model.DasTable;
+import com.intellij.database.psi.DbDataSource;
+import com.intellij.database.psi.DbPsiFacade;
+import com.intellij.database.util.DasUtil;
+import com.intellij.openapi.fileChooser.FileChooserDescriptorFactory;
+import com.intellij.openapi.project.Project;
+import com.intellij.openapi.project.ProjectManager;
+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.dbo.ColumnSelection;
+import com.rthoni.intellij.codefromds.dbo.DataSourceSelection;
+import com.rthoni.intellij.codefromds.dbo.GenerateOptions;
+import com.rthoni.intellij.codefromds.dbo.TableSelection;
+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.*;
+import java.awt.event.ActionListener;
+import java.io.File;
+import java.util.Arrays;
+import java.util.Collection;
+import java.util.List;
+import java.util.Vector;
+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 GenerateOptions _options;
+
+    public GenerateDialog(@Nullable Project project) {
+        super(project);
+        setTitle("Code FROM data source");
+        showSource(null);
+        init();
+    }
+
+    public GenerateOptions getOptions() {
+        return _options;
+    }
+
+    @Nullable
+    @Override
+    protected ValidationInfo doValidate() {
+        ValidationInfo info = null;
+        File modelDir = _options == null ? null : new File(_options.getModelsFolder());
+        if (_options == null) {
+            info = new ValidationInfo("No Data Source Selected", _listDatasources);
+        }
+        else if (!modelDir.exists() || !modelDir.isDirectory()) {
+            info = new ValidationInfo("Bad Model Folder", _textModels.getTextField());
+        }
+
+        return info;
+    }
+
+    @Nullable
+    @Override
+    protected JComponent createCenterPanel() {
+        ProjectManager pm = ProjectManager.getInstance();
+        Project[] projects = pm.getOpenProjects();
+
+        List<DbDataSource> dataSources = Arrays.stream(projects).map(project -> DbPsiFacade.getInstance(project).getDataSources()).flatMap(Collection::stream).collect(Collectors.toList());
+
+        _listDatasources.setListData(dataSources.stream().map(DasObject::getName).toArray(String[]::new));
+        _listDatasources.addListSelectionListener(e -> {
+            if (!e.getValueIsAdjusting()) {
+                int index = _listDatasources.getSelectedIndex();
+                if (index == -1) {
+                    showSource(null);
+                }
+                else {
+                    showSource(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));
+                }
+            }
+        });
+        _textModels.getTextField().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() {
+                _options.setModelsFolder(_textModels.getText());
+            }
+        });
+
+        if (dataSources.size() > 0) {
+            _listDatasources.setSelectedIndices(new int[]{0});
+        }
+
+        return _panel;
+    }
+
+    private void showSource(DbDataSource source)
+    {
+        showTable(null);
+        for (ActionListener l : _textModels.getButton().getActionListeners()) {
+            _textModels.getButton().removeActionListener(l);
+        }
+        if (source != null) {
+            _options = new GenerateOptions();
+
+            _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());
+                }
+            });
+
+            _textModels.setText(source.getProject().getBasePath() + File.separator + "Models");
+            _textModels.addBrowseFolderListener("Choose Models Destination Folder", "Choose folder",
+                    source.getProject(), FileChooserDescriptorFactory.createSingleFolderDescriptor());
+
+            _options.setSelection(new DataSourceSelection());
+            _options.getSelection().setSource(source);
+            final List<? extends DasTable> tables = DasUtil.getTables(source).toList();
+            List<TableSelection> tableSelections = new Vector<>();
+            for (DasTable table : tables) {
+                TableSelection tableSelection = new TableSelection();
+                tableSelection.setTable(table);
+                tableSelection.setColumns(DasUtil.getColumns(table).toList().stream().map(c -> {
+                    ColumnSelection selection = new ColumnSelection();
+                    selection.setColumn(c);
+                    selection.setSelected(true);
+                    return selection;
+                }).collect(Collectors.toList()));
+                tableSelections.add(tableSelection);
+            }
+            _options.getSelection().setTables(tableSelections);
+
+            _listTables.setListData(tables.stream().map(DasObject::getName).toArray(String[]::new));
+        }
+        else {
+            _options = null;
+            _listTables.setListData(new String[]{});
+        }
+    }
+
+    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 updateSelection(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 showTable(TableSelection table)
+    {
+        if (table != null) {
+            for (ListSelectionListener e : _listColumns.getListSelectionListeners()) {
+                _listColumns.removeListSelectionListener(e);
+            }
+            _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();
+                    updateSelection(table);
+                }
+            });
+        }
+        else {
+            _listColumns.setListData(new String[]{});
+        }
+    }
+}

src/com/rthoni/intellij/codefromds/ui/forms/GenerateForm.form

@@ -0,0 +1,101 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<form xmlns="http://www.intellij.com/uidesigner/form/" version="1" bind-to-class="com.rthoni.intellij.codefromds.ui.dialogs.GenerateDialog">
+  <grid id="27dc6" binding="_panel" layout-manager="GridLayoutManager" row-count="4" column-count="2" same-size-horizontally="false" same-size-vertically="false" hgap="-1" vgap="-1">
+    <margin top="0" left="0" bottom="0" right="0"/>
+    <constraints>
+      <xy x="20" y="20" width="934" height="526"/>
+    </constraints>
+    <properties/>
+    <border type="none"/>
+    <children>
+      <vspacer id="cffbd">
+        <constraints>
+          <grid row="3" column="0" row-span="1" col-span="2" vsize-policy="6" hsize-policy="1" anchor="0" fill="2" indent="0" use-parent-layout="false"/>
+        </constraints>
+      </vspacer>
+      <grid id="6946b" layout-manager="GridLayoutManager" row-count="1" column-count="2" same-size-horizontally="false" same-size-vertically="false" hgap="-1" vgap="-1">
+        <margin top="0" left="0" bottom="0" right="0"/>
+        <constraints>
+          <grid row="2" column="0" row-span="1" col-span="2" vsize-policy="3" hsize-policy="3" anchor="0" fill="3" indent="0" use-parent-layout="false"/>
+        </constraints>
+        <properties/>
+        <border type="none"/>
+        <children>
+          <component id="9779e" class="javax.swing.JLabel">
+            <constraints>
+              <grid row="0" column="0" row-span="1" col-span="1" vsize-policy="0" hsize-policy="0" anchor="8" fill="0" indent="0" use-parent-layout="false"/>
+            </constraints>
+            <properties>
+              <text value="Models Folder:"/>
+            </properties>
+          </component>
+          <component id="cba56" class="com.intellij.openapi.ui.TextFieldWithBrowseButton" binding="_textModels">
+            <constraints>
+              <grid row="0" column="1" row-span="1" col-span="1" vsize-policy="3" hsize-policy="3" anchor="0" fill="3" indent="0" use-parent-layout="false"/>
+            </constraints>
+            <properties/>
+          </component>
+        </children>
+      </grid>
+      <grid id="39adc" layout-manager="GridLayoutManager" row-count="1" column-count="1" same-size-horizontally="false" same-size-vertically="false" hgap="-1" vgap="-1">
+        <margin top="0" left="0" bottom="0" right="0"/>
+        <constraints>
+          <grid row="0" column="0" row-span="1" col-span="2" vsize-policy="3" hsize-policy="3" anchor="0" fill="3" indent="0" use-parent-layout="false"/>
+        </constraints>
+        <properties/>
+        <border type="none" title="Select Data Source"/>
+        <children>
+          <component id="ad455" class="com.intellij.ui.components.JBList" binding="_listDatasources">
+            <constraints>
+              <grid row="0" column="0" row-span="1" col-span="1" vsize-policy="6" hsize-policy="2" anchor="0" fill="3" indent="0" use-parent-layout="false">
+                <preferred-size width="150" height="50"/>
+              </grid>
+            </constraints>
+            <properties>
+              <model/>
+              <selectionMode value="0"/>
+            </properties>
+          </component>
+        </children>
+      </grid>
+      <grid id="ae3e9" layout-manager="GridLayoutManager" row-count="1" column-count="1" same-size-horizontally="false" same-size-vertically="false" hgap="-1" vgap="-1">
+        <margin top="0" left="0" bottom="0" right="0"/>
+        <constraints>
+          <grid row="1" column="0" row-span="1" col-span="1" vsize-policy="3" hsize-policy="3" anchor="0" fill="3" indent="0" use-parent-layout="false"/>
+        </constraints>
+        <properties/>
+        <border type="none" title="Select Table"/>
+        <children>
+          <component id="81c7e" class="com.intellij.ui.components.JBList" binding="_listTables">
+            <constraints>
+              <grid row="0" column="0" row-span="1" col-span="1" vsize-policy="6" hsize-policy="2" anchor="0" fill="3" indent="0" use-parent-layout="false">
+                <preferred-size width="150" height="50"/>
+              </grid>
+            </constraints>
+            <properties>
+              <selectionMode value="0"/>
+            </properties>
+          </component>
+        </children>
+      </grid>
+      <grid id="cab88" layout-manager="GridLayoutManager" row-count="1" column-count="1" same-size-horizontally="false" same-size-vertically="false" hgap="-1" vgap="-1">
+        <margin top="0" left="0" bottom="0" right="0"/>
+        <constraints>
+          <grid row="1" column="1" row-span="1" col-span="1" vsize-policy="3" hsize-policy="3" anchor="0" fill="3" indent="0" use-parent-layout="false"/>
+        </constraints>
+        <properties/>
+        <border type="none" title="Select Columns To Map"/>
+        <children>
+          <component id="15ba0" class="com.intellij.ui.components.JBList" binding="_listColumns">
+            <constraints>
+              <grid row="0" column="0" row-span="1" col-span="1" vsize-policy="6" hsize-policy="2" anchor="0" fill="3" indent="0" use-parent-layout="false">
+                <preferred-size width="150" height="50"/>
+              </grid>
+            </constraints>
+            <properties/>
+          </component>
+        </children>
+      </grid>
+    </children>
+  </grid>
+</form>