KickJava   Java API By Example, From Geeks To Geeks.

Java > Open Source Codes > org > ofbiz > webapp > ftl > SetRequestAttributeMethod


1 /*
2  * $Id: SetRequestAttributeMethod.java 5462 2005-08-05 18:35:48Z jonesde $
3  *
4  * Copyright (c) 2001-2003 The Open For Business Project - www.ofbiz.org
5  *
6  * Permission is hereby granted, free of charge, to any person obtaining a
7  * copy of this software and associated documentation files (the "Software"),
8  * to deal in the Software without restriction, including without limitation
9  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
10  * and/or sell copies of the Software, and to permit persons to whom the
11  * Software is furnished to do so, subject to the following conditions:
12  *
13  * The above copyright notice and this permission notice shall be included
14  * in all copies or substantial portions of the Software.
15  *
16  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
17  * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
18  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
19  * IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
20  * CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT
21  * OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR
22  * THE USE OR OTHER DEALINGS IN THE SOFTWARE.
23  *
24  */

25 package org.ofbiz.webapp.ftl;
26
27 import java.util.List JavaDoc;
28 import javax.servlet.http.HttpServletRequest JavaDoc;
29
30 import freemarker.core.Environment;
31 import freemarker.ext.beans.BeanModel;
32 import freemarker.template.SimpleScalar;
33 import freemarker.template.TemplateMethodModelEx;
34 import freemarker.template.TemplateModelException;
35 import freemarker.template.TemplateNumberModel;
36 import freemarker.template.TemplateScalarModel;
37
38 /**
39  * SetRequestAttributeMethod - Freemarker Method for setting request attributes
40  *
41  * @author <a HREF="mailto:jaz@ofbiz.org">Andy Zeneski</a>
42  * @version $Rev: 5462 $
43  * @since 2.1
44  */

45 public class SetRequestAttributeMethod implements TemplateMethodModelEx {
46         
47     public static final String JavaDoc module = SetRequestAttributeMethod.class.getName();
48
49     /*
50      * @see freemarker.template.TemplateMethodModel#exec(java.util.List)
51      */

52     public Object JavaDoc exec(List JavaDoc args) throws TemplateModelException {
53         if (args == null || args.size() != 2)
54             throw new TemplateModelException("Invalid number of arguements");
55         if (!(args.get(0) instanceof TemplateScalarModel))
56             throw new TemplateModelException("First argument not an instance of TemplateScalarModel");
57         if (!(args.get(1) instanceof BeanModel) && !(args.get(1) instanceof TemplateNumberModel) && !(args.get(1) instanceof TemplateScalarModel))
58             throw new TemplateModelException("Second argument not an instance of BeanModel nor TemplateNumberModel nor TemplateScalarModel");
59                           
60         Environment env = Environment.getCurrentEnvironment();
61         BeanModel req = (BeanModel)env.getVariable("request");
62         HttpServletRequest JavaDoc request = (HttpServletRequest JavaDoc) req.getWrappedObject();
63         
64         String JavaDoc name = ((TemplateScalarModel) args.get(0)).getAsString();
65         Object JavaDoc value = null;
66         if (args.get(1) instanceof TemplateScalarModel)
67             value = ((TemplateScalarModel) args.get(1)).getAsString();
68         if (args.get(1) instanceof TemplateNumberModel)
69             value = ((TemplateNumberModel) args.get(1)).getAsNumber();
70         if (args.get(1) instanceof BeanModel)
71             value = ((BeanModel) args.get(1)).getWrappedObject();
72                        
73         request.setAttribute(name, value);
74         return new SimpleScalar("");
75     }
76
77 }
78
Popular Tags