| /* |
| * Copyright 2000-2014 JetBrains s.r.o. |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| package com.intellij.openapi.actionSystem.impl; |
| |
| import com.intellij.openapi.actionSystem.*; |
| import com.intellij.openapi.project.Project; |
| import com.intellij.openapi.project.ProjectType; |
| import com.intellij.openapi.project.ProjectTypeService; |
| import org.jetbrains.annotations.NotNull; |
| import org.jetbrains.annotations.Nullable; |
| import org.jetbrains.annotations.TestOnly; |
| |
| import java.util.HashMap; |
| import java.util.Map; |
| |
| /** |
| * @author Dmitry Avdeev |
| */ |
| public class ChameleonAction extends AnAction { |
| |
| private final Map<ProjectType, AnAction> myActions = new HashMap<ProjectType, AnAction>(); |
| |
| public ChameleonAction(@NotNull AnAction first, ProjectType projectType) { |
| addAction(first, projectType); |
| copyFrom(myActions.values().iterator().next()); |
| } |
| |
| public AnAction addAction(AnAction action, ProjectType projectType) { |
| if (action instanceof ActionStub) { |
| String type = ((ActionStub)action).getProjectType(); |
| action = ActionManagerImpl.convertStub((ActionStub)action); |
| projectType = type == null ? null : new ProjectType(type); |
| } |
| return myActions.put(projectType, action); |
| } |
| |
| @Override |
| public void actionPerformed(@NotNull AnActionEvent e) { |
| AnAction action = getAction(e); |
| assert action != null; |
| action.actionPerformed(e); |
| } |
| |
| @Override |
| public void update(@NotNull AnActionEvent e) { |
| AnAction action = getAction(e); |
| if (action != null) { |
| e.getPresentation().setVisible(true); |
| action.update(e); |
| } |
| else { |
| e.getPresentation().setVisible(false); |
| } |
| } |
| |
| @Nullable |
| private AnAction getAction(AnActionEvent e) { |
| Project project = CommonDataKeys.PROJECT.getData(e.getDataContext()); |
| ProjectType projectType = ProjectTypeService.getProjectType(project); |
| AnAction action = myActions.get(projectType); |
| if (action == null) action = myActions.get(null); |
| return action; |
| } |
| |
| @TestOnly |
| public Map<ProjectType, AnAction> getActions() { |
| return myActions; |
| } |
| } |