FilesExpand file tree

main

/

format.go

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

167 lines (139 loc) · 4.56 KB

main

/

format.go

Copy path

Top

File metadata and controls

• Code
• Blame

167 lines (139 loc) · 4.56 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

91

92

93

94

95

96

97

98

99

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149

150

151

152

153

154

155

156

157

158

159

160

161

162

163

164

165

166

167

// Copyright 2023-2025 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 prototransform

import (

"google.golang.org/protobuf/encoding/protojson"

"google.golang.org/protobuf/encoding/prototext"

"google.golang.org/protobuf/proto"

)

// InputFormat provides the interface to supply the [Converter] with an input

// composition. The format provided must accept a [Resolver]. Includes

// WithResolver() method to return the [Unmarshaler] with [Resolver] Supplied.

type InputFormat interface {

WithResolver(Resolver) Unmarshaler

}

// Unmarshaler is a [Unmarshaler].

type Unmarshaler interface {

Unmarshal([]byte, proto.Message) error

}

// OutputFormat provides the interface to supply the [Converter] with an output

// composition. The format provided must accept a [Resolver]. Includes

// WithResolver() method to return the [Marshaler] with [Resolver] Supplied.

type OutputFormat interface {

WithResolver(Resolver) Marshaler

}

// Marshaler is a [Marshaler].

type Marshaler interface {

Marshal(proto.Message) ([]byte, error)

}

type binaryInputFormat struct {

proto.UnmarshalOptions

}

// BinaryInputFormat convenience method for binary input format.

func BinaryInputFormat(in proto.UnmarshalOptions) InputFormat {

return &binaryInputFormat{

UnmarshalOptions: in,

}

}

// WithResolver to supply binary input format with a prebuilt types [Resolver].

func (x binaryInputFormat) WithResolver(in Resolver) Unmarshaler {

x.Resolver = in

return x

}

// BinaryOutputFormat convenience method for binary output format.

func BinaryOutputFormat(in proto.MarshalOptions) OutputFormat {

return outputFormatWithoutResolver{Marshaler: in}

}

type jSONInputFormat struct {

protojson.UnmarshalOptions

}

// JSONInputFormat convenience method for JSON input format.

func JSONInputFormat(in protojson.UnmarshalOptions) InputFormat {

return jSONInputFormat{

UnmarshalOptions: in,

}

}

// WithResolver to supply j s o n input format with a prebuilt types [Resolver].

func (x jSONInputFormat) WithResolver(in Resolver) Unmarshaler {

x.Resolver = in

return x

}

type jSONOutputFormat struct {

protojson.MarshalOptions

}

// JSONOutputFormat convenience method for JSON output format.

func JSONOutputFormat(in protojson.MarshalOptions) OutputFormat {

return jSONOutputFormat{

MarshalOptions: in,

}

}

// WithResolver to supply j s o n output format with a prebuilt types [Resolver].

func (x jSONOutputFormat) WithResolver(in Resolver) Marshaler {

x.Resolver = in

return x

}

type textInputFormat struct {

prototext.UnmarshalOptions

}

// TextInputFormat convenience method for text input format.

func TextInputFormat(in prototext.UnmarshalOptions) InputFormat {

return textInputFormat{

UnmarshalOptions: in,

}

}

// WithResolver to supply text input format with a prebuilt types [Resolver].

func (x textInputFormat) WithResolver(in Resolver) Unmarshaler {

x.Resolver = in

return x

}

type textOutputFormat struct {

prototext.MarshalOptions

}

// TextOutputFormat convenience method for text output format.

func TextOutputFormat(in prototext.MarshalOptions) OutputFormat {

return textOutputFormat{

MarshalOptions: in,

}

}

// WithResolver to supply text output format with a prebuilt types [Resolver].

func (x textOutputFormat) WithResolver(in Resolver) Marshaler {

x.Resolver = in

return x

}

type inputFormatWithoutResolver struct {

Unmarshaler

}

// InputFormatWithoutResolver convenience method for input format without resolver.

func InputFormatWithoutResolver(in Unmarshaler) InputFormat {

return inputFormatWithoutResolver{

Unmarshaler: in,

}

}

// WithResolver to supply input format without resolver.

func (x inputFormatWithoutResolver) WithResolver(_ Resolver) Unmarshaler {

return x

}

type outputFormatWithoutResolver struct {

Marshaler

}

// OutputFormatWithoutResolver convenience method for output format without resolver.

func OutputFormatWithoutResolver(in Marshaler) OutputFormat {

return outputFormatWithoutResolver{

Marshaler: in,

}

}

// WithResolver to supply output format without resolver.

func (x outputFormatWithoutResolver) WithResolver(_ Resolver) Marshaler {

return x

}