Function Expand

common/expr.cc:210–290

namespace {
void Expand(Expr** tail, Expr* cur) {
  static common_internal::ExprEraseTag tag;
  switch (cur->kind_case()) {
    case ExprKindCase::kSelectExpr: {
      SelectExpr& select = cur->mutable_select_expr();
      if (select.has_operand()) {
        select.mutable_operand().SetNext(tag, *tail);
        *tail = &select.mutable_operand();
      }
      break;
    }
    case ExprKindCase::kCallExpr: {
      CallExpr& call = cur->mutable_call_expr();
      if (call.has_target()) {
        call.mutable_target().SetNext(tag, *tail);
        *tail = &call.mutable_target();
      }
      for (auto& arg : call.mutable_args()) {
        arg.SetNext(tag, *tail);
        *tail = &arg;
      }
      break;
    }
    case ExprKindCase::kListExpr: {
      for (auto& arg : cur->mutable_list_expr().mutable_elements()) {
        arg.mutable_expr().SetNext(tag, *tail);
        *tail = &arg.mutable_expr();
      }
      break;
    }
    case ExprKindCase::kStructExpr: {
      for (auto& field : cur->mutable_struct_expr().mutable_fields()) {
        field.mutable_value().SetNext(tag, *tail);
        *tail = &field.mutable_value();
      }
      break;
    }
    case ExprKindCase::kMapExpr: {
      for (auto& entry : cur->mutable_map_expr().mutable_entries()) {
        entry.mutable_key().SetNext(tag, *tail);
        *tail = &entry.mutable_key();
        entry.mutable_value().SetNext(tag, *tail);
        *tail = &entry.mutable_value();
      }
      break;
    }
    case ExprKindCase::kComprehensionExpr: {
      if (cur->comprehension_expr().has_accu_init()) {
        cur->mutable_comprehension_expr().mutable_accu_init().SetNext(tag,
                                                                      *tail);
        *tail = &cur->mutable_comprehension_expr().mutable_accu_init();
      }
      if (cur->comprehension_expr().has_iter_range()) {
        cur->mutable_comprehension_expr().mutable_iter_range().SetNext(tag,
                                                                       *tail);
        *tail = &cur->mutable_comprehension_expr().mutable_iter_range();
      }
      if (cur->comprehension_expr().has_loop_condition()) {