summaryrefslogtreecommitdiffstats
path: root/src/org.gluster.storage.management.console/src/org/gluster/storage/management/console/toolbar/GlusterToolbarManager.java
blob: fc8ef86506daf458ef9780f8a214bddf797f95ef (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
/*******************************************************************************
 * Copyright (c) 2011 Gluster, Inc. <http://www.gluster.com>
 * This file is part of Gluster Management Console.
 *
 * Gluster Management Console is free software; you can redistribute it and/or 
 * modify it under the terms of the GNU Affero General Public License as published
 * by the Free Software Foundation; either version 3 of the License, or
 * (at your option) any later version.
 *  
 * Gluster Management Console is distributed in the hope that it will be useful, 
 * but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
 * or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Affero General Public License
 * for more details.
 *  
 * You should have received a copy of the GNU Affero General Public License
 * along with this program. If not, see
 * <http://www.gnu.org/licenses/>.
 *******************************************************************************/
package org.gluster.storage.management.console.toolbar;

import java.util.HashMap;
import java.util.Map;

import org.eclipse.ui.IWorkbenchPage;
import org.eclipse.ui.IWorkbenchWindow;
import org.gluster.storage.management.console.actions.ActionConstants;
import org.gluster.storage.management.core.model.Cluster;
import org.gluster.storage.management.core.model.Entity;
import org.gluster.storage.management.core.model.EntityGroup;
import org.gluster.storage.management.core.model.GlusterServer;
import org.gluster.storage.management.core.model.Server;
import org.gluster.storage.management.core.model.TaskInfo;
import org.gluster.storage.management.core.model.Volume;


public class GlusterToolbarManager implements ToolbarManager {
	private enum ENTITY_TYPE {
		CLUSTER, VOLUMES, VOLUME, GLUSTER_SERVERS, GLUSTER_SERVER, DISCOVERED_SERVERS, DISCOVERED_SERVER, TASK
	};

	private IWorkbenchWindow window;
	private final Map<Class<? extends Entity>, ENTITY_TYPE> entityTypeMap = createEntityTypeMap();
	private final Map<ENTITY_TYPE, String> actionSetMap = createActionSetMap();

	private Map<Class<? extends Entity>, ENTITY_TYPE> createEntityTypeMap() {
		Map<Class<? extends Entity>, ENTITY_TYPE> entityTypeMap = new HashMap<Class<? extends Entity>, GlusterToolbarManager.ENTITY_TYPE>();
		entityTypeMap.put(Cluster.class, ENTITY_TYPE.CLUSTER);
		entityTypeMap.put(Volume.class, ENTITY_TYPE.VOLUME);
		entityTypeMap.put(Server.class, ENTITY_TYPE.DISCOVERED_SERVER);
		entityTypeMap.put(GlusterServer.class, ENTITY_TYPE.GLUSTER_SERVER);
		entityTypeMap.put(TaskInfo.class, ENTITY_TYPE.TASK);

		return entityTypeMap;
	}

	private Map<ENTITY_TYPE, String> createActionSetMap() {
		Map<ENTITY_TYPE, String> actionSetMap = new HashMap<GlusterToolbarManager.ENTITY_TYPE, String>();
		actionSetMap.put(ENTITY_TYPE.CLUSTER, ActionConstants.ACTION_SET_CLUSTER);
		actionSetMap.put(ENTITY_TYPE.VOLUMES, ActionConstants.ACTION_SET_VOLUMES);
		actionSetMap.put(ENTITY_TYPE.VOLUME, ActionConstants.ACTION_SET_VOLUME);
		actionSetMap.put(ENTITY_TYPE.GLUSTER_SERVERS, ActionConstants.ACTION_SET_GLUSTER_SERVERS);
		actionSetMap.put(ENTITY_TYPE.GLUSTER_SERVER, ActionConstants.ACTION_SET_GLUSTER_SERVER);
		actionSetMap.put(ENTITY_TYPE.DISCOVERED_SERVERS, ActionConstants.ACTION_SET_DISCOVERED_SERVERS);
		actionSetMap.put(ENTITY_TYPE.DISCOVERED_SERVER, ActionConstants.ACTION_SET_DISCOVERED_SERVER);
		actionSetMap.put(ENTITY_TYPE.TASK, ActionConstants.ACTION_SET_TASK);

		return actionSetMap;
	}

	public GlusterToolbarManager(IWorkbenchWindow window) {
		this.window = window;
	}

	@SuppressWarnings("rawtypes")
	private ENTITY_TYPE getEntityType(Entity entity) {
		if (entity instanceof EntityGroup) {
			EntityGroup entityGroup = (EntityGroup) entity;
			if (entityGroup.getEntityType() == Volume.class) {
				return ENTITY_TYPE.VOLUMES;
			} else if (entityGroup.getEntityType() == GlusterServer.class) {
				return ENTITY_TYPE.GLUSTER_SERVERS;
			} else {
				return ENTITY_TYPE.DISCOVERED_SERVERS;
			}
		}

		return entityTypeMap.get(entity.getClass());
	}

	@Override
	public void updateToolbar(Entity entity) {
		ENTITY_TYPE entityType = getEntityType(entity);
		IWorkbenchPage page = window.getActivePage();
		
		for (ENTITY_TYPE targetEntityType : actionSetMap.keySet()) {
			String actionSetId = actionSetMap.get(targetEntityType);
			if (entityType == targetEntityType) {
				// show only the action set mapped to given entity
				page.showActionSet(actionSetId);
			} else {
				page.hideActionSet(actionSetId);
			}
		}
	}
}