/*
* SoapUI, Copyright (C) 2004-2016 SmartBear Software
*
* Licensed under the EUPL, Version 1.1 or - as soon as they will be approved by the European Commission - subsequent
* versions of the EUPL (the "Licence");
* You may not use this work except in compliance with the Licence.
* You may obtain a copy of the Licence at:
*
* http://ec.europa.eu/idabc/eupl
*
* Unless required by applicable law or agreed to in writing, software distributed under the Licence is
* distributed on an "AS IS" basis, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
* express or implied. See the Licence for the specific language governing permissions and limitations
* under the Licence.
*/
package com.eviware.soapui.model.util;
import com.eviware.soapui.impl.rest.mock.RestMockAction;
import com.eviware.soapui.impl.rest.mock.RestMockResponse;
import com.eviware.soapui.impl.rest.mock.RestMockService;
import com.eviware.soapui.impl.wsdl.WsdlOperation;
import com.eviware.soapui.impl.wsdl.WsdlProject;
import com.eviware.soapui.impl.wsdl.WsdlTestSuite;
import com.eviware.soapui.impl.wsdl.loadtest.WsdlLoadTest;
import com.eviware.soapui.impl.wsdl.mock.WsdlMockOperation;
import com.eviware.soapui.impl.wsdl.mock.WsdlMockResponse;
import com.eviware.soapui.impl.wsdl.mock.WsdlMockService;
import com.eviware.soapui.impl.wsdl.testcase.WsdlTestCase;
import com.eviware.soapui.model.ModelItem;
import com.eviware.soapui.model.iface.Interface;
import com.eviware.soapui.model.iface.Operation;
import com.eviware.soapui.model.iface.Request;
import com.eviware.soapui.model.mock.MockOperation;
import com.eviware.soapui.model.mock.MockResponse;
import com.eviware.soapui.model.mock.MockService;
import com.eviware.soapui.model.project.Project;
import com.eviware.soapui.model.testsuite.LoadTest;
import com.eviware.soapui.model.testsuite.TestAssertion;
import com.eviware.soapui.model.testsuite.TestCase;
import com.eviware.soapui.model.testsuite.TestStep;
import com.eviware.soapui.model.testsuite.TestSuite;
import com.eviware.soapui.security.SecurityTest;
import com.eviware.soapui.support.UISupport;
import javax.swing.ImageIcon;
import java.util.HashMap;
import java.util.Map;
public class ModelItemIconFactory {
private static Map<Class<? extends ModelItem>, String> modelItemIcons = new HashMap<Class<? extends ModelItem>, String>();
static {
// the "class" keys here are only used for lookup - but must be implementations of ModelItem
// the icon is used in the project overview
modelItemIcons.put(Project.class, WsdlProject.ICON_NAME);
modelItemIcons.put(TestSuite.class, WsdlTestSuite.ICON_NAME);
modelItemIcons.put(TestCase.class, WsdlTestCase.ICON_NAME);
modelItemIcons.put(LoadTest.class, WsdlLoadTest.ICON_NAME);
modelItemIcons.put(MockService.class, WsdlMockService.ICON_NAME);
modelItemIcons.put(MockResponse.class, WsdlMockResponse.ICON_NAME);
modelItemIcons.put(MockOperation.class, WsdlMockOperation.ICON_NAME);
modelItemIcons.put(RestMockService.class, RestMockService.ICON_NAME);
modelItemIcons.put(RestMockAction.class, RestMockAction.getDefaultIcon());
modelItemIcons.put(RestMockResponse.class, RestMockResponse.ICON_NAME);
modelItemIcons.put(Operation.class, WsdlOperation.ICON_NAME);
modelItemIcons.put(SecurityTest.class, SecurityTest.ICON_NAME);
// the following use different icon files for the overview and in the tree
modelItemIcons.put(TestStep.class, "/teststeps.gif");
modelItemIcons.put(TestAssertion.class, "/assertions.png");
modelItemIcons.put(Request.class, "/soap_request.png");
modelItemIcons.put(Interface.class, "/interface.png");
}
public static ImageIcon getIcon(Class<? extends ModelItem> clazz) {
if (modelItemIcons.containsKey(clazz)) {
return UISupport.createImageIcon(modelItemIcons.get(clazz));
}
for (Class<?> iface : clazz.getInterfaces()) {
if (modelItemIcons.containsKey(iface)) {
return UISupport.createImageIcon(modelItemIcons.get(iface));
}
}
return null;
}
@SuppressWarnings("unchecked")
public static String getIconPath(Class<? extends ModelItem> clazz) {
if (modelItemIcons.containsKey(clazz)) {
return modelItemIcons.get(clazz);
}
for (Class<?> iface : clazz.getInterfaces()) {
if (modelItemIcons.containsKey(iface)) {
return modelItemIcons.get(iface);
}
}
while (clazz.getSuperclass() != null && ModelItem.class.isAssignableFrom(clazz.getSuperclass())) {
return getIconPath((Class<? extends ModelItem>) clazz.getSuperclass());
}
return null;
}
public static <T extends ModelItem> String getIconPath(T modelItem) {
return getIconPath(modelItem.getClass());
}
public static <T extends ModelItem> ImageIcon getIcon(T modelItem) {
return getIcon(modelItem.getClass());
}
}