// Copyright 2016 The Bazel Authors. All rights reserved.
//
// 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.google.devtools.build.lib.rules.proto;

import static com.google.devtools.build.lib.analysis.configuredtargets.RuleConfiguredTarget.Mode.HOST;
import static com.google.devtools.build.lib.analysis.configuredtargets.RuleConfiguredTarget.Mode.TARGET;
import static com.google.devtools.build.lib.collect.nestedset.Order.STABLE_ORDER;

import com.google.devtools.build.lib.actions.Artifact;
import com.google.devtools.build.lib.actions.MutableActionGraph.ActionConflictException;
import com.google.devtools.build.lib.analysis.ConfiguredTarget;
import com.google.devtools.build.lib.analysis.FileProvider;
import com.google.devtools.build.lib.analysis.FilesToRunProvider;
import com.google.devtools.build.lib.analysis.RuleConfiguredTargetBuilder;
import com.google.devtools.build.lib.analysis.RuleConfiguredTargetFactory;
import com.google.devtools.build.lib.analysis.RuleContext;
import com.google.devtools.build.lib.analysis.Runfiles;
import com.google.devtools.build.lib.analysis.RunfilesProvider;
import com.google.devtools.build.lib.collect.nestedset.NestedSetBuilder;
import com.google.devtools.build.lib.syntax.Type;

/** Implements {code proto_lang_toolchain}. */
public class ProtoLangToolchain implements RuleConfiguredTargetFactory {
  @Override
  public ConfiguredTarget create(RuleContext ruleContext)
      throws InterruptedException, RuleErrorException, ActionConflictException {
    NestedSetBuilder<Artifact> blacklistedProtos = NestedSetBuilder.stableOrder();
    for (FileProvider protos :
        ruleContext.getPrerequisites("blacklisted_protos", TARGET, FileProvider.class)) {
      blacklistedProtos.addTransitive(protos.getFilesToBuild());
    }

    return new RuleConfiguredTargetBuilder(ruleContext)
        .addProvider(
            ProtoLangToolchainProvider.create(
                ruleContext.attributes().get("command_line", Type.STRING),
                ruleContext.getPrerequisite("plugin", HOST, FilesToRunProvider.class),
                ruleContext.getPrerequisite("runtime", TARGET),
                blacklistedProtos.build()))
        .setFilesToBuild(NestedSetBuilder.<Artifact>emptySet(STABLE_ORDER))
        .addProvider(RunfilesProvider.simple(Runfiles.EMPTY))
        .build();
  }
}

@@ -1607,32 +1607,104 @@
 nfilesProvider;%0A
+import com.google.devtools.build.lib.analysis.TransitiveInfoCollection;%0A
 import com.googl
@@ -1752,32 +1752,32 @@
 stedSetBuilder;%0A
-
 import com.googl
@@ -2174,28 +2174,40 @@
    for (
-FileProvider
+TransitiveInfoCollection
  protos 
@@ -2277,28 +2277,8 @@
 RGET
-, FileProvider.class
 )) %7B
@@ -2330,24 +2330,322 @@
 .get
-FilesToBuild());
+Provider(FileProvider.class).getFilesToBuild());%0A      ProtoInfo protoInfo = protos.get(ProtoInfo.PROVIDER);%0A      // TODO(cushon): it would be nice to make this mandatory and stop adding files to build too%0A      if (protoInfo != null) %7B%0A        blacklistedProtos.addAll(protoInfo.getDirectProtoSources());%0A      %7D
 %0A   

Support proto_library in proto_lang_toolchain.blacklisted_protos in addition to output files.