source/Mir.java

   1 /*\r
   2  * Copyright (C) 2001, 2002 The Mir-coders group\r
   3  *\r
   4  * This file is part of Mir.\r
   5  *\r
   6  * Mir is free software; you can redistribute it and/or modify\r
   7  * it under the terms of the GNU General Public License as published by\r
   8  * the Free Software Foundation; either version 2 of the License, or\r
   9  * (at your option) any later version.\r
  10  *\r
  11  * Mir is distributed in the hope that it will be useful,\r
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of\r
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the\r
  14  * GNU General Public License for more details.\r
  15  *\r
  16  * You should have received a copy of the GNU General Public License\r
  17  * along with Mir; if not, write to the Free Software\r
  18  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA\r
  19  *\r
  20  * In addition, as a special exception, The Mir-coders gives permission to link\r
  21  * the code of this program with the com.oreilly.servlet library, any library\r
  22  * licensed under the Apache Software License, The Sun (tm) Java Advanced\r
  23  * Imaging library (JAI), The Sun JIMI library (or with modified versions of\r
  24  * the above that use the same license as the above), and distribute linked\r
  25  * combinations including the two.  You must obey the GNU General Public\r
  26  * License in all respects for all of the code used other than the above\r
  27  * mentioned libraries.  If you modify this file, you may extend this exception\r
  28  * to your version of the file, but you are not obligated to do so.  If you do\r
  29  * not wish to do so, delete this exception statement from your version.\r
  30  */\r
  31 \r
  32 import java.io.IOException;\r
  33 import java.io.PrintWriter;\r
  34 import java.lang.reflect.Method;\r
  35 import java.util.GregorianCalendar;\r
  36 import java.util.HashMap;\r
  37 import java.util.Locale;\r
  38 import java.util.*;\r
  39 \r
  40 import javax.servlet.ServletException;\r
  41 import javax.servlet.UnavailableException;\r
  42 import javax.servlet.http.HttpServletRequest;\r
  43 import javax.servlet.http.HttpServletResponse;\r
  44 import javax.servlet.http.HttpSession;\r
  45 \r
  46 import freemarker.template.*;\r
  47 \r
  48 import org.apache.struts.util.MessageResources;\r
  49 \r
  50 import mir.misc.*;\r
  51 import mir.servlet.*;\r
  52 import mir.util.*;\r
  53 import mir.generator.*;\r
  54 //import mir.log.Log;\r
  55 \r
  56 import mircoders.global.*;\r
  57 import mircoders.localizer.*;\r
  58 import mircoders.entity.EntityUsers;\r
  59 import mircoders.module.ModuleMessage;\r
  60 import mircoders.module.ModuleUsers;\r
  61 import mircoders.storage.DatabaseArticleType;\r
  62 import mircoders.storage.DatabaseMessages;\r
  63 import mircoders.storage.DatabaseUsers;\r
  64 \r
  65 /**\r
  66  * Mir.java - main servlet, that dispatches to servletmodules\r
  67  *\r
  68  * @author $Author: mh $\r
  69  * @version $Id: Mir.java,v 1.25 2002/12/23 03:00:11 mh Exp $\r
  70  *\r
  71  */\r
  72 \r
  73 public class Mir extends AbstractServlet {\r
  74 \r
  75   private static ModuleUsers usersModule = null;\r
  76   private static ModuleMessage messageModule = null;\r
  77   private final static HashMap servletModuleInstanceHash = new HashMap();\r
  78 \r
  79   private static List loginLanguages = null;\r
  80 \r
  81   public HttpSession session;\r
  82 \r
  83   public void doGet(HttpServletRequest req, HttpServletResponse res) throws\r
  84       ServletException, IOException {\r
  85     doPost(req, res);\r
  86   }\r
  87 \r
  88   protected TemplateModel getLoginLanguages() throws ServletException {\r
  89     synchronized (Mir.class) {\r
  90       try {\r
  91 \r
  92         if (loginLanguages == null) {\r
  93           MessageResources messageResources2 = MessageResources.\r
  94               getMessageResources("bundles.admin");\r
  95           MessageResources messageResources = MessageResources.\r
  96               getMessageResources("bundles.adminlocal");\r
  97           List languages = StringRoutines.splitString(MirGlobal.\r
  98               getConfigPropertyWithDefault("Mir.Login.Languages", "en"), ";");\r
  99 \r
 100           loginLanguages = new Vector();\r
 101           Iterator i = languages.iterator();\r
 102 \r
 103           while (i.hasNext()) {\r
 104             String code = (String) i.next();\r
 105             Locale locale = new Locale(code, "");\r
 106             String name = messageResources.getMessage(locale, "languagename");\r
 107             if (name == null)\r
 108               name = messageResources2.getMessage(locale, "languagename");\r
 109             if (name == null)\r
 110               name = code;\r
 111 \r
 112             Map record = new HashMap();\r
 113             record.put("name", name);\r
 114             record.put("code", code);\r
 115             loginLanguages.add(record);\r
 116           }\r
 117         }\r
 118 \r
 119         return FreemarkerGenerator.makeAdapter(loginLanguages);\r
 120       }\r
 121       catch (Throwable t) {\r
 122         throw new ServletException(t.getMessage());\r
 123       }\r
 124     }\r
 125   }\r
 126 \r
 127   public void doPost(HttpServletRequest req, HttpServletResponse res) throws\r
 128       ServletException, IOException, UnavailableException {\r
 129 \r
 130     long startTime = System.currentTimeMillis();\r
 131     long sessionConnectTime = 0;\r
 132     EntityUsers userEntity;\r
 133     String http = "";\r
 134 \r
 135     if (getServletContext().getAttribute("mir.confed") == null) {\r
 136       getConfig(req);\r
 137     }\r
 138 \r
 139     MirConfig.setServletName(getServletName());\r
 140 \r
 141     //*** test\r
 142      // Log.info(this, "blalalala");\r
 143 \r
 144     session = req.getSession(true);\r
 145     userEntity = (EntityUsers) session.getAttribute("login.uid");\r
 146 \r
 147     if (req.getServerPort() == 443)\r
 148       http = "https";\r
 149     else\r
 150       http = "http";\r
 151 \r
 152 \r
 153     //make sure client browsers don't cache anything\r
 154     setNoCaching(res);\r
 155 \r
 156     res.setContentType("text/html; charset=" + MirConfig.getProp("Mir.DefaultHTMLCharset"));\r
 157 \r
 158     String moduleName = req.getParameter("module");\r
 159     checkLanguage(session, req);\r
 160 \r
 161     /** @todo for cleanup and readability this should be moved to\r
 162      *  method loginIfNecessary() */\r
 163 \r
 164     if (moduleName != null && moduleName.equals("direct")) {\r
 165       //...\r
 166     }\r
 167 \r
 168     // Authentication\r
 169     if ( (moduleName != null && moduleName.equals("login")) || (userEntity == null)) {\r
 170       String user = req.getParameter("login");\r
 171       String passwd = req.getParameter("password");\r
 172       theLog.printDebugInfo("--login: evaluating for user: " + user);\r
 173       userEntity = allowedUser(user, passwd);\r
 174       if (userEntity == null) {\r
 175         // login failed: redirecting to login\r
 176         theLog.printWarning("--login: failed!");\r
 177         _sendLoginPage(res, req, res.getWriter());\r
 178         return;\r
 179       }\r
 180       else if (moduleName != null && moduleName.equals("login")) {\r
 181         // login successful\r
 182 \r
 183         theLog.printInfo("--login: successful! setting uid: " +\r
 184                          userEntity.getId());\r
 185         session.setAttribute("login.uid", userEntity);\r
 186         theLog.printDebugInfo("--login: trying to retrieve login.target");\r
 187         String target = (String) session.getAttribute("login.target");\r
 188 \r
 189         if (target != null) {\r
 190           theLog.printDebugInfo("Redirect: " + target);\r
 191           int serverPort = req.getServerPort();\r
 192           String redirect = "";\r
 193           String redirectString = "";\r
 194 \r
 195           if (serverPort == 80) {\r
 196             redirect = res.encodeURL(http + "://" + req.getServerName() +\r
 197                                      target);\r
 198             redirectString =\r
 199                 "<html><head><meta http-equiv=refresh content=\"1;URL="\r
 200                 + redirect\r
 201                 + "\"></head><body>going <a href=\"" + redirect +\r
 202                 "\">Mir</a></body></html>";\r
 203           }\r
 204           else {\r
 205             redirect = res.encodeURL(http + "://" + req.getServerName() + ":" +\r
 206                                      req.getServerPort() + target);\r
 207             redirectString =\r
 208                 "<html><head><meta http-equiv=refresh content=\"1;URL="\r
 209                 + redirect\r
 210                 + "\"></head><body>going <a href=\"" + redirect +\r
 211                 "\">Mir</a></body></html>";\r
 212           }\r
 213           res.getWriter().println(redirectString);\r
 214 \r
 215           //res.sendRedirect(redirect);\r
 216 \r
 217         }\r
 218         else {\r
 219           // redirecting to default target\r
 220           theLog.printDebugInfo("--login: no target - redirecting to default");\r
 221           _sendStartPage(res, req, res.getWriter(), userEntity);\r
 222         }\r
 223         return;\r
 224       } // if login succesful\r
 225     } // if login\r
 226 \r
 227     if (moduleName != null && moduleName.equals("logout")) {\r
 228       theLog.printDebugInfo("--logout");\r
 229       session.invalidate();\r
 230 \r
 231       //session = req.getSession(true);\r
 232       //checkLanguage(session, req);\r
 233       _sendLoginPage(res, req, res.getWriter());\r
 234       return;\r
 235     }\r
 236 \r
 237     // Check if authed!\r
 238     if (userEntity == null) {\r
 239       // redirect to loginpage\r
 240       String redirectString = req.getRequestURI();\r
 241       String queryString = req.getQueryString();\r
 242       if (queryString != null && !queryString.equals("")) {\r
 243         redirectString += "?" + req.getQueryString();\r
 244         theLog.printDebugInfo("STORING: " + redirectString);\r
 245         session.setAttribute("login.target", redirectString);\r
 246       }\r
 247       _sendLoginPage(res, req, res.getWriter());\r
 248       return;\r
 249     }\r
 250 \r
 251     // If no module is specified goto standard startpage\r
 252     if (moduleName == null || moduleName.equals("")) {\r
 253       theLog.printDebugInfo("no module: redirect to standardpage");\r
 254       _sendStartPage(res, req, res.getWriter(), userEntity);\r
 255       return;\r
 256     }\r
 257     // end of auth\r
 258 \r
 259     // From now on regular dispatching...\r
 260     try {\r
 261       // get servletmodule by parameter and continue with dispacher\r
 262       ServletModule smod = getServletModuleForName(moduleName);\r
 263       ServletModuleDispatch.dispatch(smod, req, res);\r
 264     }\r
 265     catch (ServletModuleException e) {\r
 266       handleError(req, res, res.getWriter(),\r
 267                   "ServletException in Module " + moduleName + " -- " +\r
 268                   e.getMessage());\r
 269     }\r
 270     catch (ServletModuleUserException e) {\r
 271       handleUserError(req, res, res.getWriter(), e.getMessage());\r
 272     }\r
 273 \r
 274     // timing...\r
 275     sessionConnectTime = System.currentTimeMillis() - startTime;\r
 276     theLog.printInfo("EXECTIME (" + moduleName + "): " + sessionConnectTime +\r
 277                      " ms");\r
 278   }\r
 279 \r
 280   /**\r
 281    *  Private method getServletModuleForName returns ServletModule\r
 282    *  from Cache\r
 283    *\r
 284    * @param moduleName\r
 285    * @return ServletModule\r
 286    *\r
 287    */\r
 288   private static ServletModule getServletModuleForName(String moduleName) throws\r
 289       ServletModuleException {\r
 290 \r
 291     // Instance in Map ?\r
 292     if (!servletModuleInstanceHash.containsKey(moduleName)) {\r
 293       // was not found in hash...\r
 294       try {\r
 295         Class theServletModuleClass = null;\r
 296         try {\r
 297           // first we try to get ServletModule from stern.che3.servlet\r
 298           theServletModuleClass = Class.forName(\r
 299               "mircoders.servlet.ServletModule" + moduleName);\r
 300         }\r
 301         catch (ClassNotFoundException e) {\r
 302           // on failure, we try to get it from lib-layer\r
 303           theServletModuleClass = Class.forName("mir.servlet.ServletModule" +\r
 304                                                 moduleName);\r
 305         }\r
 306         Method m = theServletModuleClass.getMethod("getInstance", null);\r
 307         ServletModule smod = (ServletModule) m.invoke(null, null);\r
 308         // we put it into map for further reference\r
 309         servletModuleInstanceHash.put(moduleName, smod);\r
 310         return smod;\r
 311       }\r
 312       catch (Exception e) {\r
 313         throw new ServletModuleException("*** error resolving classname for " +\r
 314                                          moduleName + " -- " + e.getMessage());\r
 315       }\r
 316     }\r
 317     else\r
 318       return (ServletModule) servletModuleInstanceHash.get(moduleName);\r
 319   }\r
 320 \r
 321   private void handleError(HttpServletRequest req, HttpServletResponse res,\r
 322                            PrintWriter out, String errorString) {\r
 323 \r
 324     try {\r
 325       theLog.printError(errorString);\r
 326       SimpleHash modelRoot = new SimpleHash();\r
 327       modelRoot.put("errorstring", new SimpleScalar(errorString));\r
 328       modelRoot.put("date",\r
 329                     new SimpleScalar(StringUtil.date2readableDateTime(new GregorianCalendar())));\r
 330       HTMLTemplateProcessor.process(res, MirConfig.getProp("Mir.ErrorTemplate"),\r
 331                                     modelRoot, out, getLocale(req));\r
 332       out.close();\r
 333     }\r
 334     catch (Exception e) {\r
 335       e.printStackTrace(System.out);\r
 336       System.err.println("Error in ErrorTemplate: " + e.getMessage());\r
 337     }\r
 338   }\r
 339 \r
 340   private void handleUserError(HttpServletRequest req, HttpServletResponse res,\r
 341                                PrintWriter out, String errorString) {\r
 342     try {\r
 343       theLog.printError(errorString);\r
 344       SimpleHash modelRoot = new SimpleHash();\r
 345       modelRoot.put("errorstring", new SimpleScalar(errorString));\r
 346       modelRoot.put("date",\r
 347                     new SimpleScalar(StringUtil.date2readableDateTime(new GregorianCalendar())));\r
 348       HTMLTemplateProcessor.process(res,\r
 349                                     MirConfig.getProp("Mir.UserErrorTemplate"),\r
 350                                     modelRoot, out, getLocale(req));\r
 351       out.close();\r
 352     }\r
 353     catch (Exception e) {\r
 354       System.err.println("Error in UserErrorTemplate");\r
 355     }\r
 356 \r
 357   }\r
 358 \r
 359   /**\r
 360    *  evaluate login for user / password\r
 361    */\r
 362   protected EntityUsers allowedUser(String user, String password) {\r
 363     try {\r
 364       if (usersModule == null)\r
 365         usersModule = new ModuleUsers(DatabaseUsers.getInstance());\r
 366       return usersModule.getUserForLogin(user, password);\r
 367     }\r
 368     catch (Exception e) {\r
 369       theLog.printDebugInfo(e.getMessage());\r
 370       e.printStackTrace();\r
 371       return null;\r
 372     }\r
 373   }\r
 374 \r
 375   // Redirect-methods\r
 376   private void _sendLoginPage(HttpServletResponse res, HttpServletRequest req,\r
 377                               PrintWriter out) {\r
 378     String loginTemplate = MirConfig.getProp("Mir.LoginTemplate");\r
 379     String sessionUrl = res.encodeURL("");\r
 380 \r
 381     try {\r
 382       SimpleHash mergeData = new SimpleHash();\r
 383       SimpleList languages = new SimpleList();\r
 384 \r
 385       mergeData.put("session", sessionUrl);\r
 386 \r
 387       String defaultlanguage = MirGlobal.getConfigPropertyWithDefault("Mir.Login.DefaultLanguage", "");\r
 388       if (defaultlanguage.length() == 0)\r
 389         defaultlanguage = req.getLocale().getLanguage();\r
 390       mergeData.put("defaultlanguage",  defaultlanguage);\r
 391       mergeData.put("languages", getLoginLanguages());\r
 392 \r
 393       HTMLTemplateProcessor.process(res, loginTemplate, mergeData, out, getLocale(req));\r
 394     }\r
 395     catch (Throwable e) {\r
 396       handleError(req, res, out, "Error sending login page: " + e.getMessage());\r
 397     }\r
 398   }\r
 399 \r
 400   private void _sendStartPage(HttpServletResponse res, HttpServletRequest req,\r
 401                               PrintWriter out, EntityUsers userEntity) {\r
 402     String startTemplate = "templates/admin/start_admin.template";\r
 403     String sessionUrl = res.encodeURL("");\r
 404 \r
 405     try {\r
 406       // merge with logged in user and messages\r
 407       SimpleHash mergeData = new SimpleHash();\r
 408       mergeData.put("session", sessionUrl);\r
 409       mergeData.put("login_user", userEntity);\r
 410       if (messageModule == null)\r
 411         messageModule = new ModuleMessage(DatabaseMessages.getInstance());\r
 412       mergeData.put("messages",\r
 413                     messageModule.getByWhereClause(null, "webdb_create desc", 0,\r
 414           10));\r
 415 \r
 416       mergeData.put("articletypes",\r
 417                     DatabaseArticleType.getInstance().\r
 418                     selectByWhereClause("", "id", 0, 20));\r
 419 \r
 420       HTMLTemplateProcessor.process(res, startTemplate, mergeData, out, getLocale(req));\r
 421     }\r
 422     catch (Exception e) {\r
 423       e.printStackTrace(System.out);\r
 424       handleError(req, res, out, "error while trying to send startpage. " + e.getMessage());\r
 425     }\r
 426   }\r
 427 \r
 428   public String getServletInfo() {\r
 429     return "Mir " + MirConfig.getProp("Mir.Version");\r
 430   }\r
 431 \r
 432   private void checkLanguage(HttpSession session, HttpServletRequest req) {\r
 433     // a lang parameter always sets the language\r
 434     String lang = req.getParameter("language");\r
 435     if (lang != null) {\r
 436       theLog.printInfo("selected language " + lang + " overrides accept-language");\r
 437       setLanguage(session, lang);\r
 438       setLocale(session, new Locale(lang, ""));\r
 439     }\r
 440     // otherwise store language from accept header in session\r
 441     else if (session.getAttribute("Language") == null) {\r
 442       theLog.printInfo("accept-language is " + req.getLocale().getLanguage());\r
 443       setLanguage(session, req.getLocale().getLanguage());\r
 444       setLocale(session, req.getLocale());\r
 445     }\r
 446   }\r
 447 }\r