github.com/gogo/protobuf@v1.3.2/vanity/enum.go (about)

     1  // Protocol Buffers for Go with Gadgets
     2  //
     3  // Copyright (c) 2015, The GoGo Authors.  rights reserved.
     4  // http://github.com/gogo/protobuf
     5  //
     6  // Redistribution and use in source and binary forms, with or without
     7  // modification, are permitted provided that the following conditions are
     8  // met:
     9  //
    10  //     * Redistributions of source code must retain the above copyright
    11  // notice, this list of conditions and the following disclaimer.
    12  //     * Redistributions in binary form must reproduce the above
    13  // copyright notice, this list of conditions and the following disclaimer
    14  // in the documentation and/or other materials provided with the
    15  // distribution.
    16  //
    17  // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
    18  // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
    19  // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
    20  // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
    21  // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
    22  // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
    23  // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
    24  // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
    25  // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
    26  // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
    27  // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
    28  
    29  package vanity
    30  
    31  import (
    32  	"github.com/gogo/protobuf/gogoproto"
    33  	"github.com/gogo/protobuf/proto"
    34  	descriptor "github.com/gogo/protobuf/protoc-gen-gogo/descriptor"
    35  )
    36  
    37  func EnumHasBoolExtension(enum *descriptor.EnumDescriptorProto, extension *proto.ExtensionDesc) bool {
    38  	if enum.Options == nil {
    39  		return false
    40  	}
    41  	value, err := proto.GetExtension(enum.Options, extension)
    42  	if err != nil {
    43  		return false
    44  	}
    45  	if value == nil {
    46  		return false
    47  	}
    48  	if value.(*bool) == nil {
    49  		return false
    50  	}
    51  	return true
    52  }
    53  
    54  func SetBoolEnumOption(extension *proto.ExtensionDesc, value bool) func(enum *descriptor.EnumDescriptorProto) {
    55  	return func(enum *descriptor.EnumDescriptorProto) {
    56  		if EnumHasBoolExtension(enum, extension) {
    57  			return
    58  		}
    59  		if enum.Options == nil {
    60  			enum.Options = &descriptor.EnumOptions{}
    61  		}
    62  		if err := proto.SetExtension(enum.Options, extension, &value); err != nil {
    63  			panic(err)
    64  		}
    65  	}
    66  }
    67  
    68  func TurnOffGoEnumPrefix(enum *descriptor.EnumDescriptorProto) {
    69  	SetBoolEnumOption(gogoproto.E_GoprotoEnumPrefix, false)(enum)
    70  }
    71  
    72  func TurnOffGoEnumStringer(enum *descriptor.EnumDescriptorProto) {
    73  	SetBoolEnumOption(gogoproto.E_GoprotoEnumStringer, false)(enum)
    74  }
    75  
    76  func TurnOnEnumStringer(enum *descriptor.EnumDescriptorProto) {
    77  	SetBoolEnumOption(gogoproto.E_EnumStringer, true)(enum)
    78  }