io.codemodder.codemods.DeclareVariableOnSeparateLineForVariableDeclarationExpr Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of core-codemods Show documentation

Codemods for fixing common errors across many Java projects

There is a newer version: 0.97.3

package io.codemodder.codemods;

import com.github.javaparser.ast.Node;
import com.github.javaparser.ast.NodeList;
import com.github.javaparser.ast.body.VariableDeclarator;
import com.github.javaparser.ast.expr.VariableDeclarationExpr;
import com.github.javaparser.ast.stmt.BlockStmt;
import com.github.javaparser.ast.stmt.ExpressionStmt;
import com.github.javaparser.ast.stmt.Statement;
import java.util.ArrayList;
import java.util.List;
import java.util.Objects;
import java.util.Optional;

final class DeclareVariableOnSeparateLineForVariableDeclarationExpr
    extends DeclareVariableOnSeparateLine {

  private final VariableDeclarationExpr variableDeclarationExpr;

  DeclareVariableOnSeparateLineForVariableDeclarationExpr(
      final VariableDeclarationExpr parentNode) {
    super(parentNode);
    this.variableDeclarationExpr = Objects.requireNonNull(parentNode);
  }

  /**
   * Returns a list of {@link VariableDeclarationExpr} nodes created from a list of {@link
   * VariableDeclarator}s.
   */
  protected List createVariableNodesToAdd(List inlineVariables) {
    final List nodesToAdd = new ArrayList<>();
    for (VariableDeclarator inlineVariable : inlineVariables) {
      final VariableDeclarationExpr newVariableDeclarationExpr =
          new VariableDeclarationExpr(
              variableDeclarationExpr.getModifiers(), new NodeList<>(inlineVariable));
      final ExpressionStmt expressionStmt = new ExpressionStmt(newVariableDeclarationExpr);
      nodesToAdd.add(expressionStmt);
    }
    return nodesToAdd;
  }

  /** Adds a list of nodes to the parent BlockStmt after the expressionStmt. */
  protected boolean addNewNodesToParentNode(List nodesToAdd) {
    final Optional expressionStmtOptional = variableDeclarationExpr.getParentNode();
    if (expressionStmtOptional.isPresent()
        && expressionStmtOptional.get() instanceof ExpressionStmt expressionStmt) {
      final Optional blockStmtOptional = expressionStmt.getParentNode();
      if (blockStmtOptional.isPresent() && blockStmtOptional.get() instanceof BlockStmt blockStmt) {

        final List allStmts =
            insertNodesAfterReference(blockStmt.getStatements(), expressionStmt, nodesToAdd);

        blockStmt.setStatements(new NodeList<>(allStmts));

        return true;
      }
    }

    return false;
  }
}