src/com/facebook/buck/ocaml/OCamlUtil.java

/*
 * 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.ocaml;

import com.facebook.buck.graph.DirectedAcyclicGraph;
import com.facebook.buck.graph.TopologicalSort;
import com.facebook.buck.rules.BuildRuleDependencyVisitors;
import com.facebook.buck.rules.BuildRule;
import com.facebook.buck.rules.SourcePath;
import com.facebook.buck.rules.SourcePathResolver;
import com.google.common.base.Predicate;
import com.google.common.base.Predicates;
import com.google.common.collect.FluentIterable;
import com.google.common.collect.ImmutableList;
import com.google.common.collect.ImmutableSet;

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

/**
 * Utility functions
 */
public class OCamlUtil {
  private OCamlUtil() {
  }

  /**
   * Constructs a Predicate instance which returns true if the input argument ends with
   * any String in extensions
   *
   * @param extensions for which to return true
   * @return a Predicate instance
   */
  public static Predicate<? super Path> ext(final String... extensions) {
    return new Predicate<Path>() {
      @Override
      public boolean apply(Path input) {
        String strInput = input.toString();
        for (String ext : extensions) {
          if (strInput.endsWith(ext)) {
            return true;
          }
        }
        return false;
      }
    };
  }

  public static Predicate<? super SourcePath> sourcePathExt(
      final SourcePathResolver resolver,
      final String... extensions) {
    return new Predicate<SourcePath>() {
      @Override
      public boolean apply(SourcePath input) {
        String strInput = resolver.getPath(input).toString();
        for (String ext : extensions) {
          if (strInput.endsWith(ext)) {
            return true;
          }
        }
        return false;
      }
    };
  }

  public static ImmutableList<OCamlLibrary> getTransitiveOCamlInput(
      Iterable<? extends BuildRule> inputs) {

    final DirectedAcyclicGraph<BuildRule> graph =
        BuildRuleDependencyVisitors.getBuildRuleDirectedGraphFilteredBy(
            inputs,
            Predicates.instanceOf(OCamlLibrary.class),
            Predicates.instanceOf(OCamlLibrary.class));

    final ImmutableList<BuildRule> sorted = TopologicalSort.sort(
        graph,
        Predicates.<BuildRule>alwaysTrue());

    return FluentIterable
            .from(sorted)
            .filter(OCamlLibrary.class)
            .toList();
  }


  static ImmutableSet<Path> getExtensionVariants(
      Path output,
      String...extensions) {
    String withoutExtension = stripExtension(output.toString());
    ImmutableSet.Builder<Path> builder =  ImmutableSet.builder();
    for (String ext : extensions) {
      builder.add(Paths.get(withoutExtension + ext));
    }
    return builder.build();
  }

  static String stripExtension(String fileName) {
    int index = fileName.lastIndexOf('.');

    // if dot is in the first position,
    // we are dealing with a hidden file rather than an extension
    return (index > 0) ? fileName.substring(0, index) : fileName;
  }
}