1   package com.atlassian.plugins.codegen.modules.jira;
2   
3   import com.atlassian.plugins.codegen.PluginProjectChangeset;
4   import com.atlassian.plugins.codegen.annotations.JiraPluginModuleCreator;
5   import com.atlassian.plugins.codegen.modules.AbstractPluginModuleCreator;
6   import com.atlassian.plugins.codegen.modules.common.Resource;
7   
8   import static com.atlassian.plugins.codegen.modules.Dependencies.MOCKITO_TEST;
9   
10  /**
11   * @since 3.6
12   */
13  @JiraPluginModuleCreator
14  public class CustomFieldModuleCreator extends AbstractPluginModuleCreator<CustomFieldProperties>
15  {
16      public static final String MODULE_NAME = "Custom Field";
17      private static final String TEMPLATE_PREFIX = "templates/jira/customfield/";
18  
19      //stub
20      private static final String CLASS_TEMPLATE = TEMPLATE_PREFIX + "CustomField.java.vtl";
21      private static final String UNIT_TEST_TEMPLATE = "templates/generic/GenericTest.java.vtl";
22      private static final String VIEW_TEMPLATE = "templates/common/actionview.vm.vtl";
23  
24      //examples
25      private static final String EXAMPLE_CLASS_TEMPLATE = TEMPLATE_PREFIX + "Example" + CLASS_TEMPLATE;
26  
27      private static final String PLUGIN_MODULE_TEMPLATE = TEMPLATE_PREFIX + "custom-field-plugin.xml.vtl";
28  
29      @Override
30      public PluginProjectChangeset createModule(CustomFieldProperties props) throws Exception
31      {
32          PluginProjectChangeset ret = new PluginProjectChangeset()
33              .withDependencies(MOCKITO_TEST)
34              .with(createModule(props, PLUGIN_MODULE_TEMPLATE));
35          
36          if (props.includeExamples())
37          {
38              return ret.with(createClass(props, EXAMPLE_CLASS_TEMPLATE));
39          }
40          else
41          {
42              ret = ret.with(createClassAndTests(props, CLASS_TEMPLATE, UNIT_TEST_TEMPLATE));
43              
44              //since we know resources are velocity templates, let's create them
45              for (Resource resource : props.getResources())
46              {
47                  ret = ret.with(createTemplateResource(props, resource, VIEW_TEMPLATE));
48              }
49              
50              return ret;
51          }
52      }
53  
54      @Override
55      public String getModuleName()
56      {
57          return MODULE_NAME;
58      }
59  }