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.ServletModuleExc;
  50 import mir.servlet.ServletModuleFailure;
  51 import mir.util.NullWriter;
  52 import mir.util.ResourceBundleGeneratorFunction;
  53 import mircoders.global.MirGlobal;
  54
  55 import org.apache.struts.util.MessageResources;
  56
  57 public class ServletModuleProducer extends ServletModule
  58 {
  59   private static ServletModuleProducer instance = new ServletModuleProducer();
  60   public static ServletModule getInstance() { return instance; }
  61
  62   Object comments;
  63   Map generationData;
  64   Generator generator;
  65   int totalNrComments;
  66   List producersData;
  67
  68   void generateResponse(String aGeneratorIdentifier, PrintWriter aWriter, Map aResponseData, Locale aLocale) {
  69     try {
  70       generator = MirGlobal.localizer().generators().makeAdminGeneratorLibrary().makeGenerator(aGeneratorIdentifier);
  71       MirGlobal.localizer().producerAssistant().initializeGenerationValueSet(aResponseData);
  72       aResponseData.put( "lang", new ResourceBundleGeneratorFunction( aLocale, MessageResources.getMessageResources("bundles.admin")));
  73       generator.generate(aWriter, aResponseData, new PrintWriter(new NullWriter()));
  74     }
  75     catch (Throwable t) {
  76       throw new ServletModuleFailure(t);
  77     }
  78   }
  79
  80   private ServletModuleProducer() {
  81     super();
  82     logger = new LoggerWrapper("ServletModule.Producer");
  83     defaultAction="showProducerQueueStatus";
  84   }
  85
  86   public void showMessage(PrintWriter aWriter, Locale aLocale, String aMessage, String anArgument1, String anArgument2) {
  87     Map responseData;
  88     try {
  89       responseData = new HashMap();
  90       responseData.put("message", aMessage);
  91       responseData.put("argument1", anArgument1);
  92       responseData.put("argument2", anArgument2);
  93       generateResponse("infomessage.template", aWriter, responseData, aLocale);
  94     }
  95     catch (Throwable t) {
  96       throw new ServletModuleFailure(t);
  97     }
  98   }
  99
 100
 101   public void showProducerQueueStatus(HttpServletRequest aRequest, HttpServletResponse aResponse) {
 102     Object comments;
 103     Map generationData;
 104     Generator generator;
 105     int totalNrComments;
 106     List producersData;
 107
 108     try {
 109       generator = MirGlobal.localizer().generators().makeAdminGeneratorLibrary().makeGenerator("producerqueue.template");
 110
 111       generationData = ServletHelper.makeGenerationData(getLocale(aRequest));
 112       generationData.put( "thisurl", "module=Producer&do=showProducerQueueStatus");
 113
 114       producersData = new Vector();
 115       Iterator i = MirGlobal.localizer().producers().factories().iterator();
 116       while (i.hasNext()) {
 117         ProducerFactory factory = (ProducerFactory) i.next();
 118
 119         List producerVerbs = new Vector();
 120         Iterator j = factory.verbs();
 121         while (j.hasNext()) {
 122           Map verbData = new HashMap();
 123           ProducerFactory.ProducerVerb verb = (ProducerFactory.ProducerVerb) j.next();
 124           verbData.put("name", verb.getName());
 125           verbData.put("description", verb.getDescription());
 126
 127           producerVerbs.add(verbData);
 128         }
 129
 130         Map producerData = new HashMap();
 131         producerData.put("name", factory.getName());
 132         producerData.put("verbs", producerVerbs);
 133
 134         producersData.add(producerData);
 135       }
 136       generationData.put("producers", producersData);
 137
 138       generationData.put("queue", MirGlobal.producerEngine().getQueueStatus());
 139       generator.generate(aResponse.getWriter(), generationData, new PrintWriter(new NullWriter()));
 140     }
 141     catch (Throwable t) {
 142       throw new ServletModuleFailure(t);
 143     }
 144   }
 145
 146   public void produce(HttpServletRequest req, HttpServletResponse res) {
 147     /*
 148      * This method will only be called by external scripts (e.g. from cron jobs).
 149      * The output therefore is very simple.
 150      *
 151      */
 152
 153     try {
 154       PrintWriter out = res.getWriter();
 155
 156       if (req.getParameter("producer")!=null) {
 157         String producerParam = req.getParameter("producer");
 158         String verbParam = req.getParameter("verb");
 159
 160         MirGlobal.producerEngine().addJob(producerParam, verbParam);
 161         out.println("job added");
 162       }
 163     }
 164     catch (Throwable t) {
 165       throw new ServletModuleFailure(t);
 166     }
 167   }
 168
 169   public void produceAllNew(HttpServletRequest aRequest, HttpServletResponse aResponse) {
 170     try {
 171       MirGlobal.localizer().producers().produceAllNew();
 172       showMessage(aResponse.getWriter(), getLocale(aRequest), "produceAllNewAddedToQueue", "", "");
 173     }
 174     catch (Throwable t) {
 175       throw new ServletModuleFailure(t);
 176     }
 177   }
 178
 179   public void enqueue(HttpServletRequest aRequest, HttpServletResponse aResponse) {
 180     try {
 181       if (aRequest.getParameter("producer")!=null) {
 182         String producerParam = aRequest.getParameter("producer");
 183         String verbParam = aRequest.getParameter("verb");
 184
 185         MirGlobal.producerEngine().addJob(producerParam, verbParam);
 186
 187         showProducerQueueStatus(aRequest, aResponse);
 188       }
 189     }
 190     catch (Throwable t) {
 191       throw new ServletModuleFailure(t);
 192     }
 193   }
 194
 195   public void cancelAbortJob(HttpServletRequest aRequest, HttpServletResponse aResponse)  {
 196     // ML: to be coded
 197   }
 198 }