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