1 // Protocol Buffers for Go with Gadgets 2 // 3 // Copyright (c) 2015, The GoGo Authors. All 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 main 30 31 import ( 32 "strings" 33 34 "github.com/gogo/protobuf/vanity" 35 "github.com/gogo/protobuf/vanity/command" 36 ) 37 38 func main() { 39 req := command.Read() 40 files := req.GetProtoFile() 41 files = vanity.FilterFiles(files, vanity.NotGoogleProtobufDescriptorProto) 42 43 vanity.ForEachFile(files, vanity.TurnOnMarshalerAll) 44 vanity.ForEachFile(files, vanity.TurnOnSizerAll) 45 vanity.ForEachFile(files, vanity.TurnOnUnmarshalerAll) 46 47 vanity.ForEachFile(files, vanity.TurnOffGoEnumStringerAll) 48 vanity.ForEachFile(files, vanity.TurnOnEnumStringerAll) 49 50 vanity.ForEachFile(files, vanity.TurnOnEqualAll) 51 vanity.ForEachFile(files, vanity.TurnOnGoStringAll) 52 vanity.ForEachFile(files, vanity.TurnOffGoStringerAll) 53 54 vanity.ForEachFile(files, vanity.TurnOnMessageNameAll) 55 56 for _, file := range files { 57 if strings.HasSuffix(file.GetName(), "struct.proto") { 58 // TODO struct can also get a compare method when 59 // https://github.com/gogo/protobuf/issues/221 is fixed 60 //continue 61 } 62 vanity.TurnOnCompareAll(file) 63 } 64 65 for _, file := range files { 66 if strings.HasSuffix(file.GetName(), "timestamp.proto") || 67 strings.HasSuffix(file.GetName(), "duration.proto") { 68 continue 69 } 70 vanity.TurnOnStringerAll(file) 71 vanity.TurnOnPopulateAll(file) 72 } 73 74 resp := command.Generate(req) 75 command.Write(resp) 76 } 77