View Javadoc
1   /*
2    * Copyright (c) 2002-2022, City of Paris
3    * All rights reserved.
4    *
5    * Redistribution and use in source and binary forms, with or without
6    * modification, are permitted provided that the following conditions
7    * are met:
8    *
9    *  1. Redistributions of source code must retain the above copyright notice
10   *     and the following disclaimer.
11   *
12   *  2. Redistributions in binary form must reproduce the above copyright notice
13   *     and the following disclaimer in the documentation and/or other materials
14   *     provided with the distribution.
15   *
16   *  3. Neither the name of 'Mairie de Paris' nor 'Lutece' nor the names of its
17   *     contributors may be used to endorse or promote products derived from
18   *     this software without specific prior written permission.
19   *
20   * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
21   * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
22   * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
23   * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDERS OR CONTRIBUTORS BE
24   * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
25   * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
26   * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
27   * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
28   * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
29   * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
30   * POSSIBILITY OF SUCH DAMAGE.
31   *
32   * License 1.0
33   */
34  package fr.paris.lutece.portal.service.template;
35  
36  import freemarker.core.Environment;
37  
38  import freemarker.template.TemplateDateModel;
39  import freemarker.template.TemplateMethodModelEx;
40  import freemarker.template.TemplateModel;
41  import freemarker.template.TemplateModelException;
42  import freemarker.template.TemplateNumberModel;
43  import freemarker.template.TemplateScalarModel;
44  
45  import java.text.MessageFormat;
46  
47  import java.util.List;
48  import java.util.Locale;
49  
50  public abstract class AbstractMessageFormatTemplateMethod implements TemplateMethodModelEx
51  {
52      @Override
53      public Object exec( @SuppressWarnings( "rawtypes" ) List arguments ) throws TemplateModelException
54      {
55          int argsSize = arguments.size( );
56  
57          if ( argsSize < 1 )
58          {
59              throw new TemplateModelException( "Must be called with at least one argument (the message key)" );
60          }
61  
62          String key = ( (TemplateScalarModel) arguments.get( 0 ) ).getAsString( );
63          Locale locale = Environment.getCurrentEnvironment( ).getLocale( );
64  
65          if ( argsSize == 1 )
66          {
67              // no arguments, the message is not a MessageFormat
68              return getPattern( key, locale );
69          }
70  
71          Object [ ] args = new Object [ argsSize - 1];
72  
73          for ( int i = 1; i < argsSize; i++ )
74          {
75              TemplateModel arg = (TemplateModel) arguments.get( i );
76  
77              if ( arg instanceof TemplateScalarModel )
78              {
79                  args [i - 1] = ( (TemplateScalarModel) arg ).getAsString( );
80              }
81              else
82                  if ( arg instanceof TemplateNumberModel )
83                  {
84                      args [i - 1] = ( (TemplateNumberModel) arg ).getAsNumber( );
85                  }
86                  else
87                      if ( arg instanceof TemplateDateModel )
88                      {
89                          args [i - 1] = ( (TemplateDateModel) arg ).getAsDate( );
90                      }
91                      else
92                      {
93                          throw new TemplateModelException( "Unsupported argument type : " + arg );
94                      }
95          }
96  
97          return new MessageFormat( getPattern( key, locale ), locale ).format( args, new StringBuffer( ), null ).toString( );
98      }
99  
100     protected abstract String getPattern( String key, Locale locale );
101 }