/*

 * Copyright (c) 2009 Nokia Corporation and/or its subsidiary(-ies). 

 * All rights reserved.

 * This component and the accompanying materials are made available

 * under the terms of the License "Eclipse Public License v1.0"

 * which accompanies this distribution, and is available

 * at the URL "http://www.eclipse.org/legal/epl-v10.html".

 *

 * Initial Contributors:

 * Nokia Corporation - initial contribution.

 *

 * Contributors:

 *

 * Description: 

 *

 */

package com.nokia.carbide.cpp.internal.pi.wizards.ui.util;

import java.util.ArrayList;

import java.util.HashMap;

import java.util.Iterator;

import java.util.List;

import java.util.Map;

import java.util.Map.Entry;

import org.eclipse.core.resources.IProject;

import org.eclipse.jface.viewers.IContentProvider;

import org.eclipse.jface.viewers.IElementComparer;

import org.eclipse.jface.viewers.ISelection;

import org.eclipse.jface.viewers.ITreeSelection;

import org.eclipse.jface.viewers.TreePath;

import org.eclipse.jface.viewers.TreeSelection;

import org.eclipse.jface.viewers.TreeViewer;

import org.eclipse.swt.SWT;

import org.eclipse.swt.events.SelectionAdapter;

import org.eclipse.swt.events.SelectionEvent;

import org.eclipse.swt.widgets.Composite;

import org.eclipse.swt.widgets.Item;

import org.eclipse.swt.widgets.TreeItem;

import com.nokia.carbide.cdt.builder.CarbideBuilderPlugin;

import com.nokia.carbide.cdt.builder.project.ICarbideBuildConfiguration;

import com.nokia.carbide.cdt.builder.project.ICarbideProjectInfo;

import com.nokia.carbide.cdt.builder.project.ISISBuilderInfo;

import com.nokia.carbide.cpp.pi.util.GeneralMessages;

public class PkgListTreeViewer extends TreeViewer {

	PkgListTreeContentProvider contentProvider;

	private class PkgListProjectItem extends PkgListBaseCustomItem {

		PkgListProjectItem(PkgListTreeViewer myViewer, Object myElement,

				TreeItem myTreeItem) {

			super(myViewer, myElement, myTreeItem);

		}

	}

	private class PkgListBuildConfigurationItem extends PkgListBaseCustomItem {

		PkgListBuildConfigurationItem(PkgListTreeViewer myViewer,

				Object myElement, TreeItem myTreeItem) {

			super(myViewer, myElement, myTreeItem);

		}

	}

	private class PkgListFileItem extends PkgListBaseCustomItem {

		PkgListFileItem(PkgListTreeViewer myViewer, Object myElement,

				TreeItem myTreeItem) {

			super(myViewer, myElement, myTreeItem);

		}

	}

	private class PkgListTreeViewerComparer implements IElementComparer {

		public boolean equals(Object a, Object b) {

			// if they're both build configuration items, check that their packages and configurations match

			if (   PkgListTree.isBuildConfigurationItem(a)

				&& PkgListTree.isBuildConfigurationItem(b)) {

				ICarbideBuildConfiguration configA = (ICarbideBuildConfiguration)a;

				ICarbideBuildConfiguration configB = (ICarbideBuildConfiguration)b;

				return (   configA.getCarbideProject().getProject() == configB.getCarbideProject().getProject()

				        && configA.getDisplayString().equals(configB.getDisplayString()));

			}

			else if (PkgListTree.isFileItem(a) &&

				PkgListTree.isFileItem(b)) {

					IPkgEntry fileA = (IPkgEntry)a;

					IPkgEntry fileB = (IPkgEntry)b;

					if (fileA.getPkgFile().equals(fileB.getPkgFile()) &&

							fileA.getSdk().equals(fileB.getSdk())) {

						return true;

					}

					return false;

				}

			return a.equals(b);

		}

		public int hashCode(Object element) {

			return element.hashCode();

		}

	}

	public PkgListTreeViewer(Composite arg0, int arg1) {

		// Must use NO_RADIO_GROUP, otherwise SWT automatically handle unselect for only

		// one item, I haven't find a way to create new composite for each project node.

		// Plus we want want to handle that the not so normal interaction among 

		// project checkbox/config radio button manually

		super(arg0, arg1 | SWT.NO_RADIO_GROUP | SWT.V_SCROLL | SWT.MULTI);

		super.setComparer(new PkgListTreeViewerComparer());

	}

	@Override

	public void setContentProvider(IContentProvider provider) {

		super.setContentProvider(provider);

		if (provider instanceof PkgListTreeContentProvider) {

			contentProvider = (PkgListTreeContentProvider)provider;

		} else {

			GeneralMessages.showErrorMessage(Messages.getString("PkgListTreeViewer.invalid.content.provider")); //$NON-NLS-1$

		}

	}

	private Map<Object, PkgListBaseCustomItem> itemMap = new HashMap<Object, PkgListBaseCustomItem>();

	private int style = 0;

	@Override

	protected void inputChanged(Object input, Object oldInput) {

//		collapseAll();	// just highlight the selected on, setSelectedItems() takes care of expand

		super.inputChanged(input, oldInput);

	}

	public void initializeSelectedItems(TreePath[] paths) {

		for (TreePath path : paths) {

			reveal(path);

		}

		TreeSelection elementSelection = new TreeSelection(paths);

		setSelection(elementSelection, true);

	}

	public IPkgEntry[] getSelectedPkgs() {

		List<Object> selectionList = getSelectionFromWidget();

		ArrayList<IPkgEntry> selectedPkgItems = new ArrayList<IPkgEntry>();

		for (Object item : selectionList) {

			if (item instanceof IPkgEntry) {

				selectedPkgItems.add((IPkgEntry) item);

			}

		}

		return selectedPkgItems.toArray(new IPkgEntry[selectedPkgItems.size()]);

	}

	public IProject[] getSelectedProjects() {

		List<Object> selectionList = getSelectionFromWidget();

		ArrayList<IProject> selectedProjects = new ArrayList<IProject>();

		for (Object item : selectionList) {

			if (item instanceof IProject) {

				selectedProjects.add((IProject) item);

			}

		}

		return selectedProjects.toArray(new IProject[selectedProjects.size()]);

	}

	public ICarbideBuildConfiguration[] getSelectedConfigs() {

		List<Object> selectionList = getSelectionFromWidget();

		ArrayList<ICarbideBuildConfiguration> selectedConfigs = new ArrayList<ICarbideBuildConfiguration>();

		for (Object item : selectionList) {

			if (item instanceof ICarbideBuildConfiguration) {

				selectedConfigs.add((ICarbideBuildConfiguration) item);

			}

		}

		return selectedConfigs.toArray(new ICarbideBuildConfiguration[selectedConfigs.size()]);

	}

	/**

	 * Set additional style bits for radio button appearance.

	 * Most likely style added may be FLAT when in a non-3D view.

	 * @param style int

	 * @see SWT#FLAT

	 */

	public void setStyle(int style) {

		this.style  = style;

	}

	/**

	 * We override associate to maintain each items optional

	 * radio button information.

	 */

	@Override

	protected void associate(final Object element, Item item) {

		super.associate(element, item);

		if (!(item instanceof TreeItem)) {

			return;

		}

		TreeItem treeItem = (TreeItem)item;

		PkgListBaseCustomItem baseItem = itemMap.get(getItemMapKey(element));

		if (baseItem == null) {

			int buttonStyle = style;

			SelectionAdapter selectionAdapter;

			boolean enabled = true;

			if (PkgListTree.isBuildConfigurationItem(element)) {

				baseItem = new PkgListBuildConfigurationItem(this, element, treeItem);

				buttonStyle = buttonStyle | SWT.RADIO;

				selectionAdapter = new SelectionAdapter() {

					@Override

					public void widgetSelected(SelectionEvent e) {

						toggleSelectionAndUpdateSelectionToViewer(element);

					}

				};

				// check for bad PKG, which may happen on regular Symbian stationary

				List<ISISBuilderInfo> sisBuilderInfoList = ((ICarbideBuildConfiguration)element).getSISBuilderInfoList();

				boolean allPkgGood = true && sisBuilderInfoList.size() > 0;

				for (ISISBuilderInfo sisBuilderInfo : sisBuilderInfoList)

				{

					if (sisBuilderInfo == null) {

						allPkgGood = false;

					} else if (sisBuilderInfo.getPKGFullPath().toFile().exists() == false) {

						allPkgGood = false;

					}

				}

				enabled = allPkgGood;

			} else if (PkgListTree.isProjectItem(element)) {

				baseItem = new PkgListProjectItem(this, element, treeItem);

				buttonStyle = buttonStyle | SWT.CHECK;

				selectionAdapter = new SelectionAdapter() {

					@Override

					public void widgetSelected(SelectionEvent e) {

						toggleSelectionAndUpdateSelectionToViewer(element);

					}						

				};

				ICarbideProjectInfo cpi = CarbideBuilderPlugin.getBuildManager().getProjectInfo((IProject)element);

				if (cpi != null) {

					enabled = false;

					List<ICarbideBuildConfiguration> bc = cpi.getBuildConfigurations();

					for (ICarbideBuildConfiguration config : bc) {

						// exclude configuration with (none) PKG

						List<ISISBuilderInfo> sisBuilderInfoList = ((ICarbideBuildConfiguration)config).getSISBuilderInfoList();

						for (ISISBuilderInfo sisBuilderInfo : sisBuilderInfoList)

						{

							if (sisBuilderInfo == null) {

								continue;

							} else if (sisBuilderInfo.getPKGFullPath().toFile().exists()) {

								enabled = true;

								break;

							}

						}					

					}

				}

			} else if (PkgListTree.isFileItem(element)) {

				baseItem = new PkgListFileItem(this, element, treeItem);

				buttonStyle = buttonStyle | SWT.CHECK;

				selectionAdapter = new SelectionAdapter() {

					@Override

					public void widgetSelected(SelectionEvent e) {

						toggleSelectionAndUpdateSelectionToViewer(element);

					}						

				};

			} else {

				return;	// not a custom item

			}

			baseItem.createButton(buttonStyle, selectionAdapter);

			baseItem.setEnabled(enabled);

			itemMap.put(getItemMapKey(element), baseItem);

		}

	}

	/**

	 * We override disassociate to clean up any

	 * allocated radio button information and to 

	 * help implement force selection.

	 */

	@Override

	protected void disassociate(Item item) {

		Object element = item.getData();

		if(PkgListTree.isBuildConfigurationItem(element) ||

				PkgListTree.isFileItem(element) ||

				PkgListTree.isProjectItem(element))

		{	

			PkgListBaseCustomItem baseItem = itemMap.get(getItemMapKey(element));

			if (baseItem != null) {

				if (baseItem instanceof PkgListBuildConfigurationItem ||

						baseItem instanceof PkgListProjectItem ||

						baseItem instanceof PkgListFileItem ) 

				{

					baseItem.dispose();

					itemMap.remove(getItemMapKey(element));

				}

				else {

//					shouldn't get here

				}

			}

		}

		super.disassociate(item);

	}

	// We override this underlying selection to insert UI handling of our custom

	// UI associated, can't use Listeners from TreeViewer to propagate because 

	// we are not SWT.CHECKBOX

	/*

	 * (non-Javadoc)

	 * @see org.eclipse.jface.viewers.AbstractTreeViewer#setSelectionToWidget(java.util.List, boolean)

	 */

	@Override

	protected void setSelectionToWidget(List l, boolean reveal) {

		super.setSelectionToWidget(l, reveal);

		// handle UI

		refreshSelectionOnUI();

	}

	public void refreshSelectionOnUI () {

		for (Iterator<Entry<Object, PkgListBaseCustomItem>> iter = itemMap.entrySet().iterator(); iter.hasNext();) {

			Entry<Object, PkgListBaseCustomItem> entry = iter.next();

			entry.getValue().setSelection(false);

		}

		List selectedList = getSelectionFromWidget();

		for (Object o : selectedList) {

			reveal(o);		// only expanded/visible items are available for selecting programatically

							// hidden items that never been expanded cannot

			PkgListBaseCustomItem plbci = itemMap.get(getItemMapKey(o));

			if(plbci != null) {

				plbci.setSelection(true);

			}

		}

	}

	private ArrayList<TreePath> buildArrayWithoutElement(TreePath[] selectedPaths, Object element) {

		ArrayList<TreePath> returnPath = new ArrayList<TreePath> ();

		for (TreePath path : selectedPaths) {

			boolean skip = false;

			for (int i = 0; i < path.getSegmentCount(); i++) {

				if (path.getSegment(i).equals(element)) {

					skip = true;

					break;

				}

			}

			if (skip == false) {

				returnPath.add(path);

			}

		}

		return returnPath;

	}

	public TreePath getTreePathForElement(Object element) {

		// go upward from element to root, then flip the list

		ArrayList<Object> segments = new ArrayList<Object>();

		Object current = element;

		do {

			segments.add(current);

			current = contentProvider.getParent(current);

		} while (current != null);

		Object[] reverse = segments.toArray(new Object[segments.size()]);

		Object[] inOrder = new Object[segments.size()];

		for (int i = 0; i < segments.size(); i++) {

			inOrder[i] = reverse[segments.size() - 1 - i ];

		}

		return new TreePath(inOrder);

	}

	/*

	 * Build and update the tree selection reflecting the element you want to toggle

	 */

	private void toggleSelectionAndUpdateSelectionToViewer(final Object element)

	{

		ArrayList<TreePath> myNewTreePath;

		ISelection iSelection = this.getSelection();

		ITreeSelection iTreeSelection;

		if (iSelection instanceof TreeSelection) {

			iTreeSelection = (TreeSelection)iSelection;

		} else {

			return;

		}

		TreePath[] selectedPaths = iTreeSelection.getPaths();

		if (PkgListTree.isFileItem(element)) {

			if (iTreeSelection.getPathsFor(element).length > 0) {

				// file already selected, clear self

				myNewTreePath = buildArrayWithoutElement(selectedPaths, element);

			} else {

				// file not yet selected, select self

				myNewTreePath = new ArrayList<TreePath>();

				for (TreePath path: selectedPaths) {

					myNewTreePath.add(path);

				}

				TreePath self = getTreePathForElement(element);

				if (self != null) {

					myNewTreePath.add(self);

				}

			}

		} else if (PkgListTree.isBuildConfigurationItem(element)) {

			Object projectSelection = contentProvider.getParent(element);

			myNewTreePath = buildArrayWithoutElement(selectedPaths, projectSelection);

			if (iTreeSelection.getPathsFor(element).length > 0) {

				// configuration already selected, clear self and parent project

				// do nothing, myNewTreePath already copied without parent project and self

			} else {

				// configuration not yet selected, clear siblings, select self

				TreePath self = getTreePathForElement(element);

				// myNewTreePath already copied without parent project and self

				if (self != null) {

					myNewTreePath.add(self);	// add self back

				}

				TreePath projectPath = getTreePathForElement(projectSelection);

				if (projectPath != null) {

					myNewTreePath.add(projectPath);	// add project back

				}

			}

		} else if (PkgListTree.isProjectItem(element)) {

			myNewTreePath = buildArrayWithoutElement(selectedPaths, element);

			if (iTreeSelection.getPathsFor(element).length > 0) {

				// project already selected, clear self and all children configurations

				// do nothing, myNewTreePath already copied without parent project and self

			} else {

				// project not yet selected, select self and a child configuration(default or next valid)

				TreePath self = getTreePathForElement(element);

				// so children are associated

				expandToLevel(self, 1);

				// I don't know why it didn't repaint proper without this

				collapseToLevel(self, 1);

				IProject project = (IProject) element;

				Object[] children = this.getRawChildren(project);

				ICarbideProjectInfo cpi = CarbideBuilderPlugin.getBuildManager().getProjectInfo(project);

				ICarbideBuildConfiguration configDefault = cpi.getDefaultConfiguration();

				ICarbideBuildConfiguration configDefaultOrFirstGood = null;

				// enabled default or first enabled child

				for (Object child : children) {

					if (itemMap.get(getItemMapKey(child)) == null) {

						continue;

					}

					if (itemMap.get(getItemMapKey(child)).getEnabled()) {

						if (configDefaultOrFirstGood == null || 

							((ICarbideBuildConfiguration)child).getDisplayString().equals(configDefault.getDisplayString())) {

							configDefaultOrFirstGood = (ICarbideBuildConfiguration)child;

						}

					}

				}

				if (configDefaultOrFirstGood  != null) {

					// myNewTreePath already copied without self and children

					if (self != null) {

						myNewTreePath.add(self);	// add self back

					}

					TreePath child = getTreePathForElement(configDefaultOrFirstGood);

					// myNewTreePath already copied without self and children

					if (child != null) {

						myNewTreePath.add(child);	// add child back

					}

				}

			}

		} else {

			myNewTreePath = new ArrayList<TreePath> ();

			for (TreePath path : selectedPaths) {

				myNewTreePath.add(path);

			}

		}

		// fire event

		//updateSelection(getSelection());

		// select from viewer, viewer will fire event and eventually change UI with our

		// overrode routine

		ITreeSelection treeSelection = new TreeSelection(myNewTreePath.toArray(new TreePath[myNewTreePath.size()]));

		setSelection(treeSelection, true);

	}

	public boolean getElementEnabled(Object element) {

		PkgListBaseCustomItem item = itemMap.get(getItemMapKey(element));

		if (item != null) {

			return item.getEnabled();

		}

		return false;

	}

	private Object getItemMapKey(Object element) {

		// because ICarbideBuildConfiguration says keys are equal even when their projects

		// do not match, use the project concatenated with configuration as the key for ICarbideBuildConfiguration

		if (element instanceof ICarbideBuildConfiguration) {

			ICarbideBuildConfiguration config = (ICarbideBuildConfiguration) element;

			return config.getCarbideProject().getProject() + "/" + config.getDisplayString();

		} else {

			return element;

		}

	}

}