| /* |
| * Copyright 2000-2009 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.codeInsight.actions; |
| |
| import com.intellij.codeInsight.lookup.Lookup; |
| import com.intellij.codeInsight.lookup.LookupManager; |
| import com.intellij.openapi.actionSystem.AnActionEvent; |
| import com.intellij.openapi.actionSystem.CommonDataKeys; |
| import com.intellij.openapi.actionSystem.DataContext; |
| import com.intellij.openapi.actionSystem.PlatformDataKeys; |
| import com.intellij.openapi.actionSystem.Presentation; |
| import com.intellij.openapi.editor.Editor; |
| import com.intellij.openapi.project.Project; |
| import com.intellij.psi.PsiDocumentManager; |
| import com.intellij.psi.PsiFile; |
| import com.intellij.psi.impl.source.tree.injected.InjectedLanguageUtil; |
| import org.jetbrains.annotations.NotNull; |
| import org.jetbrains.annotations.Nullable; |
| |
| public abstract class BaseCodeInsightAction extends CodeInsightAction { |
| private final boolean myLookForInjectedEditor; |
| |
| protected BaseCodeInsightAction() { |
| this(true); |
| } |
| |
| protected BaseCodeInsightAction(boolean lookForInjectedEditor) { |
| myLookForInjectedEditor = lookForInjectedEditor; |
| } |
| |
| @Override |
| @Nullable |
| protected Editor getEditor(@NotNull final DataContext dataContext, @NotNull final Project project) { |
| Editor editor = getBaseEditor(dataContext, project); |
| if (!myLookForInjectedEditor) return editor; |
| return getInjectedEditor(project, editor); |
| } |
| |
| public static Editor getInjectedEditor(@NotNull Project project, final Editor editor) { |
| return getInjectedEditor(project, editor, true); |
| } |
| |
| public static Editor getInjectedEditor(@NotNull Project project, final Editor editor, boolean commit) { |
| Editor injectedEditor = editor; |
| if (editor != null) { |
| PsiDocumentManager documentManager = PsiDocumentManager.getInstance(project); |
| PsiFile psiFile = documentManager.getCachedPsiFile(editor.getDocument()); |
| if (psiFile != null) { |
| if (commit) documentManager.commitAllDocuments(); |
| injectedEditor = InjectedLanguageUtil.getEditorForInjectedLanguageNoCommit(editor, psiFile); |
| } |
| } |
| return injectedEditor; |
| } |
| |
| @Nullable |
| protected Editor getBaseEditor(final DataContext dataContext, final Project project) { |
| return super.getEditor(dataContext, project); |
| } |
| |
| @Override |
| public void update(AnActionEvent event) { |
| Presentation presentation = event.getPresentation(); |
| DataContext dataContext = event.getDataContext(); |
| Project project = CommonDataKeys.PROJECT.getData(dataContext); |
| if (project == null){ |
| presentation.setEnabled(false); |
| return; |
| } |
| |
| final Lookup activeLookup = LookupManager.getInstance(project).getActiveLookup(); |
| if (activeLookup != null){ |
| presentation.setEnabled(isValidForLookup()); |
| } |
| else { |
| super.update(event); |
| } |
| } |
| |
| protected boolean isValidForLookup() { |
| return false; |
| } |
| } |