/*
 * Copyright 2013-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.rules;

import com.facebook.buck.util.ProjectFilesystem;
import com.google.common.base.Function;
import com.google.common.base.Preconditions;

import java.nio.file.Path;

public class DefaultBuildRuleBuilderParams implements AbstractBuildRuleBuilderParams {

  private final Function<String, Path> pathRelativizer;
  private final RuleKeyBuilderFactory ruleKeyBuilderFactory;

  public DefaultBuildRuleBuilderParams(ProjectFilesystem projectFilesystem,
      RuleKeyBuilderFactory ruleKeyBuilderFactory) {
    this(Preconditions.checkNotNull(projectFilesystem).getPathRelativizer(),
        ruleKeyBuilderFactory);
  }

  public DefaultBuildRuleBuilderParams(Function<String, Path> pathRelativizer,
      RuleKeyBuilderFactory ruleKeyBuilderFactory) {
    this.pathRelativizer = Preconditions.checkNotNull(pathRelativizer);
    this.ruleKeyBuilderFactory = Preconditions.checkNotNull(ruleKeyBuilderFactory);
  }

  @Override
  public Function<String, Path> getPathRelativizer() {
    return pathRelativizer;
  }

  @Override
  public RuleKeyBuilderFactory getRuleKeyBuilderFactory() {
    return ruleKeyBuilderFactory;
  }

}
