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 }