java/java-impl/src/com/intellij/codeInsight/editorActions/smartEnter/SwitchExpressionFixer.java

   1 /*
   2  * Copyright 2000-2009 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
  17 /*
  18  * Created by IntelliJ IDEA.
  19  * User: max
  20  * Date: Jul 30, 2006
  21  * Time: 4:55:07 PM
  22  */
  23 package com.intellij.codeInsight.editorActions.smartEnter;
  24
  25 import com.intellij.openapi.editor.Document;
  26 import com.intellij.openapi.editor.Editor;
  27 import com.intellij.psi.*;
  28 import com.intellij.util.IncorrectOperationException;
  29
  30 /**
  31  * Created by IntelliJ IDEA.
  32  * User: max
  33  * Date: Sep 5, 2003
  34  * Time: 5:32:01 PM
  35  * To change this template use Options | File Templates.
  36  */
  37 @SuppressWarnings({"HardCodedStringLiteral"})
  38 public class SwitchExpressionFixer implements Fixer {
  39   public void apply(Editor editor, JavaSmartEnterProcessor processor, PsiElement psiElement) throws IncorrectOperationException {
  40     if (psiElement instanceof PsiSwitchStatement) {
  41       final Document doc = editor.getDocument();
  42       final PsiSwitchStatement switchStatement = (PsiSwitchStatement)psiElement;
  43       final PsiJavaToken rParenth = switchStatement.getRParenth();
  44       final PsiJavaToken lParenth = switchStatement.getLParenth();
  45       final PsiExpression condition = switchStatement.getExpression();
  46
  47       if (condition == null) {
  48         if (lParenth == null || rParenth == null) {
  49           int stopOffset = doc.getLineEndOffset(doc.getLineNumber(switchStatement.getTextRange().getStartOffset()));
  50           final PsiCodeBlock block = switchStatement.getBody();
  51           if (block != null) {
  52             stopOffset = Math.min(stopOffset, block.getTextRange().getStartOffset());
  53           }
  54           doc.replaceString(switchStatement.getTextRange().getStartOffset(), stopOffset, "switch ()");
  55         } else {
  56           processor.registerUnresolvedError(lParenth.getTextRange().getEndOffset());
  57         }
  58       } else if (rParenth == null) {
  59         doc.insertString(condition.getTextRange().getEndOffset(), ")");
  60       }
  61     }
  62   }
  63 }