/*
* [The "BSD license"]
* Copyright (c) 2011 Terence Parr
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
* 3. The name of the author may not be used to endorse or promote products
* derived from this software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
* IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
* OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
* IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
* NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
* THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/
package org.stringtemplate.v4;
import org.antlr.runtime.ANTLRInputStream;
import org.stringtemplate.v4.compiler.*;
import org.stringtemplate.v4.misc.*;
import java.io.*;
import java.net.*;
// TODO: caching?
/** A directory or directory tree full of templates and/or group files.
* We load files on-demand. Dir search path: current working dir then
* CLASSPATH (as a resource). Do not look for templates outside of this dir
* subtree (except via imports).
*/
public class STGroupDir extends STGroup {
public String groupDirName;
public URL root;
public STGroupDir(String dirName) { this(dirName, '<', '>'); }
public STGroupDir(String dirName, char delimiterStartChar, char delimiterStopChar) {
super(delimiterStartChar, delimiterStopChar);
this.groupDirName = dirName;
File dir = new File(dirName);
if ( dir.exists() && dir.isDirectory() ) {
// we found the directory and it'll be file based
try {
root = dir.toURI().toURL();
}
catch (MalformedURLException e) {
throw new STException("can't load dir "+dirName, e);
}
if ( verbose ) System.out.println("STGroupDir("+dirName+") found at "+root);
}
else {
ClassLoader cl = Thread.currentThread().getContextClassLoader();
root = cl.getResource(dirName);
if ( root==null ) {
cl = this.getClass().getClassLoader();
root = cl.getResource(dirName);
}
if ( verbose ) System.out.println("STGroupDir("+dirName+") found via CLASSPATH at "+root);
if ( root==null ) {
throw new IllegalArgumentException("No such directory: "+
dirName);
}
}
}
public STGroupDir(String dirName, String encoding) {
this(dirName, encoding, '<', '>');
}
public STGroupDir(String dirName, String encoding,
char delimiterStartChar, char delimiterStopChar)
{
this(dirName, delimiterStartChar, delimiterStopChar);
this.encoding = encoding;
}
public STGroupDir(URL root, String encoding,
char delimiterStartChar, char delimiterStopChar)
{
super(delimiterStartChar, delimiterStopChar);
this.root = root;
this.encoding = encoding;
}
/** Load a template from dir or group file. Group file is given
* precedence over dir with same name. name is always fully-qualified.
*/
@Override
protected CompiledST load(String name) {
if ( verbose ) System.out.println("STGroupDir.load("+name+")");
String parent = Misc.getParent(name); // must have parent; it's fully-qualified
String prefix = Misc.getPrefix(name);
// if (parent.isEmpty()) {
// // no need to check for a group file as name has no parent
// return loadTemplateFile("/", name+".st"); // load t.st file
// }
URL groupFileURL = null;
try { // see if parent of template name is a group file
groupFileURL = new URL(root+parent+".stg");
}
catch (MalformedURLException e) {
errMgr.internalError(null, "bad URL: "+root+parent+".stg", e);
return null;
}
InputStream is = null;
try {
is = groupFileURL.openStream();
}
catch (FileNotFoundException fnfe) {
// must not be in a group file
String unqualifiedName = Misc.getFileName(name);
return loadTemplateFile(prefix, unqualifiedName+".st"); // load t.st file
}
catch (IOException ioe) {
errMgr.internalError(null, "can't load template file "+name, ioe);
}
try { // clean up
if (is!=null ) is.close();
}
catch (IOException ioe) {
errMgr.internalError(null, "can't close template file stream "+name, ioe);
}
loadGroupFile(prefix, root+parent+".stg");
return rawGetTemplate(name);
}
/** Load .st as relative file name relative to root by prefix */
public CompiledST loadTemplateFile(String prefix, String unqualifiedFileName) {
if ( verbose ) System.out.println("loadTemplateFile("+unqualifiedFileName+") in groupdir "+
"from "+root+" prefix="+prefix);
URL f = null;
try {
f = new URL(root+prefix+unqualifiedFileName);
}
catch (MalformedURLException me) {
errMgr.runTimeError(null, null, 0, ErrorType.INVALID_TEMPLATE_NAME,
me, root + unqualifiedFileName);
return null;
}
ANTLRInputStream fs;
try {
fs = new ANTLRInputStream(f.openStream(), encoding);
fs.name = unqualifiedFileName;
}
catch (IOException ioe) {
if ( verbose ) System.out.println(root+"/"+unqualifiedFileName+" doesn't exist");
//errMgr.IOError(null, ErrorType.NO_SUCH_TEMPLATE, ioe, unqualifiedFileName);
return null;
}
return loadTemplateFile(prefix, unqualifiedFileName, fs);
}
public String getName() { return groupDirName; }
public String getFileName() { return root.getFile(); }
@Override
public URL getRootDirURL() { return root; }
}