package org.eclipse.ptp.rm.jaxb.ui.dialogs;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.TreeMap;

import org.eclipse.jface.dialogs.Dialog;
import org.eclipse.jface.viewers.CheckStateChangedEvent;
import org.eclipse.jface.viewers.CheckboxTableViewer;
import org.eclipse.jface.viewers.DoubleClickEvent;
import org.eclipse.jface.viewers.ICheckStateListener;
import org.eclipse.jface.viewers.IDoubleClickListener;
import org.eclipse.jface.viewers.IStructuredSelection;
import org.eclipse.ptp.rm.jaxb.core.variables.RMVariableMap;
import org.eclipse.ptp.rm.jaxb.ui.IJAXBUINonNLSConstants;
import org.eclipse.ptp.rm.jaxb.ui.data.CheckedProperty;
import org.eclipse.ptp.rm.jaxb.ui.messages.Messages;
import org.eclipse.ptp.rm.jaxb.ui.providers.CheckedPropertyContentProvider;
import org.eclipse.ptp.rm.jaxb.ui.providers.CheckedPropertyLabelProvider;
import org.eclipse.ptp.rm.jaxb.ui.util.WidgetBuilderUtils;
import org.eclipse.swt.SWT;
import org.eclipse.swt.layout.GridData;
import org.eclipse.swt.layout.GridLayout;
import org.eclipse.swt.widgets.Composite;
import org.eclipse.swt.widgets.Control;
import org.eclipse.swt.widgets.Group;
import org.eclipse.swt.widgets.Shell;
import org.eclipse.swt.widgets.Table;

public class AttributeChoiceDialog extends Dialog implements IJAXBUINonNLSConstants {

	private CheckboxTableViewer preferences;
	private final List<CheckedProperty> allProps;
	private boolean toggle;

	public AttributeChoiceDialog(Shell parentShell) {
		super(parentShell);
		this.allProps = new ArrayList<CheckedProperty>();
		toggle = false;
	}

	public void clearChecked() {
		allProps.clear();
	}

	public Map<String, Boolean> getChecked() {
		Map<String, Boolean> map = new TreeMap<String, Boolean>();
		for (CheckedProperty p : allProps) {
			map.put(p.getName(), p.isChecked());
		}
		return map;
	}

	public void setCurrentlyVisible(Map<String, String> currentlyVisible) {
		Map<?, ?>[] vars = new Map<?, ?>[] { RMVariableMap.getActiveInstance().getVariables(),
				RMVariableMap.getActiveInstance().getDiscovered() };
		for (Map<?, ?> m : vars) {
			for (Object s : m.keySet()) {
				Object o = m.get(s);
				if (o == null) {
					o = s;
				}
				CheckedProperty p = new CheckedProperty(o);
				if (!p.isConfigurable()) {
					continue;
				}
				if (currentlyVisible == null || currentlyVisible.containsKey(s)) {
					// override default settings
					p.setChecked(true);
				}
				allProps.add(p);
			}
		}
	}

	public void toggleCheckboxes() throws Throwable {
		IStructuredSelection selection = (IStructuredSelection) preferences.getSelection();
		List<?> selected = selection.toList();
		toggle = true;
		for (Object o : selected) {
			CheckedProperty p = (CheckedProperty) o;
			boolean checked = p.isChecked();
			preferences.setChecked(p, !checked);
			p.setChecked(!checked);
		}
		toggle = false;
	}

	@Override
	protected void configureShell(Shell shell) {
		super.configureShell(shell);
		shell.setText(Messages.ConfigureLaunchSettings);
	}

	@Override
	protected Control createDialogArea(Composite parent) {
		Composite composite = (Composite) super.createDialogArea(parent);
		GridLayout layout = WidgetBuilderUtils.createGridLayout(2, true);
		GridData gdfill = WidgetBuilderUtils.createGridData(GridData.FILL_HORIZONTAL, true, true, 600, 300, 2, DEFAULT);
		Group grp = WidgetBuilderUtils.createGroup(composite, SWT.BORDER, layout, gdfill);
		int style = SWT.H_SCROLL | SWT.V_SCROLL | SWT.CHECK | SWT.MULTI | SWT.WRAP;
		Table t = WidgetBuilderUtils.createTable(grp, style, 2, 600, gdfill);
		preferences = new CheckboxTableViewer(t);
		WidgetBuilderUtils.addTableColumn(preferences, Messages.AttributeName, SWT.LEFT, null);
		WidgetBuilderUtils.addTableColumn(preferences, Messages.AttributeDescription, SWT.LEFT, null);
		preferences.setContentProvider(new CheckedPropertyContentProvider());
		preferences.setLabelProvider(new CheckedPropertyLabelProvider());
		preferences.setInput(allProps);
		preferences.addDoubleClickListener(new IDoubleClickListener() {
			public void doubleClick(DoubleClickEvent event) {
				try {
					toggleCheckboxes();
				} catch (Throwable t) {
					t.printStackTrace();
				}
			}
		});
		preferences.addCheckStateListener(new ICheckStateListener() {
			public void checkStateChanged(CheckStateChangedEvent event) {
				if (toggle) {
					return;
				}
				CheckedProperty p = (CheckedProperty) event.getElement();
				p.setChecked(event.getChecked());
			}
		});
		preferences.getTable().setHeaderVisible(true);
		applyDialogFont(parent);
		initialize();
		return composite;
	}

	private void initialize() {
		for (CheckedProperty p : allProps) {
			preferences.setChecked(p, p.isChecked());
		}
	}
}