update/org.eclipse.update.ui/src/org/eclipse/update/internal/ui/UpdateWebSitesAction.java

package org.eclipse.update.internal.ui;

import org.eclipse.jface.action.*;
import org.eclipse.jface.viewers.*;
import org.eclipse.swt.events.*;
import org.eclipse.swt.widgets.*;
import org.eclipse.ui.*;
import org.eclipse.update.internal.ui.model.*;

/**
 * @see IWorkbenchWindowActionDelegate
 */
public class UpdateWebSitesAction implements IWorkbenchWindowPulldownDelegate2, IUpdateModelChangedListener {
	private Menu fMenu;
	private boolean fRecreateMenu = false;

	/**
	 *
	 */
	public UpdateWebSitesAction() {
		UpdateUI.getDefault().getUpdateModel().addUpdateModelChangedListener(this);
	}

	/**
	 * @see IWorkbenchWindowActionDelegate#run
	 */
	public void run(IAction action)  {
	}

	/**
	 * @see IWorkbenchWindowActionDelegate#selectionChanged
	 */
	public void selectionChanged(IAction action, ISelection selection)  {
	}

	/**
	 * @see IWorkbenchWindowActionDelegate#dispose
	 */
	public void dispose()  {
		if (fMenu != null) {
			fMenu.dispose();
		}
		UpdateUI.getDefault().getUpdateModel().removeUpdateModelChangedListener(this);
	}

	/**
	 * @see IWorkbenchWindowActionDelegate#init
	 */
	public void init(IWorkbenchWindow window)  {
	}

	public Menu getMenu(Control parent) {
		setMenu(new Menu(parent));
		fillMenu(fMenu);
		initMenu();
		return fMenu;
	}
	
	public Menu getMenu(Menu parent) {
		setMenu(new Menu(parent));
		fillMenu(fMenu);
		initMenu();
		return fMenu;
	}
	
	private void setMenu(Menu menu) {
		if (fMenu != null) {
			fMenu.dispose();
		}
		fMenu = menu;
	}
	
	protected void fillMenu(Menu menu) {
		addActionToMenu(menu, new AddBookmarksAction());
		addActionToMenu(menu, new EditBookmarksAction());
		DiscoveryFolder folder = new DiscoveryFolder();
		Object[] children = folder.getChildren(folder);
		for (int i = 0; i < children.length; i++) {
			if (children[i] instanceof SiteBookmark) {
				SiteBookmark bookmark = (SiteBookmark)children[i];
				if (bookmark.isWebBookmark())
					addActionToMenu(menu, new GoToWebsiteAction(bookmark));
			}
		}
		SiteBookmark[] bookmarks = UpdateUI.getDefault().getUpdateModel().getBookmarkLeafs();
		for (int i = 0; i < bookmarks.length; i++) {
			if (bookmarks[i].isWebBookmark())
				addActionToMenu(menu, new GoToWebsiteAction(bookmarks[i]));
		}
		
		if (menu.getItemCount() > 2) 
			new Separator().fill(menu, 2);
			
	}
	
	private void initMenu() {
		fMenu.addMenuListener(new MenuAdapter() {

			public void menuShown(MenuEvent e) {
				if (fRecreateMenu) {
					Menu m = (Menu)e.widget;
					MenuItem[] items = m.getItems();
					for (int i=0; i < items.length; i++) {
						items[i].dispose();
					}
					fillMenu(m);
					fRecreateMenu= false;
				}
			}
		});
	}
	
	protected void addActionToMenu(Menu parent, Action action) {
		ActionContributionItem item = new ActionContributionItem(action);
		item.fill(parent, -1);
	}
	

	public void objectsAdded(Object parent, Object[] children) {
		fRecreateMenu = true;		
	}

	public void objectsRemoved(Object parent, Object[] children) {
		fRecreateMenu = true;		
	}

	public void objectChanged(Object object, String property) {
		fRecreateMenu = true;		
	}


}