Merge branch 'origin/master'
[idea/community.git] / python / src / com / jetbrains / python / debugger / PyDebuggerEditorsProvider.java
1 /*
2  * Copyright 2000-2014 JetBrains s.r.o.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  * http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 package com.jetbrains.python.debugger;
17
18 import com.google.common.annotations.VisibleForTesting;
19 import com.intellij.openapi.editor.Document;
20 import com.intellij.openapi.fileEditor.FileDocumentManager;
21 import com.intellij.openapi.fileTypes.FileType;
22 import com.intellij.openapi.project.Project;
23 import com.intellij.psi.*;
24 import com.intellij.xdebugger.XSourcePosition;
25 import com.intellij.xdebugger.evaluation.EvaluationMode;
26 import com.intellij.xdebugger.evaluation.XDebuggerEditorsProvider;
27 import com.jetbrains.python.PythonFileType;
28 import com.jetbrains.python.psi.impl.PyExpressionCodeFragmentImpl;
29 import com.jetbrains.python.psi.impl.PyPsiUtils;
30 import org.jetbrains.annotations.NotNull;
31 import org.jetbrains.annotations.Nullable;
32
33
34 public class PyDebuggerEditorsProvider extends XDebuggerEditorsProvider {
35
36   @NotNull
37   @Override
38   public FileType getFileType() {
39     return PythonFileType.INSTANCE;
40   }
41
42   @NotNull
43   @Override
44   public Document createDocument(@NotNull final Project project,
45                                  @NotNull String text,
46                                  @Nullable final XSourcePosition sourcePosition,
47                                  @NotNull EvaluationMode mode) {
48     text = text.trim();
49     final PyExpressionCodeFragmentImpl fragment = new PyExpressionCodeFragmentImpl(project, "fragment.py", text, true);
50
51     // Bind to context
52     final PsiElement element = getContextElement(project, sourcePosition);
53     fragment.setContext(element);
54
55     return PsiDocumentManager.getInstance(project).getDocument(fragment);
56   }
57
58   @Nullable
59   @VisibleForTesting
60   public static PsiElement getContextElement(final Project project, XSourcePosition sourcePosition) {
61     if (sourcePosition != null) {
62       final Document document = FileDocumentManager.getInstance().getDocument(sourcePosition.getFile());
63       final PsiFile psiFile = PsiDocumentManager.getInstance(project).getPsiFile(document);
64       if (psiFile != null) {
65         int offset = sourcePosition.getOffset();
66         if (offset >= 0 && offset < document.getTextLength()) {
67           final int lineEndOffset = document.getLineEndOffset(document.getLineNumber(offset));
68           do {
69             PsiElement element = psiFile.findElementAt(offset);
70             if (element != null && !(element instanceof PsiWhiteSpace || element instanceof PsiComment)) {
71               return PyPsiUtils.getStatement(element);
72             }
73             offset = element.getTextRange().getEndOffset();
74           }
75           while (offset < lineEndOffset);
76         }
77       }
78     }
79     return null;
80   }
81 }