source/mircoders/servlet/ServletModuleProducer.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 the com.oreilly.servlet library, any library
  22  * licensed under the Apache Software License, The Sun (tm) Java Advanced
  23  * Imaging library (JAI), The Sun JIMI library (or with modified versions of
  24  * the above that use the same license as the above), and distribute linked
  25  * combinations including the two.  You must obey the GNU General Public
  26  * License in all respects for all of the code used other than the above
  27  * mentioned libraries.  If you modify this file, you may extend this exception
  28  * to your version of the file, but you are not obligated to do so.  If you do
  29  * not wish to do so, delete this exception statement from your version.
  30  */
  31
  32 package mircoders.servlet;
  33
  34 import java.io.PrintWriter;
  35 import java.util.HashMap;
  36 import java.util.Iterator;
  37 import java.util.List;
  38 import java.util.Locale;
  39 import java.util.Map;
  40 import java.util.Vector;
  41
  42 import javax.servlet.http.HttpServletRequest;
  43 import javax.servlet.http.HttpServletResponse;
  44
  45 import mir.generator.Generator;
  46 import mir.log.LoggerWrapper;
  47 import mir.producer.ProducerFactory;
  48 import mir.servlet.ServletModule;
  49 import mir.servlet.ServletModuleFailure;
  50 import mir.util.NullWriter;
  51 import mir.util.ResourceBundleGeneratorFunction;
  52 import mircoders.global.MirGlobal;
  53
  54 import org.apache.struts.util.MessageResources;
  55
  56 public class ServletModuleProducer extends ServletModule
  57 {
  58   private static ServletModuleProducer instance = new ServletModuleProducer();
  59   public static ServletModule getInstance() { return instance; }
  60
  61   Object comments;
  62   Map generationData;
  63   Generator generator;
  64   int totalNrComments;
  65   List producersData;
  66
  67   void generateResponse(String aGeneratorIdentifier, PrintWriter aWriter, Map aResponseData, Locale aLocale) {
  68     try {
  69       generator = MirGlobal.localizer().generators().makeAdminGeneratorLibrary().makeGenerator(aGeneratorIdentifier);
  70       MirGlobal.localizer().producerAssistant().initializeGenerationValueSet(aResponseData);
  71       aResponseData.put( "lang", new ResourceBundleGeneratorFunction( aLocale, MessageResources.getMessageResources("bundles.admin")));
  72       generator.generate(aWriter, aResponseData, new PrintWriter(new NullWriter()));
  73     }
  74     catch (Throwable t) {
  75       throw new ServletModuleFailure(t);
  76     }
  77   }
  78
  79   private ServletModuleProducer() {
  80     super();
  81     logger = new LoggerWrapper("ServletModule.Producer");
  82     defaultAction="showProducerQueueStatus";
  83   }
  84
  85   public void showMessage(PrintWriter aWriter, Locale aLocale, String aMessage, String anArgument1, String anArgument2) {
  86     Map responseData;
  87     try {
  88       responseData = new HashMap();
  89       responseData.put("message", aMessage);
  90       responseData.put("argument1", anArgument1);
  91       responseData.put("argument2", anArgument2);
  92       generateResponse("infomessage.template", aWriter, responseData, aLocale);
  93     }
  94     catch (Throwable t) {
  95       throw new ServletModuleFailure(t);
  96     }
  97   }
  98
  99
 100   public void showProducerQueueStatus(HttpServletRequest aRequest, HttpServletResponse aResponse) {
 101     Object comments;
 102     Map generationData;
 103     Generator generator;
 104     int totalNrComments;
 105     List producersData;
 106
 107     try {
 108       generator = MirGlobal.localizer().generators().makeAdminGeneratorLibrary().makeGenerator("producerqueue.template");
 109
 110       generationData = ServletHelper.makeGenerationData(getLocale(aRequest));
 111       generationData.put( "thisurl", "module=Producer&do=showProducerQueueStatus");
 112
 113       producersData = new Vector();
 114       Iterator i = MirGlobal.localizer().producers().factories().iterator();
 115       while (i.hasNext()) {
 116         ProducerFactory factory = (ProducerFactory) i.next();
 117
 118         List producerVerbs = new Vector();
 119         Iterator j = factory.verbs();
 120         while (j.hasNext()) {
 121           Map verbData = new HashMap();
 122           ProducerFactory.ProducerVerb verb = (ProducerFactory.ProducerVerb) j.next();
 123           verbData.put("name", verb.getName());
 124           verbData.put("description", verb.getDescription());
 125
 126           producerVerbs.add(verbData);
 127         }
 128
 129         Map producerData = new HashMap();
 130         producerData.put("name", factory.getName());
 131         producerData.put("verbs", producerVerbs);
 132
 133         producersData.add(producerData);
 134       }
 135       generationData.put("producers", producersData);
 136
 137       generationData.put("queue", MirGlobal.producerEngine().getQueueStatus());
 138       generator.generate(aResponse.getWriter(), generationData, new PrintWriter(new NullWriter()));
 139     }
 140     catch (Throwable t) {
 141       throw new ServletModuleFailure(t);
 142     }
 143   }
 144
 145   public void produce(HttpServletRequest req, HttpServletResponse res) {
 146     /*
 147      * This method will only be called by external scripts (e.g. from cron jobs).
 148      * The output therefore is very simple.
 149      *
 150      */
 151
 152     try {
 153       PrintWriter out = res.getWriter();
 154
 155       if (req.getParameter("producer")!=null) {
 156         String producerParam = req.getParameter("producer");
 157         String verbParam = req.getParameter("verb");
 158
 159         MirGlobal.producerEngine().addJob(producerParam, verbParam);
 160         out.println("job added");
 161       }
 162     }
 163     catch (Throwable t) {
 164       throw new ServletModuleFailure(t);
 165     }
 166   }
 167
 168   public void produceAllNew(HttpServletRequest aRequest, HttpServletResponse aResponse) {
 169     try {
 170       MirGlobal.localizer().producers().produceAllNew();
 171       showMessage(aResponse.getWriter(), getLocale(aRequest), "produceAllNewAddedToQueue", "", "");
 172     }
 173     catch (Throwable t) {
 174       throw new ServletModuleFailure(t);
 175     }
 176   }
 177
 178   public void enqueue(HttpServletRequest aRequest, HttpServletResponse aResponse) {
 179     try {
 180       if (aRequest.getParameter("producer")!=null) {
 181         String producerParam = aRequest.getParameter("producer");
 182         String verbParam = aRequest.getParameter("verb");
 183
 184         MirGlobal.producerEngine().addJob(producerParam, verbParam);
 185
 186         showProducerQueueStatus(aRequest, aResponse);
 187       }
 188     }
 189     catch (Throwable t) {
 190       throw new ServletModuleFailure(t);
 191     }
 192   }
 193
 194   public void cancelAbortJob(HttpServletRequest aRequest, HttpServletResponse aResponse)  {
 195     // ML: to be coded
 196   }
 197 }