Application: gvSIG desktop » gvSIG scripting

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

import java.awt.BorderLayout;

import java.awt.event.ActionListener;

import java.awt.event.KeyEvent;

import java.awt.event.KeyListener;

import javax.swing.JPanel;

import javax.swing.JScrollPane;

import javax.swing.JTabbedPane;

import javax.swing.JTextArea;

import javax.swing.JViewport;

import javax.swing.text.BadLocationException;

import javax.swing.text.JTextComponent;

import org.gvsig.scripting.ScriptingManager;

import org.gvsig.scripting.ScriptingScript;

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

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

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

import org.gvsig.scripting.swing.impl.JDialogContent;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

public class EditorHelper {

    private static Logger logger = LoggerFactory.getLogger(EditorHelper.class);

    SyntaxtHighlightTextComponent editorPanel;

    private final ScriptingUIManager uimanager;

    private final ScriptingScript script;

    EditorHelper(ScriptingUIManager uimanager, ScriptingScript script) {

        this.uimanager = uimanager;

        this.script = script;

    }

    public String getText() {

        return editorPanel.getText();

    }

    public JTextComponent getJEditorPanel() {

        return this.editorPanel.getJTextComponent();

    }

    public void addDefaultActionListener(ActionListener actionlistener) {

        editorPanel.addUpdateCaretPositionActionListener(actionlistener);

    }

    public JPanel getPropertiesPanel() {

        return new JPropertiesScript(this.script);

    }

    protected JPanel getCodePanel() {

        JPanel panel = new JPanel();

        panel.setLayout(new BorderLayout());

        editorPanel = (SyntaxtHighlightTextComponent) uimanager.createSyntaxHighlightingPanel();

        editorPanel.setContentType(script.getMimeType());

        editorPanel.setText(script.getCode());

        editorPanel.addKeyListener(new MyKeyListener(this));

        panel.add(editorPanel.asJComponent(), BorderLayout.CENTER);

        panel.setVisible(true);

        return panel;

    }

    public void getSuggestions(final JTextArea editor, final String text) {

        SuggestionController suggestion = new SuggestionController(uimanager.getManager().getHelpManager(), editor, text);

        suggestion.showSuggestion();

    }

    public JDialogContent getHelpPopup(final JTextArea editor, final String text) {

        return new JDialogContent(uimanager, "help-browser", "Select JavaDoc", "Select the JavaDoc that you want to see", new JHelpSelection(uimanager, text, editor));

    }

    public void saveScript(JPropertiesScript properties) {

        if (properties != null) {

            script.setName(properties.name.getText());

            script.setDescription(properties.description.getText());

            script.setCreatedBy(properties.createdBy.getText());

            script.setVersion(properties.version.getText());

            script.setCode(this.getText());

            script.save();

        }

    }

    public static class MyKeyListener implements KeyListener {

        private final EditorHelper editorHelper;

        MyKeyListener(EditorHelper editorHelper) {

            this.editorHelper = editorHelper;

        }

        private JEditor getJEditor(KeyEvent event) {

            JEditor editor = null;

            try {

                JTextArea textArea = (JTextArea) event.getSource();

                JViewport jView = (JViewport) textArea.getParent();

                JScrollPane jScroll = (JScrollPane) jView.getParent();

                JPanel panel = (JPanel) jScroll.getParent();

                JTabbedPane tabsCode = (JTabbedPane) panel.getParent();

                if (tabsCode.getParent() instanceof DefaultJCodeEditor) {

                    editor = (DefaultJCodeEditor) tabsCode.getParent();

                } else if (tabsCode.getParent() instanceof DefaultJDialogEditor) {

                    editor = (DefaultJDialogEditor) tabsCode.getParent();

                }

            } catch (Throwable t) {

                // Do nothing

            }

            return editor;

        }

        private void setUnsavedMark(JEditor editor, boolean mark) {

            try {

                JTabbedPane tabs = (JTabbedPane) editor.getParent();

                String title = tabs.getTitleAt(tabs.getSelectedIndex());

                if (mark) {

                    if (!title.startsWith("*")) {

                        tabs.setTitleAt(tabs.getSelectedIndex(), "*" + title);

                    }

                } else {

                    if (title.startsWith("*")) {

                        title = editor.getScript().getName();

                        tabs.setTitleAt(tabs.getSelectedIndex(), title);

                    }

                }

            } catch (Throwable t) {

                // Ignore

            }

        }

        private String getWordToSearch(JTextArea textArea) throws BadLocationException {

            // Si hay un texto seleccionado devolbemos ese

            String selected = textArea.getSelectedText();

            if (selected != null) {

                return selected;

            }

            // Si no hay texto seleccionado, buscamos la palabra bajo el cursor,

            // la seleccionamos y devolbemos.

            int curpos = textArea.getSelectionStart();

            int curline = textArea.getLineOfOffset(curpos);

            int beginline = textArea.getLineStartOffset(curline);

            int endline = textArea.getLineEndOffset(curline);

            String line = textArea.getText(beginline, endline - beginline);

            // Buscamos el principio de palabra

            int beginword = curpos - beginline - 1;

            while (beginword > 0) {

                char c = line.charAt(beginword);

                if (Character.isLetterOrDigit(c)) {

                    beginword--;

                } else {

                    break;

                }

            }

            // Buscamos el final de la palabra

            int maxlen = line.length();

            int endword = curpos - beginline;

            while (endword < maxlen) {

                char c = line.charAt(endword);

                if (Character.isLetterOrDigit(c)) {

                    endword++;

                } else {

                    break;

                }

            }

            String word = line.substring(beginword + 1, endword);

            textArea.setSelectionStart(beginline + beginword + 1);

            textArea.setSelectionEnd(beginline + endword);

            return word;

        }

        private void doSave(JEditor editor) {

            editor.save();

            setUnsavedMark(editor, false);

        }

        private void doShowHelp(JEditor editor, JTextArea textArea) {

            try {

                String word = getWordToSearch(textArea);

                JDialogContent dialog = editorHelper.getHelpPopup(textArea, word);

                dialog.showDialog();

            } catch (Throwable t) {

                // Ignore

            }

        }

        private void doSuggestions(JEditor editor, JTextArea textArea) {

            try {

                String word = getWordToSearch(textArea);

                editorHelper.getSuggestions(textArea, word);

            } catch (Throwable t) {

                logger.info("Can't show suggestions.",t);

            }

        }

        private void doScriptChanged(JEditor editor) {

            editor.getScript().setSaved(false);

            setUnsavedMark(editor, true);

        }

        private void doRunScript(JEditor editor) {

            editor.getScript().runAsTask(null);

        }

        public void keyReleased(KeyEvent arg0) {

            int keyPressed = arg0.getKeyCode();

            int modifier = arg0.getModifiers();

//                        System.out.println("keyReleased:: modifier:"+Integer.toHexString(modifier)+", keyPressed:"+Integer.toHexString(keyPressed));

            JTextArea textArea = (JTextArea) arg0.getSource();

            JEditor editor = getJEditor(arg0);

            if (editor == null) {

                return;

            }

            switch (keyPressed) {

                case KeyEvent.VK_F1:

                    if (modifier == 0) {

                        doShowHelp(editor, textArea);

                    }

                    break;

                case KeyEvent.VK_F5:

                    if (modifier == 0) {

                        doRunScript(editor);

                    }

                    break;

                case KeyEvent.VK_S:

                    if (modifier == KeyEvent.CTRL_MASK) {

                        doSave(editor);

                    }

                    break;

                case KeyEvent.VK_SPACE:

                    if (modifier == KeyEvent.CTRL_MASK

                            || modifier == KeyEvent.ALT_MASK) {

                        doSuggestions(editor, textArea);

                    }

                    break;

                case KeyEvent.VK_ENTER:

                    if (modifier == KeyEvent.CTRL_MASK) {

                        doSuggestions(editor, textArea);

                    }

                    break;

                default:

                    break;

            }

        }

        public void keyPressed(KeyEvent arg0) {

            int keyPressed = arg0.getKeyCode();

            int modifier = arg0.getModifiers();

//                        System.out.println("keyPressed:: modifier:"+Integer.toHexString(modifier)+", keyPressed:"+Integer.toHexString(keyPressed));

        }

        public void keyTyped(KeyEvent arg0) {

            char keyChar = arg0.getKeyChar();

            if (keyChar != KeyEvent.CHAR_UNDEFINED) {

                JTextArea textArea = (JTextArea) arg0.getSource();

                JEditor editor = getJEditor(arg0);

                if( editor != null ) {

                    doScriptChanged(editor);

                }

            }

        }

    }

    public ScriptingManager getManager() {

        return this.uimanager.getManager();

    }

    public ScriptingUIManager getUIManager() {

        return this.uimanager;

    }

}