foreign keys

src/com/rthoni/intellij/codefromds/business/Generator.java

@@ -1,5 +1,7 @@
 package com.rthoni.intellij.codefromds.business;
 
+import com.intellij.database.model.DasForeignKey;
+import com.intellij.database.model.MultiRef;
 import com.intellij.database.psi.DbDataSource;
 import com.intellij.database.util.DasUtil;
 import com.intellij.openapi.project.Project;
@@ -8,6 +10,7 @@ import com.rthoni.intellij.codefromds.dbo.options.GenerateOptions;
 import com.rthoni.intellij.codefromds.dbo.options.TableSelection;
 import com.rthoni.intellij.codefromds.dbo.template.ColumnDataSourceDbo;
 import com.rthoni.intellij.codefromds.dbo.template.DataSourceDbo;
+import com.rthoni.intellij.codefromds.dbo.template.ForeignKeyDbo;
 import com.rthoni.intellij.codefromds.dbo.template.TableDataSourceDbo;
 import org.json.JSONObject;
 import org.jtwig.JtwigModel;
@@ -20,7 +23,10 @@ import java.nio.charset.StandardCharsets;
 import java.nio.file.Files;
 import java.nio.file.Paths;
 import java.util.HashMap;
+import java.util.List;
+import java.util.Vector;
 import java.util.stream.Collectors;
+import java.util.stream.StreamSupport;
 
 /**
  * Created by robin on 11/15/16.
@@ -54,6 +60,27 @@ public abstract class Generator {
         return options;
     }
 
+    public static ForeignKeyDbo convertForeignKey(DasForeignKey fk, DataSourceDbo source)
+    {
+        ForeignKeyDbo dbo = new ForeignKeyDbo();
+
+        dbo.setName(fk.getName());
+
+        dbo.setSourceForeignKeyName("fk_" + fk.getRefTable().getName());
+        dbo.setSourceTable(source.getTables().stream().filter(t -> t.getName().equals(fk.getTable().getName())).findFirst().get());
+        MultiRef columnsRef = fk.getColumnsRef();
+        dbo.setSourceColumns(dbo.getSourceTable().getColumns().stream().filter(c -> StreamSupport.stream(columnsRef.names().spliterator(), false).anyMatch(cc -> cc.equals(c.getName()))).collect(Collectors.toList()));
+
+        dbo.setTargetForeignKeyName(fk.getTable().getName() + "_fk");
+        dbo.setTargetTable(source.getTables().stream().filter(t -> t.getName().equals(fk.getRefTable().getName())).findFirst().get());
+        MultiRef refColumns = fk.getRefColumns();
+        dbo.setTargetColumns(dbo.getTargetTable().getColumns().stream().filter(c -> StreamSupport.stream(refColumns.names().spliterator(), false).anyMatch(cc -> cc.equals(c.getName()))).collect(Collectors.toList()));
+
+        dbo.getTargetTable().getTargetForeignKeys().add(dbo);
+
+        return dbo;
+    }
+
     public static ColumnDataSourceDbo convertColumn(ColumnSelection columnSelection)
     {
         ColumnDataSourceDbo dbo = new ColumnDataSourceDbo();
@@ -68,8 +95,9 @@ public abstract class Generator {
         TableDataSourceDbo dbo = new TableDataSourceDbo();
         dbo.setName(tableSelection.getTable().getName());
         dbo.setColumns(tableSelection.getColumns().stream().map(Generator::convertColumn).collect(Collectors.toList()));
-        dbo.setPrimaryKeys(tableSelection.getColumns().stream().filter(c -> DasUtil.isPrimary(c.getColumn())).map(Generator::convertColumn).collect(Collectors.toList()));
+        dbo.setPrimaryKeys(dbo.getColumns().stream().filter(ColumnDataSourceDbo::isPrimary).collect(Collectors.toList()));
         dbo.setHasAny(!tableSelection.hasNone());
+        dbo.setTargetForeignKeys(new Vector<>());
         return dbo;
     }
 
@@ -80,6 +108,14 @@ public abstract class Generator {
         dbo.setName(options.getSelection().getSource().getName());
         dbo.setTables(options.getSelection().getTables().stream().map(Generator::convertTable).collect(Collectors.toList()));
 
+        List<TableDataSourceDbo> tables = dbo.getTables();
+        List<TableSelection> tableSelections = options.getSelection().getTables();
+        for (int i = 0; i < tables.size(); ++i) {
+            TableDataSourceDbo table = tables.get(i);
+            TableSelection tableSelection = tableSelections.get(i);
+            table.setSourceForeignKeys(DasUtil.getForeignKeys(tableSelection.getTable()).toList().stream().map(t -> convertForeignKey(t, dbo)).collect(Collectors.toList()));
+        }
+
         return dbo;
     }
 

src/com/rthoni/intellij/codefromds/business/Helper.java

@@ -105,4 +105,12 @@ public abstract class Helper {
     {
         return getAbsolutePath(project.getBasePath(), relative);
     }
+
+    public static String pluralize(String str)
+    {
+        if (str.endsWith("y")) {
+            return str.substring(1, str.length() - 1) + "ies";
+        }
+        return str + "s";
+    }
 }

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

@@ -14,10 +14,6 @@ public class ColumnSelection {
 
     private boolean _selected;
 
-    public ColumnSelection() {
-
-    }
-
     public ColumnSelection(DasColumn column) {
         _column = column;
         _selected = true;

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

@@ -1,6 +1,5 @@
 package com.rthoni.intellij.codefromds.dbo.options;
 
-import com.intellij.database.model.DasTable;
 import com.intellij.database.psi.DbDataSource;
 import com.intellij.database.util.DasUtil;
 import com.rthoni.intellij.codefromds.business.Helper;
@@ -23,14 +22,7 @@ public class DataSourceSelection {
 
     public DataSourceSelection(DbDataSource source) {
         _source = source;
-        final List<? extends DasTable> tables = DasUtil.getTables(source).toList();
-        _tables = new Vector<>();
-        for (DasTable table : tables) {
-            TableSelection tableSelection = new TableSelection();
-            tableSelection.setTable(table);
-            tableSelection.setColumns(DasUtil.getColumns(table).toList().stream().map(ColumnSelection::new).collect(Collectors.toList()));
-            _tables.add(tableSelection);
-        }
+        _tables = DasUtil.getTables(source).toList().stream().map(TableSelection::new).collect(Collectors.toList());
     }
 
     public HashMap<String, Object> toMap()

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

@@ -36,21 +36,6 @@ public class GenerateOptions {
         _configAbsolutePath = source.getProject().getBasePath() + File.separator + "code-from-ds.json";
         _filesExtension = "cs";
         _selection = new DataSourceSelection(source);
-        _selection.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);
-        }
-        _selection.setTables(tableSelections);
     }
 
     public HashMap<String, Object> toMap()

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

@@ -1,6 +1,7 @@
 package com.rthoni.intellij.codefromds.dbo.options;
 
 import com.intellij.database.model.DasTable;
+import com.intellij.database.util.DasUtil;
 import com.rthoni.intellij.codefromds.business.Helper;
 import org.json.JSONArray;
 import org.json.JSONObject;
@@ -8,6 +9,7 @@ import org.json.JSONObject;
 import java.util.HashMap;
 import java.util.List;
 import java.util.Vector;
+import java.util.stream.Collectors;
 
 /**
  * Created by robin on 11/15/16.
@@ -18,6 +20,12 @@ public class TableSelection {
 
     private List<ColumnSelection> _columns;
 
+    public TableSelection(DasTable table)
+    {
+        _table = table;
+        _columns = DasUtil.getColumns(table).toList().stream().map(ColumnSelection::new).collect(Collectors.toList());
+    }
+
     public HashMap<String, Object> toMap()
     {
         HashMap<String, Object> map = new HashMap<>();

src/com/rthoni/intellij/codefromds/dbo/template/ForeignKeyDbo.java

@@ -0,0 +1,85 @@
+package com.rthoni.intellij.codefromds.dbo.template;
+
+import java.util.List;
+
+/**
+ * Created by robin on 11/18/16.
+ */
+public class ForeignKeyDbo {
+
+    private TableDataSourceDbo _sourceTable;
+
+    private List<ColumnDataSourceDbo> _sourceColumns;
+
+    private TableDataSourceDbo _targetTable;
+
+    private List<ColumnDataSourceDbo> _targetColumns;
+
+    private String _sourceForeignKeyName;
+
+    private String _targetForeignKeyName;
+
+    private String _name;
+
+    public TableDataSourceDbo getSourceTable() {
+        return _sourceTable;
+    }
+
+    public boolean isSelected()
+    {
+        return _sourceColumns.stream().allMatch(ColumnDataSourceDbo::isSelected) &&
+                _targetColumns.stream().allMatch(ColumnDataSourceDbo::isSelected);
+    }
+
+    public void setSourceTable(TableDataSourceDbo sourceTable) {
+        _sourceTable = sourceTable;
+    }
+
+    public List<ColumnDataSourceDbo> getSourceColumns() {
+        return _sourceColumns;
+    }
+
+    public void setSourceColumns(List<ColumnDataSourceDbo> sourceColumns) {
+        _sourceColumns = sourceColumns;
+    }
+
+    public TableDataSourceDbo getTargetTable() {
+        return _targetTable;
+    }
+
+    public void setTargetTable(TableDataSourceDbo targetTable) {
+        _targetTable = targetTable;
+    }
+
+    public List<ColumnDataSourceDbo> getTargetColumns() {
+        return _targetColumns;
+    }
+
+    public void setTargetColumns(List<ColumnDataSourceDbo> targetColumns) {
+        _targetColumns = targetColumns;
+    }
+
+    public String getSourceForeignKeyName() {
+        return _sourceForeignKeyName;
+    }
+
+    public void setSourceForeignKeyName(String sourceForeignKeyName) {
+        _sourceForeignKeyName = sourceForeignKeyName;
+    }
+
+    public String getTargetForeignKeyName() {
+        return _targetForeignKeyName;
+    }
+
+    public void setTargetForeignKeyName(String targetForeignKeyName) {
+        _targetForeignKeyName = targetForeignKeyName;
+    }
+
+    public String getName() {
+        return _name;
+    }
+
+    public void setName(String name) {
+        _name = name;
+    }
+}

src/com/rthoni/intellij/codefromds/dbo/template/TableDataSourceDbo.java

@@ -12,6 +12,10 @@ public class TableDataSourceDbo {
 
     private List<ColumnDataSourceDbo> _primaryKeys;
 
+    private List<ForeignKeyDbo> _sourceForeignKeys;
+
+    private List<ForeignKeyDbo> _targetForeignKeys;
+
     private boolean _hasAny;
 
     public String getName() {
@@ -45,4 +49,20 @@ public class TableDataSourceDbo {
     public void setHasAny(boolean hasAny) {
         _hasAny = hasAny;
     }
+
+    public List<ForeignKeyDbo> getSourceForeignKeys() {
+        return _sourceForeignKeys;
+    }
+
+    public void setSourceForeignKeys(List<ForeignKeyDbo> sourceForeignKeys) {
+        _sourceForeignKeys = sourceForeignKeys;
+    }
+
+    public List<ForeignKeyDbo> getTargetForeignKeys() {
+        return _targetForeignKeys;
+    }
+
+    public void setTargetForeignKeys(List<ForeignKeyDbo> targetForeignKeys) {
+        _targetForeignKeys = targetForeignKeys;
+    }
 }