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.io.IOException;
  34 import java.util.HashMap;
  35 import java.util.Iterator;
  36 import java.util.List;
  37 import java.util.Locale;
  38 import java.util.Map;
  39 import java.util.Vector;
  40
  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 mir.util.ResourceBundleGeneratorFunction;
  51 import mircoders.global.MirGlobal;
  52 import mircoders.global.ProducerEngine;
  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   int totalNrComments;
  63
  64   void generateResponse(String aGeneratorIdentifier, PrintWriter aWriter, Map aResponseData, Locale aLocale) {
  65     Generator generator;
  66     try {
  67       generator = MirGlobal.localizer().generators().makeAdminGeneratorLibrary().makeGenerator(aGeneratorIdentifier);
  68       MirGlobal.localizer().producerAssistant().initializeGenerationValueSet(aResponseData);
  69       aResponseData.put( "lang", new ResourceBundleGeneratorFunction( aLocale, MessageResources.getMessageResources("bundles.admin")));
  70       generator.generate(aWriter, aResponseData, logger);
  71     }
  72     catch (Throwable t) {
  73       throw new ServletModuleFailure(t);
  74     }
  75   }
  76
  77   private ServletModuleProducer() {
  78     super();
  79     logger = new LoggerWrapper("ServletModule.Producer");
  80     defaultAction="showProducerQueueStatus";
  81   }
  82
  83   public void showMessage(HttpServletRequest aRequest, HttpServletResponse aResponse, String aMessage, String anArgument1, String anArgument2) {
  84     Map responseData;
  85     try {
  86       responseData = ServletHelper.makeGenerationData(aRequest, aResponse, new Locale[] { getLocale(aRequest), getFallbackLocale(aRequest)});
  87       responseData.put("message", aMessage);
  88       responseData.put("argument1", anArgument1);
  89       responseData.put("argument2", anArgument2);
  90
  91       ServletHelper.generateResponse(aResponse.getWriter(), responseData, "infomessage.template");
  92     }
  93     catch (Throwable t) {
  94       throw new ServletModuleFailure(t);
  95     }
  96   }
  97
  98   public void showProducerQueueStatus(HttpServletRequest aRequest, HttpServletResponse aResponse) {
  99     Map generationData;
 100     Generator generator;
 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   /**
 142    * This method will only be called by external scripts (e.g. from cron jobs).
 143    * The output therefore is very simple.
 144    */
 145   public void produce(HttpServletRequest aRequest, HttpServletResponse aResponse) {
 146     try {
 147       PrintWriter out = aResponse.getWriter();
 148
 149       if (aRequest.getParameter("producer")!=null) {
 150         String producerParam = aRequest.getParameter("producer");
 151         String verbParam = aRequest.getParameter("verb");
 152
 153         try {
 154           MirGlobal.producerEngine().addJob(producerParam, verbParam);
 155           out.println("job added");
 156         }
 157         catch (Throwable t) {
 158           out.println("Can't add job: " + t.getMessage());
 159         }
 160       }
 161     }
 162     catch (IOException e) {
 163       throw new ServletModuleFailure(e);
 164     }
 165   }
 166
 167   public void producerecipe(HttpServletRequest aRequest, HttpServletResponse aResponse) {
 168     try {
 169       String recipe = aRequest.getParameter("recipe");
 170       MirGlobal.localizer().producers().produceRecipe(recipe);
 171       showMessage(aRequest, aResponse, "recipeAddedToQueue", recipe, "");
 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         ServletHelper.redirect(aResponse, "Producer", "showProducerQueueStatus");
 187       }
 188     }
 189     catch (Throwable t) {
 190       throw new ServletModuleFailure(t);
 191     }
 192   }
 193
 194   public void cancel(HttpServletRequest aRequest, HttpServletResponse aResponse)  {
 195     try {
 196       HTTPRequestParser requestParser = new HTTPRequestParser(aRequest);
 197
 198       if (requestParser.getParameter("cancelall") != null) {
 199         MirGlobal.producerEngine().cancelAllJobs();
 200       }
 201       else {
 202         List jobs = new Vector(requestParser.getParameterList("jobid"));
 203
 204         MirGlobal.producerEngine().cancelJobs(jobs);
 205       }
 206       ServletHelper.redirect(aResponse, "Producer", "showProducerQueueStatus");
 207     }
 208     catch (Throwable t) {
 209       throw new ServletModuleFailure(t);
 210     }
 211   }
 212 }