1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
|
// Protocol Buffers - Google's data interchange format
// Copyright 2023 Google LLC. All rights reserved.
//
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file or at
// https://developers.google.com/open-source/licenses/bsd
#include "google/protobuf/compiler/hpb/gen_extensions.h"
#include <string>
#include <vector>
#include "absl/strings/str_cat.h"
#include "google/protobuf/compiler/hpb/context.h"
#include "google/protobuf/compiler/hpb/gen_utils.h"
#include "google/protobuf/compiler/hpb/names.h"
#include "google/protobuf/descriptor.h"
#include "upb_generator/c/names.h"
namespace google::protobuf::hpb_generator {
namespace protobuf = ::proto2;
std::string ExtensionIdentifierBase(const protobuf::FieldDescriptor* ext) {
assert(ext->is_extension());
std::string ext_scope;
if (ext->extension_scope()) {
return upb::generator::CApiMessageType(ext->extension_scope()->full_name());
} else {
return ToCIdent(ext->file()->package());
}
}
std::string ContainingTypeName(const protobuf::FieldDescriptor* ext) {
return ext->containing_type()->file() != ext->file()
? QualifiedClassName(ext->containing_type())
: ClassName(ext->containing_type());
}
void WriteExtensionIdentifierHeader(const protobuf::FieldDescriptor* ext,
Context& ctx) {
std::string mini_table_name =
absl::StrCat(ExtensionIdentifierBase(ext), "_", ext->name(), "_ext");
std::string linkage = ext->extension_scope() ? "static" : "";
std::string ext_type = CppTypeParameterName(ext);
if (ext->is_repeated()) {
ext_type = absl::StrCat("::hpb::RepeatedField<", ext_type, ">");
}
ctx.Emit(
{{"containing_type_name", ContainingTypeName(ext)},
{"extendee_type", ContainingTypeName(ext)},
{"extension_type", ext_type},
{"default_value", DefaultValue(ext)},
{"linkage", linkage},
{"mini_table_name",
absl::StrCat(ExtensionIdentifierBase(ext), "_", ext->name(), "_ext")},
{"extension_name", ext->name()},
{"extension_number", ext->number()}},
R"cc(
inline $linkage$ constexpr ::hpb::internal::ExtensionIdentifier<
$extendee_type$, $extension_type$>
$extension_name$ =
::hpb::internal::PrivateAccess::InvokeConstructor<
::hpb::internal::ExtensionIdentifier<$containing_type_name$,
$extension_type$>>(
&$mini_table_name$, $default_value$, $extension_number$);
)cc");
}
void WriteExtensionIdentifiersHeader(
const std::vector<const protobuf::FieldDescriptor*>& extensions,
Context& ctx) {
for (const auto* ext : extensions) {
if (!ext->extension_scope()) {
WriteExtensionIdentifierHeader(ext, ctx);
}
}
}
} // namespace protobuf
} // namespace google::hpb_generator
|
