Function ExpandAllMacro2

extensions/comprehensions_v2_macros.cc:41–79

using ::google::api::expr::common::CelOperator;

absl::optional<Expr> ExpandAllMacro2(MacroExprFactory& factory, Expr& target,
                                     absl::Span<Expr> args) {
  if (args.size() != 3) {
    return factory.ReportError("all() requires 3 arguments");
  }
  if (!args[0].has_ident_expr() || args[0].ident_expr().name().empty()) {
    return factory.ReportErrorAt(
        args[0], "all() first variable name must be a simple identifier");
  }
  if (!args[1].has_ident_expr() || args[1].ident_expr().name().empty()) {
    return factory.ReportErrorAt(
        args[1], "all() second variable name must be a simple identifier");
  }
  if (args[0].ident_expr().name() == args[1].ident_expr().name()) {
    return factory.ReportErrorAt(
        args[0],
        "all() second variable must be different from the first variable");
  }
  if (args[0].ident_expr().name() == kAccumulatorVariableName) {
    return factory.ReportErrorAt(
        args[0], absl::StrCat("all() first variable name cannot be ",
                              kAccumulatorVariableName));
  }
  if (args[1].ident_expr().name() == kAccumulatorVariableName) {
    return factory.ReportErrorAt(
        args[1], absl::StrCat("all() second variable name cannot be ",
                              kAccumulatorVariableName));
  }
  auto init = factory.NewBoolConst(true);
  auto condition =
      factory.NewCall(CelOperator::NOT_STRICTLY_FALSE, factory.NewAccuIdent());
  auto step = factory.NewCall(CelOperator::LOGICAL_AND, factory.NewAccuIdent(),
                              std::move(args[2]));
  auto result = factory.NewAccuIdent();
  return factory.NewComprehension(
      args[0].ident_expr().name(), args[1].ident_expr().name(),
      std::move(target), factory.AccuVarName(), std::move(init),
      std::move(condition), std::move(step), std::move(result));
}

Macro MakeAllMacro2() {
  auto status_or_macro = Macro::Receiver(CelOperator::ALL, 3, ExpandAllMacro2);