/*******************************************************************************
 * Copyright (C) 2016, Max Hohenegger <eclipse@hohenegger.eu>
 *
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *******************************************************************************/
package org.eclipse.egit.gitflow.ui.internal.dialogs;

import java.io.IOException;

import org.eclipse.egit.ui.Activator;
import org.eclipse.egit.ui.internal.UIIcons;
import org.eclipse.jface.resource.ResourceManager;
import org.eclipse.jface.viewers.ColumnLabelProvider;
import org.eclipse.jgit.lib.Ref;
import org.eclipse.jgit.lib.Repository;
import org.eclipse.swt.graphics.Image;

class DecoratedBranchLabelProvider extends ColumnLabelProvider {
	private ResourceManager resourceManager = Activator.getDefault().getResourceManager();

	private Repository repository;

	private String prefix;

	public DecoratedBranchLabelProvider(Repository repository, String prefix) {
		this.repository = repository;
		this.prefix = prefix;
	}

	@Override
	public String getText(Object element) {
		if (element instanceof Ref) {
			String name = ((Ref) element).getName();
			return name.substring(prefix.length());
		}
		return super.getText(element);
	}

	@Override
	public Image getImage(Object element) {
		if (element instanceof Ref) {
			return decorateImage((Ref) element);
		}
		return super.getImage(element);
	}

	private Image decorateImage(Ref node) {
		String refName = node.getName();

		String branchName;
		String compareString;

		try {
			branchName = repository.getFullBranch();
			compareString = refName;
		} catch (IOException e) {
			return UIIcons.getImage(resourceManager, UIIcons.BRANCH);
		}

		if (compareString.equals(branchName)) {
			return UIIcons.getImage(resourceManager, UIIcons.CHECKED_OUT_BRANCH);
		} else {
			return UIIcons.getImage(resourceManager, UIIcons.BRANCH);
		}
	}
}