コード例 #1
0
 @Override
 public void doFix(Project project, ProblemDescriptor descriptor)
   throws IncorrectOperationException {
   final PsiElement forElement = descriptor.getPsiElement();
   final PsiElement parent = forElement.getParent();
   if (!(parent instanceof PsiForStatement)) {
     return;
   }
   final PsiForStatement forStatement = (PsiForStatement)parent;
   final String newExpression;
   if (isArrayLoopStatement(forStatement)) {
     newExpression = createArrayIterationText(forStatement);
   }
   else if (isCollectionLoopStatement(forStatement, ignoreUntypedCollections)) {
     newExpression = createCollectionIterationText(forStatement);
   }
   else if (isIndexedListLoopStatement(forStatement, ignoreUntypedCollections)) {
     newExpression = createListIterationText(forStatement);
   }
   else {
     return;
   }
   if (newExpression == null) {
     return;
   }
   replaceStatementAndShortenClassNames(forStatement, newExpression);
 }
コード例 #2
0
 @Override
 public void visitElement(PsiElement element) {
   if (element.getLanguage() != JavaLanguage.INSTANCE) {
     return;
   }
   if (!PsiUtil.isLanguageLevel5OrHigher(element)) {
     return;
   }
   super.visitElement(element);
 }
コード例 #3
0
 private void replaceCollectionGetAccess(
   PsiElement element, String contentVariableName,
   PsiVariable listVariable, String indexName,
   PsiElement childToSkip, StringBuilder out) {
   if (isListGetLookup(element, indexName, listVariable)) {
     out.append(contentVariableName);
   }
   else {
     final PsiElement[] children = element.getChildren();
     if (children.length == 0) {
       final String text = element.getText();
       if (PsiKeyword.INSTANCEOF.equals(text) &&
           out.charAt(out.length() - 1) != ' ') {
         out.append(' ');
       }
       out.append(text);
     }
     else {
       boolean skippingWhiteSpace = false;
       for (final PsiElement child : children) {
         if (child.equals(childToSkip)) {
           skippingWhiteSpace = true;
         }
         else if (child instanceof PsiWhiteSpace &&
                  skippingWhiteSpace) {
           //don't do anything
         }
         else {
           skippingWhiteSpace = false;
           replaceCollectionGetAccess(child,
                                      contentVariableName,
                                      listVariable, indexName,
                                      childToSkip, out);
         }
       }
     }
   }
 }
コード例 #4
0
 private void replaceIteratorNext(
   PsiElement element, String contentVariableName,
   String iteratorName, PsiElement childToSkip,
   StringBuilder out, PsiType contentType) {
   if (isIteratorNext(element, iteratorName, contentType)) {
     out.append(contentVariableName);
   }
   else {
     final PsiElement[] children = element.getChildren();
     if (children.length == 0) {
       final String text = element.getText();
       if (PsiKeyword.INSTANCEOF.equals(text) &&
           out.charAt(out.length() - 1) != ' ') {
         out.append(' ');
       }
       out.append(text);
     }
     else {
       boolean skippingWhiteSpace = false;
       for (final PsiElement child : children) {
         if (child.equals(childToSkip)) {
           skippingWhiteSpace = true;
         }
         else if (child instanceof PsiWhiteSpace &&
                  skippingWhiteSpace) {
           //don't do anything
         }
         else {
           skippingWhiteSpace = false;
           replaceIteratorNext(child, contentVariableName,
                               iteratorName, childToSkip, out, contentType);
         }
       }
     }
   }
 }