324b12eb2025a5a690cffdf9a4145606dccebc88
[mir.git] / source / mir / producer / reader / ScriptedProducerNode.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 the com.oreilly.servlet library, any library
22  * licensed under the Apache Software License, The Sun (tm) Java Advanced
23  * Imaging library (JAI), The Sun JIMI library (or with modified versions of
24  * the above that use the same license as the above), and distribute linked
25  * combinations including the two.  You must obey the GNU General Public
26  * License in all respects for all of the code used other than the above
27  * mentioned libraries.  If you modify this file, you may extend this exception
28  * to your version of the file, but you are not obligated to do so.  If you do
29  * not wish to do so, delete this exception statement from your version.
30  */
31
32 package mir.producer.reader;
33
34 import java.util.*;
35 import java.io.*;
36 import mir.producer.*;
37 import mir.util.*;
38
39 public class ScriptedProducerNode implements ProducerNode {
40   private ScriptedProducerNodeDefinition definition;
41   private Map integerParameterValues;
42   private Map stringParameterValues;
43   private Map nodeParameterValues;
44
45   public ScriptedProducerNode(ScriptedProducerNodeDefinition aDefinition, Map aStringParameterValues, Map anIntegerParameterValues, Map aNodeParameterValues) {
46     definition = aDefinition;
47     stringParameterValues = new HashMap();
48     stringParameterValues.putAll(aStringParameterValues);
49     integerParameterValues = new HashMap();
50     integerParameterValues.putAll(anIntegerParameterValues);
51     nodeParameterValues = new HashMap();
52     nodeParameterValues.putAll(aNodeParameterValues);
53   }
54
55   public Set buildVerbSet() {
56     return new HashSet();
57   }
58
59   public void produce(Map aValues, String aVerb, PrintWriter aLogger) throws ProducerFailure, ProducerExc {
60     try {
61       Map oldValues = new HashMap();
62       ScriptedProducerNodeTool.saveMapValues(oldValues, aValues, definition.getStringParameters().keySet());
63       ScriptedProducerNodeTool.saveMapValues(oldValues, aValues, definition.getIntegerParameters().keySet());
64       try {
65         Iterator i = stringParameterValues.entrySet().iterator();
66
67         while (i.hasNext()) {
68           Map.Entry entry = (Map.Entry) i.next();
69
70           if (entry.getValue() instanceof String) {
71             aValues.put(entry.getKey(), ParameterExpander.expandExpression(aValues, (String) entry.getValue()));
72           }
73         }
74
75         i = integerParameterValues.entrySet().iterator();
76
77         while (i.hasNext()) {
78           Map.Entry entry = (Map.Entry) i.next();
79
80           if (entry.getValue() instanceof String) {
81             aValues.put(entry.getKey(), ParameterExpander.evaluateExpression(aValues, (String) entry.getValue()));
82           }
83         }
84
85         ScriptedProducerNodeTool.pushNodeParameterValues(aValues, definition.getName(), nodeParameterValues);
86         try {
87           definition.getBody().produce(aValues, aVerb, aLogger);
88         }
89         finally {
90           ScriptedProducerNodeTool.popNodeParameterValues(aValues, definition.getName());
91         }
92       }
93       finally {
94         ScriptedProducerNodeTool.restoreMapValues(aValues, definition.getIntegerParameters().keySet(), oldValues);
95         ScriptedProducerNodeTool.restoreMapValues(aValues, definition.getStringParameters().keySet(), oldValues);
96       }
97     }
98     catch (Exception e) {
99       throw new ProducerFailure(e);
100     }
101   }
102
103 }