/*
 *  RapidMiner
 *
 *  Copyright (C) 2001-2011 by Rapid-I and the contributors
 *
 *  Complete list of developers available at our web site:
 *
 *       http://rapid-i.com
 *
 *  This program is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU Affero General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU Affero General Public License for more details.
 *
 *  You should have received a copy of the GNU Affero General Public License
 *  along with this program.  If not, see http://www.gnu.org/licenses/.
 */
package com.rapidminer.gui.tools;

import java.awt.BorderLayout;
import java.awt.Component;
import java.awt.Container;
import java.awt.Dialog;
import java.awt.Frame;
import java.awt.HeadlessException;
import java.awt.Window;
import java.io.File;
import java.util.Collection;
import java.util.LinkedList;

import javax.accessibility.AccessibleContext;
import javax.swing.AbstractButton;
import javax.swing.BorderFactory;
import javax.swing.JDialog;
import javax.swing.JFileChooser;
import javax.swing.JOptionPane;
import javax.swing.JRootPane;
import javax.swing.UIManager;
import javax.swing.event.ChangeListener;

import com.rapidminer.gui.look.fc.FileChooserUI;
import com.rapidminer.gui.tools.dialogs.ButtonDialog;

/**
 * 
 * @author Tobias Malbrecht
 */
public class ExtendedJFileChooser extends JFileChooser {
	private static final long serialVersionUID = -3457903206380227482L;

	private class FileChooserButtonDialog extends ButtonDialog {
		private static final long serialVersionUID = -553876079090407051L;

		public FileChooserButtonDialog(Dialog owner, String key, boolean modal, ExtendedJFileChooser chooser, Object ... arguments) {
			super(owner, key, modal, arguments);
			chooser.setControlButtonsAreShown(false);
			layoutDefault(chooser, NORMAL, getButtons());
		}

		public FileChooserButtonDialog(Frame window, String key, boolean modal, ExtendedJFileChooser chooser, Object ... arguments) {
			super(window, key, modal, arguments);
			chooser.setControlButtonsAreShown(false);
			layoutDefault(chooser, NORMAL, getButtons());
		}
	}
	
	private final String i18nKey;
	
	private final Object[] i18nArgs;
	
	public ExtendedJFileChooser(File directory) {
		super(directory);
		this.i18nKey = null;
		this.i18nArgs = null;
	}
	
	public ExtendedJFileChooser(String i18nKey, File directory, Object ... i18nArgs) {
		super(directory);
		this.i18nKey = i18nKey;
		this.i18nArgs = i18nArgs;
	}
    
    @Override
	protected JDialog createDialog(Component parent) throws HeadlessException {
//    	String title = getUI().getDialogTitle(this);
//    	putClientProperty(AccessibleContext.ACCESSIBLE_DESCRIPTION_PROPERTY, title);

    	JDialog dialog;
    	if (i18nKey != null) {
	    	Window window = getWindowForComponent(parent);
	    	if (window instanceof Frame) {
	    		dialog = new FileChooserButtonDialog((Frame)window, i18nKey, true, this, i18nArgs);	
	    	} else {
	    		dialog = new FileChooserButtonDialog((Dialog)window, i18nKey, true, this, i18nArgs);
	    	}
    	} else {
    		String title = getUI().getDialogTitle(this);
            putClientProperty(AccessibleContext.ACCESSIBLE_DESCRIPTION_PROPERTY, 
                              title);

            Window window = getWindowForComponent(parent);
            if (window instanceof Frame) {
                dialog = new JDialog((Frame)window, title, true);	
            } else {
                dialog = new JDialog((Dialog)window, title, true);
            }
            dialog.setComponentOrientation(this.getComponentOrientation());

            Container contentPane = dialog.getContentPane();
            contentPane.setLayout(new BorderLayout());
            this.setBorder(BorderFactory.createEmptyBorder(4, 4, 4, 4));
            contentPane.add(this, BorderLayout.CENTER);
     
            if (JDialog.isDefaultLookAndFeelDecorated()) {
                boolean supportsWindowDecorations = 
                UIManager.getLookAndFeel().getSupportsWindowDecorations();
                if (supportsWindowDecorations) {
                    dialog.getRootPane().setWindowDecorationStyle(JRootPane.FILE_CHOOSER_DIALOG);
                }
            }
            dialog.pack();
            dialog.setLocationRelativeTo(parent);
    	}

    	return dialog;
    }

    private static Window getWindowForComponent(Component parentComponent) throws HeadlessException {
	    if (parentComponent == null) {
	        return JOptionPane.getRootFrame();
	    }
	    if (parentComponent instanceof Frame || parentComponent instanceof Dialog) {
	        return (Window) parentComponent;
	    }
	    return getWindowForComponent(parentComponent.getParent());
    }
    
    private Collection<AbstractButton> getButtons() {
    	Collection<AbstractButton> buttons = new LinkedList<AbstractButton>();
    	if (getUI() instanceof FileChooserUI) {
    		buttons.add(((FileChooserUI) getUI()).getApproveButton());
    		buttons.add(((FileChooserUI) getUI()).getCancelButton());
    	}
    	return buttons;
    }
    
    public void addChangeListener(ChangeListener l) {
    	if (getUI() instanceof FileChooserUI) {
    		((FileChooserUI) getUI()).addChangeListener(l);
    	}
    }
    
    public void removeChangeListener(ChangeListener l) {
    	if (getUI() instanceof FileChooserUI) {
    		((FileChooserUI) getUI()).removeChangeListener(l);
    	}
    }
    
    public boolean isFileSelected() {
    	if (getUI() instanceof FileChooserUI) {
    		return ((FileChooserUI) getUI()).isFileSelected();
    	}
    	return true;
    }
}