1.1 restoration

[mir.git] / source / mir / generator / tal / TALTemplateEngine.java

diff --git a/source/mir/generator/tal/TALTemplateEngine.java b/source/mir/generator/tal/TALTemplateEngine.java
index 13125bf..aa3fb76 100755 (executable)
--- a/source/mir/generator/tal/TALTemplateEngine.java
+++ b/source/mir/generator/tal/TALTemplateEngine.java
@@ -33,29 +33,80 @@ package mir.generator.tal;
 import java.io.File;
 import java.io.IOException;
 import java.io.InputStream;
+import java.util.Map;
+import java.util.List;
 
 import mir.generator.tal.interfaces.TALExpressionParser;
+import mir.generator.tal.template.*;
+import mir.util.StringRoutines;
 
-public class TALTemplateEngine {
+public class TALTemplateEngine implements TemplateLibrary {
   private TALExpressionParser expressionParser;
-  private String basePath;
+  private File basePath;
   private CachingFileLoader loader;
 
-  public TALTemplateEngine(TALExpressionParser anExpressionParser, String aBasePath) {
+  private SimpleTemplateNodeLibraryRegistry registry;
+
+  public TALTemplateEngine(TALExpressionParser anExpressionParser, File aBasePath) {
     expressionParser = anExpressionParser;
     basePath = aBasePath;
     loader = new CachingFileLoader(100, new TemplateFactory());
+
+    registry = new SimpleTemplateNodeLibraryRegistry();
+
+    registerTemplateNodeLibrary("tal", "http://xml.zope.org/namespaces/tal",
+        new CoreTemplateNodeLibrary("tal", "http://xml.zope.org/namespaces/tal"));
+    registerTemplateNodeLibrary("metal", "http://xml.zope.org/namespaces/metal",
+        new MacroTemplateNodeLibrary("metal", "http://xml.zope.org/namespaces/metal"));
+  }
+
+  public void registerTemplateNodeLibrary(String aPrefix, String aUrl, TemplateNodeLibrary aLibrary) {
+    registry.registerTemplateNodeLibrary(aPrefix, aUrl, aLibrary);
+  }
+
+  public Template loadTemplate(String aName) throws IOException, TemplateProcessingException {
+    List parts = StringRoutines.splitString(aName, "#");
+
+    if (parts.size()!=1 && parts.size()!=2) {
+      throw new TemplateProcessingException("Invalid template name: " + aName);
+    }
+
+    File location = new File(basePath, (String) parts.get(0));
+
+    Template baseTemplate = (Template) loader.retrieveFile(location.getCanonicalPath());
+
+    if (baseTemplate!=null && parts.size()>1) {
+      Map definitions = (Map) baseTemplate.getContext().get(MacroTemplateNodeLibrary.MACRO_DEFINITIONS_KEY);
+      if (definitions==null || !definitions.containsKey(parts.get(1))) {
+        throw new TemplateProcessingException("No macro '"+(String) parts.get(1)+"' found in template '" + (String) parts.get(0) + "'");
+      }
+
+      return new Template(expressionParser, (TemplateNode) definitions.get(parts.get(1)), baseTemplate.getContext());
+    }
+    else {
+      return baseTemplate;
+    }
+
+
+
+//      public Template(TALExpressionParser aParser, TemplateNode aRootNode, Map aTemplateContext) {
   }
 
-  public TALTemplate loadTemplate(String aName) throws IOException {
-    File location = new File(basePath, aName);
-    return (TALTemplate) loader.retrieveFile(location.getCanonicalPath());
+  public Template lookupTemplate(String aName) throws TemplateProcessingException {
+    try {
+      return loadTemplate(aName);
+    }
+    catch (IOException e) {
+      throw new TemplateProcessingException("Can't load template " + aName + ": " + e.toString(), e);
+    }
   }
 
   private class TemplateFactory implements CachingFileLoader.CachedFileObjectFactory {
     public Object constructObject(InputStream aStream) {
       try {
-        return TALTemplateParser.parse(aStream, expressionParser);
+        TALTemplateParser parser = new TALTemplateParser(registry);
+
+        return parser.parse(aStream, expressionParser);
       }
       catch (Throwable t) {
         throw new TALFailure(t);