Application: gvSIG desktop » gvSIG scripting

package org.gvsig.scripting.swing.impl.syntaxhighlight;

import java.awt.BorderLayout;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.awt.event.KeyEvent;

import java.awt.event.KeyListener;

import java.net.URL;

import java.util.HashMap;

import java.util.Map;

import javax.swing.ImageIcon;

import javax.swing.JCheckBox;

import javax.swing.JComponent;

import javax.swing.JOptionPane;

import javax.swing.JPanel;

import javax.swing.JScrollPane;

import javax.swing.KeyStroke;

import javax.swing.UIManager;

import javax.swing.event.CaretEvent;

import javax.swing.event.CaretListener;

import javax.swing.text.JTextComponent;

import org.fife.rsta.ui.CollapsibleSectionPanel;

import org.fife.rsta.ui.search.FindToolBar;

import org.fife.rsta.ui.search.ReplaceToolBar;

import org.fife.rsta.ui.search.SearchEvent;

import org.fife.rsta.ui.search.SearchListener;

import org.fife.ui.rsyntaxtextarea.RSyntaxTextArea;

import org.fife.ui.rsyntaxtextarea.SyntaxConstants;

import org.fife.ui.rtextarea.RTextScrollPane;

import org.fife.ui.rtextarea.SearchContext;

import org.fife.ui.rtextarea.SearchEngine;

import org.fife.ui.rtextarea.SearchResult;

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

public class JRSyntaxTextArea implements SyntaxtHighlightTextComponent, SearchListener {

    public static class MyCheckBox extends JCheckBox {

        private static final String uiClassID = "ToggleButtonUI";

        @Override

        public String getUIClassID() {

            return uiClassID;

        }

    }

    private static class MyFindToolBar extends FindToolBar {

        public MyFindToolBar(SearchListener listener) {

            super(listener);

        }

        @Override

        protected JCheckBox createCB(String key) {

            JCheckBox cb = new MyCheckBox();

            cb.setToolTipText(searchMsg.getString(key));

            cb.addActionListener(listener);

            cb.addMouseListener(listener);

            cb.setIcon(this.getIcon(key));

            return cb;

        }

        private ImageIcon getIcon(String key) {

            String res = "images/search-" + key.toLowerCase() + ".png";

            URL url = this.getClass().getResource(res);

            if (url == null) {

                return new ImageIcon();

            }

            return new ImageIcon(url);

        }

    }

    private static class MyReplaceToolBar extends ReplaceToolBar {

        public MyReplaceToolBar(SearchListener listener) {

            super(listener);

        }

        @Override

        protected JCheckBox createCB(String key) {

            JCheckBox cb = new MyCheckBox();

            cb.setToolTipText(searchMsg.getString(key));

            cb.addActionListener(listener);

            cb.addMouseListener(listener);

            cb.setIcon(this.getIcon(key));

            return cb;

        }

        private ImageIcon getIcon(String key) {

            String res = "images/search-" + key.toLowerCase() + ".png";

            URL url = this.getClass().getResource(res);

            if (url == null) {

                return new ImageIcon();

            }

            return new ImageIcon(url);

        }

    }

    private final FindToolBar findToolBar;

    private final ReplaceToolBar replaceToolBar;

    private final CollapsibleSectionPanel csp;

    public class DefaultUpdateCaretPositionActionEvent extends ActionEvent implements UpdateCaretPositionActionEvent {

        /**

         *

         */

        private static final long serialVersionUID = 8238486105726094074L;

        int line = -1;

        int column = -1;

        public DefaultUpdateCaretPositionActionEvent(Object source, int id,

                String command, int line, int column) {

            super(source, id, command);

            this.line = line;

            this.column = column;

        }

        public int getLine() {

            return this.line + 1;

        }

        public int getColumn() {

            return this.column;

        }

        public boolean hasLineAndColumn() {

            return this.line >= 0 && this.column >= 0;

        }

    }

    protected ActionListener updateCaretPosition = null;

    protected RSyntaxTextArea textArea = null;

    private final RTextScrollPane scrollPanel;

    private JPanel panel;

    private Map<String, String> contentTypeAlias;

    public JRSyntaxTextArea() {

        this.contentTypeAlias = new HashMap<>();

        this.contentTypeAlias.put("text/r", SyntaxConstants.SYNTAX_STYLE_C);

        this.contentTypeAlias.put("text/ecmascript", SyntaxConstants.SYNTAX_STYLE_JAVASCRIPT);

        this.textArea = new RSyntaxTextArea(20, 60);

        this.init();

        scrollPanel = new RTextScrollPane(this.textArea);

        panel = new JPanel();

        panel.setLayout(new BorderLayout());

        panel.add(scrollPanel, BorderLayout.CENTER);

        findToolBar = new MyFindToolBar(this);

        replaceToolBar = new MyReplaceToolBar(this);

        replaceToolBar.setSearchContext(findToolBar.getSearchContext());

        csp = new CollapsibleSectionPanel();

        csp.add(scrollPanel);

        panel.add(csp, BorderLayout.CENTER);

        int ctrl = (KeyEvent.CTRL_MASK | KeyEvent.ALT_MASK);

        KeyStroke ks = KeyStroke.getKeyStroke(KeyEvent.VK_F, ctrl);

        csp.addBottomComponent(ks, findToolBar);

        ks = KeyStroke.getKeyStroke(KeyEvent.VK_H, ctrl);

        csp.addBottomComponent(ks, replaceToolBar);

        textArea.addKeyListener(new KeyListener() {

            @Override

            public void keyTyped(KeyEvent ke) {

            }

            @Override

            public void keyPressed(KeyEvent ke) {

                if (ke.getModifiers() == (KeyEvent.CTRL_MASK | KeyEvent.ALT_MASK)) {

                    if (ke.getKeyCode() == KeyEvent.VK_H) {

                        csp.showBottomComponent(replaceToolBar);

                    } else if (ke.getKeyCode() == KeyEvent.VK_F) {

                        csp.showBottomComponent(findToolBar);

                    }

                }

            }

            @Override

            public void keyReleased(KeyEvent ke) {

            }

        });

    }

    protected void init() {

        textArea.setSyntaxEditingStyle(SyntaxConstants.SYNTAX_STYLE_PYTHON);

        textArea.setCodeFoldingEnabled(true);

        textArea.setClearWhitespaceLinesEnabled(true);

        textArea.setAutoIndentEnabled(true);

        textArea.setCloseCurlyBraces(true);

        textArea.setWhitespaceVisible(true);

        textArea.setAnimateBracketMatching(true);

        textArea.setBracketMatchingEnabled(true);

        textArea.setAutoIndentEnabled(true);

        textArea.setTabsEmulated(true);

        textArea.setTabSize(2);

        textArea.setAntiAliasingEnabled(true);

        textArea.addCaretListener(new CaretListener() {

            public void caretUpdate(CaretEvent e) {

                if (updateCaretPosition == null) {

                    return;

                }

                updateCaretPosition.actionPerformed(

                        new DefaultUpdateCaretPositionActionEvent(textArea, 1, "position", textArea.getCaretLineNumber(), textArea.getCaretOffsetFromLineStart())

                );

            }

        });

    }

    private String normalizeContentType(String s) {

        s = s.toLowerCase();

        if (this.contentTypeAlias.containsKey(s)) {

            s = this.contentTypeAlias.get(s);

        }

        return s;

    }

    public void setContentType(String contentType) {

        this.textArea.setSyntaxEditingStyle(this.normalizeContentType(contentType));

    }

    public JScrollPane getJScrollPane() {

        return this.scrollPanel;

    }

    public JTextComponent getJTextComponent() {

        return this.textArea;

    }

    @Override

    public JComponent asJComponent() {

        return this.panel;

    }

    public String getContentType() {

        return this.textArea.getSyntaxEditingStyle();

    }

    public void addUpdateCaretPositionActionListener(ActionListener updateCaretPosition) {

        this.updateCaretPosition = updateCaretPosition;

    }

    public void setText(String text) {

        this.textArea.setText(text);

    }

    public String getText() {

        return this.textArea.getText();

    }

    public void addKeyListener(KeyListener keyListener) {

        this.textArea.addKeyListener(keyListener);

    }

    @Override

    public void searchEvent(SearchEvent e) {

        SearchEvent.Type type = e.getType();

        SearchContext context = e.getSearchContext();

        SearchResult result = null;

        switch (type) {

            default: // Prevent FindBugs warning later

            case MARK_ALL:

                result = SearchEngine.markAll(textArea, context);

                break;

            case FIND:

                result = SearchEngine.find(textArea, context);

                if (!result.wasFound()) {

                    UIManager.getLookAndFeel().provideErrorFeedback(textArea);

                }

                break;

            case REPLACE:

                result = SearchEngine.replace(textArea, context);

                if (!result.wasFound()) {

                    UIManager.getLookAndFeel().provideErrorFeedback(textArea);

                }

                break;

            case REPLACE_ALL:

                result = SearchEngine.replaceAll(textArea, context);

                JOptionPane.showMessageDialog(null, result.getCount()

                        + " occurrences replaced.");

                break;

        }

        String text = null;

        if (result.wasFound()) {

            text = "Text found; occurrences marked: " + result.getMarkedCount();

        } else if (type == SearchEvent.Type.MARK_ALL) {

            if (result.getMarkedCount() > 0) {

                text = "Occurrences marked: " + result.getMarkedCount();

            } else {

                text = "";

            }

        } else {

            text = "Text not found";

        }

        //setMessage(text);

    }

    @Override

    public String getSelectedText() {

        return textArea.getSelectedText();

    }

}