/*
 * SoapUI, Copyright (C) 2004-2016 SmartBear Software 
 *
 * Licensed under the EUPL, Version 1.1 or - as soon as they will be approved by the European Commission - subsequent 
 * versions of the EUPL (the "Licence"); 
 * You may not use this work except in compliance with the Licence. 
 * You may obtain a copy of the Licence at: 
 * 
 * http://ec.europa.eu/idabc/eupl 
 * 
 * Unless required by applicable law or agreed to in writing, software distributed under the Licence is 
 * distributed on an "AS IS" basis, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either 
 * express or implied. See the Licence for the specific language governing permissions and limitations 
 * under the Licence. 
 */

package com.eviware.soapui.support.components;

import com.eviware.soapui.support.UISupport;

import javax.swing.JFormattedTextField;
import javax.swing.KeyStroke;
import javax.swing.event.UndoableEditEvent;
import javax.swing.event.UndoableEditListener;
import javax.swing.undo.CannotRedoException;
import javax.swing.undo.CannotUndoException;
import javax.swing.undo.UndoManager;
import java.awt.Dimension;
import java.awt.Toolkit;
import java.awt.event.FocusEvent;
import java.awt.event.FocusListener;
import java.awt.event.KeyAdapter;
import java.awt.event.KeyEvent;

/**
 * JTextArea with Undo/Redo keyboard/popup support
 *
 * @author Ole.Matzura
 */

public class JUndoableFormattedTextField extends JFormattedTextField implements Undoable, UndoableEditListener,
        FocusListener {
    public static final int UNDO_LIMIT = 100;

    private UndoManager undoManager;
    private boolean discardEditsOnSet = false;

    public JUndoableFormattedTextField() {
        super();
        init();
    }

    private void init() {
        getDocument().addUndoableEditListener(this);
        addFocusListener(this);

        setMinimumSize(new Dimension(50, 50));
        addKeyListener(new KeyAdapter() {

            public void keyPressed(KeyEvent e) {
                if (KeyStroke.getKeyStrokeForEvent(e).equals(UISupport.getKeyStroke("menu Z"))) {
                    undo();
                } else if (KeyStroke.getKeyStrokeForEvent(e).equals(UISupport.getKeyStroke("menu Y"))) {
                    redo();
                } else if (KeyStroke.getKeyStrokeForEvent(e).equals(UISupport.getKeyStroke("menu X"))) {
                    cut();
                } else if (KeyStroke.getKeyStrokeForEvent(e).equals(UISupport.getKeyStroke("menu C"))) {
                    copy();
                } else if (KeyStroke.getKeyStrokeForEvent(e).equals(UISupport.getKeyStroke("menu V"))) {
                    paste();
                } else {
                    return;
                }

                e.consume();
            }
        });
    }

    public JUndoableFormattedTextField(String text) {
        super(text);
        init();
    }

    public JUndoableFormattedTextField(int columns) {
        super(columns);
        init();
    }

    public JUndoableFormattedTextField(AbstractFormatter formatter) {
        super(formatter);
        init();
    }

    public void setText(String text) {
        ensureUndoManager();
        super.setText(text == null ? "" : text);

        if (discardEditsOnSet && undoManager != null) {
            undoManager.discardAllEdits();
        }
    }

    public boolean isDiscardEditsOnSet() {
        return discardEditsOnSet;
    }

    public void setDiscardEditsOnSet(boolean discardEditsOnSet) {
        this.discardEditsOnSet = discardEditsOnSet;
    }

    public UndoManager getUndoManager() {
        return undoManager;
    }

    private void ensureUndoManager() {
        if (isEditable() && undoManager == null) {
            undoManager = new UndoManager();
            undoManager.setLimit(UNDO_LIMIT);
        }
    }

    public void focusGained(FocusEvent fe) {
        ensureUndoManager();
    }

    public void focusLost(FocusEvent fe) {
        // removeUndoMananger();
    }

    public void undoableEditHappened(UndoableEditEvent e) {
        if (undoManager != null) {
            undoManager.addEdit(e.getEdit());
        }
    }

    public void undo() {
        if (!isEditable()) {
            getToolkit().beep();
            return;
        }

        try {
            if (undoManager != null) {
                undoManager.undo();
            }
        } catch (CannotUndoException cue) {
            Toolkit.getDefaultToolkit().beep();
        }
    }

    public void redo() {
        if (!isEditable()) {
            getToolkit().beep();
            return;
        }

        try {
            if (undoManager != null) {
                undoManager.redo();
            }
        } catch (CannotRedoException cue) {
            Toolkit.getDefaultToolkit().beep();
        }
    }

    public void setSelectedText(String txt) {
        replaceSelection(txt);
    }

    public boolean canRedo() {
        return undoManager != null && undoManager.canRedo();
    }

    public boolean canUndo() {
        return undoManager != null && undoManager.canUndo();
    }
}