org.codehaus.gmaven.runtime.v1_7.ConsoleFeature Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of gmaven-runtime-1.7 Show documentation
Support for Groovy 1.7.x.
The newest version!
/*
 * Copyright (C) 2006-2007 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.codehaus.gmaven.runtime.v1_7;

import groovy.lang.Binding;
import org.codehaus.gmaven.feature.Component;
import org.codehaus.gmaven.feature.support.ComponentSupport;
import org.codehaus.gmaven.feature.support.FeatureSupport;
import org.codehaus.gmaven.runtime.Console;
import org.codehaus.gmaven.runtime.support.util.NoExitSecurityManager;
import org.sonatype.gshell.io.StreamSet;

import java.lang.reflect.Field;
import java.util.EventObject;

/**
 * Provides the GUI console feature.
 *
 * @version $Id$
 * @author Jason Dillon
 */
public class ConsoleFeature
    extends FeatureSupport
{
    public ConsoleFeature() {
        super(Console.KEY);
    }

    @Override
    protected Component doCreate() throws Exception {
        return new ConsoleImpl();
    }

    //
    // ConsoleImpl
    //

    private class ConsoleImpl
        extends ComponentSupport
        implements Console
    {
        private final Object lock = new Object();

        private ConsoleImpl() throws Exception {
            super(ConsoleFeature.this);
        }

        public void execute(final ClassLoader classLoader) throws Exception {
            assert classLoader != null;

            final StreamSet streams = StreamSet.system();

            final SecurityManager sm = System.getSecurityManager();

            System.setSecurityManager(new NoExitSecurityManager());

            try {
                final groovy.ui.Console console = new groovy.ui.Console(classLoader, new Binding()) {
                    public void exit(final EventObject event) {
                        try {
                            super.exit(event);
                        }
                        finally {
                            synchronized (lock) {
                                lock.notifyAll();
                            }
                        }
                    }
                    public boolean isScriptRunning() {
                        try {
                            Field scriptRunningField = groovy.ui.Console.class.getDeclaredField("scriptRunning");
                            scriptRunningField.setAccessible(true);
                            return (Boolean) scriptRunningField.get(this);
                        } catch (NoSuchFieldException e) {
                            log.error("Unable to get field scriptRunning.  Defaulted to false.");
                            return false;
                        } catch (IllegalAccessException e) {
                            log.error("Unable to get field scriptRunning.  Defaulted to false.");
                            return false;
                        }
                    }
                    public void setScriptRunning(boolean scriptRunning) {
                        try {
                            Field scriptRunningField = groovy.ui.Console.class.getDeclaredField("scriptRunning");
                            scriptRunningField.setAccessible(true);
                            scriptRunningField.set(this, scriptRunning);
                        } catch (NoSuchFieldException e) {
                            log.error("Unable to set field scriptRunning.");
                        } catch (IllegalAccessException e) {
                            log.error("Unable to set field scriptRunning.");
                        }
                    }
                };

                console.run();

                synchronized (lock) {
                    lock.wait();
                }
            }
            finally {
                System.setSecurityManager(sm);
                StreamSet.system(streams);
            }
        }
    }
}