src/main/java/com/google/devtools/build/lib/starlarkbuildapi/StarlarkAspectPropagationContextApi.java

// Copyright 2025 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.devtools.build.lib.starlarkbuildapi;

import com.google.devtools.build.docgen.annot.DocCategory;
import com.google.devtools.build.lib.cmdline.Label;
import com.google.devtools.build.lib.starlarkbuildapi.core.StructApi;
import javax.annotation.Nullable;
import net.starlark.java.annot.StarlarkBuiltin;
import net.starlark.java.annot.StarlarkMethod;
import net.starlark.java.eval.EvalException;
import net.starlark.java.eval.StarlarkValue;

/** Interface for the context object given to aspect propagation functions. */
@StarlarkBuiltin(
    name = "propagation_ctx",
    category = DocCategory.BUILTIN,
    doc =
        "A context object that is passed to the <code>propagation_predicate</code>,"
            + " <code>attr_aspects</code> and <code>toolchains_aspects</code> functions of"
            + " aspects. It provides access to the information needed to determine whether the"
            + " aspect should be propagated to the target and what attributes or toolchain types it"
            + " should be propagated to next.")
public interface StarlarkAspectPropagationContextApi extends StarlarkValue {

  @StarlarkMethod(
      name = "attr",
      structField = true,
      doc =
          "A struct to access only the public parameters of the aspect. The keys and values of the"
              + " struct are the parameters names and values.")
  StructApi getAttr() throws EvalException;

  @StarlarkMethod(
      name = "rule",
      structField = true,
      doc = "Allows access to the details of the rule.")
  StarlarkAspectPropagationRuleApi getRule() throws EvalException;

  /** Interface for the rule details provided to the aspect propagation functions. */
  @StarlarkBuiltin(name = "starlark_aspect_propagation_rule", documented = false)
  static interface StarlarkAspectPropagationRuleApi extends StarlarkValue {
    @StarlarkMethod(name = "label", structField = true, doc = "The label of the target.")
    Label getLabel() throws EvalException;

    @StarlarkMethod(
        name = "attr",
        structField = true,
        doc =
            "A struct to access the attributes of the target. Attribute names are the keys, and"
                + " each value is a struct with <code>value</code> (the attribute's value) and"
                + " <code>is_tool</code> (a boolean indicating if the attribute is a tool)."
                + " Dependency attributes are represented by labels because they"
                + " are not analyzed at this stage.")
    StructApi getAttr() throws EvalException;

    @StarlarkMethod(
        name = "qualified_kind",
        structField = true,
        doc =
            "The rule kind of the target broken down into 2 fields; <code>file_label</code>: the"
                + " label of the file containing the rule definition and <code>rule_name</code>:"
                + " the name of the rule.")
    QualifiedRuleKindApi getQualifiedKind() throws EvalException;
  }

  /** Interface for the qualified rule kind of the target. */
  @StarlarkBuiltin(name = "qualified_rule_kind", documented = false)
  static interface QualifiedRuleKindApi extends StarlarkValue {
    @StarlarkMethod(
        name = "file_label",
        structField = true,
        allowReturnNones = true, // for native rules
        doc = "The label of the file containing the rule definition.")
    @Nullable
    Label getFileLabel() throws EvalException;

    @StarlarkMethod(name = "rule_name", structField = true, doc = "The name of the rule.")
    String getRuleName() throws EvalException;
  }

  /**
   * Interface for the target's attribute. It can be extended to include more metadata about the
   * attribute like its annotations.
   */
  @StarlarkBuiltin(name = "rule_attribute", documented = false)
  static interface RuleAttributeApi extends StarlarkValue {
    @StarlarkMethod(name = "value", structField = true, doc = "The value of the attribute.")
    Object getValue() throws EvalException;

    @StarlarkMethod(name = "is_tool", structField = true, doc = "Whether the attribute is a tool.")
    boolean isTool() throws EvalException;
  }
}