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

package org.gvsig.tools.swing.impl.pickercontroller;

import java.awt.Cursor;

import java.awt.event.ActionEvent;

import java.io.File;

import java.net.URL;

import java.nio.charset.StandardCharsets;

import javax.swing.ImageIcon;

import javax.swing.JButton;

import javax.swing.JFileChooser;

import javax.swing.JOptionPane;

import javax.swing.text.JTextComponent;

import org.apache.commons.io.FileUtils;

import org.apache.commons.lang3.ArrayUtils;

import org.apache.commons.lang3.StringUtils;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.dataTypes.DataTypes;

import org.gvsig.tools.folders.FoldersManager;

import org.gvsig.tools.i18n.I18nManager;

import org.gvsig.tools.swing.api.ToolsSwingLocator;

import org.gvsig.tools.swing.api.pickercontroller.AbstractPickerController;

import org.gvsig.tools.swing.api.threadsafedialogs.ThreadSafeDialogsManager;

import org.gvsig.tools.swing.api.windowmanager.WindowManager;

import org.gvsig.tools.swing.api.windowmanager.WindowManager_v2;

import org.gvsig.tools.swing.icontheme.IconTheme;

import org.gvsig.tools.swing.icontheme.IconThemeManager;

import org.gvsig.tools.swing.impl.DefaultZoomDialog;

/**

 *

 * @author jjdelcerro

 */

@SuppressWarnings("UseSpecificCatch")

public class BytearrayPickerControllerImpl  

        extends AbstractPickerController<byte[]> 

    {

    private final JTextComponent jtext;

    private final JButton btnUpload;

    private final JButton btnDownload;

    private final JButton btnEditAsText;

    private byte[] value;

    private final String fileChooserID;

    private final File initialPath;

    public BytearrayPickerControllerImpl(

            JTextComponent jtext,

            JButton btnUpload,

            JButton btnDownload,

            JButton btnEdiAsText,

            String fileChooserID, 

            File initialPath

        ) {

        this.jtext = jtext;

        this.btnUpload = btnUpload;

        this.btnDownload = btnDownload;

        if( btnEdiAsText == null ) {

            this.btnEditAsText = new JButton();

        } else {

            this.btnEditAsText = btnEdiAsText;

        }

        this.fileChooserID = fileChooserID;

        this.initialPath = initialPath;

        I18nManager i18n = ToolsLocator.getI18nManager();

        this.btnDownload.setCursor(Cursor.getPredefinedCursor(Cursor.HAND_CURSOR));

        this.btnDownload.setText("");

        if( StringUtils.isBlank(this.btnDownload.getToolTipText()) ) {

            this.btnDownload.setToolTipText(i18n.getTranslation("_Download_data"));

        }

        this.btnDownload.setIcon(this.getIcon("picker-bytearray-download"));

        this.btnDownload.addActionListener((ActionEvent e) -> {

          doDownload();

        });

        this.btnUpload.setCursor(Cursor.getPredefinedCursor(Cursor.HAND_CURSOR));

        this.btnUpload.setText("");

        if( StringUtils.isBlank(this.btnUpload.getToolTipText()) ) {

            this.btnUpload.setToolTipText(i18n.getTranslation("_Upload_data"));

        }

        this.btnUpload.setIcon(this.getIcon("picker-bytearray-upload"));

        this.btnUpload.addActionListener((ActionEvent e) -> {

          doUpload();

        });

        this.btnEditAsText.setCursor(Cursor.getPredefinedCursor(Cursor.HAND_CURSOR));

        this.btnEditAsText.setText("");

        if( StringUtils.isBlank(this.btnEditAsText.getToolTipText()) ) {

            this.btnEditAsText.setToolTipText(i18n.getTranslation("_Edit_as_text"));

        }

        this.btnEditAsText.setIcon(this.getIcon("picker-bytearray-editastext"));

        this.btnEditAsText.addActionListener((ActionEvent e) -> {

          doEditAsText();

        });

        this.jtext.setEditable(false);

        ToolsSwingLocator.getToolsSwingManager().addClearButton(jtext, (ActionEvent e) -> {

            this.value = null;

            this.jtext.setText("");

            this.btnDownload.setEnabled(false);

            this.btnEditAsText.setEnabled(true);

        });

    }

    @Override

    public byte[] get() {

        return this.value;

    }

    @Override

    public void set(byte[] value) {

        this.value = value;

        if( this.value==null ) {

            this.btnDownload.setEnabled(false);

            this.btnEditAsText.setEnabled(true);

            this.jtext.setText("");

        } else {

            this.btnDownload.setEnabled(true);

            this.btnEditAsText.setEnabled(true);

            int size = this.value.length;

            this.jtext.setText(String.format("%d bytes", size));

        }

        this.fireChangeEvent();

    }

    @Override

    public void coerceAndSet(Object value) {

        this.set((byte[]) this.coerce(DataTypes.BYTEARRAY, value, null));

    }

    @Override

    public void setEnabled(boolean enabled) {

        this.btnDownload.setEnabled(true);

        this.btnUpload.setEnabled(enabled);

        this.btnEditAsText.setEnabled(true);

    }

    @Override

    public boolean isEnabled() {

        return this.btnUpload.isEnabled();

    }

    private File fileChooser(String title, int mode) {

        ThreadSafeDialogsManager dialogs = ToolsSwingLocator.getThreadSafeDialogsManager();

        FoldersManager folderManager = ToolsLocator.getFoldersManager();

        File[] f = dialogs.showChooserDialog(

                title, 

                mode, 

                JFileChooser.FILES_ONLY, 

                true, 

                folderManager.getLastPath(

                        this.fileChooserID, 

                        this.initialPath

                ),

                null, 

                true

        );

        if( ArrayUtils.isEmpty(f) ) {

            return null;

        }

        return f[0];

    }

    private void doDownload() {

        I18nManager i18n = ToolsLocator.getI18nManager();

        String title = i18n.getTranslation("_Download_data");

        File f = this.fileChooser(title, JFileChooser.SAVE_DIALOG);

        if( f ==null ) {

            return;

        }

        try {

            FileUtils.writeByteArrayToFile(f, value);

        } catch(Exception ex) {

            LOG.warn("Can't save bytes from file '"+f.getAbsolutePath()+"'.",ex);

        }

    }

    private void doUpload() {

        I18nManager i18n = ToolsLocator.getI18nManager();

        String title = i18n.getTranslation("_Upload_data");

        File f = this.fileChooser(title, JFileChooser.OPEN_DIALOG);

        if( f == null ) {

            return;

        }

        try {

            byte[] bytes = FileUtils.readFileToByteArray(f);

            this.set(bytes);

            this.fireChangeEvent();

        } catch (Exception ex) {

            LOG.warn("Can't load bytes from file '"+f.getAbsolutePath()+"'.",ex);

        }

    }

    private void doEditAsText() {

        DefaultZoomDialog dialog;

        I18nManager i18n = ToolsLocator.getI18nManager();

        boolean isEditable = this.isEditable() && this.isEnabled();

        String title = isEditable?i18n.getTranslation("_Text_editor"):i18n.getTranslation("_Text_viewer");

        try {

            String text = "";

            if( value!=null ) {

                text = new String(value, StandardCharsets.UTF_8);

            }

            dialog = new DefaultZoomDialog(

                    this.jtext, 

                    title, 

                    text, 

                    WindowManager.MODE.DIALOG

            );

            dialog.setEditable(isEditable);

            dialog.setAlwaysOnTop(true);

            dialog.setVisible(true);

        } catch(Exception ex) {

            LOG.warn("Can't show text editor.",ex);

            ThreadSafeDialogsManager dialogs = ToolsSwingLocator.getThreadSafeDialogsManager();

            dialogs.messageDialog(

                    i18n.getTranslation("_Cant_show_text_editor"), 

                    title, 

                    JOptionPane.WARNING_MESSAGE

            );

            return;

        }

        try {

            if ( isEditable && dialog.getAction()==WindowManager_v2.BUTTON_OK) {

                String text = dialog.getText();

                this.set(text.getBytes(StandardCharsets.UTF_8));

            }

        } catch(Exception ex) {

            LOG.warn("Can't convert text to raw bytes.",ex);

            ThreadSafeDialogsManager dialogs = ToolsSwingLocator.getThreadSafeDialogsManager();

            dialogs.messageDialog(

                    i18n.getTranslation("_Cant_convert_text_to_raw_bytes"), 

                    title, 

                    JOptionPane.WARNING_MESSAGE

            );

            return;

        }

    }

    public static void selfRegister() {

        URL imageResource = BytearrayPickerControllerImpl.class.getClassLoader().getResource("org/gvsig/tools/swing/picker/picker-bytearray-upload.png");

        if (imageResource != null) {

            IconThemeManager iconThemeManager = ToolsSwingLocator.getIconThemeManager();

            IconTheme theme = iconThemeManager.getCurrent();

            ImageIcon icon = new ImageIcon(imageResource);

            theme.registerDefault("tools", "picker", "picker-bytearray-upload", icon, imageResource);

        }

        imageResource = BytearrayPickerControllerImpl.class.getClassLoader().getResource("org/gvsig/tools/swing/picker/picker-bytearray-download.png");

        if (imageResource != null) {

            IconThemeManager iconThemeManager = ToolsSwingLocator.getIconThemeManager();

            IconTheme theme = iconThemeManager.getCurrent();

            ImageIcon icon = new ImageIcon(imageResource);

            theme.registerDefault("tools", "picker", "picker-bytearray-download", icon, imageResource);

        }

        imageResource = BytearrayPickerControllerImpl.class.getClassLoader().getResource("org/gvsig/tools/swing/picker/picker-bytearray-editastext.png");

        if (imageResource != null) {

            IconThemeManager iconThemeManager = ToolsSwingLocator.getIconThemeManager();

            IconTheme theme = iconThemeManager.getCurrent();

            ImageIcon icon = new ImageIcon(imageResource);

            theme.registerDefault("tools", "picker", "picker-bytearray-editastext", icon, imageResource);

        }

        imageResource = BytearrayPickerControllerImpl.class.getClassLoader().getResource("org/gvsig/tools/swing/picker/picker-bytearray-setnull.png");

        if (imageResource != null) {

            IconThemeManager iconThemeManager = ToolsSwingLocator.getIconThemeManager();

            IconTheme theme = iconThemeManager.getCurrent();

            ImageIcon icon = new ImageIcon(imageResource);

            theme.registerDefault("tools", "picker", "picker-bytearray-setnull", icon, imageResource);

        }

    }

}