forked from bufbuild/buf
/
cc_enable_arenas.go
90 lines (83 loc) · 2.83 KB
/
cc_enable_arenas.go
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
82
83
84
85
86
87
88
89
90
// Copyright 2020-2024 Buf Technologies, Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package bufimagemodify
import (
"context"
"github.com/alis-exchange/buf/private/bufpkg/bufimage"
"go.uber.org/zap"
"google.golang.org/protobuf/proto"
"google.golang.org/protobuf/types/descriptorpb"
)
// CcEnableArenasID is the ID of the cc_enable_arenas modifier.
const CcEnableArenasID = "CC_ENABLE_ARENAS"
// ccEnableArenas is the SourceCodeInfo path for the cc_enable_arenas option.
// https://github.com/protocolbuffers/protobuf/blob/29152fbc064921ca982d64a3a9eae1daa8f979bb/src/google/protobuf/descriptor.proto#L420
var ccEnableArenasPath = []int32{8, 31}
func ccEnableArenas(
logger *zap.Logger,
sweeper Sweeper,
value bool,
overrides map[string]bool,
) Modifier {
return ModifierFunc(
func(ctx context.Context, image bufimage.Image) error {
seenOverrideFiles := make(map[string]struct{}, len(overrides))
for _, imageFile := range image.Files() {
modifierValue := value
if overrideValue, ok := overrides[imageFile.Path()]; ok {
modifierValue = overrideValue
seenOverrideFiles[imageFile.Path()] = struct{}{}
}
if err := ccEnableArenasForFile(ctx, sweeper, imageFile, modifierValue); err != nil {
return err
}
}
for overrideFile := range overrides {
if _, ok := seenOverrideFiles[overrideFile]; !ok {
logger.Sugar().Warnf("%s override for %q was unused", CcEnableArenasID, overrideFile)
}
}
return nil
},
)
}
func ccEnableArenasForFile(
ctx context.Context,
sweeper Sweeper,
imageFile bufimage.ImageFile,
value bool,
) error {
descriptor := imageFile.FileDescriptorProto()
options := descriptor.GetOptions()
switch {
case isWellKnownType(ctx, imageFile):
// The file is a well-known type, don't do anything.
return nil
case options != nil && options.GetCcEnableArenas() == value:
// The option is already set to the same value, don't do anything.
return nil
case options == nil && descriptorpb.Default_FileOptions_CcEnableArenas == value:
// The option is not set, but the value we want to set is the
// same as the default, don't do anything.
return nil
}
if options == nil {
descriptor.Options = &descriptorpb.FileOptions{}
}
descriptor.Options.CcEnableArenas = proto.Bool(value)
if sweeper != nil {
sweeper.mark(imageFile.Path(), ccEnableArenasPath)
}
return nil
}