media handling fixes, gotten rid of StorageObject, set the default method for blots...
[mir.git] / source / mircoders / localizer / basic / MirBasicProducerAssistantLocalizer.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.localizer.basic;
31
32 import mir.config.MirPropertiesConfiguration;
33 import mir.entity.adapter.EntityAdapter;
34 import mir.entity.adapter.EntityIteratorAdapter;
35 import mir.generator.Generator;
36 import mir.generator.GeneratorExc;
37 import mir.generator.GeneratorFailure;
38 import mir.log.LoggerWrapper;
39 import mir.misc.StringUtil;
40 import mir.util.*;
41 import mir.util.generator.ReflectionGeneratorFunctionsAdapter;
42 import mircoders.global.MirGlobal;
43 import mircoders.localizer.MirLocalizerExc;
44 import mircoders.localizer.MirLocalizerFailure;
45 import mircoders.localizer.MirProducerAssistantLocalizer;
46 import org.w3c.dom.Document;
47 import org.w3c.dom.NamedNodeMap;
48 import org.w3c.dom.Node;
49 import org.w3c.dom.NodeList;
50 import org.w3c.tidy.Tidy;
51 import org.w3c.tidy.Configuration;
52
53 import java.io.ByteArrayInputStream;
54 import java.io.IOException;
55 import java.io.StringWriter;
56 import java.util.GregorianCalendar;
57 import java.util.HashMap;
58 import java.util.Iterator;
59 import java.util.List;
60 import java.util.Map;
61
62 public class MirBasicProducerAssistantLocalizer implements MirProducerAssistantLocalizer {
63   protected LoggerWrapper logger;
64
65   public void initializeGenerationValueSet(Map aValueSet) throws MirLocalizerExc, MirLocalizerFailure  {
66     try {
67       Iterator i;
68
69       Map configMap = new HashMap();
70
71       logger = new LoggerWrapper("Localizer.ProducerAssistant");
72
73 // obsolete:
74       configMap.put("producerDocRoot", MirGlobal.config().getString("Producer.DocRoot"));
75       configMap.put("storageRoot", MirGlobal.config().getString("Producer.StorageRoot"));
76       configMap.put("productionHost", MirGlobal.config().getString("Producer.ProductionHost"));
77       configMap.put("openAction", MirGlobal.config().getString("Producer.OpenAction"));
78       configMap.put("docRoot", MirGlobal.config().getString("RootUri"));
79       configMap.put("actionRoot", MirGlobal.config().getString("RootUri") + "/servlet/Mir");
80       configMap.put("now", new GeneratorFormatAdapters.DateFormatAdapter(new GregorianCalendar().getTime(), MirGlobal.config().getString("Mir.DefaultTimezone")));
81       configMap.put("videoHost", MirGlobal.config().getString("Producer.Video.Host"));
82       configMap.put("audioHost", MirGlobal.config().getString("Producer.Audio.Host"));
83       configMap.put("imageHost", MirGlobal.config().getString("Producer.Image.Host"));
84       configMap.put("imagePath", MirGlobal.config().getString("Producer.Image.Path"));
85       configMap.put("mirVersion", MirGlobal.config().getString("Mir.Version"));
86       configMap.put("defEncoding", MirGlobal.config().getString("Mir.DefaultEncoding"));
87
88 // "new":
89       configMap.putAll(MirPropertiesConfiguration.instance().allSettings());
90
91       aValueSet.put("config", configMap);
92
93       aValueSet.put("utility", new Utility()); 
94
95       aValueSet.put("languages",
96         new EntityIteratorAdapter("", "", 20, MirGlobal.localizer().dataModel().adapterModel(), "language"));
97
98       aValueSet.put("topics",
99         new EntityIteratorAdapter("", "", 20, MirGlobal.localizer().dataModel().adapterModel(), "topic"));
100
101       Map articleTypeMap = new HashMap();
102       articleTypeMap.put("openposting", "0");
103       articleTypeMap.put("newswire", "1");
104       articleTypeMap.put("feature", "2");
105       articleTypeMap.put("topicspecial", "3");
106       articleTypeMap.put("startspecial", "4");
107
108       i = new EntityIteratorAdapter("", "", 20, MirGlobal.localizer().dataModel().adapterModel(), "articleType");
109       while (i.hasNext()) {
110         EntityAdapter articleType = (EntityAdapter) i.next();
111
112         articleTypeMap.put(articleType.get("name"), articleType.get("id"));
113       }
114       aValueSet.put("articletype", articleTypeMap);
115
116       Map commentStatusMap = new HashMap();
117       i = new EntityIteratorAdapter("", "", 20, MirGlobal.localizer().dataModel().adapterModel(), "commentStatus");
118       while (i.hasNext()) {
119         EntityAdapter commentStatus = (EntityAdapter) i.next();
120
121         commentStatusMap.put(commentStatus.get("name"), commentStatus.get("id"));
122       }
123       aValueSet.put("commentstatus", commentStatusMap);
124       aValueSet.put("languageCodeToId", new getLanguageIdFunction());
125     }
126     catch (Throwable t) {
127       logger.error("initializeGenerationValueSet: Exception while collecting comment statuses" + t.getMessage());
128
129       throw new MirLocalizerFailure(t);
130     }
131
132   };
133   public static class getLanguageIdFunction implements Generator.Function {
134     private Map languageCodeToId;
135     private String otherLanguageId;
136     private LoggerWrapper logger = new LoggerWrapper("Localizer.Earth.getLanguageIdFunction");
137
138     public getLanguageIdFunction() throws MirLocalizerFailure {
139       try {
140         otherLanguageId = "";
141         languageCodeToId = new HashMap();
142
143         Iterator i = new EntityIteratorAdapter("", "", 20, MirGlobal.localizer().dataModel().adapterModel(), "language");
144         while (i.hasNext()) {
145           EntityAdapter language = (EntityAdapter) i.next();
146           if (language.get("code").equals("ot"))
147             otherLanguageId = (String) language.get("id");
148
149           languageCodeToId.put(language.get("code"), language.get("id"));
150         }
151       }
152       catch (Throwable t) {
153         logger.error(t.toString());
154
155         throw new MirLocalizerFailure(t);
156       }
157     }
158
159     public Object perform(List aParameters) throws GeneratorExc, GeneratorFailure {
160       try {
161         if (aParameters.size() != 1)
162           throw new GeneratorExc("getLanguageIdFunction: 1 parameter expected: language-code");
163
164         String result = (String) languageCodeToId.get(aParameters.get(0));
165         if (result == null)
166           result = otherLanguageId;
167
168         return result;
169       }
170       catch (GeneratorExc e) {
171         throw e;
172       }
173       catch (Throwable t) {
174         throw new GeneratorFailure("getLanguageIdFunction: " + t.getMessage(), t);
175       }
176     };
177   }
178
179
180   public String filterNonHTMLText(String aText) {
181
182     logger.debug("about to filter non HTML Text of length " + aText.length());
183     try {
184       String result =
185           StringUtil.createHTML(
186           StringUtil.removeHTMLTags(aText),
187           MirGlobal.config().getString("Producer.ImageRoot"),
188           MirGlobal.config().getString("Producer.MailLinkName"),
189           MirGlobal.config().getString("Producer.ExtLinkName"),
190           MirGlobal.config().getString("Producer.IntLinkName")
191           );
192       logger.debug("done filtering non-HTML text ");
193       return result;
194     }
195     catch (Throwable t) {
196       logger.error("error while filtering non-HTML text: " + t.toString());
197
198       throw new RuntimeException(t.toString());
199     }
200   }
201   public String filterHTMLText(String aText) {
202     try {
203       StringWriter out = new StringWriter();
204       Tidy tidy = new Tidy();
205       ByteArrayInputStream in = new ByteArrayInputStream(aText.getBytes("UTF8"));
206       tidy.setMakeClean(true);
207       tidy.setCharEncoding(Configuration.UTF8);
208       tidy.setErrout(logger.asPrintWriter(LoggerWrapper.DEBUG_MESSAGE));
209       print(tidy.parseDOM(in, null), out);
210       
211       return out.toString();
212     }
213     catch (IOException e) {
214       return e.getMessage();
215     }
216   }
217
218   private boolean checkAttr(String attrName) {
219     if (attrName.equals("onLoad") || attrName.equals("onClick") || attrName.equals("onFocus") || attrName.equals("onBlur") || attrName.equals("onMouseOver") || attrName.equals("onMouseOut") || attrName.equals("style") || attrName.equals("STYLE") || attrName.equals("height") || attrName.equals("width") || attrName.equals("HEIGHT") || attrName.equals("WIDTH"))
220       return false;
221     else
222       return true;
223
224   }
225
226   private boolean checkNode(String nodeName) {
227     if (nodeName.equals("a") ||
228         nodeName.equals("img") ||
229         nodeName.equals("h1") ||
230         nodeName.equals("h2") ||
231         nodeName.equals("h3") ||
232         nodeName.equals("h4") ||
233         nodeName.equals("h5") ||
234         nodeName.equals("h6") ||
235         nodeName.equals("br") ||
236         nodeName.equals("form") ||
237         nodeName.equals("input") ||
238         nodeName.equals("hr") ||
239         nodeName.equals("strong") ||
240         nodeName.equals("font") ||
241         nodeName.equals("b") ||
242         nodeName.equals("i") ||
243         nodeName.equals("em") ||
244         nodeName.equals("p") ||
245         nodeName.equals("table") ||
246         nodeName.equals("tr") ||
247         nodeName.equals("td") ||
248         nodeName.equals("th") ||
249         nodeName.equals("ul") ||
250         nodeName.equals("ol") ||
251         nodeName.equals("li")
252     ) {
253       return true;
254     } else {
255
256       return false;
257     }
258   }
259
260   private void print(Node node, StringWriter out) throws IOException {
261     if (node == null) {
262       return;
263     }
264     int type = node.getNodeType();
265     boolean canOutput = checkNode(node.getNodeName());
266
267     switch (type) {
268
269       case Node.DOCUMENT_NODE:
270
271         print(((Document) node).getDocumentElement(), out);
272         out.flush();
273         break;
274
275       case Node.ELEMENT_NODE:
276         if (canOutput) {
277           out.write('<');
278
279           out.write(node.getNodeName());
280           NamedNodeMap attrs = node.getAttributes();
281
282           for (int i = 0; i < attrs.getLength(); i++) {
283             String attrName = attrs.item(i).getNodeName();
284             if (checkAttr(attrName)) {
285               out.write(' ');
286               out.write(attrs.item(i).getNodeName());
287               out.write("=\"");
288
289               out.write(attrs.item(i).getNodeValue());
290               out.write('"');
291             }
292           }
293
294           if (node.getChildNodes()==null || node.getChildNodes().getLength()==0) {
295             out.write("/");
296           }
297           out.write('>');
298         }
299         NodeList children = node.getChildNodes();
300         if (children != null) {
301           int len = children.getLength();
302           for (int i = 0; i < len; i++) {
303             print(children.item(i), out);
304           }
305         }
306         break;
307
308       case Node.TEXT_NODE:
309         out.write(node.getNodeValue());
310         break;
311
312     }
313
314     if (type == Node.ELEMENT_NODE && canOutput && node.getChildNodes()!=null && node.getChildNodes().getLength()>0) {
315       out.write("</");
316       out.write(node.getNodeName());
317       out.write('>');
318     }
319
320     out.flush();
321   }
322
323   public static class Utility extends ReflectionGeneratorFunctionsAdapter {
324     public Utility () {
325       super(new MirBasicUtilityFunctions());
326     }
327     public Object getDatetime() {
328       return new GeneratorDateTimeFunctions.DateTimeFunctions(
329         MirPropertiesConfiguration.instance().getString("Mir.DefaultTimezone"));
330     }
331
332     public Object getCompressWhitespace() {
333       return new freemarker.template.utility.CompressWhitespace();
334     }
335   }
336 }