org.eclipse.mylyn.context.ui/src/org/eclipse/mylyn/ui/actions/ToggleRelationshipProviderAction.java

/*******************************************************************************
 * Copyright (c) 2004 - 2005 University Of British Columbia 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:
 *     University Of British Columbia - initial API and implementation
 *******************************************************************************/
/*
 * Created on Jul 27, 2004
  */
package org.eclipse.mylar.ui.actions;

import org.eclipse.jface.action.Action;
import org.eclipse.jface.action.ActionContributionItem;
import org.eclipse.jface.action.IMenuCreator;
import org.eclipse.mylar.core.IDegreeOfSeparation;
import org.eclipse.mylar.core.IMylarStructureBridge;
import org.eclipse.mylar.core.MylarPlugin;
import org.eclipse.swt.SWT;
import org.eclipse.swt.widgets.Control;
import org.eclipse.swt.widgets.Menu;
import org.eclipse.swt.widgets.MenuItem;

/**
 * @author Mik Kersten
 */
public class ToggleRelationshipProviderAction extends Action implements IMenuCreator{

//    private final String PREFIX = "org.eclipse.mylar.ui.relatedElements.providers";
//    private String prefId = "org.eclipse.mylar.ui.relatedElements.providers";
    
	private static final String LABEL_DEGREE_OF_SEPARATION = "Select Degree Of Separation:";

	public static final String ID = "org.eclipse.mylar.ui.actions.active.search.toggle";
		
    private IMylarStructureBridge bridge;
    private Menu dropDownMenu = null;
        
	public ToggleRelationshipProviderAction(IMylarStructureBridge bridge) {
		super();
        this.bridge = bridge;
        setImageDescriptor(MylarPlugin.getDefault().getActiveSearchIcon(bridge));
        setId(ID);
        setText(MylarPlugin.getDefault().getActiveSearchLabel(bridge));
        setToolTipText(MylarPlugin.getDefault().getActiveSearchLabel(bridge));
        setMenuCreator(this);	
		
		degreeOfSeparation = bridge.getRelationshipProviders().get(0).getCurrentDegreeOfSeparation();
		
		if (degreeOfSeparation > 0) run();
	}   
	 
	@Override
	public void run() {
		MylarPlugin.getContextManager().updateDegreesOfSeparation(bridge.getRelationshipProviders(), degreeOfSeparation);
	}
	
	public void dispose() {			
		if (dropDownMenu != null) {
			dropDownMenu.dispose();
			dropDownMenu = null;
		}
	}

	public Menu getMenu(Control parent) {			
		if (dropDownMenu != null) {
			dropDownMenu.dispose();
		}
		dropDownMenu = new Menu(parent);
		addActionsToMenu();
		return dropDownMenu;
	}

	public Menu getMenu(Menu parent) {
		if (dropDownMenu != null) {
			dropDownMenu.dispose();
		}
		dropDownMenu = new Menu(parent);
		addActionsToMenu();
		return dropDownMenu;
	}    
	
	private int degreeOfSeparation = 0;
	
	public void addActionsToMenu() {
		// HACK: if there are multiple providers, all store the same current DOS
		degreeOfSeparation = bridge.getRelationshipProviders().get(0).getCurrentDegreeOfSeparation();
		
		MenuItem menuItem = new MenuItem(dropDownMenu, SWT.NONE);
		menuItem.setText(LABEL_DEGREE_OF_SEPARATION);
//		menuItem.setEnabled(false);
		
		menuItem = new MenuItem(dropDownMenu, SWT.SEPARATOR);
		
		for(IDegreeOfSeparation separation: bridge.getDegreesOfSeparation()){
								
			Action degreeOfSeparationSelectionAction = new Action(separation.getDegree() + ": " + separation.getLabel(), AS_CHECK_BOX) {	    		
	    		@Override
				public void run() {
	    			try{
		    			degreeOfSeparation = Integer.parseInt(getId());
		    			MylarPlugin.getContextManager().updateDegreesOfSeparation(bridge.getRelationshipProviders(), degreeOfSeparation);
	    			} catch (NumberFormatException e){
	    				// ignore this for now
	    				MylarPlugin.fail(e, "invalid degree of separation", false);
	    			}
				}
			};  
			degreeOfSeparationSelectionAction.setId(""+separation.getDegree());
			degreeOfSeparationSelectionAction.setEnabled(true);
			degreeOfSeparationSelectionAction.setToolTipText(separation.getLabel());
			ActionContributionItem item= new ActionContributionItem(degreeOfSeparationSelectionAction);
			item.fill(dropDownMenu, -1);
			
			degreeOfSeparationSelectionAction.setChecked(false);
			if(degreeOfSeparation == 0 && separation.getDegree() == 0){
				degreeOfSeparationSelectionAction.setChecked(true);
			} else if (degreeOfSeparation != 0 && separation.getDegree() != 0 && degreeOfSeparation == separation.getDegree()) {
				degreeOfSeparationSelectionAction.setChecked(true);
			}
		}
	}
}