feat(orm): add query proto codegen

orm/Makefile

@@ -1,3 +1,7 @@
 codegen:
 	go install ./cmd/protoc-gen-go-cosmos-orm
+	go install ./cmd/protoc-gen-go-cosmos-orm-proto
+	# generate .proto files first
+	(cd internal; buf generate --template buf.proto.gen.yaml)
+	# generate go code
 	(cd internal; buf generate)

orm/cmd/protoc-gen-go-cosmos-orm-proto/main.go

@@ -0,0 +1,11 @@
+package main
+
+import (
+	"google.golang.org/protobuf/compiler/protogen"
+
+	"github.com/cosmos/cosmos-sdk/orm/internal/codegen"
+)
+
+func main() {
+	protogen.Options{}.Run(codegen.QueryProtoPluginRunner)
+}

orm/cmd/protoc-gen-go-cosmos-orm/main.go

@@ -7,5 +7,5 @@ import (
 )
 
 func main() {
-	protogen.Options{}.Run(codegen.PluginRunner)
+	protogen.Options{}.Run(codegen.ORMPluginRunner)
 }

orm/internal/buf.gen.yaml

@@ -6,7 +6,10 @@ managed:
     override:
       buf.build/cosmos/cosmos-sdk: cosmossdk.io/api
 plugins:
-  - name: go-pulsar
+  - name: go
+    out: .
+    opt: paths=source_relative
+  - name: go-grpc
     out: .
     opt: paths=source_relative
   - name: go-cosmos-orm

orm/internal/buf.proto.gen.yaml

@@ -0,0 +1,11 @@
+version: v1
+managed:
+  enabled: true
+  go_package_prefix:
+    default: github.com/cosmos/cosmos-sdk/orm/internal
+    override:
+      buf.build/cosmos/cosmos-sdk: cosmossdk.io/api
+plugins:
+  - name: go-cosmos-orm-proto
+    out: .
+    opt: paths=source_relative

orm/internal/codegen/codegen.go

@@ -2,9 +2,11 @@ package codegen
 
 import (
 	"fmt"
+	"os"
 
 	"google.golang.org/protobuf/compiler/protogen"
 	"google.golang.org/protobuf/proto"
+	"google.golang.org/protobuf/types/pluginpb"
 
 	ormv1 "cosmossdk.io/api/cosmos/orm/v1"
 	"github.com/cosmos/cosmos-proto/generator"
@@ -17,7 +19,8 @@ const (
 	ormTablePkg = protogen.GoImportPath("github.com/cosmos/cosmos-sdk/orm/model/ormtable")
 )
 
-func PluginRunner(p *protogen.Plugin) error {
+func ORMPluginRunner(p *protogen.Plugin) error {
+	p.SupportedFeatures = uint64(pluginpb.CodeGeneratorResponse_FEATURE_PROTO3_OPTIONAL)
 	for _, f := range p.Files {
 		if !f.Generate {
 			continue
@@ -32,12 +35,42 @@ func PluginRunner(p *protogen.Plugin) error {
 			GeneratedFile: gen,
 			LocalPackages: map[string]bool{},
 		}
-		f := fileGen{GeneratedFile: cgen, file: f}
-		err := f.gen()
+		fgen := fileGen{GeneratedFile: cgen, file: f}
+		err := fgen.gen()
+		if err != nil {
+			return err
+		}
+	}
+
+	return nil
+}
+
+func QueryProtoPluginRunner(p *protogen.Plugin) error {
+	p.SupportedFeatures = uint64(pluginpb.CodeGeneratorResponse_FEATURE_PROTO3_OPTIONAL)
+	for _, f := range p.Files {
+		if !f.Generate {
+			continue
+		}
+
+		if !hasTables(f) {
+			continue
+		}
+
+		out, err := os.OpenFile(fmt.Sprintf("%s_query.proto", f.GeneratedFilenamePrefix), os.O_RDWR|os.O_TRUNC|os.O_CREATE, 0644)
 		if err != nil {
 			return err
 		}
 
+		err = queryProtoGen{
+			File:    f,
+			svc:     newWriter(),
+			msgs:    newWriter(),
+			outFile: out,
+			imports: map[string]bool{},
+		}.gen()
+		if err != nil {
+			return err
+		}
 	}
 
 	return nil

orm/internal/codegen/file.go

@@ -87,7 +87,11 @@ func (f fileGen) storeStructName() string {
 }
 
 func (f fileGen) fileShortName() string {
-	filename := f.file.Proto.GetName()
+	return fileShortName(f.file)
+}
+
+func fileShortName(file *protogen.File) string {
+	filename := file.Proto.GetName()
 	shortName := filepath.Base(filename)
 	i := strings.Index(shortName, ".")
 	if i > 0 {
@@ -155,11 +159,20 @@ func (f fileGen) genStoreConstructor(stores []*protogen.Message) {
 	f.P("}")
 }
 
-func (f fileGen) fieldsToCamelCase(fields string) string {
+func fieldsToCamelCase(fields string) string {
 	splitFields := strings.Split(fields, ",")
 	camelFields := make([]string, len(splitFields))
 	for i, field := range splitFields {
 		camelFields[i] = strcase.ToCamel(field)
 	}
 	return strings.Join(camelFields, "")
 }
+
+func fieldsToSnakeCase(fields string) string {
+	splitFields := strings.Split(fields, ",")
+	camelFields := make([]string, len(splitFields))
+	for i, field := range splitFields {
+		camelFields[i] = strcase.ToSnake(field)
+	}
+	return strings.Join(camelFields, "_")
+}