/*
 * Copyright (c) 2005-2009 Nokia Corporation and/or its subsidiary(-ies). 
 * All rights reserved.
 * This component and the accompanying materials are made available
 * under the terms of "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.testfw.resultview.view;

import java.util.Hashtable;
import java.util.Stack;

import org.eclipse.swt.SWT;
import org.eclipse.swt.graphics.Image;
import org.eclipse.swt.layout.GridData;
import org.eclipse.swt.widgets.Composite;
import org.eclipse.swt.widgets.Tree;
import org.eclipse.swt.widgets.TreeItem;

import com.nokia.testfw.core.model.result.TestCaseResult;
import com.nokia.testfw.core.model.result.TestResult;
import com.nokia.testfw.core.model.result.TestRunResult;
import com.nokia.testfw.core.model.result.TestSuiteResult;
import com.nokia.testfw.resultview.ResultViewPlugin;

public class TestResultTree {
	private static final Image passedIcon = ResultViewPlugin
			.getImage("obj16/testok.gif"); //$NON-NLS-1$
	private static final Image failedIcon = ResultViewPlugin
			.getImage("obj16/testfail.gif"); //$NON-NLS-1$
	private static final Image testIcon = ResultViewPlugin
			.getImage("obj16/test.gif"); //$NON-NLS-1$
	private static final Image skipIcon = ResultViewPlugin
			.getImage("obj16/testskip.gif"); //$NON-NLS-1$
	private static final Image runIcon = ResultViewPlugin
			.getImage("obj16/testrun.gif"); //$NON-NLS-1$

	Tree tree;
	Hashtable<String, TreeItem> treeItems;

	public TestResultTree(Composite parent) {
		treeItems = new Hashtable<String, TreeItem>();
		tree = new Tree(parent, SWT.V_SCROLL | SWT.SINGLE);
		GridData gridData = new GridData(GridData.FILL_BOTH
				| GridData.GRAB_HORIZONTAL | GridData.GRAB_VERTICAL);

		tree.setLayoutData(gridData);
	}

	public Tree getTree() {
		return tree;
	}

	/**
	 * initial the tree with empty test result
	 */
	public void init(TestRunResult runResult) {
		// create a root node
		for (TestResult result : runResult.getResults()) {
			createTreeItems(null, result);
		}
	}

	/**
	 * update one test case result
	 * 
	 * @param result
	 */
	public void update(TestCaseResult result) {
		// since it is using same result object, just redraw the tree
		TreeItem item = treeItems.get(result.getUniqueName());
		updateTreeItem(item);
		// make sure to show this item in the tree
		tree.showItem(item);
		// update parent suite is any
		TestSuiteResult suite;
		TestResult child = result;
		while ((suite = child.getParent()) != null) {
			updateTreeItem(treeItems.get(suite.getUniqueName()));
			child = suite;
		}
	}

	public boolean addTestCase(TestCaseResult result) {
		if (treeItems.get(result.getUniqueName()) != null)
			return false;

		TestSuiteResult parent = result.getParent();
		if (parent == null) {
			createNewTreeItem(null, result);
		} else {
			Stack<TestResult> stack = new Stack<TestResult>();
			TreeItem parentItem = null;
			stack.push(result);
			while (parent != null) {
				parentItem = treeItems.get(parent.getUniqueName());
				if (parentItem == null) {
					stack.push(parent);
				} else {
					break;
				}
				parent = parent.getParent();
			}
			while (!stack.isEmpty()) {
				TestResult test = stack.pop();
				parentItem = createNewTreeItem(parentItem, test);
			}
		}
		return true;
	}

	/**
	 * reset the result tree and clean it is contents
	 */
	public void reset() {
		tree.removeAll();
		treeItems.clear();
	}

	/**
	 * create one tree item to show a testcase(suite) result
	 * 
	 * @param testResult
	 */
	protected void createTreeItems(TreeItem parent, TestResult testResult) {
		TreeItem item = createNewTreeItem(parent, testResult);
		// create items for its children is any
		if (testResult instanceof TestSuiteResult) {
			TestSuiteResult suite = (TestSuiteResult) testResult;
			for (TestResult child : suite.getChildren()) {
				createTreeItems(item, child);
			}
		}
	}

	private TreeItem createNewTreeItem(TreeItem parentItem, TestResult result) {
		TreeItem treeItem = null;
		if (parentItem == null) {
			treeItem = new TreeItem(tree, SWT.NONE);
		} else {
			treeItem = new TreeItem(parentItem, SWT.NONE);
		}
		treeItem.setData(result);
		treeItem.setExpanded(true);
		updateTreeItem(treeItem);
		// set to show all items. may change to only show suite later
		tree.showItem(treeItem);
		treeItems.put(result.getUniqueName(), treeItem);
		return treeItem;
	}

	private void updateTreeItem(TreeItem item) {
		TestResult result = (TestResult) item.getData();
		item.setText(result.getDesc());

		switch (result.getStatus()) {
		case SUCCESS:
			item.setImage(passedIcon);
			break;
		case FAILURE:
			item.setImage(failedIcon);
			break;
		case SKIP:
			item.setImage(skipIcon);
			break;
		case STARTED:
			item.setImage(runIcon);
			break;
		default:
			item.setImage(testIcon);
		}

	}

	// // handle the double click selection
	// private void setupSelection() {
	// tree.addListener(SWT.Selection, new Listener() {
	//
	// public void handleEvent(Event event) {
	// TreeItem item = tree.getSelection()[0];
	// if (item == null) {
	// return;
	// }
	// TestResult result = (TestResult) item.getData();
	// if (result == null) {
	// ResultViewPlugin.log(IStatus.WARNING,
	// "can't find selected test result");
	// }
	// // find the file
	// if (result.getFile() == null) {
	// ResultViewPlugin.log(IStatus.OK,
	// "no locaiton information associated with the test");
	// return;
	// }
	// // find file in workspace
	// WorkspaceUtils.openFile(result.getFile(), result.getLine(),
	// result.getColumn());
	// }
	//
	// });
	// }
}