TestRoots/watchdog

package nl.tudelft.watchdog.eclipse.ui.handlers;

import java.util.Map;

import nl.tudelft.watchdog.core.util.WatchDogGlobals;
import nl.tudelft.watchdog.eclipse.ui.InfoDialog;
import nl.tudelft.watchdog.eclipse.ui.preferences.Preferences;
import nl.tudelft.watchdog.eclipse.ui.util.UIUtils;

import org.eclipse.core.commands.AbstractHandler;
import org.eclipse.core.commands.ExecutionEvent;
import org.eclipse.core.commands.ExecutionException;
import org.eclipse.jface.dialogs.MessageDialog;
import org.eclipse.ui.commands.IElementUpdater;
import org.eclipse.ui.handlers.HandlerUtil;
import org.eclipse.ui.menus.UIElement;

/** Handler for displaying an {@link InfoDialog}. */
public class InfoHandler extends AbstractHandler implements IElementUpdater {

    @Override
    public Object execute(ExecutionEvent event) throws ExecutionException {
        InfoDialog infoDialog = new InfoDialog(
                HandlerUtil.getActiveShell(event));
        infoDialog.setBlockOnOpen(false);
        infoDialog.open();
        return null;
    }

    @Override
    public void updateElement(UIElement element,
            @SuppressWarnings("rawtypes") Map parameters) {
        if (WatchDogGlobals.isActive) {
            Preferences preferences = Preferences.getInstance();

            displayUpdateQuestionDialog(preferences);

            if (WatchDogGlobals.lastTransactionFailed
                    || preferences.isOldVersion()) {
                element.setTooltip(WatchDogGlobals.ACTIVE_WATCHDOG_TEXT);
                element.setIcon(UIUtils.WATCHDOG_ICON_WARNING);
            } else {
                element.setTooltip(WatchDogGlobals.ACTIVE_WATCHDOG_TEXT);
                element.setIcon(UIUtils.WATCHDOG_ICON);
            }

            if (!preferences.isOldVersion()) {
                preferences.setBigUpdateAnswered(false);
            }

        } else {
            element.setTooltip(WatchDogGlobals.INACTIVE_WATCHDOG_TEXT);
            element.setIcon(UIUtils.WATCHDOG_ICON_DISABLED);
        }

    }

    private void displayUpdateQuestionDialog(Preferences preferences) {
        if (preferences.isBigUpdateAvailable()
                && !preferences.isBigUpdateAnswered()) {
            preferences.setBigUpdateAnswered(true);
            boolean wantsToUpdate = MessageDialog.openQuestion(null,
                    "Major WatchDog Update Available!",
                    "We improved WatchDog a lot for you! Update?");
            if (wantsToUpdate) {
                UIUtils.updateWatchDog();
            }
        }
    }

}