1 /**
2 * This file Copyright (c) 2003-2010 Magnolia International
3 * Ltd. (http://www.magnolia-cms.com). All rights reserved.
4 *
5 *
6 * This file is dual-licensed under both the Magnolia
7 * Network Agreement and the GNU General Public License.
8 * You may elect to use one or the other of these licenses.
9 *
10 * This file is distributed in the hope that it will be
11 * useful, but AS-IS and WITHOUT ANY WARRANTY; without even the
12 * implied warranty of MERCHANTABILITY or FITNESS FOR A
13 * PARTICULAR PURPOSE, TITLE, or NONINFRINGEMENT.
14 * Redistribution, except as permitted by whichever of the GPL
15 * or MNA you select, is prohibited.
16 *
17 * 1. For the GPL license (GPL), you can redistribute and/or
18 * modify this file under the terms of the GNU General
19 * Public License, Version 3, as published by the Free Software
20 * Foundation. You should have received a copy of the GNU
21 * General Public License, Version 3 along with this program;
22 * if not, write to the Free Software Foundation, Inc., 51
23 * Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
24 *
25 * 2. For the Magnolia Network Agreement (MNA), this file
26 * and the accompanying materials are made available under the
27 * terms of the MNA which accompanies this distribution, and
28 * is available at http://www.magnolia-cms.com/mna.html
29 *
30 * Any modifications to this file must keep this entire header
31 * intact.
32 *
33 */
34 package info.magnolia.module.groovy.console;
35
36 import groovy.lang.Binding;
37 import groovy.lang.GroovyCodeSource;
38 import groovy.lang.GroovyShell;
39 import groovy.lang.Script;
40 import info.magnolia.context.MgnlContext;
41 import info.magnolia.module.groovy.support.HierarchyManagerProvider;
42 import info.magnolia.module.groovy.support.classes.MgnlGroovyClassLoader;
43
44 import java.io.InputStream;
45 import java.io.InputStreamReader;
46 import java.io.UnsupportedEncodingException;
47 import java.io.Writer;
48
49 import org.apache.commons.lang.CharEncoding;
50 import org.codehaus.groovy.control.CompilationFailedException;
51 import org.codehaus.groovy.runtime.InvokerHelper;
52
53
54 /**
55 * Groovy console using {@link MgnlGroovyClassLoader} to parse Groovy source.
56 * @author fgrilli
57 * @version $Id$
58 */
59 public class MgnlGroovyConsole extends GroovyShell {
60
61 private static final MgnlGroovyClassLoader cl = new MgnlGroovyClassLoader(new HierarchyManagerProvider.SystemContextHierarchyManagerProvider());
62
63 public MgnlGroovyConsole(Binding binding) {
64 super(binding);
65 }
66
67 public Object evaluate(InputStream in, String fileName, Writer out) throws CompilationFailedException {
68 Script script = null;
69 try {
70 script = parse(new InputStreamReader(in, CharEncoding.UTF_8), fileName);
71 script.setProperty("out", out);
72 script.setProperty("err", out);
73 script.setProperty("ctx", MgnlContext.getInstance());
74 return script.run();
75 }
76 catch (UnsupportedEncodingException e) {
77 throw new RuntimeException(e);
78 }
79 finally {
80 if (script != null) {
81 InvokerHelper.removeClass(script.getClass());
82 }
83 }
84 }
85
86 /**
87 * Parses the given script and returns it ready to be run. When running in a secure environment
88 * (-Djava.security.manager) codeSource.getCodeSource() determines what policy grants should be given to the script.
89 * @param codeSource
90 * @return ready to run script
91 */
92 @Override
93 public Script parse(final GroovyCodeSource codeSource) throws CompilationFailedException {
94 return InvokerHelper.createScript(parseClass(codeSource), getContext());
95 }
96
97 @Override
98 public String generateScriptName() {
99 return super.generateScriptName();
100 }
101
102 /**
103 * Parses the groovy code contained in codeSource and returns a java class.
104 */
105 private Class parseClass(final GroovyCodeSource codeSource) throws CompilationFailedException {
106 // Don't cache scripts
107 return cl.parseClass(codeSource, false);
108 }
109 }