| // 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.java.proto; |
| |
| import static com.google.devtools.build.lib.collect.nestedset.Order.STABLE_ORDER; |
| import static com.google.devtools.build.lib.rules.java.proto.JplCcLinkParams.createCcLinkingInfo; |
| import static com.google.devtools.build.lib.rules.java.proto.StrictDepsUtils.constructJcapFromAspectDeps; |
| |
| import com.google.common.collect.ImmutableList; |
| import com.google.common.collect.ImmutableMultimap; |
| 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.OutputGroupInfo; |
| 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.analysis.TransitiveInfoCollection; |
| import com.google.devtools.build.lib.analysis.configuredtargets.RuleConfiguredTarget.Mode; |
| import com.google.devtools.build.lib.collect.nestedset.NestedSet; |
| import com.google.devtools.build.lib.collect.nestedset.NestedSetBuilder; |
| import com.google.devtools.build.lib.rules.java.JavaCommon; |
| import com.google.devtools.build.lib.rules.java.JavaCompilationArgsProvider; |
| import com.google.devtools.build.lib.rules.java.JavaConfiguration; |
| import com.google.devtools.build.lib.rules.java.JavaInfo; |
| import com.google.devtools.build.lib.rules.java.JavaRuleOutputJarsProvider; |
| import com.google.devtools.build.lib.rules.java.JavaRunfilesProvider; |
| import com.google.devtools.build.lib.rules.java.JavaSkylarkApiProvider; |
| import com.google.devtools.build.lib.rules.java.JavaSourceJarsProvider; |
| import com.google.devtools.build.lib.rules.java.JavaStrictCompilationArgsProvider; |
| import com.google.devtools.build.lib.rules.java.ProguardLibrary; |
| import com.google.devtools.build.lib.rules.java.ProguardSpecProvider; |
| |
| /** Implementation of the java_lite_proto_library rule. */ |
| public class JavaLiteProtoLibrary implements RuleConfiguredTargetFactory { |
| |
| @Override |
| public ConfiguredTarget create(final RuleContext ruleContext) |
| throws InterruptedException, RuleErrorException, ActionConflictException { |
| JavaCommon.checkRuleLoadedThroughMacro(ruleContext); |
| |
| Iterable<JavaProtoLibraryAspectProvider> javaProtoLibraryAspectProviders = |
| ruleContext.getPrerequisites("deps", Mode.TARGET, JavaProtoLibraryAspectProvider.class); |
| |
| JavaCompilationArgsProvider dependencyArgsProviders = |
| constructJcapFromAspectDeps(ruleContext, javaProtoLibraryAspectProviders); |
| |
| // We assume that the runtime jars will not have conflicting artifacts |
| // with the same root relative path |
| Runfiles runfiles = |
| new Runfiles.Builder(ruleContext.getWorkspaceName()) |
| .addTransitiveArtifactsWrappedInStableOrder(dependencyArgsProviders.getRuntimeJars()) |
| .build(); |
| |
| JavaSourceJarsProvider sourceJarsProvider = |
| JavaSourceJarsProvider.merge( |
| ruleContext.getPrerequisites("deps", Mode.TARGET, JavaSourceJarsProvider.class)); |
| |
| NestedSetBuilder<Artifact> filesToBuild = NestedSetBuilder.stableOrder(); |
| |
| filesToBuild.addAll(sourceJarsProvider.getSourceJars()); |
| |
| for (JavaProtoLibraryAspectProvider provider : javaProtoLibraryAspectProviders) { |
| filesToBuild.addTransitive(provider.getJars()); |
| } |
| |
| JavaRunfilesProvider javaRunfilesProvider = new JavaRunfilesProvider(runfiles); |
| |
| JavaInfo.Builder javaInfoBuilder = |
| JavaInfo.Builder.create() |
| .addProvider(JavaCompilationArgsProvider.class, dependencyArgsProviders); |
| if (ruleContext.getFragment(JavaConfiguration.class).isJlplStrictDepsEnforced() |
| && !ruleContext.getDisabledFeatures().contains("jlpl_strict_deps")) { |
| JavaStrictCompilationArgsProvider strictDependencyArgsProviders = |
| new JavaStrictCompilationArgsProvider( |
| constructJcapFromAspectDeps( |
| ruleContext, javaProtoLibraryAspectProviders, /* alwaysStrict= */ true)); |
| javaInfoBuilder.addProvider( |
| JavaStrictCompilationArgsProvider.class, strictDependencyArgsProviders); |
| } |
| JavaInfo javaInfo = |
| javaInfoBuilder |
| .addProvider(JavaSourceJarsProvider.class, sourceJarsProvider) |
| .addProvider(JavaRuleOutputJarsProvider.class, JavaRuleOutputJarsProvider.EMPTY) |
| .addProvider(JavaRunfilesProvider.class, javaRunfilesProvider) |
| .setJavaConstraints(ImmutableList.of("android")) |
| .build(); |
| |
| return new RuleConfiguredTargetBuilder(ruleContext) |
| .setFilesToBuild(filesToBuild.build()) |
| .addSkylarkTransitiveInfo( |
| JavaSkylarkApiProvider.NAME, JavaSkylarkApiProvider.fromRuleContext()) |
| .addProvider(RunfilesProvider.withData(Runfiles.EMPTY, runfiles)) |
| .addOutputGroup(OutputGroupInfo.DEFAULT, NestedSetBuilder.<Artifact>emptySet(STABLE_ORDER)) |
| .addNativeDeclaredProvider(getJavaLiteRuntimeSpec(ruleContext)) |
| .addNativeDeclaredProvider(javaInfo) |
| .addNativeDeclaredProvider(createCcLinkingInfo(ruleContext, ImmutableList.of())) |
| .build(); |
| } |
| |
| private ProguardSpecProvider getJavaLiteRuntimeSpec(RuleContext ruleContext) { |
| NestedSet<Artifact> specs = |
| new ProguardLibrary(ruleContext).collectProguardSpecs(ImmutableMultimap.<Mode, String>of()); |
| |
| TransitiveInfoCollection runtime = |
| JavaProtoAspectCommon.getLiteProtoToolchainProvider(ruleContext).runtime(); |
| if (runtime == null) { |
| return new ProguardSpecProvider(specs); |
| } |
| |
| ProguardSpecProvider specProvider = runtime.get(ProguardSpecProvider.PROVIDER); |
| if (specProvider == null) { |
| return new ProguardSpecProvider(specs); |
| } |
| |
| return new ProguardSpecProvider( |
| NestedSetBuilder.fromNestedSet(specs) |
| .addTransitive(specProvider.getTransitiveProguardSpecs()) |
| .build()); |
| } |
| } |