/*
* Copyright 2000-2017 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.byteCodeViewer;
import com.intellij.codeInsight.lookup.LookupManager;
import com.intellij.icons.AllIcons;
import com.intellij.openapi.actionSystem.AnAction;
import com.intellij.openapi.actionSystem.AnActionEvent;
import com.intellij.openapi.actionSystem.CommonDataKeys;
import com.intellij.openapi.actionSystem.DataContext;
import com.intellij.openapi.application.ReadAction;
import com.intellij.openapi.compiler.CompilerManager;
import com.intellij.openapi.editor.Editor;
import com.intellij.openapi.progress.ProgressIndicator;
import com.intellij.openapi.progress.ProgressManager;
import com.intellij.openapi.progress.Task;
import com.intellij.openapi.project.Project;
import com.intellij.openapi.roots.ProjectRootManager;
import com.intellij.openapi.ui.Messages;
import com.intellij.openapi.ui.popup.JBPopup;
import com.intellij.openapi.ui.popup.JBPopupFactory;
import com.intellij.openapi.util.Disposer;
import com.intellij.openapi.vfs.VirtualFile;
import com.intellij.psi.*;
import com.intellij.psi.impl.source.tree.injected.InjectedLanguageUtil;
import com.intellij.psi.util.PsiUtilBase;
import com.intellij.psi.util.PsiUtilCore;
import com.intellij.ui.awt.RelativePoint;
import com.intellij.util.Processor;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
/**
* @author anna
* @since 5/4/12
*/
public class ShowByteCodeAction extends AnAction {
@Override
public void update(AnActionEvent e) {
e.getPresentation().setEnabled(false);
e.getPresentation().setIcon(AllIcons.Toolwindows.Documentation);
final Project project = e.getData(CommonDataKeys.PROJECT);
if (project != null) {
final PsiElement psiElement = getPsiElement(e.getDataContext(), project, e.getData(CommonDataKeys.EDITOR));
if (psiElement != null) {
if (psiElement.getContainingFile() instanceof PsiClassOwner) {
e.getPresentation().setEnabled(true);
}
}
}
}
@Override
public void actionPerformed(AnActionEvent e) {
final DataContext dataContext = e.getDataContext();
final Project project = e.getProject();
if (project == null) return;
final Editor editor = e.getData(CommonDataKeys.EDITOR);
final PsiElement psiElement = getPsiElement(dataContext, project, editor);
if (psiElement == null) return;
if (ByteCodeViewerManager.getContainingClass(psiElement) == null) {
Messages.showWarningDialog(project, "The selection should contain a class", "Unable to Find Class to Show Bytecode");
return;
}
final String psiElementTitle = ByteCodeViewerManager.getInstance(project).getTitle(psiElement);
final VirtualFile virtualFile = PsiUtilCore.getVirtualFile(psiElement);
if (virtualFile == null) return;
final RelativePoint bestPopupLocation = JBPopupFactory.getInstance().guessBestPopupLocation(dataContext);
final SmartPsiElementPointer element = SmartPointerManager.getInstance(project).createSmartPsiElementPointer(psiElement);
ProgressManager.getInstance().run(new Task.Backgroundable(project, "Looking for Bytecode...") {
private String myByteCode;
private String myErrorMessage;
private String myErrorTitle;
@Override
public void run(@NotNull ProgressIndicator indicator) {
if (ProjectRootManager.getInstance(project).getFileIndex().isInContent(virtualFile) && isMarkedForCompilation(project, virtualFile)) {
myErrorMessage = "Unable to show bytecode for '" + psiElementTitle + "'. Class file does not exist or is out-of-date.";
myErrorTitle = "Class File Out-Of-Date";
}
else {
myByteCode = ReadAction.compute(() -> ByteCodeViewerManager.getByteCode(psiElement));
}
}
@Override
public void onSuccess() {
if (project.isDisposed()) return;
if (myErrorMessage != null && myTitle != null) {
Messages.showWarningDialog(project, myErrorMessage, myErrorTitle);
return;
}
final PsiElement targetElement = element.getElement();
if (targetElement == null) return;
final ByteCodeViewerManager codeViewerManager = ByteCodeViewerManager.getInstance(project);
if (codeViewerManager.hasActiveDockedDocWindow()) {
codeViewerManager.doUpdateComponent(targetElement, myByteCode);
}
else {
if (myByteCode == null) {
Messages.showErrorDialog(project, "Unable to parse class file for '" + psiElementTitle + "'.", "Bytecode not Found");
return;
}
final ByteCodeViewerComponent component = new ByteCodeViewerComponent(project, null);
component.setText(myByteCode, targetElement);
Processor<JBPopup> pinCallback = popup -> {
codeViewerManager.recreateToolWindow(targetElement, targetElement);
popup.cancel();
return false;
};
final JBPopup popup = JBPopupFactory.getInstance().createComponentPopupBuilder(component, null)
.setProject(project)
.setDimensionServiceKey(project, ShowByteCodeAction.class.getName(), false)
.setResizable(true)
.setMovable(true)
.setRequestFocus(LookupManager.getActiveLookup(editor) == null)
.setTitle(psiElementTitle + " Bytecode")
.setCouldPin(pinCallback)
.createPopup();
Disposer.register(popup, component);
if (editor != null) {
popup.showInBestPositionFor(editor);
} else {
popup.show(bestPopupLocation);
}
}
}
});
}
private static boolean isMarkedForCompilation(Project project, VirtualFile virtualFile) {
final CompilerManager compilerManager = CompilerManager.getInstance(project);
return !compilerManager.isUpToDate(compilerManager.createFilesCompileScope(new VirtualFile[]{virtualFile}));
}
@Nullable
private static PsiElement getPsiElement(DataContext dataContext, Project project, @Nullable Editor editor) {
PsiElement psiElement = null;
if (editor == null) {
psiElement = dataContext.getData(CommonDataKeys.PSI_ELEMENT);
} else {
final PsiFile file = PsiUtilBase.getPsiFileInEditor(editor, project);
final Editor injectedEditor = InjectedLanguageUtil.getEditorForInjectedLanguageNoCommit(editor, file);
if (injectedEditor != null) {
psiElement = findElementInFile(PsiUtilBase.getPsiFileInEditor(injectedEditor, project), injectedEditor);
}
if (file != null && psiElement == null) {
psiElement = findElementInFile(file, editor);
}
}
return psiElement;
}
private static PsiElement findElementInFile(@Nullable PsiFile psiFile, @NotNull Editor editor) {
return psiFile != null ? psiFile.findElementAt(editor.getCaretModel().getOffset()) : null;
}
}