com.facebook.buck.features.project.intellij.lang.java.JavaLibraryModuleRule.java Source code

Introduction

Here is the source code for com.facebook.buck.features.project.intellij.lang.java.JavaLibraryModuleRule.java

Source

/*
 * Copyright 2015-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.features.project.intellij.lang.java;

import com.facebook.buck.core.model.targetgraph.DescriptionWithTargetGraph;
import com.facebook.buck.core.model.targetgraph.TargetNode;
import com.facebook.buck.core.sourcepath.SourcePath;
import com.facebook.buck.features.project.intellij.BaseIjModuleRule;
import com.facebook.buck.features.project.intellij.ModuleBuildContext;
import com.facebook.buck.features.project.intellij.aggregation.AggregationContext;
import com.facebook.buck.features.project.intellij.model.IjModuleFactoryResolver;
import com.facebook.buck.features.project.intellij.model.IjModuleType;
import com.facebook.buck.features.project.intellij.model.IjProjectConfig;
import com.facebook.buck.features.project.intellij.model.folders.IjResourceFolderType;
import com.facebook.buck.io.filesystem.ProjectFilesystem;
import com.facebook.buck.jvm.core.JavaPackageFinder;
import com.facebook.buck.jvm.java.JavaLibraryDescription;
import com.google.common.collect.ImmutableMultimap;
import com.google.common.collect.ImmutableSet;
import com.google.common.collect.ImmutableSortedSet;
import java.nio.file.Path;
import java.util.Optional;

public class JavaLibraryModuleRule extends BaseIjModuleRule<JavaLibraryDescription.CoreArg> {

    private final JavaPackageFinder packageFinder;

    public JavaLibraryModuleRule(ProjectFilesystem projectFilesystem, IjModuleFactoryResolver moduleFactoryResolver,
            IjProjectConfig projectConfig, JavaPackageFinder packageFinder) {
        super(projectFilesystem, moduleFactoryResolver, projectConfig);
        this.packageFinder = packageFinder;
    }

    @Override
    public Class<? extends DescriptionWithTargetGraph<?>> getDescriptionClass() {
        return JavaLibraryDescription.class;
    }

    @Override
    public void apply(TargetNode<JavaLibraryDescription.CoreArg> target, ModuleBuildContext context) {
        Optional<Path> presetResourcesRoot = target.getConstructorArg().getResourcesRoot();
        ImmutableSortedSet<SourcePath> resources = target.getConstructorArg().getResources();
        ImmutableSet<Path> resourcePaths;
        if (presetResourcesRoot.isPresent()) {
            resourcePaths = getResourcePaths(resources, presetResourcesRoot.get());
            addResourceFolders(IjResourceFolderType.JAVA_RESOURCE, resourcePaths, presetResourcesRoot.get(),
                    context);
        } else {
            resourcePaths = getResourcePaths(resources);
            ImmutableMultimap<Path, Path> resourcesRootsToResources = getResourcesRootsToResources(packageFinder,
                    resourcePaths);
            for (Path resourcesRoot : resourcesRootsToResources.keySet()) {
                addResourceFolders(IjResourceFolderType.JAVA_RESOURCE, resourcesRootsToResources.get(resourcesRoot),
                        resourcesRoot, context);
            }
        }

        addDepsAndSources(target, true /* wantsPackagePrefix */, context, resourcePaths);
        JavaLibraryRuleHelper.addCompiledShadowIfNeeded(projectConfig, target, context);
        JavaLibraryRuleHelper.addNonSourceBuildTargets(target, context);
        context.setJavaLanguageLevel(JavaLibraryRuleHelper.getLanguageLevel(projectConfig, target));
        context.setCompilerOutputPath(moduleFactoryResolver.getCompilerOutputPath(target));
    }

    @Override
    public IjModuleType detectModuleType(TargetNode<JavaLibraryDescription.CoreArg> targetNode) {
        return IjModuleType.JAVA_MODULE;
    }

    @Override
    public void applyDuringAggregation(AggregationContext context,
            TargetNode<JavaLibraryDescription.CoreArg> targetNode) {
        super.applyDuringAggregation(context, targetNode);
        JavaLibraryRuleHelper.addLanguageAggregationKeyIfNeeded(projectConfig, targetNode, context);
    }
}