aa3fb763c05eeae2fb64ab5d7283c47c0b587b38
[mir.git] / source / mir / generator / tal / TALTemplateEngine.java
1 /*
2  * Copyright (C) 2001, 2002 The Mir-coders group
3  *
4  * This file is part of Mir.
5  *
6  * Mir is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License as published by
8  * the Free Software Foundation; either version 2 of the License, or
9  * (at your option) any later version.
10  *
11  * Mir is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License
17  * along with Mir; if not, write to the Free Software
18  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
19  *
20  * In addition, as a special exception, The Mir-coders gives permission to link
21  * the code of this program with  any library licensed under the Apache Software License,
22  * The Sun (tm) Java Advanced Imaging library (JAI), The Sun JIMI library
23  * (or with modified versions of the above that use the same license as the above),
24  * and distribute linked combinations including the two.  You must obey the
25  * GNU General Public License in all respects for all of the code used other than
26  * the above mentioned libraries.  If you modify this file, you may extend this
27  * exception to your version of the file, but you are not obligated to do so.
28  * If you do not wish to do so, delete this exception statement from your version.
29  */
30
31 package mir.generator.tal;
32
33 import java.io.File;
34 import java.io.IOException;
35 import java.io.InputStream;
36 import java.util.Map;
37 import java.util.List;
38
39 import mir.generator.tal.interfaces.TALExpressionParser;
40 import mir.generator.tal.template.*;
41 import mir.util.StringRoutines;
42
43 public class TALTemplateEngine implements TemplateLibrary {
44   private TALExpressionParser expressionParser;
45   private File basePath;
46   private CachingFileLoader loader;
47
48   private SimpleTemplateNodeLibraryRegistry registry;
49
50   public TALTemplateEngine(TALExpressionParser anExpressionParser, File aBasePath) {
51     expressionParser = anExpressionParser;
52     basePath = aBasePath;
53     loader = new CachingFileLoader(100, new TemplateFactory());
54
55     registry = new SimpleTemplateNodeLibraryRegistry();
56
57     registerTemplateNodeLibrary("tal", "http://xml.zope.org/namespaces/tal",
58         new CoreTemplateNodeLibrary("tal", "http://xml.zope.org/namespaces/tal"));
59     registerTemplateNodeLibrary("metal", "http://xml.zope.org/namespaces/metal",
60         new MacroTemplateNodeLibrary("metal", "http://xml.zope.org/namespaces/metal"));
61   }
62
63   public void registerTemplateNodeLibrary(String aPrefix, String aUrl, TemplateNodeLibrary aLibrary) {
64     registry.registerTemplateNodeLibrary(aPrefix, aUrl, aLibrary);
65   }
66
67   public Template loadTemplate(String aName) throws IOException, TemplateProcessingException {
68     List parts = StringRoutines.splitString(aName, "#");
69
70     if (parts.size()!=1 && parts.size()!=2) {
71       throw new TemplateProcessingException("Invalid template name: " + aName);
72     }
73
74     File location = new File(basePath, (String) parts.get(0));
75
76     Template baseTemplate = (Template) loader.retrieveFile(location.getCanonicalPath());
77
78     if (baseTemplate!=null && parts.size()>1) {
79       Map definitions = (Map) baseTemplate.getContext().get(MacroTemplateNodeLibrary.MACRO_DEFINITIONS_KEY);
80       if (definitions==null || !definitions.containsKey(parts.get(1))) {
81         throw new TemplateProcessingException("No macro '"+(String) parts.get(1)+"' found in template '" + (String) parts.get(0) + "'");
82       }
83
84       return new Template(expressionParser, (TemplateNode) definitions.get(parts.get(1)), baseTemplate.getContext());
85     }
86     else {
87       return baseTemplate;
88     }
89
90
91
92 //      public Template(TALExpressionParser aParser, TemplateNode aRootNode, Map aTemplateContext) {
93   }
94
95   public Template lookupTemplate(String aName) throws TemplateProcessingException {
96     try {
97       return loadTemplate(aName);
98     }
99     catch (IOException e) {
100       throw new TemplateProcessingException("Can't load template " + aName + ": " + e.toString(), e);
101     }
102   }
103
104   private class TemplateFactory implements CachingFileLoader.CachedFileObjectFactory {
105     public Object constructObject(InputStream aStream) {
106       try {
107         TALTemplateParser parser = new TALTemplateParser(registry);
108
109         return parser.parse(aStream, expressionParser);
110       }
111       catch (Throwable t) {
112         throw new TALFailure(t);
113       }
114     }
115   }
116 }