dataflow/src/main/java/org/checkerframework/dataflow/cfg/node/ObjectCreationNode.java

package org.checkerframework.dataflow.cfg.node;

import com.sun.source.tree.NewClassTree;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.Objects;
import javax.lang.model.element.ExecutableElement;
import org.checkerframework.checker.nullness.qual.Nullable;
import org.checkerframework.javacutil.TreeUtils;
import org.plumelib.util.StringsPlume;

/**
 * A node for new object creation.
 *
 * <pre>
 *   <em>new constructor(arg1, arg2, ...)</em>
 * </pre>
 */
public class ObjectCreationNode extends Node {

  protected final NewClassTree tree;
  protected final Node constructor;
  protected final List<Node> arguments;

  // Class body for anonymous classes, otherwise null.
  protected final @Nullable ClassDeclarationNode classbody;

  public ObjectCreationNode(
      NewClassTree tree,
      Node constructor,
      List<Node> arguments,
      @Nullable ClassDeclarationNode classbody) {
    super(TreeUtils.typeOf(tree));
    this.tree = tree;
    this.constructor = constructor;
    this.arguments = arguments;
    this.classbody = classbody;

    // set assignment contexts for parameters
    int i = 0;
    ExecutableElement elem = TreeUtils.elementFromUse(tree);
    if (elem != null) {
      for (Node arg : arguments) {
        AssignmentContext ctx = new AssignmentContext.MethodParameterContext(elem, i++);
        arg.setAssignmentContext(ctx);
      }
    }
  }

  public Node getConstructor() {
    return constructor;
  }

  public List<Node> getArguments() {
    return arguments;
  }

  public Node getArgument(int i) {
    return arguments.get(i);
  }

  public @Nullable Node getClassBody() {
    return classbody;
  }

  @Override
  public NewClassTree getTree() {
    return tree;
  }

  @Override
  public <R, P> R accept(NodeVisitor<R, P> visitor, P p) {
    return visitor.visitObjectCreation(this, p);
  }

  @Override
  public String toString() {
    StringBuilder sb = new StringBuilder();
    sb.append("new " + constructor + "(");
    sb.append(StringsPlume.join(", ", arguments));
    sb.append(")");
    if (classbody != null) {
      // TODO: maybe this can be done nicer...
      sb.append(" ");
      sb.append(classbody.toString());
    }
    return sb.toString();
  }

  @Override
  public boolean equals(@Nullable Object obj) {
    if (!(obj instanceof ObjectCreationNode)) {
      return false;
    }
    ObjectCreationNode other = (ObjectCreationNode) obj;
    if (constructor == null && other.getConstructor() != null) {
      return false;
    }

    return getConstructor().equals(other.getConstructor())
        && getArguments().equals(other.getArguments());
  }

  @Override
  public int hashCode() {
    return Objects.hash(constructor, arguments);
  }

  @Override
  public Collection<Node> getOperands() {
    ArrayList<Node> list = new ArrayList<>(1 + arguments.size());
    list.add(constructor);
    list.addAll(arguments);
    return list;
  }
}