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 }