python/src/com/jetbrains/python/codeInsight/postfix/PyIfPostfixTemplate.java

   1 /*
   2  * Copyright 2000-2016 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.codeInsight.postfix;
  17
  18 import com.intellij.codeInsight.template.postfix.templates.SurroundPostfixTemplateBase;
  19 import com.intellij.lang.surroundWith.Surrounder;
  20 import com.intellij.openapi.editor.Editor;
  21 import com.intellij.openapi.project.Project;
  22 import com.intellij.openapi.util.TextRange;
  23 import com.intellij.psi.PsiDocumentManager;
  24 import com.intellij.psi.codeStyle.CodeStyleManager;
  25 import com.intellij.util.IncorrectOperationException;
  26 import com.jetbrains.python.psi.*;
  27 import com.jetbrains.python.refactoring.surround.surrounders.expressions.PyExpressionSurrounder;
  28 import org.jetbrains.annotations.NotNull;
  29
  30 public class PyIfPostfixTemplate extends SurroundPostfixTemplateBase {
  31
  32   public static final String TEMPLATE_DESCRIPTION = "if expr";
  33
  34   public PyIfPostfixTemplate() {
  35     super("if", TEMPLATE_DESCRIPTION, PyPostfixUtils.PY_PSI_INFO, PyPostfixUtils.selectorTopmost());
  36   }
  37
  38   @NotNull
  39   @Override
  40   protected Surrounder getSurrounder() {
  41     return new PyIfSurrounder();
  42   }
  43
  44   private static class PyIfSurrounder extends PyExpressionSurrounder {
  45     @Override
  46     public boolean isApplicable(@NotNull PyExpression expr) {
  47       return true;
  48     }
  49
  50     @Override
  51     public TextRange surroundExpression(@NotNull Project project, @NotNull Editor editor, @NotNull PyExpression expression)
  52       throws IncorrectOperationException {
  53       String text = "if a:\n pass";
  54       PyIfStatement ifStatement = PyElementGenerator.getInstance(project).
  55         createFromText(LanguageLevel.getDefault(), PyIfStatement.class, text);
  56       final PyExpression condition = ifStatement.getIfPart().getCondition();
  57       if (condition != null) {
  58         condition.replace(expression);
  59       }
  60       ifStatement = (PyIfStatement)CodeStyleManager.getInstance(project).reformat(ifStatement);
  61       ifStatement = (PyIfStatement)expression.getParent().replace(ifStatement);
  62       PsiDocumentManager.getInstance(project).doPostponedOperationsAndUnblockDocument(editor.getDocument());
  63       PyStatementList statementList = ifStatement.getIfPart().getStatementList();
  64       PyStatement[] statements = statementList.getStatements();
  65       final TextRange range = statements[0].getTextRange();
  66       editor.getDocument().deleteString(range.getStartOffset(), range.getEndOffset());
  67       return TextRange.from(range.getStartOffset(), 0);
  68     }
  69
  70     @Override
  71     public String getTemplateDescription() {
  72       return TEMPLATE_DESCRIPTION;
  73     }
  74   }
  75 }