generate code

.idea/misc.xml

@@ -3,13 +3,6 @@
   <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" />
@@ -20,7 +13,7 @@
     <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">
+  <component name="ProjectRootManager" version="2" languageLevel="JDK_1_3" default="false" assert-keyword="false" jdk-15="false" project-jdk-name="IntelliJ IDEA RS-163.7608" project-jdk-type="IDEA JDK">
     <output url="file://$PROJECT_DIR$/out" />
   </component>
 </project>

code-from-ds.iml

@@ -1,7 +1,7 @@
 <?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">
+  <component name="NewModuleRootManager" LANGUAGE_LEVEL="JDK_1_8" inherit-compiler-output="true">
     <exclude-output />
     <content url="file://$MODULE_DIR$">
       <sourceFolder url="file://$MODULE_DIR$/src" isTestSource="false" />
@@ -9,6 +9,7 @@
     </content>
     <orderEntry type="inheritedJdk" />
     <orderEntry type="sourceFolder" forTests="false" />
-    <orderEntry type="library" name="json-20160810" level="project" />
+    <orderEntry type="library" name="org.json:json:20160810" level="project" />
+    <orderEntry type="library" name="jtwig-core" level="project" />
   </component>
 </module>

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

@@ -2,9 +2,14 @@ package com.rthoni.intellij.codefromds.business;
 
 import com.intellij.database.psi.DbDataSource;
 import com.rthoni.intellij.codefromds.dbo.GenerateOptions;
+import com.rthoni.intellij.codefromds.dbo.TableSelection;
 import org.json.JSONObject;
+import org.jtwig.JtwigModel;
+import org.jtwig.JtwigTemplate;
 
+import java.io.File;
 import java.io.FileOutputStream;
+import java.io.IOException;
 import java.nio.charset.StandardCharsets;
 import java.nio.file.Files;
 import java.nio.file.Paths;
@@ -15,7 +20,8 @@ import java.util.HashMap;
  */
 public abstract class Generator {
 
-    public static void saveOptions(GenerateOptions options) throws Exception {
+    public static void saveOptions(GenerateOptions options) throws Exception
+    {
         HashMap<String, Object> map = options.toMap();
         JSONObject obj = new JSONObject(map);
         FileOutputStream file = new FileOutputStream(options.getConfigPath());
@@ -23,8 +29,9 @@ public abstract class Generator {
         file.close();
     }
 
-    public static GenerateOptions loadOptions(String configPath) throws Exception {
-        String data = Files.readAllLines(Paths.get(configPath), StandardCharsets.UTF_8).stream().reduce("", (s1, s2) -> s1 + s2);
+    public static GenerateOptions loadOptions(String configPath) throws Exception
+    {
+        String data = Files.readAllLines(Paths.get(configPath), StandardCharsets.UTF_8).stream().reduce("", (s1, s2) -> s1 + s2 + "\n");
 
         JSONObject obj = new JSONObject(data);
         String src = obj.getJSONObject("selection").getString("source");
@@ -40,8 +47,27 @@ public abstract class Generator {
         return options;
     }
 
-    public static void generate(GenerateOptions options)
+    public static void generateFile(String templatePath, String outputFile, GenerateOptions options, TableSelection table) throws IOException
     {
+        String data = Helper.readFile(templatePath);
+
+        FileOutputStream file = new FileOutputStream(outputFile);
+
+        JtwigTemplate template = JtwigTemplate.inlineTemplate(data);
+        JtwigModel model = JtwigModel.newModel().with("options", options).with("table", table);
+
+        template.render(model, file);
 
+        file.close();
+    }
+
+    public static void generate(GenerateOptions options) throws IOException
+    {
+        generateFile(options.getDataSourceTemplatePath(), options.getModelsPath() + File.separator + "Database.cs", options, null);
+        for (TableSelection table : options.getSelection().getTables()) {
+            if (!table.hasNone()) {
+                generateFile(options.getModelsTemplatePath(), options.getModelsPath() + File.separator + table.getTable().getName() + ".cs", options, table);
+            }
+        }
     }
 }

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

@@ -8,6 +8,10 @@ import com.rthoni.intellij.codefromds.dbo.TableSelection;
 import org.json.JSONArray;
 import org.json.JSONObject;
 
+import java.io.IOException;
+import java.nio.charset.StandardCharsets;
+import java.nio.file.Files;
+import java.nio.file.Paths;
 import java.util.Arrays;
 import java.util.Collection;
 import java.util.List;
@@ -55,4 +59,9 @@ public abstract class Helper {
         }
         return null;
     }
+
+    public static String readFile(String path) throws IOException {
+        return Files.readAllLines(Paths.get(path), StandardCharsets.UTF_8)
+                .stream().reduce("", (s1, s2) -> s1 + s2 + "\n");
+    }
 }

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

@@ -42,7 +42,11 @@ public class GenerateAction extends AnAction {
             } catch (Exception e1) {
                 e1.printStackTrace();
             }
-            Generator.generate(options);
+            try {
+                Generator.generate(options);
+            } catch (Exception e1) {
+                e1.printStackTrace();
+            }
         }
     }