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