anti-abuse filter upgrade
[mir.git] / source / mircoders / servlet / ServletHelper.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 package mircoders.servlet;
31
32 import java.io.PrintWriter;
33 import java.util.HashMap;
34 import java.util.Locale;
35 import java.util.Map;
36 import javax.servlet.http.HttpServletRequest;
37 import javax.servlet.http.HttpServletResponse;
38
39 import org.apache.struts.util.MessageResources;
40
41 import mir.config.MirPropertiesConfiguration;
42 import mir.entity.adapter.EntityIteratorAdapter;
43 import mir.generator.Generator;
44 import mir.log.LoggerWrapper;
45 import mir.servlet.ServletModuleExc;
46 import mir.servlet.ServletModuleFailure;
47 import mir.util.CachingRewindableIterator;
48 import mir.util.ResourceBundleGeneratorFunction;
49 import mircoders.entity.EntityUsers;
50 import mircoders.global.MirGlobal;
51
52
53 public class ServletHelper {
54   static LoggerWrapper logger = new LoggerWrapper("ServletModule.Helper");
55
56
57   public static Map makeGenerationData(HttpServletRequest aRequest, HttpServletResponse aResponse, Locale[] aLocales) throws ServletModuleExc {
58     return makeGenerationData(aRequest, aResponse, aLocales, "bundles.adminlocal", "bundles.admin");
59   }
60
61   public static Map makeGenerationData(HttpServletRequest aRequest, HttpServletResponse aResponse, Locale[] aLocales, String aBundle) throws ServletModuleExc {
62     return makeGenerationData(aRequest, aResponse, aLocales, aBundle, aBundle);
63   }
64
65   public static Map makeGenerationData(HttpServletRequest aRequest, HttpServletResponse aResponse, Locale[] aLocales, String aBundle, String aDefaultBundle) throws ServletModuleExc {
66
67     try {
68       Map result = new HashMap();
69
70       MirGlobal.localizer().producerAssistant().initializeGenerationValueSet(result);
71
72       // ML: hackish
73       ((Map) result.get("config")).put("actionRoot",
74              aResponse.encodeURL(MirGlobal.config().getString("RootUri") + "/servlet/Mir"));
75
76       result.put("returnurl", null);
77       result.put("login_user", getUser(aRequest));
78
79       Object languages =
80           new CachingRewindableIterator(
81             new EntityIteratorAdapter( "", "id", 30,
82                MirGlobal.localizer().dataModel().adapterModel(), "language"));
83
84       Object topics =
85           new CachingRewindableIterator(
86             new EntityIteratorAdapter("", "id", 30,
87                MirGlobal.localizer().dataModel().adapterModel(), "topic"));
88
89       Object articleTypes =
90           new CachingRewindableIterator(
91             new EntityIteratorAdapter( "", "id", 30,
92                MirGlobal.localizer().dataModel().adapterModel(), "articleType"));
93
94       Object commentStatuses =
95           new CachingRewindableIterator(
96             new EntityIteratorAdapter( "", "id", 30,
97                MirGlobal.localizer().dataModel().adapterModel(), "commentStatus"));
98
99       result.put("commentstatuses", commentStatuses);
100       result.put("articletypes", articleTypes);
101       result.put("languages", languages);
102       result.put("topics", topics);
103
104       result.put( "lang",
105           new ResourceBundleGeneratorFunction( aLocales,
106              new MessageResources[] { MessageResources.getMessageResources(aBundle),
107                 MessageResources.getMessageResources(aDefaultBundle)}));
108
109       return result;
110     }
111     catch (Throwable t) {
112       throw new ServletModuleFailure(t);
113     }
114   }
115
116   public static void generateResponse(PrintWriter aWriter, Map aGenerationData, String aGenerator) throws ServletModuleExc {
117     Generator generator;
118
119     try {
120       generator = MirGlobal.localizer().generators().makeAdminGeneratorLibrary().makeGenerator(aGenerator);
121
122       generator.generate(aWriter, aGenerationData, logger);
123     }
124     catch (Throwable t) {
125       throw new ServletModuleFailure(t);
126     }
127   }
128
129   public static void generateOpenPostingResponse(PrintWriter aWriter, Map aGenerationData, String aGenerator) throws ServletModuleExc {
130     Generator generator;
131
132     try {
133       generator = MirGlobal.localizer().generators().makeOpenPostingGeneratorLibrary().makeGenerator(aGenerator);
134
135       generator.generate(aWriter, aGenerationData, logger);
136     }
137     catch (Throwable t) {
138       throw new ServletModuleFailure(t);
139     }
140   }
141
142   public static void redirect(HttpServletResponse aResponse, String aQuery) throws ServletModuleExc, ServletModuleFailure {
143     try {
144       aResponse.sendRedirect(aResponse.encodeRedirectURL(MirPropertiesConfiguration.instance().getString("RootUri") + "/servlet/Mir?"+aQuery));
145     }
146     catch (Throwable t) {
147       throw new ServletModuleFailure("ServletModule.redirect: " +t.getMessage(), t);
148     }
149   }
150
151   public static void redirect(HttpServletResponse aResponse, String aModule, String aMethod) throws ServletModuleExc, ServletModuleFailure {
152     redirect(aResponse, "module="+aModule+"&do="+aMethod);
153   }
154
155   public static void setUser(HttpServletRequest aRequest, EntityUsers aUser) {
156     if (aUser!=null)
157       aRequest.getSession().setAttribute("login.uid", aUser);
158     else
159       aRequest.getSession().removeAttribute("login.uid");
160   }
161
162   public static EntityUsers getUser(HttpServletRequest aRequest) {
163     return (EntityUsers) aRequest.getSession().getAttribute("login.uid");
164   }
165   public static String getUserName(HttpServletRequest aRequest) {
166     EntityUsers user = getUser(aRequest);
167
168     if (user!=null)
169       return user.getValue("login");
170     else
171       return "nobody";
172   }
173 }