public TextRange getElementRange(@NotNull ASTNode astNode) {
    if (BasicJavaAstTreeUtil.is(astNode, CLASS_SET)) {
      ASTNode lBrace = BasicJavaAstTreeUtil.getLBrace(astNode);
      ASTNode rBrace = BasicJavaAstTreeUtil.getRBrace(astNode);
      if (lBrace != null && rBrace != null) {
        return new TextRange(BasicJavaAstTreeUtil.getTextOffset(lBrace), rBrace.getTextRange().getEndOffset());
      }
    }

    return astNode.getTextRange();
  }
public TextRange getElementRange(@NotNull ASTNode astNode)