Function ExpandTransformMapEntry3Macro

extensions/comprehensions_v2_macros.cc:385–426

}

absl::optional<Expr> ExpandTransformMapEntry3Macro(MacroExprFactory& factory,
                                                   Expr& target,
                                                   absl::Span<Expr> args) {
  if (args.size() != 3) {
    return factory.ReportError("transformMapEntry() requires 3 arguments");
  }
  if (!args[0].has_ident_expr() || args[0].ident_expr().name().empty()) {
    return factory.ReportErrorAt(
        args[0],
        "transformMapEntry() 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],
        "transformMapEntry() second variable name must be a simple identifier");
  }
  if (args[0].ident_expr().name() == args[1].ident_expr().name()) {
    return factory.ReportErrorAt(args[0],
                                 "transformMapEntry() second variable must be "
                                 "different from the first variable");
  }
  if (args[0].ident_expr().name() == kAccumulatorVariableName) {
    return factory.ReportErrorAt(
        args[0],
        absl::StrCat("transformMapEntry() first variable name cannot be ",
                     kAccumulatorVariableName));
  }
  if (args[1].ident_expr().name() == kAccumulatorVariableName) {
    return factory.ReportErrorAt(
        args[1],
        absl::StrCat("transformMapEntry() second variable name cannot be ",
                     kAccumulatorVariableName));
  }
  std::string iter_var = args[0].ident_expr().name();
  std::string iter_var2 = args[1].ident_expr().name();
  Expr step = factory.NewCall("cel.@mapInsert", factory.NewAccuIdent(),
                              std::move(args[2]));
  return factory.NewComprehension(std::move(iter_var), std::move(iter_var2),
                                  std::move(target), factory.AccuVarName(),
                                  factory.NewMap(), factory.NewBoolConst(true),
                                  std::move(step), factory.NewAccuIdent());
}

Macro MakeTransformMap3EntryMacro() {
  auto status_or_macro =