Method addFieldToDescriptorEnumType

lib/generation/ProtobufDescriptor.cpp:114–157

}

google::protobuf::FieldDescriptorProto *
ProtobufDescriptor::addFieldToDescriptorEnumType(
    google::protobuf::DescriptorProto &Desc, const std::string &FieldName,
    const Type &T) {
  assert(T.isEnumType() && "Unexpected Program State");
  auto EnumName = T.getTypeName();
  // TypeName might contain ':' due to namespace or nested.
  // Replacing them as ':' is not valid character for identifier
  std::replace(EnumName.begin(), EnumName.end(), ':', '_');
  if (Enums.find(EnumName) == Enums.end()) {
    const auto *EnumDef = T.getGlobalDef();
    if (EnumDef == nullptr)
      return nullptr;
    auto *EnumDesc = ProtoFile.add_enum_type();
    EnumDesc->set_name(EnumName);
    // First value of protobuf enum should be 0.
    // Set default name as UNKNOWN and overwrite it if target enum has 0 value.
    // Enum value should be globally unique in protobuf.
    // Adding name of enum as prefix to avoid collision.
    auto *DefaultVal = EnumDesc->add_value();
    DefaultVal->set_name(EnumName + "_UNKNOWN");
    DefaultVal->set_number(0);
    std::set<int64_t> EnumValues;
    bool HaveAlias = false;
    for (auto EnumVal : EnumDef->getElements()) {
      auto ElemVal = EnumVal.getValue();
      auto *ValDesc =
          ElemVal == 0 && EnumDesc->value(0).name() == EnumName + "_UNKNOWN"
              ? EnumDesc->mutable_value(0)
              : EnumDesc->add_value();
      ValDesc->set_name(EnumName + "_" + EnumVal.getName());
      ValDesc->set_number(ElemVal);
      auto Result = EnumValues.insert(ElemVal);
      if (!Result.second)
        HaveAlias = true;
    }
    if (HaveAlias)
      EnumDesc->mutable_options()->set_allow_alias(true);
    Enums.emplace(EnumName);
  }
  return addFieldToDescriptor(Desc, FieldName,
                              google::protobuf::FieldDescriptorProto::TYPE_ENUM,
                              EnumName);
}

google::protobuf::FieldDescriptorProto *
ProtobufDescriptor::addFieldToDescriptorPointerType(