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.List;
  37 import java.util.Map;
  38
  39 import mir.generator.tal.interfaces.TALExpressionParser;
  40 import mir.generator.tal.template.CoreTemplateNodeLibrary;
  41 import mir.generator.tal.template.MacroTemplateNodeLibrary;
  42 import mir.generator.tal.template.Template;
  43 import mir.generator.tal.template.TemplateLibrary;
  44 import mir.generator.tal.template.TemplateNode;
  45 import mir.generator.tal.template.TemplateNodeLibrary;
  46 import mir.generator.tal.template.TemplateProcessingException;
  47 import mir.util.StringRoutines;
  48
  49 public class TALTemplateEngine implements TemplateLibrary {
  50   private TALExpressionParser expressionParser;
  51   private File basePath;
  52   private CachingFileLoader loader;
  53
  54   private SimpleTemplateNodeLibraryRegistry registry;
  55
  56   public TALTemplateEngine(TALExpressionParser anExpressionParser, File aBasePath) {
  57     expressionParser = anExpressionParser;
  58     basePath = aBasePath;
  59     loader = new CachingFileLoader(100, new TemplateFactory());
  60
  61     registry = new SimpleTemplateNodeLibraryRegistry();
  62
  63     registerTemplateNodeLibrary("tal", "http://xml.zope.org/namespaces/tal",
  64         new CoreTemplateNodeLibrary("tal", "http://xml.zope.org/namespaces/tal"));
  65     registerTemplateNodeLibrary("metal", "http://xml.zope.org/namespaces/metal",
  66         new MacroTemplateNodeLibrary("metal", "http://xml.zope.org/namespaces/metal"));
  67   }
  68
  69   public void registerTemplateNodeLibrary(String aPrefix, String aUrl, TemplateNodeLibrary aLibrary) {
  70     registry.registerTemplateNodeLibrary(aPrefix, aUrl, aLibrary);
  71   }
  72
  73   public Template loadTemplate(String aName) throws IOException, TemplateProcessingException {
  74     List parts = StringRoutines.splitString(aName, "#");
  75
  76     if (parts.size()!=1 && parts.size()!=2) {
  77       throw new TemplateProcessingException("Invalid template name: " + aName);
  78     }
  79
  80     File location = new File(basePath, (String) parts.get(0));
  81
  82     Template baseTemplate = (Template) loader.retrieveFile(location.getCanonicalPath());
  83
  84     if (baseTemplate!=null && parts.size()>1) {
  85       Map definitions = (Map) baseTemplate.getContext().get(MacroTemplateNodeLibrary.MACRO_DEFINITIONS_KEY);
  86       if (definitions==null || !definitions.containsKey(parts.get(1))) {
  87         throw new TemplateProcessingException("No macro '"+(String) parts.get(1)+"' found in template '" + (String) parts.get(0) + "'");
  88       }
  89
  90       return new Template(expressionParser, (TemplateNode) definitions.get(parts.get(1)), baseTemplate.getContext());
  91     }
  92                 return baseTemplate;
  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 }