org.eclipse.mylyn.reviews.ui/src/org/eclipse/mylyn/internal/reviews/ui/ReviewColumnLabelProvider.java

/*******************************************************************************
 * Copyright (c) 2015, 2016 Landon Butterworth and others.
 * 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
 *
 * Contributors:
 *     Landon Butterworth - initial API and implementation
 *******************************************************************************/

package org.eclipse.mylyn.internal.reviews.ui;

import org.eclipse.jface.viewers.ColumnLabelProvider;
import org.eclipse.jface.viewers.ViewerCell;
import org.eclipse.mylyn.commons.ui.CommonImages;
import org.eclipse.mylyn.internal.reviews.ui.editors.parts.TaskReview;
import org.eclipse.mylyn.tasks.core.ITask.SynchronizationState;
import org.eclipse.swt.graphics.Image;

/**
 * @author Landon Butterworth
 */
public class ReviewColumnLabelProvider extends ColumnLabelProvider {

	private final int VERIFIED = 1;

	private final int NOT_VERIFIED = -1;

	private static final int DESCRIPTION_COLUMN = 0;

	private static final int CODE_REVIEW_COLUMN = 1;

	private static final int VERIFIED_COLUMN = 2;

	private static final int STATUS_COLUMN = 3;

	private static final int MINUS_TWO = -2;

	private static final int MINUS_ONE = -1;

	private static final int PLUS_ONE = 1;

	private static final int PLUS_TWO = 2;

	public ReviewColumnLabelProvider() {
	}

	public Image getColumnImage(Object element, int columnIndex) {
		TaskReview reviewContainer = (TaskReview) element;

		switch (columnIndex) {
		case DESCRIPTION_COLUMN:
			return getIncomingChangesImage(reviewContainer);
		case CODE_REVIEW_COLUMN:
			return getReviewStateImage(reviewContainer);
		case VERIFIED_COLUMN:
			return getVerifiedStateImage(reviewContainer);
		}

		return null;
	}

	private Image getIncomingChangesImage(TaskReview reviewContainer) {
		if (reviewContainer.getSyncState() == SynchronizationState.INCOMING) {
			return CommonImages.getImage(CommonImages.OVERLAY_SYNC_INCOMMING);
		} else if (reviewContainer.getSyncState() == SynchronizationState.INCOMING_NEW) {
			return CommonImages.getImage(CommonImages.OVERLAY_SYNC_INCOMMING_NEW);
		}

		return CommonImages.getImage(CommonImages.OVERLAY_CLEAR);
	}

	private Image getReviewStateImage(TaskReview reviewContainer) {
		int reviewScore = reviewContainer.getCodeReviewScore();

		switch (reviewScore) {
		case MINUS_TWO:
			return CommonImages.getImage(ReviewsImages.RED_NOT);
		case MINUS_ONE:
			return CommonImages.getImage(ReviewsImages.MINUS_ONE);
		case PLUS_ONE:
			return CommonImages.getImage(ReviewsImages.PLUS_ONE);
		case PLUS_TWO:
			return CommonImages.getImage(ReviewsImages.GREEN_CHECK);
		}

		return CommonImages.getImage(CommonImages.OVERLAY_CLEAR);
	}

	private Image getVerifiedStateImage(TaskReview reviewContainer) {
		int verifiedState = reviewContainer.getVerifiedScore();

		if (verifiedState >= VERIFIED) {
			return CommonImages.getImage(ReviewsImages.GREEN_CHECK);
		} else if (verifiedState <= NOT_VERIFIED) {
			return CommonImages.getImage(ReviewsImages.RED_NOT);
		}

		return CommonImages.getImage(CommonImages.OVERLAY_CLEAR);
	}

	public String getColumnText(Object element, int columnIndex) {
		TaskReview reviewContainer = (TaskReview) element;

		switch (columnIndex) {
		case DESCRIPTION_COLUMN:
			return reviewContainer.getSummary();
		case STATUS_COLUMN:
			return reviewContainer.getStatus();
		default:
			return ""; //$NON-NLS-1$
		}
	}

	public String getSortString(Object element, int columnIndex) {
		TaskReview reviewContainer = (TaskReview) element;

		switch (columnIndex) {
		case CODE_REVIEW_COLUMN:
			return Integer.toString(reviewContainer.getCodeReviewScore());
		case VERIFIED_COLUMN:
			return Integer.toString(reviewContainer.getVerifiedScore());
		default:
			return getColumnText(element, columnIndex);
		}
	}

	@Override
	public String getToolTipText(Object element) {
		TaskReview review = (TaskReview) element;
		StringBuilder sb = new StringBuilder();

		sb.append("Description: ");
		sb.append(review.getSummary());

		sb.append(System.getProperty("line.separator"));
		sb.append("Verified: ");
		if (review.getVerifiedScore() >= 1) {
			sb.append("YES");
		} else {
			sb.append("NO");
		}

		sb.append(System.getProperty("line.separator"));
		sb.append("Code Review: ");
		sb.append(review.getCodeReviewScore());

		return sb.toString();
	}

	@Override
	public void update(ViewerCell cell) {
		Object element = cell.getElement();

		int columnNumber = cell.getColumnIndex();

		cell.setImage(getColumnImage(element, columnNumber));
		cell.setText(getColumnText(element, columnNumber));
	}
}