| /* |
| * 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. |
| */ |
| |
| /** |
| * class ToggleMethodBreakpointAction |
| * @author Jeka |
| */ |
| package com.intellij.debugger.actions; |
| |
| import com.intellij.debugger.DebuggerManagerEx; |
| import com.intellij.debugger.engine.requests.RequestManagerImpl; |
| import com.intellij.debugger.impl.DebuggerUtilsEx; |
| import com.intellij.debugger.ui.breakpoints.Breakpoint; |
| import com.intellij.debugger.ui.breakpoints.BreakpointManager; |
| import com.intellij.debugger.ui.breakpoints.MethodBreakpoint; |
| import com.intellij.openapi.actionSystem.*; |
| import com.intellij.openapi.editor.Document; |
| import com.intellij.openapi.editor.Editor; |
| import com.intellij.openapi.fileEditor.FileEditorManager; |
| import com.intellij.openapi.fileTypes.FileType; |
| import com.intellij.openapi.fileTypes.StdFileTypes; |
| import com.intellij.openapi.project.Project; |
| import com.intellij.openapi.vfs.VirtualFile; |
| import com.intellij.psi.PsiDocumentManager; |
| import com.intellij.psi.PsiElement; |
| import com.intellij.psi.PsiFile; |
| import com.intellij.psi.PsiMethod; |
| import com.intellij.util.DocumentUtil; |
| import com.intellij.util.text.CharArrayUtil; |
| import org.jetbrains.annotations.Nullable; |
| |
| public class ToggleMethodBreakpointAction extends AnAction { |
| |
| public void update(AnActionEvent event){ |
| boolean toEnable = getPlace(event) != null; |
| |
| if (ActionPlaces.isPopupPlace(event.getPlace())) { |
| event.getPresentation().setVisible(toEnable); |
| } |
| else { |
| event.getPresentation().setEnabled(toEnable); |
| } |
| } |
| |
| |
| public void actionPerformed(AnActionEvent e) { |
| Project project = e.getData(CommonDataKeys.PROJECT); |
| if (project == null) { |
| return; |
| } |
| DebuggerManagerEx debugManager = DebuggerManagerEx.getInstanceEx(project); |
| if (debugManager == null) { |
| return; |
| } |
| final BreakpointManager manager = debugManager.getBreakpointManager(); |
| final PlaceInDocument place = getPlace(e); |
| if(place != null && DocumentUtil.isValidOffset(place.getOffset(), place.getDocument())) { |
| Breakpoint breakpoint = manager.findBreakpoint(place.getDocument(), place.getOffset(), MethodBreakpoint.CATEGORY); |
| if(breakpoint == null) { |
| final int methodLine = place.getDocument().getLineNumber(place.getOffset()); |
| MethodBreakpoint methodBreakpoint = manager.addMethodBreakpoint(place.getDocument(), methodLine); |
| if(methodBreakpoint != null) { |
| RequestManagerImpl.createRequests(methodBreakpoint); |
| } |
| } |
| else { |
| manager.removeBreakpoint(breakpoint); |
| } |
| } |
| } |
| |
| @Nullable |
| private static PlaceInDocument getPlace(AnActionEvent event) { |
| final Project project = event.getData(CommonDataKeys.PROJECT); |
| if(project == null) { |
| return null; |
| } |
| |
| PsiElement method = null; |
| Document document = null; |
| |
| if (ActionPlaces.PROJECT_VIEW_POPUP.equals(event.getPlace()) || |
| ActionPlaces.STRUCTURE_VIEW_POPUP.equals(event.getPlace()) || |
| ActionPlaces.FAVORITES_VIEW_POPUP.equals(event.getPlace()) || |
| ActionPlaces.NAVIGATION_BAR_POPUP.equals(event.getPlace())) { |
| final PsiElement psiElement = event.getData(CommonDataKeys.PSI_ELEMENT); |
| if(psiElement instanceof PsiMethod) { |
| final PsiFile containingFile = psiElement.getContainingFile(); |
| if (containingFile != null) { |
| method = psiElement; |
| document = PsiDocumentManager.getInstance(project).getDocument(containingFile); |
| } |
| } |
| } |
| else { |
| Editor editor = event.getData(CommonDataKeys.EDITOR); |
| if(editor == null) { |
| editor = FileEditorManager.getInstance(project).getSelectedTextEditor(); |
| } |
| if (editor != null) { |
| document = editor.getDocument(); |
| PsiFile file = PsiDocumentManager.getInstance(project).getPsiFile(document); |
| if (file != null) { |
| final VirtualFile virtualFile = file.getVirtualFile(); |
| FileType fileType = virtualFile != null ? virtualFile.getFileType() : null; |
| if (StdFileTypes.JAVA == fileType || StdFileTypes.CLASS == fileType) { |
| method = findMethod(project, editor); |
| } |
| } |
| } |
| } |
| |
| return method != null ? new PlaceInDocument(document, method.getTextOffset()) : null; |
| } |
| |
| @Nullable |
| private static PsiMethod findMethod(Project project, Editor editor) { |
| if (editor == null) { |
| return null; |
| } |
| PsiFile psiFile = PsiDocumentManager.getInstance(project).getPsiFile(editor.getDocument()); |
| if(psiFile == null) { |
| return null; |
| } |
| final int offset = CharArrayUtil.shiftForward(editor.getDocument().getCharsSequence(), editor.getCaretModel().getOffset(), " \t"); |
| return DebuggerUtilsEx.findPsiMethod(psiFile, offset); |
| } |
| } |