Application: gvSIG desktop » Library: org.gvsig.tools

package org.gvsig.texteditor;

import java.awt.BorderLayout;

import java.awt.Font;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.io.File;

import java.io.IOException;

import java.util.Objects;

import javax.swing.JFileChooser;

import javax.swing.JMenu;

import javax.swing.JMenuBar;

import javax.swing.JMenuItem;

import javax.swing.JOptionPane;

import javax.swing.JScrollPane;

import javax.swing.JTextArea;

import javax.swing.event.UndoableEditEvent;

import javax.swing.text.JTextComponent;

import javax.swing.undo.CannotUndoException;

import javax.swing.undo.UndoManager;

import org.apache.commons.io.FileUtils;

/**

 *

 * @author gvSIG Team

 */

public class DefaultJTextEditor extends AbstractJTextEditor {

    private JTextArea textArea;

    // Is File Saved/Opened

    private boolean opened = false;

    private boolean saved = false;

    private File openedFile;

    // Undo manager for managing the storage of the undos

    // so that the can be redone if requested

    private UndoManager undo;

    private String mimeType;

    public DefaultJTextEditor() {

        super();

        JMenuBar menuBar = new JMenuBar();

        JMenu fileMenu = new JMenu("File");

        addMenuItem(fileMenu, "Open...", (ActionEvent e) -> {

            doOpenFile();

        });

        addMenuItem(fileMenu, "Save", (ActionEvent e) -> {

            doSaveFile();

        });

        addMenuItem(fileMenu, "Save as...", (ActionEvent e) -> {

            doSaveAsFile();

        });

        addMenuItem(fileMenu, "Close", (ActionEvent e) -> {

            doClose();

        });

        JMenu editMenu = new JMenu("Edit");

        addMenuItem(editMenu, "Undo", (ActionEvent e) -> {

            doUndo();

        });

        addMenuItem(editMenu, "Redo", (ActionEvent e) -> {

            doRedo();

        });

        addMenuItem(editMenu, "Select all", (ActionEvent e) -> {

            doSelectAll();

        });

        addMenuItem(editMenu, "Cut", (ActionEvent e) -> {

            doCut();

        });

        addMenuItem(editMenu, "Copy", (ActionEvent e) -> {

            doCopy();

        });

        addMenuItem(editMenu, "Paste", (ActionEvent e) -> {

            doPaste();

        });

        menuBar.add(fileMenu);

        menuBar.add(editMenu);

        undo = new UndoManager();

        textArea.getDocument().addUndoableEditListener((UndoableEditEvent e) -> {

            undo.addEdit(e.getEdit());

        });

        this.setLayout(new BorderLayout());

        this.add(menuBar, BorderLayout.NORTH);

        JScrollPane scroll = new JScrollPane(

                getJTextComponent(), 

                JScrollPane.VERTICAL_SCROLLBAR_ALWAYS, 

                JScrollPane.HORIZONTAL_SCROLLBAR_ALWAYS

        );

        this.add(scroll, BorderLayout.CENTER);

    }

    private void addMenuItem(JMenu menu, String text, ActionListener listener) {

        JMenuItem item = new JMenuItem(text);

        item.addActionListener(listener);

        item.setEnabled(true);

        menu.add(item);

    }

    private void saveFile(File file) {

        try {

            FileUtils.write(file, textArea.getText());

            saved = true;

        } catch (IOException ex) {

            LOGGER.warn("Can't save file (" + Objects.toString(file) + ")", ex);

        }

    }

    private void doOpenFile() {

        JFileChooser open = new JFileChooser();

        open.showOpenDialog(null);

        File file = open.getSelectedFile();

        try {

            openedFile = file;

            String s = FileUtils.readFileToString(file);

            textArea.setText(s);

            opened = true;

        } catch (IOException ex) {

            LOGGER.warn("Can't open file (" + Objects.toString(file) + ")", ex);

        }

    }

    private void doSaveFile() {

        JFileChooser save = new JFileChooser();

        File file = save.getSelectedFile();

        if (opened == false && saved == false) {

            save.showSaveDialog(null);

            int confirmationResult;

            if (file.exists()) {

                confirmationResult = JOptionPane.showConfirmDialog(this, "Replace existing file?");

                if (confirmationResult == JOptionPane.YES_OPTION) {

                    saveFile(file);

                }

            } else {

                saveFile(file);

            }

        } else {

            try {

                FileUtils.write(openedFile, textArea.getText());

            } catch (IOException ex) {

                LOGGER.warn("Can't save file (" + Objects.toString(file) + ")", ex);

            }

        }

    }

    private void doSaveAsFile() {

        JFileChooser saveAs = new JFileChooser();

        saveAs.showSaveDialog(null);

        File filename = saveAs.getSelectedFile();

        int confirmationResult;

        if (filename.exists()) {

            confirmationResult = JOptionPane.showConfirmDialog(this, "Replace existing file?");

            if (confirmationResult == JOptionPane.YES_OPTION) {

                saveFile(filename);

            }

        } else {

            saveFile(filename);

        }

    }

    private void doClose() {

        this.setVisible(false);

    }

    private void doUndo() {

        try {

            undo.undo();

        } catch (CannotUndoException ex) {

            LOGGER.warn("Can't undo", ex);

        }

    }

    private void doRedo() {

        try {

            undo.redo();

        } catch (CannotUndoException ex) {

            LOGGER.warn("Can't redo", ex);

        }

    }

    private void doSelectAll() {

        textArea.selectAll();

    }

    private void doCopy() {

        textArea.copy();

    }

    private void doPaste() {

        textArea.paste();

    }

    private void doCut() {

        textArea.cut();

    }

    @Override

    public void setMimetype(String mimeType) {

        this.mimeType = mimeType;

    }

    @Override

    public String getMimetype() {

        return this.mimeType;

    }

    @Override

    public void setText(String text) {

        this.textArea.setText(text);

    }

    @Override

    public String getText() {

        return this.textArea.getText();

    }

    @Override

    public void clean() {

        this.setText("");

    }

    @Override

    public JTextComponent getJTextComponent() {

        if (this.textArea == null) {

            textArea = new JTextArea(30, 50);

            textArea.setEditable(true);

            Font textFont = new Font("Verdana", 0, 14);

            textArea.setFont(textFont);

        }

        return this.textArea;

    }

}