1   package com.atlassian.plugins.codegen.modules.jira;
2   
3   import com.atlassian.plugins.codegen.annotations.Dependencies;
4   import com.atlassian.plugins.codegen.annotations.Dependency;
5   import com.atlassian.plugins.codegen.annotations.JiraPluginModuleCreator;
6   import com.atlassian.plugins.codegen.modules.AbstractPluginModuleCreator;
7   import com.atlassian.plugins.codegen.modules.PluginModuleLocation;
8   
9   /**
10   * @since 3.6
11   */
12  @JiraPluginModuleCreator
13  @Dependencies({
14          @Dependency(groupId = "org.mockito", artifactId = "mockito-all", version = "1.8.5", scope = "test")
15  })
16  public class JqlFunctionModuleCreator extends AbstractPluginModuleCreator<JqlFunctionProperties>
17  {
18  
19      public static final String MODULE_NAME = "JQL Function";
20      private static final String TEMPLATE_PREFIX = "templates/jira/jql/";
21  
22      //stub
23      private static final String CLASS_TEMPLATE = TEMPLATE_PREFIX + "JqlFunction.java.vtl";
24      private static final String UNIT_TEST_TEMPLATE = TEMPLATE_PREFIX + "JqlFunctionTest.java.vtl";
25      private static final String FUNC_TEST_TEMPLATE = TEMPLATE_PREFIX + "JqlFunctionFuncTest.java.vtl";
26  
27      //examples
28      private static final String EXAMPLE_CLASS_TEMPLATE = TEMPLATE_PREFIX + "Example" + CLASS_TEMPLATE;
29  
30      private static final String PLUGIN_MODULE_TEMPLATE = TEMPLATE_PREFIX + "jql-function-plugin.xml.vtl";
31  
32      @Override
33      public void createModule(PluginModuleLocation location, JqlFunctionProperties props) throws Exception
34      {
35          String packageName = props.getPackage();
36  
37          String classname = props.getClassname();
38  
39          if (props.includeExamples())
40          {
41              templateHelper.writeJavaClassFromTemplate(EXAMPLE_CLASS_TEMPLATE, classname, location.getSourceDirectory(), packageName, props);
42          } else
43          {
44              //main class
45              templateHelper.writeJavaClassFromTemplate(CLASS_TEMPLATE, classname, location.getSourceDirectory(), packageName, props);
46  
47              //unit test
48              templateHelper.writeJavaClassFromTemplate(UNIT_TEST_TEMPLATE, testClassname(classname), location.getTestDirectory(), packageName, props);
49  
50              //func test
51              //templateHelper.writeJavaClassFromTemplate(FUNC_TEST_TEMPLATE, funcTestClassname(classname), location.getTestDirectory(), funcTestPackageName(packageName), props);
52          }
53  
54  
55          addModuleToPluginXml(PLUGIN_MODULE_TEMPLATE, location, props);
56      }
57  
58  
59      @Override
60      public String getModuleName()
61      {
62          return MODULE_NAME;
63      }
64  }