blaze-base/src/com/google/idea/blaze/base/model/primitives/Label.java

/*
 * Copyright 2016 The Bazel Authors. All rights reserved.
 *
 * 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.google.idea.blaze.base.model.primitives;

import com.google.common.collect.Lists;
import com.google.idea.blaze.base.ui.BlazeValidationError;
import com.intellij.openapi.diagnostic.Logger;

import javax.annotation.Nullable;
import javax.annotation.concurrent.Immutable;
import java.util.Collection;
import java.util.Comparator;
import java.util.List;

/**
 * Wrapper around a string for a blaze label (//package:rule).
 */
@Immutable
public final class Label extends TargetExpression {
  private static final Logger LOG = Logger.getInstance(Label.class);

  public static final Comparator<Label> COMPARATOR = (o1, o2) -> String.CASE_INSENSITIVE_ORDER.compare(o1.toString(), o2.toString());

  public static final long serialVersionUID = 2L;

  /**
   * Silently returns null if this is not a valid Label
   */
  @Nullable
  public static Label createIfValid(String label) {
    if (validate(label)) {
      return new Label(label);
    }
    return null;
  }

  public Label(String label) {
    super(label);
    List<BlazeValidationError> errors = Lists.newArrayList();
    if (!validate(label, errors)) {
      BlazeValidationError.throwError(errors);
    }
  }

  public Label(
    WorkspacePath packageName,
    RuleName newRuleName) {
    this("//" + packageName.toString() + ":" + newRuleName.toString());
  }

  public static boolean validate(String label) {
    return validate(label, null);
  }

  public static boolean validate(String label, @Nullable Collection<BlazeValidationError> errors) {
    int colonIndex = label.indexOf(':');
    if (label.startsWith("//") && colonIndex >= 0) {
      String packageName = label.substring("//".length(), colonIndex);
      if (!validatePackagePath(packageName, errors)) {
        return false;
      }
      String ruleName = label.substring(colonIndex + 1);
      if (!RuleName.validate(ruleName, errors)) {
        return false;
      }
      return true;
    }
    if (label.startsWith("@") && colonIndex >= 0) {
      // a bazel-specific label pointing to a different repository
      int slashIndex = label.indexOf("//");
      if (slashIndex >= 0) {
        return validate(label.substring(slashIndex), errors);
      }
    }
    if (errors != null) {
      errors.add(new BlazeValidationError("Not a valid label, no target name found: " + label));
    }
    return false;
  }

  /**
   * Extract the rule name from a label. The rule name follows a colon at the end of the label.
   *
   * @return the rule name
   */
  public RuleName ruleName() {
    String labelStr = toString();
    int colonLocation = labelStr.lastIndexOf(':');
    int ruleNameStart = colonLocation + 1;
    String ruleNameStr = labelStr.substring(ruleNameStart);
    return RuleName.create(ruleNameStr);
  }

  /**
   * Return the workspace path for the package label for the given label. For example, if the
   * package is //j/c/g/a/apps/docs:release, it returns j/c/g/a/apps/docs.
   */
  public WorkspacePath blazePackage() {
    String labelStr = toString();
    int startIndex = labelStr.indexOf("//") + "//".length();
    int colonIndex = labelStr.lastIndexOf(':');
    LOG.assertTrue(colonIndex >= 0);
    return new WorkspacePath(labelStr.substring(startIndex, colonIndex));
  }

  public static boolean validatePackagePath(String path) {
    return validatePackagePath(path, null);
  }

  public static boolean validatePackagePath(String path, @Nullable Collection<BlazeValidationError> errors) {
    // Empty packages are legal but not recommended
    if (path.isEmpty()) {
      return true;
    }

    if (path.charAt(0) < 'a' || path.charAt(0) > 'z') {
      BlazeValidationError.collect(errors, new BlazeValidationError(
        "Invalid package name: " + path + "\n "
        + "Package names must start with a lowercase ASCII letter"
      ));
      return false;
    }
    if (path.contains("//")) {
      BlazeValidationError.collect(errors, new BlazeValidationError(
        "Invalid package name: " + path + "\n "
        + "package names may not contain \"//\" path separators."
      ));
      return false;
    }
    if (path.endsWith("/")) {
      BlazeValidationError.collect(errors, new BlazeValidationError(
        "Invalid package name: " + path + "\n "
        + "package names may not end with \"/\""
      ));
      return false;
    }
    return true;
  }

}