com.facebook.buck.core.model.targetgraph.impl.TargetGraphAndTargets.java Source code

Introduction

Here is the source code for com.facebook.buck.core.model.targetgraph.impl.TargetGraphAndTargets.java

Source

/*
 * Copyright 2018-present Facebook, Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); you may
 * not use this file except in compliance with the License. You may obtain
 * a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 * License for the specific language governing permissions and limitations
 * under the License.
 */

package com.facebook.buck.core.model.targetgraph.impl;

import com.facebook.buck.core.model.BuildTarget;
import com.facebook.buck.core.model.targetgraph.TargetGraph;
import com.facebook.buck.core.model.targetgraph.TargetNode;
import com.facebook.buck.util.RichStream;
import com.google.common.collect.ImmutableSet;
import com.google.common.collect.Iterables;
import java.util.Iterator;

public class TargetGraphAndTargets {
    private final TargetGraph targetGraph;
    private final ImmutableSet<TargetNode<?>> projectRoots;

    public TargetGraphAndTargets(TargetGraph targetGraph, Iterable<TargetNode<?>> projectRoots) {
        this.targetGraph = targetGraph;
        this.projectRoots = ImmutableSet.copyOf(projectRoots);
    }

    public TargetGraph getTargetGraph() {
        return targetGraph;
    }

    public ImmutableSet<TargetNode<?>> getProjectRoots() {
        return projectRoots;
    }

    /**
     * @param nodes Nodes whose test targets we would like to find
     * @return A set of all test targets that test the targets in {@code nodes}.
     */
    public static ImmutableSet<BuildTarget> getExplicitTestTargets(Iterator<TargetNode<?>> nodes) {
        return RichStream.from(nodes).flatMap(node -> TargetNodes.getTestTargetsForNode(node).stream())
                .toImmutableSet();
    }

    public static TargetGraphAndTargets create(ImmutableSet<BuildTarget> graphRoots, TargetGraph projectGraph,
            boolean isWithTests, ImmutableSet<BuildTarget> explicitTests) {
        // Get the roots of the main graph. This contains all the targets in the project slice, or all
        // the valid project roots if a project slice is not specified.
        Iterable<TargetNode<?>> projectRoots = projectGraph.getAll(graphRoots);

        // Optionally get the roots of the test graph. This contains all the tests that cover the roots
        // of the main graph or their dependencies.
        Iterable<TargetNode<?>> associatedTests = ImmutableSet.of();
        if (isWithTests) {
            associatedTests = projectGraph.getAll(explicitTests);
        }

        TargetGraph targetGraph = projectGraph.getSubgraph(Iterables.concat(projectRoots, associatedTests));

        return new TargetGraphAndTargets(targetGraph, projectRoots);
    }
}