xmltooling/util/TemplateEngine.h

   1 /*
   2  *  Copyright 2001-2007 Internet2
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *     http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 /**
  18  * @file xmltooling/util/TemplateEngine.h
  19  *
  20  * Simple template replacement engine.
  21  */
  22
  23 #ifndef __xmltooling_template_h__
  24 #define __xmltooling_template_h__
  25
  26 #include <xmltooling/base.h>
  27
  28 #include <map>
  29 #include <string>
  30 #include <iostream>
  31
  32 #if defined (_MSC_VER)
  33     #pragma warning( push )
  34     #pragma warning( disable : 4251 )
  35 #endif
  36
  37 namespace xmltooling {
  38
  39     /**
  40      * Simple template replacement engine. Supports the following:
  41      * <ul>
  42      *  <li> &lt;mlp key/&gt; </li>
  43      *  <li> &lt;mlpif key&gt; stuff &lt;/mlpif&gt;</li>
  44      *  <li> &lt;mlpifnot key&gt; stuff &lt;/mlpifnot&gt;</li>
  45      * </ul>
  46      *
  47      * The default tag prefix is "mlp". This can be overridden for
  48      * compatibility.
  49      */
  50     class XMLTOOL_API TemplateEngine
  51     {
  52         MAKE_NONCOPYABLE(TemplateEngine);
  53     public:
  54
  55         TemplateEngine() {
  56             setTagPrefix("mlp");
  57         }
  58
  59         virtual ~TemplateEngine() {}
  60
  61         /**
  62          * Sets the tag name to use when locating template replacement tags.
  63          *
  64          * @param tagPrefix base prefix for tags
  65          */
  66         void setTagPrefix(const char* tagPrefix);
  67
  68         /**
  69          * Interface to parameters to plug into templates.
  70          * Allows callers to supply a more dynamic lookup mechanism to supplement a basic map.
  71          */
  72         class XMLTOOL_API TemplateParameters {
  73             MAKE_NONCOPYABLE(TemplateParameters);
  74         public:
  75             TemplateParameters() {}
  76             virtual ~TemplateParameters() {}
  77
  78             /** Map of known parameters to supply to template. */
  79             std::map<std::string,std::string> m_map;
  80
  81             /**
  82              * Returns the value of a parameter to plug into the template.
  83              *
  84              * @param name  name of parameter
  85              * @return value of parameter, or NULL
  86              */
  87             virtual const char* getParameter(const char* name) const {
  88                 std::map<std::string,std::string>::const_iterator i=m_map.find(name);
  89                 return (i!=m_map.end() ? i->second.c_str() : NULL);
  90             }
  91         };
  92
  93         /**
  94          * Processes template from an input stream and executes replacements and
  95          * conditional logic based on parameters.
  96          *
  97          * @param is            input stream providing template
  98          * @param os            output stream to send results of executing template
  99          * @param parameters    parameters to plug into template
 100          * @param e             optional exception to extract parameters from
 101          */
 102         virtual void run(
 103             std::istream& is,
 104             std::ostream& os,
 105             const TemplateParameters& parameters,
 106             const XMLToolingException* e=NULL
 107             ) const;
 108
 109     private:
 110         void trimspace(std::string& s) const;
 111         void html_encode(std::ostream& os, const char* start) const;
 112         void process(
 113             bool visible,
 114             const std::string& buf,
 115             const char*& lastpos,
 116             std::ostream& os,
 117             const TemplateParameters& parameters,
 118             const XMLToolingException* e
 119             ) const;
 120
 121         std::string keytag,iftag,ifendtag,ifnottag,ifnotendtag;
 122     };
 123 };
 124
 125 #if defined (_MSC_VER)
 126     #pragma warning( pop )
 127 #endif
 128
 129 #endif /* __xmltooling_template_h__ */