/*
 * Copyright (c) Facebook, Inc. and its affiliates.
 *
 * 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.coercer;

import com.facebook.buck.core.cell.CellPathResolver;
import com.facebook.buck.core.model.BuildTarget;
import com.facebook.buck.core.model.InternalFlavor;
import com.facebook.buck.core.model.TargetConfiguration;
import com.facebook.buck.core.path.ForwardRelativePath;
import com.facebook.buck.io.filesystem.ProjectFilesystem;
import com.facebook.buck.rules.macros.LocationPlatformMacro;
import com.facebook.buck.util.types.Pair;
import com.google.common.collect.ImmutableList;
import com.google.common.collect.ImmutableSet;
import java.util.Optional;

/** Coerces `$(location-platform ...)` macros into {@link LocationPlatformMacro}. */
class LocationPlatformMacroTypeCoercer
    extends AbstractLocationMacroTypeCoercer<LocationPlatformMacro> {

  public LocationPlatformMacroTypeCoercer(TypeCoercer<BuildTarget> buildTargetTypeCoercer) {
    super(buildTargetTypeCoercer);
  }

  @Override
  public Class<LocationPlatformMacro> getOutputClass() {
    return LocationPlatformMacro.class;
  }

  @Override
  public LocationPlatformMacro coerce(
      CellPathResolver cellRoots,
      ProjectFilesystem filesystem,
      ForwardRelativePath pathRelativeToProjectRoot,
      TargetConfiguration targetConfiguration,
      TargetConfiguration hostConfiguration,
      ImmutableList<String> args)
      throws CoerceFailedException {
    if (args.size() < 1) {
      throw new CoerceFailedException(
          String.format("expected at least one argument (found %d)", args.size()));
    }
    Pair<BuildTarget, Optional<String>> target =
        coerceTarget(
            cellRoots,
            filesystem,
            pathRelativeToProjectRoot,
            targetConfiguration,
            hostConfiguration,
            args.get(0));
    return LocationPlatformMacro.of(
        target.getFirst(),
        target.getSecond(),
        args.subList(1, args.size()).stream()
            .map(InternalFlavor::of)
            .collect(ImmutableSet.toImmutableSet()));
  }
}
