Class BindingGraphConverter

java/dagger/internal/codegen/BindingGraphConverter.java:46–166

Converts dagger.internal.codegen.BindingGraphs to dagger.model.BindingGraphs.

/** Converts {@link dagger.internal.codegen.BindingGraph}s to {@link dagger.model.BindingGraph}s. */
final class BindingGraphConverter extends ComponentTreeTraverser {

  private final MutableNetwork<Node, Edge> network =
      NetworkBuilder.directed().allowsParallelEdges(true).allowsSelfLoops(true).build();

  private ComponentNode parentComponent;
  private ComponentNode currentComponent;

  private BindingGraphConverter(BindingGraph graph) {
    super(graph);
  }

  /**
   * Creates the external {@link dagger.model.BindingGraph} representing the given internal root
   * {@link dagger.internal.codegen.BindingGraph}.
   */
  static dagger.model.BindingGraph convert(BindingGraph graph) {
    BindingGraphConverter converter = new BindingGraphConverter(graph);
    converter.traverseComponents();
    return BindingGraphProxies.bindingGraph(converter.network);
  }

  @Override
  protected void visitComponent(BindingGraph graph) {
    ComponentNode grandparentNode = parentComponent;
    parentComponent = currentComponent;
    currentComponent = componentNode(componentTreePath().toComponentPath());
    network.addNode(currentComponent);
    super.visitComponent(graph);
    currentComponent = parentComponent;
    parentComponent = grandparentNode;
  }

  @Override
  protected void visitSubcomponentFactoryMethod(
      BindingGraph graph, BindingGraph parent, ExecutableElement factoryMethod) {
    network.addEdge(parentComponent, currentComponent, childFactoryMethodEdge(factoryMethod));
    super.visitSubcomponentFactoryMethod(graph, parent, factoryMethod);
  }

  @Override
  protected BindingGraphTraverser bindingGraphTraverser(
      ComponentTreePath componentTreePath, ComponentMethodDescriptor entryPointMethod) {
    return new BindingGraphVisitor(componentTreePath, entryPointMethod);
  }

  private final class BindingGraphVisitor extends BindingGraphTraverser {

    private Node current;

    BindingGraphVisitor(
        ComponentTreePath componentTreePath, ComponentMethodDescriptor entryPointMethod) {
      super(componentTreePath, entryPointMethod);
      current = currentComponent;
      network.addNode(current);
    }

    @Override