| /* |
| * 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; |
| } |
| } |