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 29package vanity 30 31import ( 32 "github.com/gogo/protobuf/gogoproto" 33 "github.com/gogo/protobuf/proto" 34 descriptor "github.com/gogo/protobuf/protoc-gen-gogo/descriptor" 35) 36 37func FieldHasBoolExtension(field *descriptor.FieldDescriptorProto, extension *proto.ExtensionDesc) bool { 38 if field.Options == nil { 39 return false 40 } 41 value, err := proto.GetExtension(field.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 54func SetBoolFieldOption(extension *proto.ExtensionDesc, value bool) func(field *descriptor.FieldDescriptorProto) { 55 return func(field *descriptor.FieldDescriptorProto) { 56 if FieldHasBoolExtension(field, extension) { 57 return 58 } 59 if field.Options == nil { 60 field.Options = &descriptor.FieldOptions{} 61 } 62 if err := proto.SetExtension(field.Options, extension, &value); err != nil { 63 panic(err) 64 } 65 } 66} 67 68func TurnOffNullable(field *descriptor.FieldDescriptorProto) { 69 if field.IsRepeated() && !field.IsMessage() { 70 return 71 } 72 SetBoolFieldOption(gogoproto.E_Nullable, false)(field) 73} 74 75func TurnOffNullableForNativeTypes(field *descriptor.FieldDescriptorProto) { 76 if field.IsRepeated() || field.IsMessage() { 77 return 78 } 79 SetBoolFieldOption(gogoproto.E_Nullable, false)(field) 80} 81 82func TurnOffNullableForNativeTypesWithoutDefaultsOnly(field *descriptor.FieldDescriptorProto) { 83 if field.IsRepeated() || field.IsMessage() { 84 return 85 } 86 if field.DefaultValue != nil { 87 return 88 } 89 SetBoolFieldOption(gogoproto.E_Nullable, false)(field) 90} 91