1 /*
2 * $Source: /usr/cvsroot/melati/melati/src/main/java/org/melati/template/velocity/WebMacroClasspathResourceLoader.java,v $
3 * $Revision: 1.12 $
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 /**
71 * Get an InputStream so that the Runtime can build a
72 * template with it, munge it if it is a WM template.
73 *
74 * @param templateName name of template to get
75 * @return InputStream containing the template
76 * @throws ResourceNotFoundException if template not found
77 * in the file template path.
78 * @see org.apache.velocity.runtime.resource.loader.ResourceLoader
79 */
80 public InputStream getResourceStream(String templateName)
81 throws ResourceNotFoundException {
82 if (templateName.endsWith(".wm")) {
83 InputStream converted = WebMacroConverter.convert(super.getResourceStream(templateName));
84
85 try {
86 String wmName = "/dist/melati/melati/src/main/java/" + templateName;
87 File wmFile = new File(wmName);
88 String vmName;
89 if (wmFile.exists()) {
90 vmName = "/dist/melati/melati/src/main/java/" +
91 (templateName.endsWith(".wm")?
92 templateName.substring(0,templateName.length() -3) + ".vm" :
93 templateName);
94 } else {
95 wmName = "/dist/melati/melati/src/test/java/" + templateName;
96 wmFile = new File(wmName);
97 if (wmFile.exists()) {
98 vmName = "/dist/melati/melati/src/test/java/" +
99 (templateName.endsWith(".wm")?
100 templateName.substring(0,templateName.length() -3) + ".vm" :
101 templateName);
102 } else
103 throw new RuntimeException("Cannot find file " + wmName);
104 }
105 File convertedFile = new File(vmName).getCanonicalFile();
106 System.err.println(vmName);
107 convertedFile.createNewFile();
108 PrintStream ps = new PrintStream(new FileOutputStream(convertedFile));
109 int nextChar = converted.read();
110 while (nextChar > 0) {
111 ps.write((char)nextChar);
112 nextChar = converted.read();
113 }
114 converted.reset();
115 ps.close();
116 } catch (IOException e) {
117 e.printStackTrace();
118 throw new RuntimeException(e);
119 }
120 return converted;
121 } else
122 return super.getResourceStream(templateName);
123 }
124
125 }