View Javadoc

1   /**
2    * Copyright 2005-2013 The Kuali Foundation
3    *
4    * Licensed under the Educational Community License, Version 2.0 (the "License");
5    * you may not use this file except in compliance with the License.
6    * You may obtain a copy of the License at
7    *
8    * http://www.opensource.org/licenses/ecl2.php
9    *
10   * Unless required by applicable law or agreed to in writing, software
11   * distributed under the License is distributed on an "AS IS" BASIS,
12   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13   * See the License for the specific language governing permissions and
14   * limitations under the License.
15   */
16  package edu.sampleu.common;
17  
18  import freemarker.cache.ClassTemplateLoader;
19  import freemarker.cache.TemplateLoader;
20  import freemarker.template.Configuration;
21  
22  import java.io.File;
23  import java.io.InputStream;
24  
25  /**
26   * @author Kuali Rice Team (rice.collab@kuali.org)
27   */
28  public class FreemarkerAftGenerator {
29      private static Configuration cfg = new Configuration();
30  
31      // Templates for File Generation
32      private static String DIR_TMPL = "/Gen/";
33  
34      //Configuration
35      private static TemplateLoader templateLoader = new ClassTemplateLoader(FreemarkerAftGenerator.class, DIR_TMPL);
36  
37      private static String STJUNITBASE_TMPL = "STJUnitBase.ftl";
38      private static String STJUNITBKMRKGEN_TMPL = "STJUnitBkMrkGen.ftl";
39      private static String STJUNITNAVGEN_TMPL = "STJUnitNavGen.ftl";
40      private static String STNGBASE_TMPL = "STNGBase.ftl";
41      private static String STNGBKMRKGEN_TMPL = "STNGBkMrkGen.ftl";
42      private static String STNGNAVGEN_TMPL = "STNGNavGen.ftl";
43  
44      public static void main(String[] args) throws Exception {
45          cfg.setTemplateLoader(templateLoader);
46  
47          String propsLocation = "/GenFiles/Group.properties";
48          if (args.length > 0) {
49              propsLocation = "/GenFiles/" + args[0];
50          }
51  
52          //Here we can prepare a list of template & properties file and can iterate to generate files dynamically on single run.
53          createFile(propsLocation, STJUNITBASE_TMPL);
54          createFile(propsLocation, STJUNITBKMRKGEN_TMPL);
55          createFile(propsLocation, STJUNITNAVGEN_TMPL);
56          createFile(propsLocation, STNGBASE_TMPL);
57          createFile(propsLocation, STNGBKMRKGEN_TMPL);
58          createFile(propsLocation, STNGNAVGEN_TMPL);
59      }
60  
61      private static void createFile(String propLocation, String template) throws Exception {
62          try {
63              InputStream in = FreemarkerAftGenerator.class.getResourceAsStream(propLocation);
64              File f1 = new File("src" + File.separatorChar + "it" + File.separatorChar + "resources"
65                      + File.separatorChar + "GenFiles" + File.separatorChar
66                      + propLocation.substring(propLocation.lastIndexOf("/"), propLocation.lastIndexOf("."))
67                      + template.substring(0, template.length() - 4) + ".java");
68              FreemarkerUtil.ftlWrite(f1, cfg.getTemplate(template), in);
69  
70          } catch (Exception e) {
71              e.printStackTrace();
72              throw new Exception("Unable to generate files", e);
73          }
74      }
75  }