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