Merge pull request #1 from lotabout/topic-workflow

Topic workflow

Author: lotabout

.gitignore

@@ -1,2 +1,3 @@
 .idea
 out
+target

CodeGenerator.iml

@@ -9,6 +9,5 @@
     </content>
     <orderEntry type="inheritedJdk" />
     <orderEntry type="sourceFolder" forTests="false" />
-    <orderEntry type="library" name="innerbuilder" level="project" />
   </component>
-</module>
+</module>

resources/META-INF/plugin.xml

@@ -1,7 +1,7 @@
 <idea-plugin>
   <id>me.lotabout.codegenerator</id>
   <name>CodeGenerator</name>
-  <version>1.0</version>
+  <version>1.1</version>
   <vendor email="lotabout@gmail.com" url="https://lotabout.me">Personal</vendor>
 
   <description><![CDATA[
@@ -20,6 +20,7 @@
 
   <change-notes><![CDATA[
   <ul>
+  <li>version 1.1 <p>Able to define workflows for selecting members/classes</p></li>
   <li>version 1.0 <p>First release</p></li>
   </ul>
     ]]>

resources/template/default.vm

@@ -118,6 +118,7 @@
 ## - PsiShortNamesCache: Class   utility to search classes
 ## - PsiJavaPsiFacade: Class     Java specific utility to search classes
 ## - GlobalSearchScope: Class    class to create search scopes, used by above utilities
+## - EntryFactory: Class         EntryFactory.of(...) to turn PsiXXX to XXXEntry.
 ##
 ## References
 ## ----------

src/me/lotabout/codegenerator/CodeGeneratorSettings.java

@@ -7,6 +7,8 @@
 import com.intellij.openapi.util.io.FileUtil;
 import com.intellij.util.xmlb.XmlSerializerUtil;
 import me.lotabout.codegenerator.config.CodeTemplate;
+import me.lotabout.codegenerator.config.MemberSelectionConfig;
+import me.lotabout.codegenerator.config.PipelineStep;
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.annotations.Nullable;
 
@@ -19,13 +21,13 @@
 public class CodeGeneratorSettings implements PersistentStateComponent<CodeGeneratorSettings> {
 
     private static final Logger LOGGER = Logger.getInstance(CodeGeneratorSettings.class);
-    private Map<String, CodeTemplate> codeTemplates;
+    private List<CodeTemplate> codeTemplates;
 
     public CodeGeneratorSettings() {
 
     }
 
-    public CodeGeneratorSettings setCodeTemplates(Map<String, CodeTemplate> codeTemplates) {
+    public CodeGeneratorSettings setCodeTemplates(List<CodeTemplate> codeTemplates) {
         this.codeTemplates = codeTemplates;
         return this;
     }
@@ -42,39 +44,42 @@ public CodeGeneratorSettings setCodeTemplates(Map<String, CodeTemplate> codeTemp
         XmlSerializerUtil.copyBean(codeGeneratorSettings, this);
     }
 
-    public Map<String, CodeTemplate> getCodeTemplates() {
+    public List<CodeTemplate> getCodeTemplates() {
         if (codeTemplates == null) {
             codeTemplates = loadDefaultTemplates();
         }
         return codeTemplates;
     }
 
     public Optional<CodeTemplate> getCodeTemplate(String templateId) {
-        return Optional.ofNullable(codeTemplates.get(templateId));
+        return codeTemplates.stream()
+                .filter(t -> t!= null && t.getId().equals(templateId))
+                .findFirst();
     }
 
     public void removeCodeTemplate(String templateId) {
-        codeTemplates.remove(templateId);
+        codeTemplates.removeIf(template -> template.name.equals(templateId));
     }
 
-    private Map<String, CodeTemplate> loadDefaultTemplates() {
+    private List<CodeTemplate> loadDefaultTemplates() {
         List<CodeTemplate> templates = new ArrayList<>();
         try {
-            templates.add(createTemplate("HUESerialization", "body"));
+            templates.add(createTemplate("HUESerialization", "body", Collections.singletonList(new MemberSelectionConfig())));
         } catch (Exception e) {
             LOGGER.error("loadDefaultTemplates failed", e);
         }
 
-        return templates.stream().collect(Collectors.toMap(CodeTemplate::getId, Function.identity()));
+        return templates;
     }
 
     @NotNull
-    private CodeTemplate createTemplate(String name, String type) throws IOException {
+    private CodeTemplate createTemplate(String name, String type, List<PipelineStep> pipeline) throws IOException {
         String velocityTemplate = FileUtil.loadTextAndClose(CodeGeneratorSettings.class.getResourceAsStream("/template/" + name + ".vm"));
         CodeTemplate codeTemplate = new CodeTemplate();
         codeTemplate.type = type;
         codeTemplate.name = name;
         codeTemplate.template = velocityTemplate;
+        codeTemplate.pipeline.addAll(pipeline);
         return codeTemplate;
     }