Application: gvSIG desktop » gvSIG scripting

package org.gvsig.scripting.swing.impl;

import org.gvsig.scripting.swing.api.JScriptingConsole;

import java.awt.BorderLayout;

import java.awt.Color;

import java.awt.Component;

import java.awt.Font;

import java.awt.event.ComponentEvent;

import java.awt.event.ComponentListener;

import java.io.IOException;

import java.io.OutputStream;

import java.io.PrintStream;

import javax.swing.JComponent;

import javax.swing.JPanel;

import javax.swing.JScrollPane;

import javax.swing.JTextPane;

import javax.swing.plaf.ComponentUI;

import javax.swing.text.AttributeSet;

import javax.swing.text.SimpleAttributeSet;

import javax.swing.text.StyleConstants;

import javax.swing.text.StyleContext;

public class DefaultJScriptingConsole extends JPanel implements JScriptingConsole {

    private static final long serialVersionUID = -1181983617272717357L;

    private JTextAreaOutputStream stdout = null;

    private JTextAreaOutputStream stderr = null;

    private JTextPane console;

    private boolean captureOutput;

    public DefaultJScriptingConsole(boolean captureOutput) {

        this.captureOutput = captureOutput;

        this.initComponents();

        this.stdout = new JTextAreaOutputStream(System.out, Color.BLACK);

        this.stderr = new JTextAreaOutputStream(System.err, Color.RED.darker());

        this.addComponentListener(new ComponentListener() {

            @Override

            public void componentResized(ComponentEvent e) {

            }

            @Override

            public void componentMoved(ComponentEvent e) {

            }

            @Override

            public void componentShown(ComponentEvent e) {

                if( canCaptureOutput() ) {

                    if ( System.out != stdout.getPrintStream()) {

                        System.setOut(stdout.getPrintStream(System.out));

                    }

                    if ( System.err != stderr.getPrintStream()) {

                        System.setErr(stderr.getPrintStream(System.err));

                    }

                }

            }

            @Override

            public void componentHidden(ComponentEvent e) {

                if( canCaptureOutput() ) {

                    if ( System.out != stdout.getPrintStream()) {

                        System.setOut(stdout.getOriginalPS());

                    }

                    if ( System.err != stderr.getPrintStream()) {

                        System.setErr(stderr.getOriginalPS());

                    }

                }

            }

        });

    }

    private boolean canCaptureOutput() {

        return this.captureOutput;

    }

    private void initComponents() {

        this.setLayout(new BorderLayout());

        console = new JTextPane() {

            private static final long serialVersionUID = -1309476113717086242L;

            // https://tips4java.wordpress.com/2009/01/25/no-wrap-text-pane/

            @Override

            public boolean getScrollableTracksViewportWidth() {

                Component parent = getParent();

                ComponentUI ui = getUI();

                return parent != null ? (ui.getPreferredSize(this).width <= parent

                        .getSize().width) : true;

            }

        };

        console.setFont(new Font("monospaced", Font.PLAIN, 13));

        JScrollPane scrollConsole = new JScrollPane(console);

        this.add(scrollConsole, BorderLayout.CENTER);

    }

    @Override

    public void clear() {

        this.console.setText("");

    }

    @Override

    public void append(String s) {

        append(s, stdout.color);

    }

    @Override

    public void append(String msg, Color color) {

        StyleContext sc = StyleContext.getDefaultStyleContext();

        AttributeSet aset = sc.addAttribute(SimpleAttributeSet.EMPTY, StyleConstants.Foreground, color);

        int len = console.getDocument().getLength();

        console.setCaretPosition(len);

        console.setCharacterAttributes(aset, false);

        console.replaceSelection(msg);

    }

    @Override

    public PrintStream getStdout() {

        return this.stdout.getPrintStream();

    }

    @Override

    public PrintStream getStderr() {

        return this.stderr.getPrintStream();

    }

    @Override

    public JComponent asJComponent() {

        return this;

    }

    private class JTextAreaOutputStream extends OutputStream {

        PrintStream printStream = null;

        PrintStream originalPS;

        private final Color color;

        public JTextAreaOutputStream(PrintStream originalPS, Color color) {

            super();

            this.originalPS = originalPS;

            this.color = color;

        }

        @Override

        public void write(int i) throws IOException {

            append(Character.toString((char) i), color);

            originalPS.write(i);

        }

        @Override

        public void write(byte[] buf, int off, int len) throws IOException {

            String s = new String(buf, off, len);

            append(s, color);

            originalPS.write(buf, off, len);

        }

        public PrintStream getPrintStream() {

            if (this.printStream == null) {

                this.printStream = new PrintStream(this);

            }

            return this.printStream;

        }

        public PrintStream getPrintStream(PrintStream originalPS) {

            this.originalPS = originalPS;

            return this.getPrintStream();

        }

        public PrintStream getOriginalPS() {

            return this.originalPS;

        }

    }

}