JBTable: use a singleton list
[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.intellij.openapi.editor.Document;
19 import com.intellij.openapi.fileEditor.FileDocumentManager;
20 import com.intellij.openapi.fileTypes.FileType;
21 import com.intellij.openapi.project.Project;
22 import com.intellij.psi.*;
23 import com.intellij.xdebugger.XSourcePosition;
24 import com.intellij.xdebugger.evaluation.EvaluationMode;
25 import com.intellij.xdebugger.evaluation.XDebuggerEditorsProvider;
26 import com.jetbrains.python.PythonFileType;
27 import com.jetbrains.python.psi.impl.PyExpressionCodeFragmentImpl;
28 import com.jetbrains.python.psi.impl.PyPsiUtils;
29 import org.jetbrains.annotations.NotNull;
30 import org.jetbrains.annotations.Nullable;
31
32
33 public class PyDebuggerEditorsProvider extends XDebuggerEditorsProvider {
34
35   @NotNull
36   @Override
37   public FileType getFileType() {
38     return PythonFileType.INSTANCE;
39   }
40
41   @NotNull
42   @Override
43   public Document createDocument(@NotNull final Project project,
44                                  @NotNull String text,
45                                  @Nullable final XSourcePosition sourcePosition,
46                                  @NotNull EvaluationMode mode) {
47     text = text.trim();
48     final PyExpressionCodeFragmentImpl fragment = new PyExpressionCodeFragmentImpl(project, "fragment.py", text, true);
49
50     // Bind to context
51     final PsiElement element = getContextElement(project, sourcePosition);
52     fragment.setContext(element);
53
54     return PsiDocumentManager.getInstance(project).getDocument(fragment);
55   }
56
57   @Nullable
58   private static PsiElement getContextElement(final Project project, XSourcePosition sourcePosition) {
59     if (sourcePosition != null) {
60       final Document document = FileDocumentManager.getInstance().getDocument(sourcePosition.getFile());
61       final PsiFile psiFile = PsiDocumentManager.getInstance(project).getPsiFile(document);
62       if (psiFile != null) {
63         int offset = sourcePosition.getOffset();
64         if (offset >= 0 && offset < document.getTextLength()) {
65           final int lineEndOffset = document.getLineEndOffset(document.getLineNumber(offset));
66           do {
67             PsiElement element = psiFile.findElementAt(offset);
68             if (element != null && !(element instanceof PsiWhiteSpace || element instanceof PsiComment)) {
69               return PyPsiUtils.getStatement(element);
70             }
71             offset = element.getTextRange().getEndOffset() + 1;
72           }
73           while (offset < lineEndOffset);
74         }
75       }
76     }
77     return null;
78   }
79 }