View Javadoc
1   /*
2    * $Source$
3    * $Revision$
4    *
5    * Copyright (C) 2006 Tim Pizey
6    *
7    * Part of Melati (http://melati.org), a framework for the rapid
8    * development of clean, maintainable web applications.
9    *
10   * Melati is free software; Permission is granted to copy, distribute
11   * and/or modify this software under the terms either:
12   *
13   * a) the GNU General Public License as published by the Free Software
14   *    Foundation; either version 2 of the License, or (at your option)
15   *    any later version,
16   *
17   *    or
18   *
19   * b) any version of the Melati Software License, as published
20   *    at http://melati.org
21   *
22   * You should have received a copy of the GNU General Public License and
23   * the Melati Software License along with this program;
24   * if not, write to the Free Software Foundation, Inc.,
25   * 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA to obtain the
26   * GNU General Public License and visit http://melati.org to obtain the
27   * Melati Software License.
28   *
29   * Feel free to contact the Developers of Melati (http://melati.org),
30   * if you would like to work out a different arrangement than the options
31   * outlined here.  It is our intention to allow Melati to be used by as
32   * wide an audience as possible.
33   *
34   * This program is distributed in the hope that it will be useful,
35   * but WITHOUT ANY WARRANTY; without even the implied warranty of
36   * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
37   * GNU General Public License for more details.
38   *
39   * Contact details for copyright holder:
40   *
41   *     Tim Pizey <timp At paneris.org>
42   *     http://paneris.org/~timp
43   */
44  package org.melati.template.velocity;
45  
46  import java.io.File;
47  import java.io.FileOutputStream;
48  import java.io.IOException;
49  import java.io.InputStream;
50  import java.io.PrintStream;
51  
52  import org.apache.velocity.exception.ResourceNotFoundException;
53  import org.apache.velocity.runtime.resource.loader.ClasspathResourceLoader;
54  
55  /**
56   * Loads templates from the classpath, mungs them if they are WM 
57   * templates.
58   * 
59   * 
60   * NOTE This does not allow modern WebMacro syntax with
61   * optional #begin in #foreach.
62   * 
63   * NOTE Only if a curly bracket is at the start of a line will it not be turned into an #end
64   *
65   * @author Tim Pizey based on work by Jason van Zyl and Tim Joyce.
66   */
67  public class WebMacroClasspathResourceLoader 
68      extends ClasspathResourceLoader {
69    
70    private static final String MELATI_SRC_TEST_JAVA = "/dist/melati/melati/src/test/java/";
71    private static final String MELATI_SRC_MAIN_JAVA = "/dist/melati/melati/src/main/java/";
72  
73    /**
74     * Get an InputStream so that the Velocity Runtime can build a
75     * template with it, munge it if it is a WM template.
76     *
77     * FIXME Hardcoded paths 
78     *
79     * @param templateName name of template to get
80     * @return InputStream containing the template
81     * @throws ResourceNotFoundException if template not found
82     *         in the file template path.
83     * @see org.apache.velocity.runtime.resource.loader.ResourceLoader
84     */
85    public InputStream getResourceStream(String templateName)
86        throws ResourceNotFoundException {
87      if (templateName.endsWith(".wm")) {
88        InputStream converted = WebMacroConverter.convert(super.getResourceStream(templateName));
89        
90        try {
91          String wmName = MELATI_SRC_MAIN_JAVA + templateName;
92          File wmFile = new File(wmName);
93          String vmName;
94          if (wmFile.exists()) {
95            vmName = MELATI_SRC_MAIN_JAVA + 
96                  templateName.substring(0,templateName.length() -3) + ".vm";
97          } else { 
98            wmName = MELATI_SRC_TEST_JAVA + templateName;
99            wmFile = new File(wmName);
100           if (wmFile.exists()) {
101             vmName = MELATI_SRC_TEST_JAVA + 
102                   templateName.substring(0,templateName.length() -3) + ".vm";
103           } else 
104             throw new RuntimeException("Cannot find file " + wmName);
105         }
106         File convertedFile = new File(vmName).getCanonicalFile();
107         // System.err.println(vmName);
108         convertedFile.createNewFile();
109         PrintStream ps = new PrintStream(new FileOutputStream(convertedFile));
110         int nextChar = converted.read();
111         while (nextChar > 0) {
112           ps.write((char)nextChar);
113           nextChar = converted.read();
114         }
115         converted.reset();
116         ps.close();
117       } catch (IOException e) {
118         e.printStackTrace();
119         throw new RuntimeException(e);
120       }
121       return converted;
122     } else
123       return super.getResourceStream(templateName);
124   }
125 
126 }