com.facebook.buck.apple.CoreDataModel.java Source code

Introduction

Here is the source code for com.facebook.buck.apple.CoreDataModel.java

Source

/*
 * Copyright 2016-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.apple;

import com.facebook.buck.model.BuildTargets;
import com.facebook.buck.model.Flavor;
import com.facebook.buck.model.ImmutableFlavor;
import com.facebook.buck.rules.AbstractBuildRule;
import com.facebook.buck.rules.AddToRuleKey;
import com.facebook.buck.rules.BuildContext;
import com.facebook.buck.rules.BuildRuleParams;
import com.facebook.buck.rules.BuildableContext;
import com.facebook.buck.rules.SourcePath;
import com.facebook.buck.rules.Tool;
import com.facebook.buck.shell.ShellStep;
import com.facebook.buck.step.Step;
import com.facebook.buck.step.ExecutionContext;
import com.facebook.buck.step.fs.MakeCleanDirectoryStep;
import com.google.common.collect.ImmutableList;
import com.google.common.collect.ImmutableMap;
import com.google.common.collect.ImmutableSet;

import java.nio.file.Path;
import java.nio.file.Paths;

import javax.annotation.Nullable;

public class CoreDataModel extends AbstractBuildRule {

    public static final Flavor FLAVOR = ImmutableFlavor.of("core-data-model");

    @AddToRuleKey
    private final String moduleName;

    @AddToRuleKey
    private final Tool momc;

    @AddToRuleKey
    private final ImmutableSet<SourcePath> dataModelPaths;

    @AddToRuleKey
    private final String sdkName;

    @AddToRuleKey
    private final String minOSVersion;

    private final Path sdkRoot;
    private final Path outputDir;

    CoreDataModel(BuildRuleParams params, AppleCxxPlatform appleCxxPlatform, String moduleName,
            ImmutableSet<SourcePath> dataModelPaths) {
        super(params);
        this.moduleName = moduleName;
        this.dataModelPaths = dataModelPaths;
        String outputDirString = BuildTargets.getGenPath(getProjectFilesystem(), params.getBuildTarget(), "%s")
                .toString().replace('#', '-'); // momc doesn't like # in paths
        this.outputDir = Paths.get(outputDirString);
        this.sdkName = appleCxxPlatform.getAppleSdk().getName();
        this.sdkRoot = appleCxxPlatform.getAppleSdkPaths().getSdkPath();
        this.minOSVersion = appleCxxPlatform.getMinVersion();
        this.momc = appleCxxPlatform.getMomc();
    }

    @Override
    public ImmutableList<Step> getBuildSteps(BuildContext context, BuildableContext buildableContext) {
        ImmutableList.Builder<Step> stepsBuilder = ImmutableList.builder();
        stepsBuilder.add(new MakeCleanDirectoryStep(getProjectFilesystem(), outputDir));
        for (SourcePath dataModelPath : dataModelPaths) {
            stepsBuilder.add(new ShellStep(getProjectFilesystem().getRootPath()) {
                @Override
                protected ImmutableList<String> getShellCommandInternal(ExecutionContext executionContext) {
                    ImmutableList.Builder<String> commandBuilder = ImmutableList.builder();

                    commandBuilder.addAll(momc.getCommandPrefix(context.getSourcePathResolver()));
                    commandBuilder.add("--sdkroot", sdkRoot.toString(), "--" + sdkName + "-deployment-target",
                            minOSVersion, "--module", moduleName,
                            context.getSourcePathResolver().getAbsolutePath(dataModelPath).toString(),
                            getProjectFilesystem().resolve(outputDir).toString());

                    return commandBuilder.build();
                }

                @Override
                public ImmutableMap<String, String> getEnvironmentVariables(ExecutionContext context) {
                    return momc.getEnvironment();
                }

                @Override
                public String getShortName() {
                    return "momc";
                }
            });
        }
        buildableContext.recordArtifact(getOutputDir());
        return stepsBuilder.build();
    }

    @Nullable
    @Override
    public Path getPathToOutput() {
        return outputDir;
    }

    public Path getOutputDir() {
        return outputDir;
    }
}