src/com/facebook/buck/rules/BuckPyFunction.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.rules;

import com.facebook.buck.util.HumanReadableException;
import com.google.common.base.Preconditions;
import com.google.common.collect.Iterables;
import com.google.common.collect.Sets;

import java.util.SortedSet;

/**
 * Used to generate a function for use within buck.py for the rule described by a
 * {@link Description}.
 */
class BuckPyFunction {

  private final ArgObjectPopulatomatic argsInspector;

  public BuckPyFunction(ArgObjectPopulatomatic argsInspector) {
    this.argsInspector = Preconditions.checkNotNull(argsInspector);
  }

  String toPythonFunction(BuildRuleType type, Object dto) {
    Preconditions.checkNotNull(type);
    Preconditions.checkNotNull(dto);

    StringBuilder builder = new StringBuilder();

    SortedSet<ParamInfo> mandatory = Sets.newTreeSet();
    SortedSet<ParamInfo> optional = Sets.newTreeSet();

    for (ParamInfo param : argsInspector.getAllParamInfo(dto)) {
      if (isSkippable(param)) {
        continue;
      }

      if (param.isOptional()) {
        optional.add(param);
      } else {
        mandatory.add(param);
      }
    }

    builder.append("@provide_for_build\n")
        .append("def ").append(type.getName()).append("(name, ");

    // Construct the args.
    for (ParamInfo param : Iterables.concat(mandatory, optional)) {
      appendPythonParameter(builder, param);
    }
    builder.append("visibility=[], build_env=None):\n")

        // Define the rule.
        .append("  add_rule({\n")
        .append("    'type' : '").append(type.getName()).append("',\n")
        .append("    'name' : name,\n");

    // Iterate over args.
    for (ParamInfo param : Iterables.concat(mandatory, optional)) {
      builder.append("    '")
          .append(param.getName())
          .append("' : ")
          .append(param.getName())
          .append(",\n");
    }

    builder.append("    'visibility' : visibility,\n");
    builder.append("  }, build_env)\n\n");

    return builder.toString();
  }

  private void appendPythonParameter(StringBuilder builder, ParamInfo param) {
    builder.append(param.getName());
    if (param.isOptional()) {
      builder.append("=").append(getPythonDefault(param));
    }
    builder.append(", ");
  }

  private String getPythonDefault(ParamInfo param) {
    if (param.getContainerType() != null) {
      return "[]";
    }

    if (Boolean.class.equals(param.getType())) {
      return "False";
    }

    if (Number.class.isAssignableFrom(param.getType())) {
      return "0";
    }

    if (String.class.equals(param.getType())) {
      return "''";
    }

    return "None";
  }

  private boolean isSkippable(ParamInfo param) {
    if ("name".equals(param.getName())) {
      if (!String.class.equals(param.getType())) {
        throw new HumanReadableException("'name' parameter must be a java.lang.String");
      }
      return true;
    }

    if ("visibility".equals(param.getName())) {
      throw new HumanReadableException(
          "'visibility' parameter must be omitted. It will be passed to the rule at run time.");
    }

    return false;
  }
}