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.samplu.common;
17  
18  import freemarker.template.Configuration;
19  import freemarker.template.Template;
20  import freemarker.template.TemplateException;
21  import org.apache.commons.io.FileUtils;
22  import org.kuali.rice.testtools.common.PropertiesUtils;
23  import org.springframework.ui.freemarker.FreeMarkerTemplateUtils;
24  
25  import java.io.File;
26  import java.io.IOException;
27  import java.io.InputStream;
28  import java.util.Properties;
29  
30  /**
31   * FreemarkerUtil loads properties from user defined InputStream.  systemPropertiesOverride should be used to override properties
32   * using a given key to identify them from JVM args. (i.e. -Dkey.name to override the name property in the key file.)
33   * TODO setup so the loading and overriding of properties is done for the user rather then them having to call it.
34   * @author Kuali Rice Team (rice.collab@kuali.org)
35   */
36  public class FreemarkerUtil {
37  
38      protected Configuration cfg;
39  
40      /**
41       * Calls ftlWrite that also accepts a key, using the output getName as the key.
42       * @param output
43       * @param template
44       * @return
45       * @throws java.io.IOException
46       * @throws freemarker.template.TemplateException
47       */
48      public static File ftlWrite(File output, Template template, InputStream inputStream) throws IOException, TemplateException {
49  
50          return ftlWrite(output.getName(), output, template, inputStream);
51      }
52  
53      /**
54       * Loads properties from user defined properties file, if not available uses resource file
55       *
56       * writes processed template  to file
57       * @param key
58       * @param output
59       * @param template
60       * @throws IOException
61       * @throws TemplateException
62       */
63      public static File ftlWrite(String key, File output, Template template, InputStream inputStream) throws IOException, TemplateException {
64          Properties props = PropertiesUtils.loadProperties(inputStream);
65          props.put("baseName", output.getName().substring(0, output.getName().indexOf("ST")));
66          props.put("className", output.getName().substring(0, output.getName().indexOf("ST"))); // backwards compatibility
67          if (output.getName().contains("TmplMthd")) { // Template method pattern
68              props.put("className", output.getName().substring(0, output.getName().indexOf("TmplMthd")));
69          }
70  
71          if (props.get("test1") == null ) { // backwards compatibility for Smoke Test Freemarker Generation
72              props.put("test1", "test" + props.get("className") + "Bookmark");
73              props.put("test2", "test" + props.get("className") + "Nav");
74          }
75  
76          PropertiesUtils.systemPropertiesOverride(props, key);
77          PropertiesUtils.transformNumberedPropertiesToList(props);
78          File outputFile = writeTemplateToFile(output, template, props);
79  
80          return outputFile;
81      }
82  
83      protected static Properties loadProperties(InputStream inputStream) throws IOException {
84          Properties props = new Properties();
85  
86          if(inputStream != null) {
87              props.load(inputStream);
88          }
89  
90          return props;
91      }
92  
93      /**
94       *
95       * @param file
96       * @param template
97       * @param props
98       * @return
99       * @throws IOException
100      * @throws freemarker.template.TemplateException
101      */
102     protected static File writeTemplateToFile(File file, Template template, Properties props) throws IOException, TemplateException {
103         String output = FreeMarkerTemplateUtils.processTemplateIntoString(template, props);
104         FileUtils.writeStringToFile(file, output);
105 
106         return file;
107     }
108 }