update deps

Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>

.github/dependabot.yml

@@ -0,0 +1,19 @@
+# To get started with Dependabot version updates, you'll need to specify which
+# package ecosystems to update and where the package manifests are located.
+# Please see the documentation for all configuration options:
+# https://help.github.com/github/administering-a-repository/configuration-options-for-dependency-updates
+
+version: 2
+updates:
+
+  # Maintain dependencies for GitHub Actions
+  - package-ecosystem: "github-actions"
+    directory: "/"
+    schedule:
+      interval: "daily"
+
+  # Maintain dependencies for Golang
+  - package-ecosystem: "gomod"
+    directory: "/"
+    schedule:
+      interval: "daily"

.github/renovate.json

@@ -1,20 +0,0 @@
-{
-  "extends": [
-    "config:base"
-  ],
-  "postUpdateOptions": ["gomodTidy"],
-  "packageRules": [
-    {
-      "matchUpdateTypes": ["minor", "patch", "pin", "digest"],
-      "automerge": true
-    },
-    {
-      "groupName": "all deps",
-      "separateMajorMinor": true,
-      "groupSlug": "all",
-      "packagePatterns": [
-        "*"
-      ]
-    }
-  ]
-}

.github/workflows/autoapprove.yml

@@ -0,0 +1,20 @@
+name: "autoapprove"
+
+on:
+  pull_request_target:
+    types: [assigned, opened, synchronize, reopened]
+
+permissions:
+  pull-requests: write
+  contents: write
+
+jobs:
+  autoapprove:
+    runs-on: ubuntu-latest
+    steps:
+      - name: approve
+        uses: hmarr/auto-approve-action@v3
+        if: github.actor == 'vtolstov' || github.actor == 'dependabot[bot]'
+        id: approve
+        with:
+          github-token: ${{ secrets.GITHUB_TOKEN }}

.github/workflows/automerge.yml

@@ -0,0 +1,21 @@
+name: "automerge"
+
+on:
+  pull_request_target:
+    types: [assigned, opened, synchronize, reopened]
+
+permissions:
+  pull-requests: write
+  contents: write
+
+jobs:
+  automerge:
+    runs-on: ubuntu-latest
+    if: github.actor == 'vtolstov'
+    steps:
+      - name: merge
+        id: merge
+        run: gh pr merge --auto --merge "$PR_URL"
+        env:
+          PR_URL: ${{github.event.pull_request.html_url}}
+          GITHUB_TOKEN: ${{secrets.TOKEN}}

.github/workflows/build.yml

@@ -3,19 +3,20 @@ on:
  push:
     branches:
     - master
+    - v3
 jobs:
   test:
     name: test
     runs-on: ubuntu-latest
     steps:
     - name: setup
-      uses: actions/setup-go@v2
+      uses: actions/setup-go@v3
       with:
-        go-version: 1.16
+        go-version: 1.17
     - name: checkout
-      uses: actions/checkout@v2
+      uses: actions/checkout@v3
     - name: cache
-      uses: actions/cache@v2
+      uses: actions/cache@v3
       with:
         path: ~/go/pkg/mod
         key: ${{ runner.os }}-go-${{ hashFiles('**/go.sum') }}
@@ -31,9 +32,9 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - name: checkout
-        uses: actions/checkout@v2
+        uses: actions/checkout@v3
       - name: lint
-        uses: golangci/golangci-lint-action@v2
+        uses: golangci/golangci-lint-action@v3.4.0
         continue-on-error: true
         with:
           # Required: the version of golangci-lint is required and must be specified without patch version: we always use the latest patch version.

.github/workflows/codeql-analysis.yml

@@ -0,0 +1,78 @@
+# For most projects, this workflow file will not need changing; you simply need
+# to commit it to your repository.
+#
+# You may wish to alter this file to override the set of languages analyzed,
+# or to provide custom queries or build logic.
+#
+# ******** NOTE ********
+# We have attempted to detect the languages in your repository. Please check
+# the `language` matrix defined below to confirm you have the correct set of
+# supported CodeQL languages.
+#
+name: "codeql"
+
+on:
+  workflow_run:
+    workflows: ["prbuild"]
+    types:
+      - completed
+  push:
+    branches: [ master, v3 ]
+  pull_request:
+    # The branches below must be a subset of the branches above
+    branches: [ master, v3 ]
+  schedule:
+    - cron: '34 1 * * 0'
+
+jobs:
+  analyze:
+    name: analyze
+    runs-on: ubuntu-latest
+    permissions:
+      actions: read
+      contents: read
+      security-events: write
+
+    strategy:
+      fail-fast: false
+      matrix:
+        language: [ 'go' ]
+        # CodeQL supports [ 'cpp', 'csharp', 'go', 'java', 'javascript', 'python' ]
+        # Learn more:
+        # https://docs.github.com/en/free-pro-team@latest/github/finding-security-vulnerabilities-and-errors-in-your-code/configuring-code-scanning#changing-the-languages-that-are-analyzed
+
+    steps:
+    - name: checkout
+      uses: actions/checkout@v3
+    - name: setup
+      uses: actions/setup-go@v3
+      with:
+        go-version: 1.17
+    # Initializes the CodeQL tools for scanning.
+    - name: init
+      uses: github/codeql-action/init@v2
+      with:
+        languages: ${{ matrix.language }}
+        # If you wish to specify custom queries, you can do so here or in a config file.
+        # By default, queries listed here will override any specified in a config file.
+        # Prefix the list here with "+" to use these queries and those in the config file.
+        # queries: ./path/to/local/query, your-org/your-repo/queries@main
+
+    # Autobuild attempts to build any compiled languages  (C/C++, C#, or Java).
+    # If this step fails, then you should remove it and run the build manually (see below)
+    - name: autobuild
+      uses: github/codeql-action/autobuild@v2
+
+    # ℹ️ Command-line programs to run using the OS shell.
+    # 📚 https://git.io/JvXDl
+
+    # ✏️ If the Autobuild fails above, remove it and uncomment the following three lines
+    #    and modify them (or add more) to build your code if your project
+    #    uses a compiled language
+
+    #- run: |
+    #   make bootstrap
+    #   make release
+
+    - name: analyze
+      uses: github/codeql-action/analyze@v2

.github/workflows/dependabot-automerge.yml

@@ -0,0 +1,27 @@
+name: "dependabot-automerge"
+
+on:
+  pull_request_target:
+    types: [assigned, opened, synchronize, reopened]
+
+permissions:
+  pull-requests: write
+  contents: write
+
+jobs:
+  automerge:
+    runs-on: ubuntu-latest
+    if: github.actor == 'dependabot[bot]'
+    steps:
+      - name: metadata
+        id: metadata
+        uses: dependabot/fetch-metadata@v1.3.6
+        with:
+          github-token: "${{ secrets.TOKEN }}"
+      - name: merge
+        id: merge
+        if: ${{contains(steps.metadata.outputs.dependency-names, 'go.unistack.org')}}
+        run: gh pr merge --auto --merge "$PR_URL"
+        env:
+          PR_URL: ${{github.event.pull_request.html_url}}
+          GITHUB_TOKEN: ${{secrets.TOKEN}}

.github/workflows/pr.yml

@@ -3,19 +3,20 @@ on:
   pull_request:
     branches:
     - master
+    - v3
 jobs:
   test:
     name: test
     runs-on: ubuntu-latest
     steps:
     - name: setup
-      uses: actions/setup-go@v2
+      uses: actions/setup-go@v3
       with:
-        go-version: 1.16
+        go-version: 1.17
     - name: checkout
-      uses: actions/checkout@v2
+      uses: actions/checkout@v3
     - name: cache
-      uses: actions/cache@v2
+      uses: actions/cache@v3
       with:
         path: ~/go/pkg/mod
         key: ${{ runner.os }}-go-${{ hashFiles('**/go.sum') }}
@@ -31,9 +32,9 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - name: checkout
-        uses: actions/checkout@v2
+        uses: actions/checkout@v3
       - name: lint
-        uses: golangci/golangci-lint-action@v2
+        uses: golangci/golangci-lint-action@v3.4.0
         continue-on-error: true
         with:
           # Required: the version of golangci-lint is required and must be specified without patch version: we always use the latest patch version.

.gitignore

@@ -0,0 +1,38 @@
+# Develop tools
+/.vscode/
+/.idea/
+
+# Binaries for programs and plugins
+*.exe
+*.exe~
+*.dll
+*.so
+*.dylib
+
+# Folders
+_obj
+_test
+_build
+
+# Architecture specific extensions/prefixes
+*.[568vq]
+[568vq].out
+
+*.cgo1.go
+*.cgo2.c
+_cgo_defun.c
+_cgo_gotypes.go
+_cgo_export.*
+
+# Test binary, build with `go test -c`
+*.test
+
+# Output of the go coverage tool, specifically when used with LiteIDE
+*.out
+
+# vim temp files
+*~
+*.swp
+*.swo
+/protoc-gen-go-micro
+

README.md

@@ -1,4 +1,4 @@
-# `protoc-gen-micro`
+# `protoc-gen-go-micro`
 protobuf plugin to generate helper code for micro framework
 
 A generic **code**/script/data generator based on [Protobuf](https://developers.google.com/protocol-buffers/).
@@ -10,16 +10,17 @@ This project is a generator plugin for the Google Protocol Buffers compiler (`pr
 ## Usage
 
 ```console
-$> protoc --micro_out=debug=true,components="micro|http":. input.proto
+$> protoc --go_micro_out=debug=true,components="micro|http":. input.proto
 ```
 
 | Option                | Default Value | Accepted Values           | Description
 |-----------------------|---------------|---------------------------|-----------------------
+| `tag_path`            | `.`           | `any local path`          | path contains generated protobuf code that needs to be tagged
 | `debug`               | *false*       | `true` or `false`         | if *true*, `protoc` will generate a more verbose output
-| `components`          | `micro`       | `micro rpc http chi gorilla` | some values cant coexists like gorilla/chi or rpc/http, values must be concatinated with pipe symbol
+| `components`          | `micro`       | `micro rpc http chi gorilla client server` | some values can't coexists like gorilla/chi or rpc/http, values must be concatinated with pipe symbol
 
 ## Install
 
 * Install the **go** compiler and tools from https://golang.org/doc/install
-* Install **protoc-gen-go**: `go get google.golang.org/protobuf/cmd/protoc-gen-go`
-* Install **protoc-gen-micro**: `go get github.com/unistack-org/protoc-gen-micro/v3`
+* Install **protoc-gen-go**: `go install google.golang.org/protobuf/cmd/protoc-gen-go`
+* Install **protoc-gen-go-micro**: `go install go.unistack.org/protoc-gen-go-micro/v3`

ast.go

@@ -0,0 +1,170 @@
+package main
+
+import (
+	"go/ast"
+	"go/format"
+	"go/parser"
+	"go/token"
+	"os"
+	"path/filepath"
+	"strings"
+
+	"github.com/fatih/structtag"
+	tag_options "go.unistack.org/micro-proto/v3/tag"
+	"google.golang.org/protobuf/compiler/protogen"
+	"google.golang.org/protobuf/proto"
+)
+
+var astFields = make(map[string]map[string]map[string]*structtag.Tags) // map proto file with proto message ast struct
+
+func (g *Generator) astFill(file *protogen.File, message *protogen.Message) error {
+	for _, field := range message.Fields {
+		if field.Desc.Options() == nil {
+			continue
+		}
+		if !proto.HasExtension(field.Desc.Options(), tag_options.E_Tags) {
+			continue
+		}
+
+		opts := proto.GetExtension(field.Desc.Options(), tag_options.E_Tags)
+		if opts != nil {
+			fpath := filepath.Join(g.tagPath, file.GeneratedFilenamePrefix+".pb.go")
+			mp, ok := astFields[fpath]
+			if !ok {
+				mp = make(map[string]map[string]*structtag.Tags)
+			}
+			nmp, ok := mp[message.GoIdent.GoName]
+			if !ok {
+				nmp = make(map[string]*structtag.Tags)
+			}
+			tags, err := structtag.Parse(opts.(string))
+			if err != nil {
+				return err
+			}
+			nmp[field.GoName] = tags
+			mp[message.GoIdent.GoName] = nmp
+			astFields[fpath] = mp
+		}
+	}
+	for _, nmessage := range message.Messages {
+		if err := g.astFill(file, nmessage); err != nil {
+			return err
+		}
+	}
+
+	return nil
+}
+
+func (g *Generator) astGenerate(plugin *protogen.Plugin) error {
+	if g.tagPath == "" {
+		return nil
+	}
+
+	for _, file := range plugin.Files {
+		if !file.Generate {
+			continue
+		}
+		for _, message := range file.Messages {
+			if err := g.astFill(file, message); err != nil {
+				return err
+			}
+		}
+	}
+
+	for file, mp := range astFields {
+		fset := token.NewFileSet()
+		pf, err := parser.ParseFile(fset, file, nil, parser.AllErrors|parser.ParseComments)
+		if err != nil {
+			return err
+		}
+
+		r := retag{}
+		f := func(n ast.Node) ast.Visitor {
+			if r.err != nil {
+				return nil
+			}
+
+			if v, ok := n.(*ast.TypeSpec); ok {
+				r.fields = mp[v.Name.Name]
+				return r
+			}
+
+			return nil
+		}
+
+		ast.Walk(structVisitor{f}, pf)
+
+		if r.err != nil {
+			return err
+		}
+
+		fp, err := os.OpenFile(file, os.O_WRONLY|os.O_TRUNC, os.FileMode(0644))
+		if err != nil {
+			return err
+		}
+		if err = format.Node(fp, fset, pf); err != nil {
+			fp.Close()
+			return err
+		}
+		if err = fp.Close(); err != nil {
+			return err
+		}
+	}
+
+	return nil
+}
+
+type retag struct {
+	err    error
+	fields map[string]*structtag.Tags
+}
+
+func (v retag) Visit(n ast.Node) ast.Visitor {
+	if v.err != nil {
+		return nil
+	}
+	if f, ok := n.(*ast.Field); ok {
+		if len(f.Names) == 0 {
+			return nil
+		}
+
+		newTags := v.fields[f.Names[0].String()]
+		if newTags == nil {
+			return nil
+		}
+		if f.Tag == nil {
+			f.Tag = &ast.BasicLit{
+				Kind: token.STRING,
+			}
+		}
+
+		oldTags, err := structtag.Parse(strings.Trim(f.Tag.Value, "`"))
+		if err != nil {
+			v.err = err
+			return nil
+		}
+		for _, t := range newTags.Tags() {
+			oldTags.Set(t)
+		}
+
+		f.Tag.Value = "`" + oldTags.String() + "`"
+
+		return nil
+	}
+
+	return v
+}
+
+type structVisitor struct {
+	visitor func(n ast.Node) ast.Visitor
+}
+
+func (v structVisitor) Visit(n ast.Node) ast.Visitor {
+	if tp, ok := n.(*ast.TypeSpec); ok {
+		if _, ok := tp.Type.(*ast.StructType); ok {
+			ast.Walk(v.visitor(n), n)
+			return nil // This will ensure this struct is no longer traversed
+		}
+	}
+	return v
+}

chi.go

@@ -4,9 +4,7 @@ import (
 	"google.golang.org/protobuf/compiler/protogen"
 )
 
-var (
-	chiPackageFiles map[protogen.GoPackageName]struct{}
-)
+var chiPackageFiles map[protogen.GoPackageName]struct{}
 
 func (g *Generator) chiGenerate(component string, plugin *protogen.Plugin) error {
 	chiPackageFiles = make(map[protogen.GoPackageName]struct{})
@@ -23,53 +21,54 @@ func (g *Generator) chiGenerate(component string, plugin *protogen.Plugin) error
 
 		chiPackageFiles[file.GoPackageName] = struct{}{}
 		gname := "micro" + "_" + component + ".pb.go"
-		gfile := plugin.NewGeneratedFile(gname, ".")
 
-		gfile.P("// Code generated by protoc-gen-micro")
+		path := file.GoImportPath
+		if g.standalone {
+			path = "."
+		}
+		gfile := plugin.NewGeneratedFile(gname, path)
+
+		gfile.P("// Code generated by protoc-gen-go-micro. DO NOT EDIT.")
+		gfile.P("// protoc-gen-go-micro version: " + versionComment)
+		gfile.P()
 		gfile.P("package ", file.GoPackageName)
+		gfile.P()
 
-		gfile.P()
-		gfile.P("import (")
-		gfile.P(`"context"`)
-		gfile.P(`"fmt"`)
-		gfile.P(`"net/http"`)
-		gfile.P(`"reflect"`)
-		gfile.P(`"strings"`)
-		gfile.P(`chi "github.com/go-chi/chi/v4"`)
-		gfile.P(`middleware "github.com/go-chi/chi/v4/middleware"`)
-		gfile.P(`micro_api "github.com/unistack-org/micro/v3/api"`)
-		gfile.P(")")
-		gfile.P()
+		gfile.Import(contextPackage)
+		gfile.Import(fmtPackage)
+		gfile.Import(httpPackage)
+		gfile.Import(reflectPackage)
+		gfile.Import(stringsPackage)
+		gfile.Import(chiPackage)
+		gfile.Import(chiMiddlewarePackage)
 
 		gfile.P("type routeKey struct{}")
 
-		gfile.P("func RouteName(ctx context.Context) (string, bool) {")
+		gfile.P("func RouteName(ctx ", contextPackage.Ident("Context"), ") (string, bool) {")
 		gfile.P("value, ok := ctx.Value(routeKey{}).(string)")
 		gfile.P("return value, ok")
 		gfile.P("}")
 		gfile.P()
-		gfile.P("func RegisterHandlers(r *chi.Mux, h interface{}, eps []*micro_api.Endpoint) error {")
-		gfile.P("v := reflect.ValueOf(h)")
+		gfile.P("func RegisterHandlers(r *", chiPackage.Ident("Mux"), ", h interface{}, eps []", microServerHttpPackage.Ident("EndpointMetadata"), ") error {")
+		gfile.P("v := ", reflectPackage.Ident("ValueOf"), "(h)")
 		gfile.P("if v.NumMethod() < 1 {")
-		gfile.P(`return fmt.Errorf("handler has no methods: %T", h)`)
+		gfile.P(`return `, fmtPackage.Ident("Errorf"), `("handler has no methods: %T", h)`)
 		gfile.P("}")
 		gfile.P("for _, ep := range eps {")
-		gfile.P(`idx := strings.Index(ep.Name, ".")`)
-		gfile.P("if idx < 1 || len(ep.Name) <= idx {")
-		gfile.P(`return fmt.Errorf("invalid api.Endpoint name: %s", ep.Name)`)
+		gfile.P(`idx := `, stringsPackage.Ident("Index"), `(ep.Name, ".")`)
+		gfile.P(`if idx < 1 || len(ep.Name) <= idx {`)
+		gfile.P(`return `, fmtPackage.Ident("Errorf"), `("invalid endpoint name: %s", ep.Name)`)
 		gfile.P("}")
-		gfile.P("name := ep.Name[idx+1:]")
+		gfile.P(`name := ep.Name[idx+1:]`)
 		gfile.P("m := v.MethodByName(name)")
 		gfile.P("if !m.IsValid() || m.IsZero() {")
-		gfile.P(`return fmt.Errorf("invalid handler, method %s not found", name)`)
+		gfile.P(`return `, fmtPackage.Ident("Errorf"), `("invalid handler, method %s not found", name)`)
 		gfile.P("}")
-		gfile.P("rh, ok := m.Interface().(func(http.ResponseWriter, *http.Request))")
+		gfile.P("rh, ok := m.Interface().(func(", httpPackage.Ident("ResponseWriter"), ", *", httpPackage.Ident("Request"), "))")
 		gfile.P("if !ok {")
-		gfile.P(`return fmt.Errorf("invalid handler: %#+v", m.Interface())`)
-		gfile.P("}")
-		gfile.P("for _, method := range ep.Method {")
-		gfile.P("r.With(middleware.WithValue(routeKey{}, ep.Name)).MethodFunc(method, ep.Path[0], rh)")
+		gfile.P(`return `, fmtPackage.Ident("Errorf"), `("invalid handler: %#+v", m.Interface())`)
 		gfile.P("}")
+		gfile.P("r.With(", chiMiddlewarePackage.Ident("WithValue"), `(routeKey{}, ep.Name)).MethodFunc(ep.Method, ep.Path, rh)`)
 		gfile.P("}")
 		gfile.P("return nil")
 		gfile.P("}")

example/example.proto

@@ -0,0 +1,40 @@
+syntax = "proto3";
+
+package example;
+
+option go_package = "github.com/unistack-org/protoc-gen-go-micro/v3/example;examplepb";
+
+import "tag/tag.proto";
+import "api/annotations.proto";
+import "openapiv3/annotations.proto";
+//import "google/protobuf/wrappers.proto";
+import "graphql/graphql.proto";
+
+service Example {
+	rpc Call(CallReq) returns (CallRsp) {
+    option (micro.graphql.rpc) = {type: QUERY};
+    option (micro.openapiv3.openapiv3_operation) = {
+      operation_id: "Call";
+      responses: {
+        default: {
+          reference: {_ref: ".example.Error"};
+        };
+      };
+    };
+		option (micro.api.http) = { post: "/v1/example/call/{name}"; body: "*"; };
+		option (micro.api.micro_method) = { timeout: "5s"; };
+  };
+};
+
+message CallReq {
+  string name = 1 [(micro.graphql.field) = {required: true}]; 
+  string req = 2;
+};
+
+message CallRsp {
+  string rsp = 2;
+};
+
+message Error {
+	string msg = 1;
+};

example/schema.graphql

@@ -0,0 +1,11 @@
+directive @Example on FIELD_DEFINITION
+input CallReqInput {
+	name: String!
+	req: String
+}
+type CallRsp {
+	rsp: String
+}
+type Query {
+	exampleCall(in: CallReqInput): CallRsp
+}

fiealaligment.go

@@ -0,0 +1,24 @@
+package main
+
+import (
+	"log"
+	"os"
+
+	"golang.org/x/tools/go/analysis/passes/fieldalignment"
+	"golang.org/x/tools/go/analysis/singlechecker"
+	"google.golang.org/protobuf/compiler/protogen"
+)
+
+func (g *Generator) fieldAlign(plugin *protogen.Plugin) error {
+	if !g.fieldaligment {
+		return nil
+	}
+
+	log.Printf("%v\n", []string{"fieldalignment", "-fix", g.tagPath})
+	origArgs := os.Args
+	os.Args = []string{"fieldalignment", "-fix", g.tagPath}
+	singlechecker.Main(fieldalignment.Analyzer)
+	os.Args = origArgs
+
+	return nil
+}

generate.go

@@ -0,0 +1,3 @@
+package main
+
+//go:generate sh -xc "protoc -I./example -I. -I$(go list -f '{{ .Dir }}' -m go.unistack.org/micro-proto/v3) --go-micro_out=components=graphqls,graphql_file=./schema.graphql:./example example/example.proto"

go.mod

@@ -1,9 +1,35 @@
-module github.com/unistack-org/protoc-gen-micro/v3
+module go.unistack.org/protoc-gen-go-micro/v3
 
-go 1.16
+go 1.22.0
+
+toolchain go1.23.4
 
 require (
-	github.com/grpc-ecosystem/grpc-gateway/v2 v2.2.0
-	google.golang.org/genproto v0.0.0-20210207032614-bba0dbe2a9ea
-	google.golang.org/protobuf v1.25.0
+	github.com/fatih/structtag v1.2.0
+	github.com/jhump/protoreflect v1.17.0
+	github.com/vektah/gqlparser/v2 v2.5.20
+	go.unistack.org/micro-proto/v3 v3.4.1
+	golang.org/x/tools v0.28.0
+	google.golang.org/protobuf v1.35.2
+)
+
+require (
+	github.com/agnivade/levenshtein v1.2.0 // indirect
+	github.com/bufbuild/protocompile v0.14.1 // indirect
+	github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc // indirect
+	github.com/golang/protobuf v1.5.4 // indirect
+	github.com/google/gnostic v0.7.0 // indirect
+	github.com/google/gnostic-models v0.6.9 // indirect
+	github.com/kr/pretty v0.3.1 // indirect
+	github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2 // indirect
+	github.com/rogpeppe/go-internal v1.11.0 // indirect
+	github.com/sergi/go-diff v1.3.1 // indirect
+	github.com/stretchr/testify v1.10.0 // indirect
+	golang.org/x/mod v0.22.0 // indirect
+	golang.org/x/sync v0.10.0 // indirect
+	golang.org/x/text v0.21.0 // indirect
+	google.golang.org/genproto v0.0.0-20240227224415-6ceb2ff114de // indirect
+	google.golang.org/genproto/googleapis/rpc v0.0.0-20241209162323-e6fa225c2576 // indirect
+	google.golang.org/grpc v1.68.1 // indirect
+	gopkg.in/yaml.v3 v3.0.1 // indirect
 )

gorilla.go

@@ -4,9 +4,7 @@ import (
 	"google.golang.org/protobuf/compiler/protogen"
 )
 
-var (
-	gorillaPackageFiles map[protogen.GoPackageName]struct{}
-)
+var gorillaPackageFiles map[protogen.GoPackageName]struct{}
 
 func (g *Generator) gorillaGenerate(component string, plugin *protogen.Plugin) error {
 	gorillaPackageFiles = make(map[protogen.GoPackageName]struct{})
@@ -23,43 +21,45 @@ func (g *Generator) gorillaGenerate(component string, plugin *protogen.Plugin) e
 
 		gorillaPackageFiles[file.GoPackageName] = struct{}{}
 		gname := "micro" + "_" + component + ".pb.go"
-		gfile := plugin.NewGeneratedFile(gname, ".")
 
-		gfile.P("// Code generated by protoc-gen-micro")
+		path := file.GoImportPath
+		if g.standalone {
+			path = "."
+		}
+		gfile := plugin.NewGeneratedFile(gname, path)
+
+		gfile.P("// Code generated by protoc-gen-go-micro. DO NOT EDIT.")
+		gfile.P("// protoc-gen-go-micro version: " + versionComment)
+		gfile.P()
 		gfile.P("package ", file.GoPackageName)
 		gfile.P()
 
-		gfile.P("import (")
-		gfile.P(`"fmt"`)
-		gfile.P(`"net/http"`)
-		gfile.P(`"reflect"`)
-		gfile.P(`"strings"`)
-		gfile.P()
-		gfile.P(`micro_api "github.com/unistack-org/micro/v3/api"`)
-		gfile.P(`mux "github.com/gorilla/mux"`)
-		gfile.P(")")
-		gfile.P()
+		gfile.Import(fmtPackage)
+		gfile.Import(httpPackage)
+		gfile.Import(reflectPackage)
+		gfile.Import(stringsPackage)
+		gfile.Import(gorillaMuxPackage)
 
-		gfile.P("func RegisterHandlers(r *mux.Router, h interface{}, eps []*micro_api.Endpoint) error {")
-		gfile.P("v := reflect.ValueOf(h)")
+		gfile.P("func RegisterHandlers(r *", gorillaMuxPackage.Ident("Router"), ", h interface{}, eps []", microServerHttpPackage.Ident("EndpointMetadata"), ") error {")
+		gfile.P("v := ", reflectPackage.Ident("ValueOf"), "(h)")
 		gfile.P("if v.NumMethod() < 1 {")
-		gfile.P(`return fmt.Errorf("handler has no methods: %T", h)`)
+		gfile.P(`return `, fmtPackage.Ident("Errorf"), `("handler has no methods: %T", h)`)
 		gfile.P("}")
 		gfile.P("for _, ep := range eps {")
-		gfile.P(`idx := strings.Index(ep.Name, ".")`)
-		gfile.P("if idx < 1 || len(ep.Name) <= idx {")
-		gfile.P(`return fmt.Errorf("invalid api.Endpoint name: %s", ep.Name)`)
+		gfile.P(`idx := `, stringsPackage.Ident("Index"), `(ep.Name, ".")`)
+		gfile.P(`if idx < 1 || len(ep.Name) <= idx {`)
+		gfile.P(`return `, fmtPackage.Ident("Errorf"), `("invalid endpoint name: %s", ep.Name)`)
 		gfile.P("}")
-		gfile.P("name := ep.Name[idx+1:]")
+		gfile.P(`name := ep.Name[idx+1:]`)
 		gfile.P("m := v.MethodByName(name)")
 		gfile.P("if !m.IsValid() || m.IsZero() {")
-		gfile.P(`return fmt.Errorf("invalid handler, method %s not found", name)`)
+		gfile.P(`return `, fmtPackage.Ident("Errorf"), `("invalid handler, method %s not found", name)`)
 		gfile.P("}")
-		gfile.P("rh, ok := m.Interface().(func(http.ResponseWriter, *http.Request))")
+		gfile.P("rh, ok := m.Interface().(func(", httpPackage.Ident("ResponseWriter"), ", *", httpPackage.Ident("Request"), "))")
 		gfile.P("if !ok {")
-		gfile.P(`return fmt.Errorf("invalid handler: %#+v", m.Interface())`)
+		gfile.P(`return `, fmtPackage.Ident("Errorf"), `("invalid handler: %#+v", m.Interface())`)
 		gfile.P("}")
-		gfile.P("r.HandleFunc(ep.Path[0], rh).Methods(ep.Method...).Name(ep.Name)")
+		gfile.P(`r.HandleFunc(ep.Path, rh).Methods(ep.Method).Name(ep.Name)`)
 		gfile.P("}")
 		gfile.P("return nil")
 		gfile.P("}")

graphql.go

@@ -0,0 +1,9 @@
+package main
+
+import (
+	"google.golang.org/protobuf/compiler/protogen"
+)
+
+func (g *Generator) graphqlGenerate(plugin *protogen.Plugin) error {
+	return nil
+}

graphql/callstack.go

@@ -0,0 +1,32 @@
+package generator
+
+type Callstack interface {
+	Push(entry interface{})
+	Pop(entry interface{})
+	Has(entry interface{}) bool
+}
+
+func NewCallstack() Callstack {
+	return &callstack{stack: make(map[interface{}]int), index: 0}
+}
+
+type callstack struct {
+	stack  map[interface{}]int
+	sorted []string
+	index  int
+}
+
+func (c *callstack) Pop(entry interface{}) {
+	delete(c.stack, entry)
+	c.index--
+}
+
+func (c *callstack) Push(entry interface{}) {
+	c.stack[entry] = c.index
+	c.index++
+}
+
+func (c *callstack) Has(entry interface{}) bool {
+	_, ok := c.stack[entry]
+	return ok
+}

graphql/descriptors.go

@@ -0,0 +1,77 @@
+package generator
+
+import (
+	"strings"
+
+	"github.com/jhump/protoreflect/desc"
+	"github.com/vektah/gqlparser/v2/ast"
+	any "google.golang.org/protobuf/types/known/anypb"
+)
+
+type ObjectDescriptor struct {
+	*ast.Definition
+	desc.Descriptor
+
+	types      []*ObjectDescriptor
+	fields     []*FieldDescriptor
+	fieldNames map[string]*FieldDescriptor
+}
+
+func (o *ObjectDescriptor) AsGraphql() *ast.Definition {
+	return o.Definition
+}
+
+func (o *ObjectDescriptor) uniqueName(f *desc.FieldDescriptor) string {
+	return strings.Title(f.GetName())
+}
+
+func (o *ObjectDescriptor) IsInput() bool {
+	return o.Kind == ast.InputObject
+}
+
+func (o *ObjectDescriptor) GetFields() []*FieldDescriptor {
+	return o.fields
+}
+
+func (o *ObjectDescriptor) GetTypes() []*ObjectDescriptor {
+	return o.types
+}
+
+func (o *ObjectDescriptor) IsMessage() bool {
+	_, ok := o.Descriptor.(*desc.MessageDescriptor)
+	return ok
+}
+
+// same isEmpty but for mortals
+func IsEmpty(o *desc.MessageDescriptor) bool { return isEmpty(o, NewCallstack()) }
+
+// make sure objects are fulled with all objects
+func isEmpty(o *desc.MessageDescriptor, callstack Callstack) bool {
+	callstack.Push(o)
+	defer callstack.Pop(o)
+
+	if len(o.GetFields()) == 0 {
+		return true
+	}
+	for _, f := range o.GetFields() {
+		objType := f.GetMessageType()
+		if objType == nil {
+			return false
+		}
+
+		// check if the call stack already contains a reference to this type and prevent it from calling itself again
+		if callstack.Has(objType) {
+			return true
+		}
+		if !isEmpty(objType, callstack) {
+			return false
+		}
+	}
+
+	return true
+}
+
+// TODO maybe not compare by strings
+func IsAny(o *desc.MessageDescriptor) bool {
+	return string((&any.Any{}).ProtoReflect().Descriptor().FullName()) == o.GetFullyQualifiedName()
+}

graphql/generator.go

@@ -0,0 +1,761 @@
+package generator
+
+import (
+	"fmt"
+	"strconv"
+	"strings"
+
+	"github.com/jhump/protoreflect/desc"
+	"github.com/vektah/gqlparser/v2/ast"
+	"google.golang.org/protobuf/compiler/protogen"
+	descriptor "google.golang.org/protobuf/types/descriptorpb"
+
+	gqlpb "go.unistack.org/micro-proto/v3/graphql"
+)
+
+const (
+	fieldPrefix        = "Field"
+	inputSuffix        = "Input"
+	typeSep            = "_"
+	packageSep         = "."
+	anyTypeDescription = "Any is any json type"
+	scalarBytes        = "Bytes"
+	goFieldDirective   = "goField"
+
+	DefaultExtension = "graphql"
+)
+
+func NewSchemas(descs []*desc.FileDescriptor, mergeSchemas, genServiceDesc bool, plugin *protogen.Plugin) (schemas SchemaDescriptorList, err error) {
+	var files []*descriptor.FileDescriptorProto
+	for _, d := range descs {
+		files = append(files, d.AsFileDescriptorProto())
+	}
+	var goref GoRef
+	if plugin != nil {
+		goref, err = NewGoRef(plugin)
+		if err != nil {
+			return nil, err
+		}
+	}
+
+	if mergeSchemas {
+		schema := NewSchemaDescriptor(genServiceDesc, goref)
+		for _, file := range descs {
+			err := generateFile(file, schema)
+			if err != nil {
+				return nil, err
+			}
+		}
+
+		return []*SchemaDescriptor{schema}, nil
+	}
+
+	for _, file := range descs {
+		schema := NewSchemaDescriptor(genServiceDesc, goref)
+		err := generateFile(file, schema)
+		if err != nil {
+			return nil, err
+		}
+
+		schemas = append(schemas, schema)
+	}
+
+	return
+}
+
+func generateFile(file *desc.FileDescriptor, schema *SchemaDescriptor) error {
+	schema.FileDescriptors = append(schema.FileDescriptors, file)
+
+	for _, svc := range file.GetServices() {
+		svcOpts := GraphqlServiceOptions(svc.AsServiceDescriptorProto().GetOptions())
+		if svcOpts != nil && svcOpts.Ignore != nil && *svcOpts.Ignore {
+			continue
+		}
+		for _, rpc := range svc.GetMethods() {
+			rpcOpts := GraphqlMethodOptions(rpc.AsMethodDescriptorProto().GetOptions())
+			if rpcOpts != nil && rpcOpts.Ignore != nil && *rpcOpts.Ignore {
+				continue
+			}
+			in, err := schema.CreateObjects(rpc.GetInputType(), true)
+			if err != nil {
+				return err
+			}
+
+			out, err := schema.CreateObjects(rpc.GetOutputType(), false)
+			if err != nil {
+				return err
+			}
+
+			if rpc.IsServerStreaming() && rpc.IsClientStreaming() {
+				schema.GetMutation().addMethod(svc, rpc, in, out)
+			}
+
+			if rpc.IsServerStreaming() {
+				schema.GetSubscription().addMethod(svc, rpc, in, out)
+			} else {
+				switch GetRequestType(rpcOpts, svcOpts) {
+				case gqlpb.Type_QUERY:
+					schema.GetQuery().addMethod(svc, rpc, in, out)
+				default:
+					schema.GetMutation().addMethod(svc, rpc, in, out)
+				}
+			}
+		}
+	}
+
+	return nil
+}
+
+type SchemaDescriptorList []*SchemaDescriptor
+
+func (s SchemaDescriptorList) AsGraphql() (astSchema []*ast.Schema) {
+	for _, ss := range s {
+		astSchema = append(astSchema, ss.AsGraphql())
+	}
+	return
+}
+
+func (s SchemaDescriptorList) GetForDescriptor(file *protogen.File) *SchemaDescriptor {
+	for _, schema := range s {
+		for _, d := range schema.FileDescriptors {
+			if d.AsFileDescriptorProto() == file.Proto {
+				return schema
+			}
+		}
+	}
+	return nil
+}
+
+func NewSchemaDescriptor(genServiceDesc bool, goref GoRef) *SchemaDescriptor {
+	sd := &SchemaDescriptor{
+		Directives:                 map[string]*ast.DirectiveDefinition{},
+		reservedNames:              map[string]desc.Descriptor{},
+		createdObjects:             map[createdObjectKey]*ObjectDescriptor{},
+		generateServiceDescriptors: genServiceDesc,
+		goRef:                      goref,
+	}
+	for _, name := range graphqlReservedNames {
+		sd.reservedNames[name] = nil
+	}
+	return sd
+}
+
+type SchemaDescriptor struct {
+	Directives      map[string]*ast.DirectiveDefinition
+	FileDescriptors []*desc.FileDescriptor
+
+	files []*desc.FileDescriptor
+
+	query        *RootDefinition
+	mutation     *RootDefinition
+	subscription *RootDefinition
+
+	objects []*ObjectDescriptor
+
+	reservedNames  map[string]desc.Descriptor
+	createdObjects map[createdObjectKey]*ObjectDescriptor
+
+	generateServiceDescriptors bool
+
+	goRef GoRef
+}
+
+type createdObjectKey struct {
+	desc  desc.Descriptor
+	input bool
+}
+
+func (s *SchemaDescriptor) AsGraphql() *ast.Schema {
+	queryDef := *s.GetQuery().Definition
+	mutationDef := *s.GetMutation().Definition
+	subscriptionsDef := *s.GetSubscription().Definition
+	schema := &ast.Schema{Types: map[string]*ast.Definition{}, Directives: s.Directives}
+	schema.Query = &queryDef
+	schema.Types["Query"] = &queryDef
+	if s.query.methods == nil {
+		schema.Query.Fields = append(schema.Query.Fields, &ast.FieldDefinition{
+			Name: "dummy",
+			Type: ast.NamedType("Boolean", &ast.Position{}),
+		})
+	}
+	if s.mutation.methods != nil {
+		schema.Mutation = &mutationDef
+		schema.Types["Mutation"] = &mutationDef
+	}
+	if s.subscription.methods != nil {
+		schema.Subscription = &subscriptionsDef
+		schema.Types["Subscription"] = &subscriptionsDef
+	}
+
+	for _, o := range s.objects {
+		def := o.AsGraphql()
+		schema.Types[def.Name] = def
+	}
+	return schema
+}
+
+func (s *SchemaDescriptor) Objects() []*ObjectDescriptor {
+	return s.objects
+}
+
+func (s *SchemaDescriptor) GetMutation() *RootDefinition {
+	if s.mutation == nil {
+		s.mutation = NewRootDefinition(Mutation, s)
+	}
+	return s.mutation
+}
+
+func (s *SchemaDescriptor) GetSubscription() *RootDefinition {
+	if s.subscription == nil {
+		s.subscription = NewRootDefinition(Subscription, s)
+	}
+	return s.subscription
+}
+
+func (s *SchemaDescriptor) GetQuery() *RootDefinition {
+	if s.query == nil {
+		s.query = NewRootDefinition(Query, s)
+	}
+
+	return s.query
+}
+
+// make name be unique
+// just create a map and register every name
+func (s *SchemaDescriptor) uniqueName(d desc.Descriptor, input bool) (name string) {
+	var collisionPrefix string
+	var suffix string
+	if _, ok := d.(*desc.MessageDescriptor); input && ok {
+		suffix = inputSuffix
+	}
+	name = strings.Title(CamelCaseSlice(strings.Split(strings.TrimPrefix(d.GetFullyQualifiedName(), d.GetFile().GetPackage()+packageSep), packageSep)) + suffix)
+
+	if _, ok := d.(*desc.FieldDescriptor); ok {
+		collisionPrefix = fieldPrefix
+		name = CamelCaseSlice(strings.Split(strings.Trim(d.GetParent().GetName()+packageSep+strings.Title(d.GetName()), packageSep), packageSep))
+	} else {
+		collisionPrefix = CamelCaseSlice(strings.Split(d.GetFile().GetPackage(), packageSep))
+	}
+
+	originalName := name
+	for uniqueSuffix := 0; ; uniqueSuffix++ {
+		d2, ok := s.reservedNames[name]
+		if !ok {
+			break
+		}
+		if d2 == d {
+			return name
+		}
+		if uniqueSuffix == 0 {
+			name = collisionPrefix + typeSep + originalName
+			continue
+		}
+		name = collisionPrefix + typeSep + originalName + strconv.Itoa(uniqueSuffix)
+	}
+
+	s.reservedNames[name] = d
+	return
+}
+
+func (s *SchemaDescriptor) CreateObjects(d desc.Descriptor, input bool) (obj *ObjectDescriptor, err error) {
+	// the case if trying to resolve a primitive as a object. In this case we just return nil
+	if d == nil {
+		return
+	}
+	if obj, ok := s.createdObjects[createdObjectKey{d, input}]; ok {
+		return obj, nil
+	}
+
+	obj = &ObjectDescriptor{
+		Definition: &ast.Definition{
+			Description: getDescription(d),
+			Name:        s.uniqueName(d, input),
+			Position:    &ast.Position{},
+		},
+		Descriptor: d,
+	}
+
+	s.createdObjects[createdObjectKey{d, input}] = obj
+
+	switch dd := d.(type) {
+	case *desc.MessageDescriptor:
+		if IsEmpty(dd) {
+			return obj, nil
+		}
+		if IsAny(dd) {
+			// TODO find a better way to handle any types
+			delete(s.createdObjects, createdObjectKey{d, input})
+			any := s.createScalar(s.uniqueName(dd, false), anyTypeDescription)
+			return any, nil
+		}
+
+		kind := ast.Object
+		if input {
+			kind = ast.InputObject
+		}
+		fields := FieldDescriptorList{}
+		outputOneofRegistrar := map[*desc.OneOfDescriptor]struct{}{}
+
+		for _, df := range dd.GetFields() {
+			fieldOpts := GraphqlFieldOptions(df.AsFieldDescriptorProto().GetOptions())
+			if fieldOpts != nil && fieldOpts.Ignore != nil && *fieldOpts.Ignore {
+				continue
+			}
+			var fieldDirective []*ast.Directive
+			if df.GetType() == descriptor.FieldDescriptorProto_TYPE_MESSAGE && IsEmpty(df.GetMessageType()) {
+				continue
+			}
+
+			// Internally `optional` fields are represented as a oneof, and as such should be skipped.
+			if oneof := df.GetOneOf(); oneof != nil && !df.AsFieldDescriptorProto().GetProto3Optional() {
+				opts := GraphqlOneofOptions(oneof.AsOneofDescriptorProto().GetOptions())
+				if opts.GetIgnore() {
+					continue
+				}
+				if !input {
+					if _, ok := outputOneofRegistrar[oneof]; ok {
+						continue
+					}
+					outputOneofRegistrar[oneof] = struct{}{}
+					field, err := s.createUnion(oneof)
+					if err != nil {
+						return nil, err
+					}
+					fields = append(fields, field)
+					continue
+				}
+
+				// create oneofs as directives for input objects
+				directive := &ast.DirectiveDefinition{
+					Description: getDescription(oneof),
+					Name:        s.uniqueName(oneof, input),
+					Locations:   []ast.DirectiveLocation{ast.LocationInputFieldDefinition},
+					Position:    &ast.Position{Src: &ast.Source{}},
+				}
+				s.Directives[directive.Name] = directive
+				fieldDirective = append(fieldDirective, &ast.Directive{
+					Name:     directive.Name,
+					Position: &ast.Position{Src: &ast.Source{}},
+					// ParentDefinition: obj.Definition, TODO
+					Definition: directive,
+					Location:   ast.LocationInputFieldDefinition,
+				})
+			}
+
+			fieldObj, err := s.CreateObjects(resolveFieldType(df), input)
+			if err != nil {
+				return nil, err
+			}
+			if fieldObj == nil && df.GetMessageType() != nil {
+				continue
+			}
+			f, err := s.createField(df, fieldObj)
+			if err != nil {
+				return nil, err
+			}
+			f.Directives = append(f.Directives, fieldDirective...)
+			fields = append(fields, f)
+		}
+
+		obj.Definition.Fields = fields.AsGraphql()
+		obj.Definition.Kind = kind
+		obj.fields = fields
+	case *desc.EnumDescriptor:
+		obj.Definition.Kind = ast.Enum
+		obj.Definition.EnumValues = enumValues(dd.GetValues())
+	default:
+		panic(fmt.Sprintf("received unexpected value %v of type %T", dd, dd))
+	}
+
+	s.objects = append(s.objects, obj)
+	return obj, nil
+}
+
+func resolveFieldType(field *desc.FieldDescriptor) desc.Descriptor {
+	msgType := field.GetMessageType()
+	enumType := field.GetEnumType()
+	if msgType != nil {
+		return msgType
+	}
+	if enumType != nil {
+		return enumType
+	}
+	return nil
+}
+
+func enumValues(evals []*desc.EnumValueDescriptor) (vlist ast.EnumValueList) {
+	for _, eval := range evals {
+		vlist = append(vlist, &ast.EnumValueDefinition{
+			Description: getDescription(eval),
+			Name:        eval.GetName(),
+			Position:    &ast.Position{},
+		})
+	}
+
+	return vlist
+}
+
+type FieldDescriptorList []*FieldDescriptor
+
+func (fl FieldDescriptorList) AsGraphql() (dl []*ast.FieldDefinition) {
+	for _, f := range fl {
+		dl = append(dl, f.FieldDefinition)
+	}
+	return dl
+}
+
+type FieldDescriptor struct {
+	*ast.FieldDefinition
+	*desc.FieldDescriptor
+
+	typ *ObjectDescriptor
+}
+
+func (f *FieldDescriptor) GetType() *ObjectDescriptor {
+	return f.typ
+}
+
+type MethodDescriptor struct {
+	*desc.ServiceDescriptor
+	*desc.MethodDescriptor
+
+	*ast.FieldDefinition
+
+	input  *ObjectDescriptor
+	output *ObjectDescriptor
+}
+
+func (m *MethodDescriptor) AsGraphql() *ast.FieldDefinition {
+	return m.FieldDefinition
+}
+
+func (m *MethodDescriptor) GetInput() *ObjectDescriptor {
+	return m.input
+}
+
+func (m *MethodDescriptor) GetOutput() *ObjectDescriptor {
+	return m.output
+}
+
+type RootDefinition struct {
+	*ast.Definition
+
+	Parent *SchemaDescriptor
+
+	methods       []*MethodDescriptor
+	reservedNames map[string]ServiceAndMethod
+}
+
+type ServiceAndMethod struct {
+	svc *descriptor.ServiceDescriptorProto
+	rpc *descriptor.MethodDescriptorProto
+}
+
+func (r *RootDefinition) UniqueName(svc *descriptor.ServiceDescriptorProto, rpc *descriptor.MethodDescriptorProto) (name string) {
+	rpcOpts := GraphqlMethodOptions(rpc.GetOptions())
+	svcOpts := GraphqlServiceOptions(svc.GetOptions())
+	if rpcOpts != nil && rpcOpts.Name != nil {
+		name = *rpcOpts.Name
+	} else if svcOpts != nil && svcOpts.Name != nil {
+		if *svcOpts.Name == "" {
+			name = ToLowerFirst(rpc.GetName())
+		} else {
+			name = *svcOpts.Name + strings.Title(rpc.GetName())
+		}
+	} else {
+		name = ToLowerFirst(svc.GetName()) + strings.Title(rpc.GetName())
+	}
+
+	originalName := name
+	for uniqueSuffix := 0; ; uniqueSuffix++ {
+		snm, ok := r.reservedNames[name]
+		if !ok {
+			break
+		}
+		if svc == snm.svc && snm.rpc == rpc {
+			return name
+		}
+		name = originalName + strconv.Itoa(uniqueSuffix)
+	}
+
+	r.reservedNames[name] = ServiceAndMethod{svc, rpc}
+	return
+}
+
+func (r *RootDefinition) Methods() []*MethodDescriptor {
+	return r.methods
+}
+
+func (r *RootDefinition) addMethod(svc *desc.ServiceDescriptor, rpc *desc.MethodDescriptor, in, out *ObjectDescriptor) {
+	var args ast.ArgumentDefinitionList
+
+	if in != nil && (in.Descriptor != nil && !IsEmpty(in.Descriptor.(*desc.MessageDescriptor)) || in.Definition.Kind == ast.Scalar) {
+		args = append(args, &ast.ArgumentDefinition{
+			Name:     "in",
+			Type:     ast.NamedType(in.Name, &ast.Position{}),
+			Position: &ast.Position{},
+		})
+	}
+	objType := ast.NamedType("Boolean", &ast.Position{})
+	if out != nil && (out.Descriptor != nil && !IsEmpty(out.Descriptor.(*desc.MessageDescriptor)) || in.Definition.Kind == ast.Scalar) {
+		objType = ast.NamedType(out.Name, &ast.Position{})
+	}
+
+	svcDir := &ast.DirectiveDefinition{
+		Description: getDescription(svc),
+		Name:        svc.GetName(),
+		Locations:   []ast.DirectiveLocation{ast.LocationFieldDefinition},
+		Position:    &ast.Position{Src: &ast.Source{}},
+	}
+	r.Parent.Directives[svcDir.Name] = svcDir
+
+	m := &MethodDescriptor{
+		ServiceDescriptor: svc,
+		MethodDescriptor:  rpc,
+		FieldDefinition: &ast.FieldDefinition{
+			Description: getDescription(rpc),
+			Name:        r.UniqueName(svc.AsServiceDescriptorProto(), rpc.AsMethodDescriptorProto()),
+			Arguments:   args,
+			Type:        objType,
+			Position:    &ast.Position{},
+		},
+		input:  in,
+		output: out,
+	}
+	if r.Parent.generateServiceDescriptors {
+		m.Directives = []*ast.Directive{{
+			Name:       svcDir.Name,
+			Position:   &ast.Position{},
+			Definition: svcDir,
+			Location:   svcDir.Locations[0],
+		}}
+	}
+
+	r.methods = append(r.methods, m)
+	// TODO maybe not do it here?
+	r.Definition.Fields = append(r.Definition.Fields, m.FieldDefinition)
+}
+
+type rootName string
+
+const (
+	Mutation     rootName = "Mutation"
+	Query        rootName = "Query"
+	Subscription rootName = "Subscription"
+)
+
+func NewRootDefinition(name rootName, parent *SchemaDescriptor) *RootDefinition {
+	return &RootDefinition{Definition: &ast.Definition{
+		Kind:     ast.Object,
+		Name:     string(name),
+		Position: &ast.Position{},
+	}, Parent: parent, reservedNames: map[string]ServiceAndMethod{}}
+}
+
+func getDescription(descs ...desc.Descriptor) string {
+	var description []string
+	for _, d := range descs {
+		info := d.GetSourceInfo()
+		if info == nil {
+			continue
+		}
+		if info.LeadingComments != nil {
+			description = append(description, *info.LeadingComments)
+		}
+		if info.TrailingComments != nil {
+			description = append(description, *info.TrailingComments)
+		}
+	}
+
+	return strings.Join(description, "\n")
+}
+
+func (s *SchemaDescriptor) createField(field *desc.FieldDescriptor, obj *ObjectDescriptor) (_ *FieldDescriptor, err error) {
+	fieldAst := &ast.FieldDefinition{
+		Description: getDescription(field),
+		Name:        ToLowerFirst(CamelCase(field.GetName())),
+		Type:        &ast.Type{Position: &ast.Position{}},
+		Position:    &ast.Position{},
+	}
+	fieldOpts := GraphqlFieldOptions(field.AsFieldDescriptorProto().GetOptions())
+	if fieldOpts != nil && fieldOpts.Name != nil {
+		fieldAst.Name = *fieldOpts.Name
+		directive := &ast.DirectiveDefinition{
+			Name: goFieldDirective,
+			Arguments: []*ast.ArgumentDefinition{{
+				Name:     "forceResolver",
+				Type:     ast.NamedType("Boolean", &ast.Position{}),
+				Position: &ast.Position{},
+			}, {
+				Name:     "name",
+				Type:     ast.NamedType("String", &ast.Position{}),
+				Position: &ast.Position{},
+			}},
+			Locations: []ast.DirectiveLocation{ast.LocationInputFieldDefinition, ast.LocationFieldDefinition},
+			Position:  &ast.Position{Src: &ast.Source{}},
+		}
+		s.Directives[directive.Name] = directive
+		if s.goRef != nil {
+			fieldAst.Directives = []*ast.Directive{{
+				Name: directive.Name,
+				Arguments: []*ast.Argument{{
+					Name: "name",
+					Value: &ast.Value{
+						Raw:      s.goRef.FindGoField(field.GetFullyQualifiedName()).GoName,
+						Kind:     ast.StringValue,
+						Position: &ast.Position{},
+					},
+					Position: &ast.Position{},
+				}},
+				Position: &ast.Position{},
+				// ParentDefinition: nil, TODO
+				Definition: directive,
+			}}
+		}
+	}
+	switch field.GetType() {
+	case descriptor.FieldDescriptorProto_TYPE_DOUBLE,
+		descriptor.FieldDescriptorProto_TYPE_FLOAT:
+		fieldAst.Type.NamedType = ScalarFloat
+
+	case descriptor.FieldDescriptorProto_TYPE_BYTES:
+		scalar := s.createScalar(scalarBytes, "")
+		fieldAst.Type.NamedType = scalar.Name
+
+	case descriptor.FieldDescriptorProto_TYPE_INT64,
+		descriptor.FieldDescriptorProto_TYPE_SINT64,
+		descriptor.FieldDescriptorProto_TYPE_SFIXED64,
+		descriptor.FieldDescriptorProto_TYPE_INT32,
+		descriptor.FieldDescriptorProto_TYPE_SINT32,
+		descriptor.FieldDescriptorProto_TYPE_SFIXED32,
+		descriptor.FieldDescriptorProto_TYPE_UINT32,
+		descriptor.FieldDescriptorProto_TYPE_FIXED32,
+		descriptor.FieldDescriptorProto_TYPE_UINT64,
+		descriptor.FieldDescriptorProto_TYPE_FIXED64:
+		fieldAst.Type.NamedType = ScalarInt
+
+	case descriptor.FieldDescriptorProto_TYPE_BOOL:
+		fieldAst.Type.NamedType = ScalarBoolean
+
+	case descriptor.FieldDescriptorProto_TYPE_STRING:
+		fieldAst.Type.NamedType = ScalarString
+
+	case descriptor.FieldDescriptorProto_TYPE_GROUP:
+		return nil, fmt.Errorf("proto2 groups are not supported please use proto3 syntax")
+
+	case descriptor.FieldDescriptorProto_TYPE_ENUM:
+		fieldAst.Type.NamedType = obj.Name
+
+	case descriptor.FieldDescriptorProto_TYPE_MESSAGE:
+		fieldAst.Type.NamedType = obj.Name
+
+	default:
+		panic("unknown proto field type")
+	}
+
+	if isRepeated(field) {
+		fieldAst.Type = ast.ListType(fieldAst.Type, &ast.Position{})
+		fieldAst.Type.Elem.NonNull = true
+	}
+	if isRequired(field) {
+		fieldAst.Type.NonNull = true
+	}
+
+	return &FieldDescriptor{
+		FieldDefinition: fieldAst,
+		FieldDescriptor: field,
+		typ:             obj,
+	}, nil
+}
+
+func (s *SchemaDescriptor) createScalar(name string, description string) *ObjectDescriptor {
+	obj := &ObjectDescriptor{
+		Definition: &ast.Definition{
+			Kind:        ast.Scalar,
+			Description: description,
+			Name:        name,
+			Position:    &ast.Position{},
+		},
+	}
+	s.objects = append(s.objects, obj)
+	return obj
+}
+
+func (s *SchemaDescriptor) createUnion(oneof *desc.OneOfDescriptor) (*FieldDescriptor, error) {
+	var types []string
+	var objTypes []*ObjectDescriptor
+	for _, choice := range oneof.GetChoices() {
+		obj, err := s.CreateObjects(resolveFieldType(choice), false)
+		if err != nil {
+			return nil, err
+		}
+		f, err := s.createField(choice, obj)
+		if err != nil {
+			return nil, err
+		}
+
+		obj = &ObjectDescriptor{
+			Definition: &ast.Definition{
+				Kind:        ast.Object,
+				Description: getDescription(f),
+				Name:        s.uniqueName(choice, false),
+				Fields:      ast.FieldList{f.FieldDefinition},
+				Position:    &ast.Position{},
+			},
+			Descriptor: f,
+			fields:     []*FieldDescriptor{f},
+			fieldNames: map[string]*FieldDescriptor{},
+		}
+		s.objects = append(s.objects, obj)
+		types = append(types, obj.Name)
+		objTypes = append(objTypes, obj)
+	}
+	obj := &ObjectDescriptor{
+		Definition: &ast.Definition{
+			Kind:        ast.Union,
+			Description: getDescription(oneof),
+			Name:        s.uniqueName(oneof, false),
+			Types:       types,
+			Position:    &ast.Position{},
+		},
+		Descriptor: oneof,
+		types:      objTypes,
+	}
+	s.objects = append(s.objects, obj)
+	name := ToLowerFirst(CamelCase(oneof.GetName()))
+	opts := GraphqlOneofOptions(oneof.AsOneofDescriptorProto().GetOptions())
+	if opts.GetName() != "" {
+		name = opts.GetName()
+	}
+	return &FieldDescriptor{
+		FieldDefinition: &ast.FieldDefinition{
+			Description: getDescription(oneof),
+			Name:        name,
+			Type:        ast.NamedType(obj.Name, &ast.Position{}),
+			Position:    &ast.Position{},
+		},
+		FieldDescriptor: nil,
+		typ:             obj,
+	}, nil
+}
+
+func isRepeated(field *desc.FieldDescriptor) bool {
+	return field.GetLabel() == descriptor.FieldDescriptorProto_LABEL_REPEATED
+}
+
+func isRequired(field *desc.FieldDescriptor) bool {
+	if v := GraphqlFieldOptions(field.AsFieldDescriptorProto().GetOptions()); v != nil {
+		return v.GetRequired()
+	}
+	return false
+}
+
+const (
+	ScalarInt     = "Int"
+	ScalarFloat   = "Float"
+	ScalarString  = "String"
+	ScalarBoolean = "Boolean"
+	ScalarID      = "ID"
+)
+
+var graphqlReservedNames = []string{"__Directive", "__Type", "__Field", "__EnumValue", "__InputValue", "__Schema", "Int", "Float", "String", "Boolean", "ID"}

graphql/goreference.go

@@ -0,0 +1,30 @@
+package generator
+
+import (
+	"google.golang.org/protobuf/compiler/protogen"
+)
+
+type GoRef interface {
+	FindGoField(field string) *protogen.Field
+}
+
+func NewGoRef(p *protogen.Plugin) (GoRef, error) {
+	return goRef{p}, nil
+}
+
+type goRef struct {
+	*protogen.Plugin
+}
+
+func (g goRef) FindGoField(field string) *protogen.Field {
+	for _, file := range g.Files {
+		for _, msg := range file.Messages {
+			for _, f := range msg.Fields {
+				if string(f.Desc.FullName()) == field {
+					return f
+				}
+			}
+		}
+	}
+	return nil
+}

graphql/registry.go

@@ -0,0 +1,139 @@
+package generator
+
+import (
+	"sync"
+
+	"github.com/jhump/protoreflect/desc"
+	"github.com/vektah/gqlparser/v2/ast"
+)
+
+type Registry interface {
+	FindMethodByName(op ast.Operation, name string) *desc.MethodDescriptor
+	FindObjectByName(name string) *desc.MessageDescriptor
+
+	// FindObjectByFullyQualifiedName TODO maybe find a better way to get ast definition
+	FindObjectByFullyQualifiedName(fqn string) (*desc.MessageDescriptor, *ast.Definition)
+
+	// FindFieldByName given the proto Descriptor and the graphql field name get the the proto FieldDescriptor
+	FindFieldByName(msg desc.Descriptor, name string) *desc.FieldDescriptor
+
+	// FindUnionFieldByMessageFQNAndName given the proto Descriptor and the graphql field name get the the proto FieldDescriptor
+	FindUnionFieldByMessageFQNAndName(fqn, name string) *desc.FieldDescriptor
+
+	FindGraphqlFieldByProtoField(msg *ast.Definition, name string) *ast.FieldDefinition
+}
+
+func NewRegistry(files SchemaDescriptorList) Registry {
+	v := &repository{
+		mu:                       &sync.RWMutex{},
+		methodsByName:            map[ast.Operation]map[string]*desc.MethodDescriptor{},
+		objectsByName:            map[string]*desc.MessageDescriptor{},
+		objectsByFQN:             map[string]*ObjectDescriptor{},
+		graphqlFieldsByName:      map[desc.Descriptor]map[string]*desc.FieldDescriptor{},
+		graphqlUnionFieldsByName: map[string]map[string]*desc.FieldDescriptor{},
+		protoFieldsByName:        map[*ast.Definition]map[string]*ast.FieldDefinition{},
+	}
+	for _, f := range files {
+		v.methodsByName[ast.Mutation] = map[string]*desc.MethodDescriptor{}
+		for _, m := range f.GetMutation().Methods() {
+			v.methodsByName[ast.Mutation][m.Name] = m.MethodDescriptor
+		}
+		v.methodsByName[ast.Query] = map[string]*desc.MethodDescriptor{}
+		for _, m := range f.GetQuery().Methods() {
+			v.methodsByName[ast.Query][m.Name] = m.MethodDescriptor
+		}
+		v.methodsByName[ast.Subscription] = map[string]*desc.MethodDescriptor{}
+		for _, m := range f.GetSubscription().Methods() {
+			v.methodsByName[ast.Subscription][m.Name] = m.MethodDescriptor
+		}
+	}
+	for _, f := range files {
+		for _, m := range f.Objects() {
+			switch m.Kind {
+			case ast.Union:
+				fqn := m.Descriptor.GetParent().GetFullyQualifiedName()
+				if _, ok := v.graphqlUnionFieldsByName[fqn]; !ok {
+					v.graphqlUnionFieldsByName[fqn] = map[string]*desc.FieldDescriptor{}
+				}
+				for _, tt := range m.GetTypes() {
+					for _, f := range tt.GetFields() {
+						v.graphqlUnionFieldsByName[fqn][f.Name] = f.FieldDescriptor
+					}
+				}
+			case ast.Object:
+				v.protoFieldsByName[m.Definition] = map[string]*ast.FieldDefinition{}
+				for _, f := range m.GetFields() {
+					if f.FieldDescriptor != nil {
+						v.protoFieldsByName[m.Definition][f.FieldDescriptor.GetName()] = f.FieldDefinition
+					}
+				}
+			case ast.InputObject:
+				v.graphqlFieldsByName[m.Descriptor] = map[string]*desc.FieldDescriptor{}
+				for _, f := range m.GetFields() {
+					v.graphqlFieldsByName[m.Descriptor][f.Name] = f.FieldDescriptor
+				}
+			}
+			switch msgDesc := m.Descriptor.(type) {
+			case *desc.MessageDescriptor:
+				v.objectsByFQN[m.GetFullyQualifiedName()] = m
+				v.objectsByName[m.Name] = msgDesc
+			}
+		}
+	}
+	return v
+}
+
+type repository struct {
+	files SchemaDescriptorList
+	mu    *sync.RWMutex
+
+	methodsByName            map[ast.Operation]map[string]*desc.MethodDescriptor
+	objectsByName            map[string]*desc.MessageDescriptor
+	objectsByFQN             map[string]*ObjectDescriptor
+	graphqlFieldsByName      map[desc.Descriptor]map[string]*desc.FieldDescriptor
+	protoFieldsByName        map[*ast.Definition]map[string]*ast.FieldDefinition
+	graphqlUnionFieldsByName map[string]map[string]*desc.FieldDescriptor
+}
+
+func (r repository) FindMethodByName(op ast.Operation, name string) *desc.MethodDescriptor {
+	r.mu.RLock()
+	defer r.mu.RUnlock()
+	m, _ := r.methodsByName[op][name]
+	return m
+}
+
+func (r repository) FindObjectByName(name string) *desc.MessageDescriptor {
+	r.mu.RLock()
+	defer r.mu.RUnlock()
+	o, _ := r.objectsByName[name]
+	return o
+}
+
+func (r repository) FindObjectByFullyQualifiedName(fqn string) (*desc.MessageDescriptor, *ast.Definition) {
+	r.mu.RLock()
+	defer r.mu.RUnlock()
+	o, _ := r.objectsByFQN[fqn]
+	msg, _ := o.Descriptor.(*desc.MessageDescriptor)
+	return msg, o.Definition
+}
+
+func (r repository) FindFieldByName(msg desc.Descriptor, name string) *desc.FieldDescriptor {
+	r.mu.RLock()
+	defer r.mu.RUnlock()
+	f, _ := r.graphqlFieldsByName[msg][name]
+	return f
+}
+
+func (r repository) FindUnionFieldByMessageFQNAndName(fqn, name string) *desc.FieldDescriptor {
+	r.mu.RLock()
+	defer r.mu.RUnlock()
+	f, _ := r.graphqlUnionFieldsByName[fqn][name]
+	return f
+}
+
+func (r repository) FindGraphqlFieldByProtoField(msg *ast.Definition, name string) *ast.FieldDefinition {
+	r.mu.RLock()
+	defer r.mu.RUnlock()
+	f, _ := r.protoFieldsByName[msg][name]
+	return f
+}

graphql/utils.go

@@ -0,0 +1,291 @@
+package generator
+
+import (
+	"strings"
+	"unicode"
+	"unicode/utf8"
+
+	"github.com/jhump/protoreflect/desc"
+	"google.golang.org/protobuf/proto"
+	descriptor "google.golang.org/protobuf/types/descriptorpb"
+	"google.golang.org/protobuf/types/pluginpb"
+
+	gqlpb "go.unistack.org/micro-proto/v3/graphql"
+)
+
+func GraphqlMethodOptions(opts proto.Message) *gqlpb.Rpc {
+	if opts != nil {
+		v := proto.GetExtension(opts, gqlpb.E_Rpc)
+		if v != nil {
+			return v.(*gqlpb.Rpc)
+		}
+	}
+	return nil
+}
+
+func GraphqlServiceOptions(opts proto.Message) *gqlpb.Svc {
+	if opts != nil {
+		v := proto.GetExtension(opts, gqlpb.E_Svc)
+		if v != nil {
+			return v.(*gqlpb.Svc)
+		}
+	}
+	return nil
+}
+
+func GraphqlFieldOptions(opts proto.Message) *gqlpb.Field {
+	if opts != nil {
+		v := proto.GetExtension(opts, gqlpb.E_Field)
+		if v != nil && v.(*gqlpb.Field) != nil {
+			return v.(*gqlpb.Field)
+		}
+	}
+	return nil
+}
+
+func GraphqlOneofOptions(opts proto.Message) *gqlpb.Oneof {
+	if opts != nil {
+		v := proto.GetExtension(opts, gqlpb.E_Oneof)
+		if v != nil && v.(*gqlpb.Oneof) != nil {
+			return v.(*gqlpb.Oneof)
+		}
+	}
+	return nil
+}
+
+// GoCamelCase camel-cases a protobuf name for use as a Go identifier.
+//
+// If there is an interior underscore followed by a lower case letter,
+// drop the underscore and convert the letter to upper case.
+func GoCamelCase(s string) string {
+	// Invariant: if the next letter is lower case, it must be converted
+	// to upper case.
+	// That is, we process a word at a time, where words are marked by _ or
+	// upper case letter. Digits are treated as words.
+	var b []byte
+	for i := 0; i < len(s); i++ {
+		c := s[i]
+		switch {
+		case c == '.' && i+1 < len(s) && isASCIILower(s[i+1]):
+			// Skip over '.' in ".{{lowercase}}".
+		case c == '.':
+			b = append(b, '_') // convert '.' to '_'
+		case c == '_' && (i == 0 || s[i-1] == '.'):
+			// Convert initial '_' to ensure we start with a capital letter.
+			// Do the same for '_' after '.' to match historic behavior.
+			b = append(b, 'X') // convert '_' to 'X'
+		case c == '_' && i+1 < len(s) && isASCIILower(s[i+1]):
+			// Skip over '_' in "_{{lowercase}}".
+		case isASCIIDigit(c):
+			b = append(b, c)
+		default:
+			// Assume we have a letter now - if not, it's a bogus identifier.
+			// The next word is a sequence of characters that must start upper case.
+			if isASCIILower(c) {
+				c -= 'a' - 'A' // convert lowercase to uppercase
+			}
+			b = append(b, c)
+
+			// Accept lower case sequence that follows.
+			for ; i+1 < len(s) && isASCIILower(s[i+1]); i++ {
+				b = append(b, s[i+1])
+			}
+		}
+	}
+	return string(b)
+}
+
+func GetRequestType(rpcOpts *gqlpb.Rpc, svcOpts *gqlpb.Svc) gqlpb.Type {
+	if rpcOpts != nil && rpcOpts.Type != nil {
+		return *rpcOpts.Type
+	}
+	if svcOpts != nil && svcOpts.Type != nil {
+		return *svcOpts.Type
+	}
+	return gqlpb.Type_DEFAULT
+}
+
+func CreateDescriptorsFromProto(req *pluginpb.CodeGeneratorRequest) (descs []*desc.FileDescriptor, err error) {
+	dd, err := desc.CreateFileDescriptorsFromSet(&descriptor.FileDescriptorSet{
+		File: req.GetProtoFile(),
+	})
+	if err != nil {
+		return nil, err
+	}
+	for _, d := range dd {
+		for _, filename := range req.FileToGenerate {
+			if filename == d.GetName() {
+				descs = append(descs, d)
+			}
+		}
+	}
+	return
+}
+
+func ResolveProtoFilesRecursively(descs []*desc.FileDescriptor) (files FileDescriptors) {
+	for _, f := range descs {
+		files = append(files, ResolveProtoFilesRecursively(f.GetDependencies())...)
+		files = append(files, f)
+	}
+
+	return files
+}
+
+type FileDescriptors []*desc.FileDescriptor
+
+func (ds FileDescriptors) AsFileDescriptorProto() (files []*descriptor.FileDescriptorProto) {
+	for _, d := range ds {
+		files = append(files, d.AsFileDescriptorProto())
+	}
+	return
+}
+
+// Split splits the camelcase word and returns a list of words. It also
+// supports digits. Both lower camel case and upper camel case are supported.
+// For more info please check: http://en.wikipedia.org/wiki/CamelCase
+//
+// Examples
+//
+//	"" =>                     [""]
+//	"lowercase" =>            ["lowercase"]
+//	"Class" =>                ["Class"]
+//	"MyClass" =>              ["My", "Class"]
+//	"MyC" =>                  ["My", "C"]
+//	"HTML" =>                 ["HTML"]
+//	"PDFLoader" =>            ["PDF", "Loader"]
+//	"AString" =>              ["A", "String"]
+//	"SimpleXMLParser" =>      ["Simple", "XML", "Parser"]
+//	"vimRPCPlugin" =>         ["vim", "RPC", "Plugin"]
+//	"GL11Version" =>          ["GL", "11", "Version"]
+//	"99Bottles" =>            ["99", "Bottles"]
+//	"May5" =>                 ["May", "5"]
+//	"BFG9000" =>              ["BFG", "9000"]
+//	"BöseÜberraschung" =>     ["Böse", "Überraschung"]
+//	"Two  spaces" =>          ["Two", "  ", "spaces"]
+//	"BadUTF8\xe2\xe2\xa1" =>  ["BadUTF8\xe2\xe2\xa1"]
+//
+// Splitting rules
+//
+//  1. If string is not valid UTF-8, return it without splitting as
+//     single item array.
+//  2. Assign all unicode characters into one of 4 sets: lower case
+//     letters, upper case letters, numbers, and all other characters.
+//  3. Iterate through characters of string, introducing splits
+//     between adjacent characters that belong to different sets.
+//  4. Iterate through array of split strings, and if a given string
+//     is upper case:
+//     if subsequent string is lower case:
+//     move last character of upper case string to beginning of
+//     lower case string
+func SplitCamelCase(src string) (entries []string) {
+	// don't split invalid utf8
+	if !utf8.ValidString(src) {
+		return []string{src}
+	}
+	entries = []string{}
+	var runes [][]rune
+	lastClass := 0
+	class := 0
+	// split into fields based on class of unicode character
+	for _, r := range src {
+		switch true {
+		case unicode.IsLower(r):
+			class = 1
+		case unicode.IsUpper(r):
+			class = 2
+		case unicode.IsDigit(r):
+			class = 3
+		default:
+			class = 4
+		}
+		if class == lastClass {
+			runes[len(runes)-1] = append(runes[len(runes)-1], r)
+		} else {
+			runes = append(runes, []rune{r})
+		}
+		lastClass = class
+	}
+	// handle upper case -> lower case sequences, e.g.
+	// "PDFL", "oader" -> "PDF", "Loader"
+	for i := 0; i < len(runes)-1; i++ {
+		if unicode.IsUpper(runes[i][0]) && unicode.IsLower(runes[i+1][0]) {
+			runes[i+1] = append([]rune{runes[i][len(runes[i])-1]}, runes[i+1]...)
+			runes[i] = runes[i][:len(runes[i])-1]
+		}
+	}
+	// construct []string from results
+	for _, s := range runes {
+		if len(s) > 0 {
+			entries = append(entries, string(s))
+		}
+	}
+	return
+}
+
+// CamelCase returns the CamelCased name.
+// If there is an interior underscore followed by a lower case letter,
+// drop the underscore and convert the letter to upper case.
+// There is a remote possibility of this rewrite causing a name collision,
+// but it's so remote we're prepared to pretend it's nonexistent - since the
+// C++ generator lowercases names, it's extremely unlikely to have two fields
+// with different capitalizations.
+// In short, _my_field_name_2 becomes XMyFieldName_2.
+func CamelCase(s string) string {
+	if s == "" {
+		return ""
+	}
+	t := make([]byte, 0, 32)
+	i := 0
+	if s[0] == '_' {
+		// Need a capital letter; drop the '_'.
+		t = append(t, 'X')
+		i++
+	}
+	// Invariant: if the next letter is lower case, it must be converted
+	// to upper case.
+	// That is, we process a word at a time, where words are marked by _ or
+	// upper case letter. Digits are treated as words.
+	for ; i < len(s); i++ {
+		c := s[i]
+		if c == '_' && i+1 < len(s) && isASCIILower(s[i+1]) {
+			continue // Skip the underscore in s.
+		}
+		if isASCIIDigit(c) {
+			t = append(t, c)
+			continue
+		}
+		// Assume we have a letter now - if not, it's a bogus identifier.
+		// The next word is a sequence of characters that must start upper case.
+		if isASCIILower(c) {
+			c ^= ' ' // Make it a capital letter.
+		}
+		t = append(t, c) // Guaranteed not lower case.
+		// Accept lower case sequence that follows.
+		for i+1 < len(s) && isASCIILower(s[i+1]) {
+			i++
+			t = append(t, s[i])
+		}
+	}
+	return string(t)
+}
+
+// CamelCaseSlice is like CamelCase, but the argument is a slice of strings to
+// be joined with "_".
+func CamelCaseSlice(elem []string) string { return CamelCase(strings.Join(elem, "_")) }
+
+// Is c an ASCII lower-case letter?
+func isASCIILower(c byte) bool {
+	return 'a' <= c && c <= 'z'
+}
+
+// Is c an ASCII digit?
+func isASCIIDigit(c byte) bool {
+	return '0' <= c && c <= '9'
+}
+
+func ToLowerFirst(s string) string {
+	if len(s) > 0 {
+		return string(unicode.ToLower(rune(s[0]))) + s[1:]
+	}
+	return ""
+}

graphqls.go

@@ -0,0 +1,56 @@
+package main
+
+import (
+	"bytes"
+	"os"
+
+	"github.com/vektah/gqlparser/v2/formatter"
+	generator "go.unistack.org/protoc-gen-go-micro/v3/graphql"
+	"google.golang.org/protobuf/compiler/protogen"
+	"google.golang.org/protobuf/proto"
+	"google.golang.org/protobuf/types/pluginpb"
+)
+
+func (g *Generator) graphqlsGenerate(plugin *protogen.Plugin) error {
+	descs, err := generator.CreateDescriptorsFromProto(plugin.Request)
+	if err != nil {
+		return err
+	}
+
+	gqlDesc, err := generator.NewSchemas(descs, false, false, plugin)
+	if err != nil {
+		return err
+	}
+
+	var outFiles []*pluginpb.CodeGeneratorResponse_File
+
+	for _, schema := range gqlDesc {
+		buf := &bytes.Buffer{}
+		formatter.NewFormatter(buf).FormatSchema(schema.AsGraphql())
+
+		outFiles = append(outFiles, &pluginpb.CodeGeneratorResponse_File{
+			Name:    proto.String(g.graphqlFile),
+			Content: proto.String(buf.String()),
+		})
+	}
+
+	res := &pluginpb.CodeGeneratorResponse{
+		File:              outFiles,
+		SupportedFeatures: proto.Uint64(uint64(pluginpb.CodeGeneratorResponse_FEATURE_PROTO3_OPTIONAL)),
+	}
+
+	if err != nil {
+		res.Error = proto.String(err.Error())
+	}
+
+	out, err := proto.Marshal(res)
+	if err != nil {
+		return err
+	}
+
+	if _, err := os.Stdout.Write(out); err != nil {
+		return err
+	}
+
+	return nil
+}

http.go

@@ -1,21 +1,10 @@
 package main
 
 import (
-	"strings"
-
-	openapi_options "github.com/grpc-ecosystem/grpc-gateway/v2/protoc-gen-openapiv2/options"
 	"google.golang.org/protobuf/compiler/protogen"
-	"google.golang.org/protobuf/proto"
 )
 
-type Error struct {
-	packageName string
-	types       []string
-}
-
-func (g *Generator) httpGenerate(component string, plugin *protogen.Plugin) error {
-	errors := make(map[string]struct{})
-
+func (g *Generator) httpGenerate(component string, plugin *protogen.Plugin, genClient bool, genServer bool) error {
 	for _, file := range plugin.Files {
 		if !file.Generate {
 			continue
@@ -25,113 +14,42 @@ func (g *Generator) httpGenerate(component string, plugin *protogen.Plugin) erro
 		}
 
 		gname := file.GeneratedFilenamePrefix + "_micro_" + component + ".pb.go"
-		gfile := plugin.NewGeneratedFile(gname, ".")
+		path := file.GoImportPath
+		if g.standalone {
+			path = "."
+		}
+		gfile := plugin.NewGeneratedFile(gname, path)
 
-		gfile.P("// Code generated by protoc-gen-micro")
-		gfile.P("// source: ", *file.Proto.Name)
+		gfile.P("// Code generated by protoc-gen-go-micro. DO NOT EDIT.")
+		gfile.P("// protoc-gen-go-micro version: " + versionComment)
+		gfile.P("// source: ", file.Proto.GetName())
+		gfile.P()
 		gfile.P("package ", file.GoPackageName)
-
-		gfile.P()
-		gfile.P("import (")
-		gfile.P(`"context"`)
-		gfile.P()
-		gfile.P(`micro_api "github.com/unistack-org/micro/v3/api"`)
-		gfile.P(`micro_client_http "github.com/unistack-org/micro-client-http/v3"`)
-		gfile.P(`micro_client "github.com/unistack-org/micro/v3/client"`)
-		gfile.P(`micro_server "github.com/unistack-org/micro/v3/server"`)
-		gfile.P(")")
 		gfile.P()
 
-		gfile.P("// Reference imports to suppress errors if they are not otherwise used.")
-		gfile.P("var (")
-		gfile.P("_ ", "micro_api.Endpoint")
-		gfile.P("_ ", "context.Context")
-		gfile.P(" _ ", "micro_client.Option")
-		gfile.P(" _ ", "micro_server.Option")
-		gfile.P(")")
-		gfile.P()
+		gfile.Import(contextPackage)
+
+		if genClient {
+			gfile.Import(microClientPackage)
+			gfile.Import(microClientHttpPackage)
+		}
+		if genServer {
+			gfile.Import(microServerPackage)
+		}
 
 		for _, service := range file.Services {
-			generateServiceClient(gfile, service)
-			generateServiceClientMethods(gfile, service, true)
-			generateServiceServer(gfile, service)
-			generateServiceServerMethods(gfile, service)
-			generateServiceRegister(gfile, service)
-			if component == "http" {
-				for k, v := range getErrors(service) {
-					errors[k] = v
-				}
+			g.generateServiceEndpoints(gfile, service, component)
+			if genClient {
+				g.generateServiceClient(gfile, file, service)
+				g.generateServiceClientMethods(gfile, service, component)
+			}
+			if genServer {
+				g.generateServiceServer(gfile, file, service)
+				g.generateServiceServerMethods(gfile, service)
+				g.generateServiceRegister(gfile, file, service, component)
 			}
 		}
 	}
 
-	files := make(map[string]*Error)
-	for _, file := range plugin.Files {
-		if !file.Generate {
-			continue
-		}
-
-		err, ok := files[file.GeneratedFilenamePrefix]
-		if !ok {
-			err = &Error{packageName: string(file.GoPackageName)}
-		}
-		fok := false
-		for _, message := range file.Messages {
-			if _, ok := errors["."+string(message.Desc.FullName())]; ok {
-				fok = true
-				err.types = append(err.types, string(message.Desc.FullName()))
-			}
-		}
-		if fok {
-			files[file.GeneratedFilenamePrefix] = err
-		}
-	}
-
-	for file, err := range files {
-		gfile := plugin.NewGeneratedFile(file+"_micro_errors.pb.go", ".")
-		generateServiceErrors(gfile, err)
-	}
-
 	return nil
 }
-
-func getErrors(service *protogen.Service) map[string]struct{} {
-	errors := make(map[string]struct{})
-
-	for _, method := range service.Methods {
-		if method.Desc.Options() == nil {
-			continue
-		}
-		if !proto.HasExtension(method.Desc.Options(), openapi_options.E_Openapiv2Operation) {
-			continue
-		}
-
-		opts := proto.GetExtension(method.Desc.Options(), openapi_options.E_Openapiv2Operation)
-		if opts == nil {
-			continue
-		}
-
-		r := opts.(*openapi_options.Operation)
-		for _, response := range r.Responses {
-			if response.Schema == nil || response.Schema.JsonSchema == nil {
-				continue
-			}
-			errors[response.Schema.JsonSchema.Ref] = struct{}{}
-		}
-	}
-
-	return errors
-}
-
-func generateServiceErrors(gfile *protogen.GeneratedFile, err *Error) {
-	gfile.P("package ", err.packageName)
-	gfile.P("import (")
-	gfile.P(`"fmt"`)
-	gfile.P(")")
-	for _, typ := range err.types {
-		gfile.P("func (err *", typ[strings.LastIndex(typ, ".")+1:], ") Error() string {")
-		gfile.P(`return fmt.Sprintf("%#v", err)`)
-		gfile.P("}")
-		gfile.P()
-	}
-}

main.go

@@ -3,29 +3,36 @@ package main
 import (
 	"flag"
 	"fmt"
+	"os"
 	"strings"
 
 	"google.golang.org/protobuf/compiler/protogen"
+	"google.golang.org/protobuf/types/pluginpb"
 )
 
 var (
-	flags          flag.FlagSet
-	flagDebug      *bool
-	flagComponents *string
-	flagPaths      *string
-	flagModule     *string
+	flagSet           = flag.NewFlagSet(os.Args[0], flag.ExitOnError)
+	flagDebug         = flagSet.Bool("debug", false, "debug output")
+	flagStandalone    = flagSet.Bool("standalone", false, "generate file to standalone dir")
+	flagFieldaligment = flagSet.Bool("fieldaligment", false, "align struct fields in generated code")
+	flagComponents    = flagSet.String("components", "micro|rpc|http|client|server|openapiv3|graphql", "specify components to generate")
+	flagTagPath       = flagSet.String("tag_path", "", "tag rewriting dir")
+	flagOpenapiFile   = flagSet.String("openapi_file", "apidocs.swagger.yaml", "openapi file name")
+	flagGraphqlFile   = flagSet.String("graphql_file", "schema.graphqls", "graphql file name")
+	flagReflection    = flagSet.Bool("reflection", false, "enable server reflection support")
+	flagHelp          = flagSet.Bool("help", false, "display help message")
 )
 
-func init() {
-	flagDebug = flags.Bool("debug", false, "")
-	flagComponents = flags.String("components", "micro", "")
-	flagPaths = flag.String("paths", "", "")
-	flagModule = flag.String("module", "", "")
-}
-
 func main() {
 	opts := &protogen.Options{
-		ParamFunc: flags.Set,
+		ParamFunc: flagSet.Set,
+	}
+
+	_ = flagSet.Parse(os.Args[1:])
+
+	if *flagHelp {
+		flagSet.PrintDefaults()
+		return
 	}
 
 	g := &Generator{}
@@ -34,24 +41,77 @@ func main() {
 }
 
 type Generator struct {
+	components    string
+	standalone    bool
+	debug         bool
+	fieldaligment bool
+	tagPath       string
+	openapiFile   string
+	graphqlFile   string
+	reflection    bool
+	plugin        *protogen.Plugin
 }
 
 func (g *Generator) Generate(plugin *protogen.Plugin) error {
 	var err error
 
+	g.plugin = plugin
+	g.standalone = *flagStandalone
+	g.debug = *flagDebug
+	g.components = *flagComponents
+	g.fieldaligment = *flagFieldaligment
+	g.tagPath = *flagTagPath
+	g.openapiFile = *flagOpenapiFile
+	g.graphqlFile = *flagGraphqlFile
+	g.reflection = *flagReflection
+	plugin.SupportedFeatures = uint64(pluginpb.CodeGeneratorResponse_FEATURE_PROTO3_OPTIONAL)
+
+	genClient := true
+	genServer := true
+	var genNone bool
+
+	if strings.Contains(g.components, "server") {
+		genServer = true
+	}
+	if strings.Contains(g.components, "client") {
+		genClient = true
+	}
+	if strings.Contains(g.components, "none") {
+		genNone = true
+	}
+	if strings.Contains(g.components, "rpc") || strings.Contains(g.components, "http") {
+		if !genServer && !genClient && !genNone {
+			genServer = true
+			genClient = true
+		}
+	}
+
 	// Protoc passes a slice of File structs for us to process
-	for _, component := range strings.Split(*flagComponents, "|") {
+	for _, component := range strings.Split(g.components, "|") {
 		switch component {
+		case "server", "client":
+			continue
 		case "micro":
-			err = g.microGenerate(component, plugin)
+			err = g.microGenerate(component, plugin, genClient, genServer)
+			if err == nil {
+				err = g.writeErrors(plugin)
+			}
 		case "http":
-			err = g.httpGenerate(component, plugin)
-		case "grpc", "rpc":
-			err = g.rpcGenerate(component, plugin)
+			err = g.httpGenerate(component, plugin, genClient, genServer)
+		case "grpc", "drpc", "rpc":
+			err = g.rpcGenerate(component, plugin, genClient, genServer)
 		case "gorilla":
 			err = g.gorillaGenerate(component, plugin)
 		case "chi":
 			err = g.chiGenerate(component, plugin)
+		case "openapiv3":
+			err = g.openapiv3Generate(plugin)
+		case "graphqls":
+			err = g.graphqlsGenerate(plugin)
+		case "graphql":
+			err = g.graphqlGenerate(plugin)
+		case "none":
+			break
 		default:
 			err = fmt.Errorf("unknown component: %s", component)
 		}
@@ -63,5 +123,15 @@ func (g *Generator) Generate(plugin *protogen.Plugin) error {
 
 	}
 
+	if err = g.astGenerate(plugin); err != nil {
+		plugin.Error(err)
+		return err
+	}
+
+	if err = g.fieldAlign(plugin); err != nil {
+		plugin.Error(err)
+		return err
+	}
+
 	return nil
 }

micro.go

@@ -1,10 +1,12 @@
 package main
 
 import (
+	"fmt"
+
 	"google.golang.org/protobuf/compiler/protogen"
 )
 
-func (g *Generator) microGenerate(component string, plugin *protogen.Plugin) error {
+func (g *Generator) microGenerate(component string, plugin *protogen.Plugin, genClient bool, genServer bool) error {
 	for _, file := range plugin.Files {
 		if !file.Generate {
 			continue
@@ -14,39 +16,53 @@ func (g *Generator) microGenerate(component string, plugin *protogen.Plugin) err
 		}
 
 		gname := file.GeneratedFilenamePrefix + "_" + component + ".pb.go"
-		gfile := plugin.NewGeneratedFile(gname, ".")
 
-		gfile.P("// Code generated by protoc-gen-micro")
-		gfile.P("// source: ", *file.Proto.Name)
+		path := file.GoImportPath
+		if g.standalone {
+			path = "."
+		}
+		gfile := plugin.NewGeneratedFile(gname, path)
+
+		gfile.P("// Code generated by protoc-gen-go-micro. DO NOT EDIT.")
+		gfile.P("// versions:")
+		gfile.P("// - protoc-gen-go-micro " + versionComment)
+		gfile.P("// - protoc              ", protocVersion(plugin))
+		gfile.P("// source: ", file.Desc.Path())
+		gfile.P()
 		gfile.P("package ", file.GoPackageName)
-
-		gfile.P()
-		gfile.P("import (")
-		gfile.P(`"context"`)
-		gfile.P()
-		gfile.P(`micro_api "github.com/unistack-org/micro/v3/api"`)
-		gfile.P(`micro_client "github.com/unistack-org/micro/v3/client"`)
-		gfile.P(")")
 		gfile.P()
 
-		gfile.P("// Reference imports to suppress errors if they are not otherwise used.")
-		gfile.P("var (")
-		gfile.P("_ ", "micro_api.Endpoint")
-		gfile.P("_ ", "context.Context")
-		gfile.P(" _ ", "micro_client.Option")
-		gfile.P(")")
-		gfile.P()
+		gfile.Import(contextPackage)
 
+		if genClient {
+			gfile.Import(microClientPackage)
+		}
 		// generate services
 		for _, service := range file.Services {
-			generateServiceEndpoints(gfile, service)
-			generateServiceClientInterface(gfile, service)
-			generateServiceClientStreamInterface(gfile, service)
-			generateServiceServerInterface(gfile, service)
-			generateServiceServerStreamInterface(gfile, service)
+			g.generateServiceName(gfile, service)
+			if genClient {
+				g.generateServiceClientInterface(gfile, service)
+				g.generateServiceClientStreamInterface(gfile, service)
+			}
+			if genServer {
+				g.generateServiceServerInterface(gfile, service)
+				g.generateServiceServerStreamInterface(gfile, service)
+			}
 		}
 
 	}
 
 	return nil
 }
+
+func protocVersion(plugin *protogen.Plugin) string {
+	v := plugin.Request.GetCompilerVersion()
+	if v == nil {
+		return "(unknown)"
+	}
+	var suffix string
+	if s := v.GetSuffix(); s != "" {
+		suffix = "-" + s
+	}
+	return fmt.Sprintf("v%d.%d.%d%s", v.GetMajor(), v.GetMinor(), v.GetPatch(), suffix)
+}

openapiv3_util.go

@@ -0,0 +1,39 @@
+package main
+
+import v3 "go.unistack.org/micro-proto/v3/openapiv3"
+
+func getMediaType(eopt interface{}) string {
+	ctype := "application/json"
+
+	if eopt == nil {
+		return ctype
+	}
+
+	if eopt == v3.E_Openapiv3Operation.InterfaceOf(v3.E_Openapiv3Operation.Zero()) {
+		return ctype
+	}
+
+	opt, ok := eopt.(*v3.Operation)
+	if !ok || opt.RequestBody == nil {
+		return ctype
+	}
+
+	if opt.GetRequestBody() == nil {
+		return ctype
+	}
+
+	if opt.GetRequestBody().GetRequestBody() == nil {
+		return ctype
+	}
+
+	c := opt.GetRequestBody().GetRequestBody().GetContent()
+	if c == nil {
+		return ctype
+	}
+
+	for _, prop := range c.GetAdditionalProperties() {
+		ctype = prop.Name
+	}
+
+	return ctype
+}

rpc.go

@@ -4,7 +4,7 @@ import (
 	"google.golang.org/protobuf/compiler/protogen"
 )
 
-func (g *Generator) rpcGenerate(component string, plugin *protogen.Plugin) error {
+func (g *Generator) rpcGenerate(component string, plugin *protogen.Plugin, genClient bool, genServer bool) error {
 	for _, file := range plugin.Files {
 		if !file.Generate {
 			continue
@@ -14,37 +14,40 @@ func (g *Generator) rpcGenerate(component string, plugin *protogen.Plugin) error
 		}
 
 		gname := file.GeneratedFilenamePrefix + "_micro_" + component + ".pb.go"
-		gfile := plugin.NewGeneratedFile(gname, ".")
+		path := file.GoImportPath
+		if g.standalone {
+			path = "."
+		}
+		gfile := plugin.NewGeneratedFile(gname, path)
 
-		gfile.P("// Code generated by protoc-gen-micro")
-		gfile.P("// source: ", *file.Proto.Name)
+		gfile.P("// Code generated by protoc-gen-go-micro. DO NOT EDIT.")
+		gfile.P("// protoc-gen-go-micro version: " + versionComment)
+		gfile.P("// source: ", file.Proto.GetName())
+		gfile.P()
 		gfile.P("package ", file.GoPackageName)
-
-		gfile.P()
-		gfile.P("import (")
-		gfile.P(`"context"`)
-		gfile.P()
-		gfile.P(`micro_api "github.com/unistack-org/micro/v3/api"`)
-		gfile.P(`micro_client "github.com/unistack-org/micro/v3/client"`)
-		gfile.P(`micro_server "github.com/unistack-org/micro/v3/server"`)
-		gfile.P(")")
 		gfile.P()
 
-		gfile.P("// Reference imports to suppress errors if they are not otherwise used.")
-		gfile.P("var (")
-		gfile.P("_ ", "micro_api.Endpoint")
-		gfile.P("_ ", "context.Context")
-		gfile.P(" _ ", "micro_client.Option")
-		gfile.P(" _ ", "micro_server.Option")
-		gfile.P(")")
-		gfile.P()
+		gfile.Import(contextPackage)
 
+		if genClient {
+			gfile.Import(microClientPackage)
+		}
+		if genServer {
+			gfile.Import(microServerPackage)
+		}
 		for _, service := range file.Services {
-			generateServiceClient(gfile, service)
-			generateServiceClientMethods(gfile, service, false)
-			generateServiceServer(gfile, service)
-			generateServiceServerMethods(gfile, service)
-			generateServiceRegister(gfile, service)
+			if genClient {
+				g.generateServiceClient(gfile, file, service)
+				g.generateServiceClientMethods(gfile, service, component)
+			}
+			if genServer {
+				g.generateServiceServer(gfile, file, service)
+				g.generateServiceServerMethods(gfile, service)
+				g.generateServiceRegister(gfile, file, service, component)
+			}
+			if component == "grpc" && g.reflection {
+				g.generateServiceDesc(gfile, file, service)
+			}
 		}
 	}
 

tools.go

@@ -0,0 +1,7 @@
+//go:build ignore
+
+package tools
+
+import (
+	_ "go.unistack.org/micro-proto/v3/graphql"
+)

variables.go

@@ -0,0 +1,28 @@
+package main
+
+import "google.golang.org/protobuf/compiler/protogen"
+
+var (
+	ioPackage              = protogen.GoImportPath("io")
+	graphqlPackage         = protogen.GoImportPath("github.com/99designs/gqlgen/graphql")
+	reflectPackage         = protogen.GoImportPath("reflect")
+	stringsPackage         = protogen.GoImportPath("strings")
+	fmtPackage             = protogen.GoImportPath("fmt")
+	contextPackage         = protogen.GoImportPath("context")
+	httpPackage            = protogen.GoImportPath("net/http")
+	gorillaMuxPackage      = protogen.GoImportPath("github.com/gorilla/mux")
+	chiPackage             = protogen.GoImportPath("github.com/go-chi/chi/v5")
+	chiMiddlewarePackage   = protogen.GoImportPath("github.com/go-chi/chi/v5/middleware")
+	microMetadataPackage   = protogen.GoImportPath("go.unistack.org/micro/v3/metadata")
+	microClientPackage     = protogen.GoImportPath("go.unistack.org/micro/v3/client")
+	microServerPackage     = protogen.GoImportPath("go.unistack.org/micro/v3/server")
+	microClientHttpPackage = protogen.GoImportPath("go.unistack.org/micro-client-http/v3")
+	microServerHttpPackage = protogen.GoImportPath("go.unistack.org/micro-server-http/v3")
+	microCodecPackage      = protogen.GoImportPath("go.unistack.org/micro-proto/v3/codec")
+	microErrorsPackage     = protogen.GoImportPath("go.unistack.org/micro/v3/errors")
+	grpcPackage            = protogen.GoImportPath("google.golang.org/grpc")
+	protojsonPackage       = protogen.GoImportPath("google.golang.org/protobuf/encoding/protojson")
+	timePackage            = protogen.GoImportPath("time")
+	deprecationComment     = "// Deprecated: Do not use."
+	versionComment         = "v3.10.4"
+)