mirror of
https://github.com/zitadel/zitadel.git
synced 2024-12-17 13:27:34 +00:00
095ec21678
* chore(proto): update versions * change protoc plugin * some cleanups * define api for setting emails in new api * implement user.SetEmail * move SetEmail buisiness logic into command * resuse newCryptoCode * command: add ChangeEmail unit tests Not complete, was not able to mock the generator. * Revert "resuse newCryptoCode" This reverts commit c89e90ae35ae924a3f706a0a7394f933910c2e65. * undo change to crypto code generators * command: use a generator so we can test properly * command: reorganise ChangeEmail improve test coverage * implement VerifyEmail including unit tests * add URL template tests * proto: change context to object * remove old auth option * remove old auth option * fix linting errors run gci on modified files * add permission checks and fix some errors * comments * comments --------- Co-authored-by: Livio Spring <livio.a@gmail.com> Co-authored-by: Tim Möhlmann <tim+github@zitadel.com>
98 lines
2.3 KiB
Go
98 lines
2.3 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
_ "embed"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"text/template"
|
|
|
|
"google.golang.org/protobuf/compiler/protogen"
|
|
"google.golang.org/protobuf/proto"
|
|
"google.golang.org/protobuf/types/descriptorpb"
|
|
"google.golang.org/protobuf/types/pluginpb"
|
|
|
|
"github.com/zitadel/zitadel/internal/protoc/protoc-gen-authoption/authoption"
|
|
)
|
|
|
|
var (
|
|
//go:embed auth_method_mapping.go.tmpl
|
|
authTemplate []byte
|
|
)
|
|
|
|
type authMethods struct {
|
|
GoPackageName string
|
|
ProtoPackageName string
|
|
ServiceName string
|
|
AuthOptions []authOption
|
|
}
|
|
|
|
type authOption struct {
|
|
Name string
|
|
Permission string
|
|
CheckFieldName string
|
|
}
|
|
|
|
func main() {
|
|
|
|
input, _ := io.ReadAll(os.Stdin)
|
|
var req pluginpb.CodeGeneratorRequest
|
|
err := proto.Unmarshal(input, &req)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
opts := protogen.Options{}
|
|
plugin, err := opts.New(&req)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
plugin.SupportedFeatures = uint64(pluginpb.CodeGeneratorResponse_FEATURE_PROTO3_OPTIONAL)
|
|
|
|
authTemp := loadTemplate(authTemplate)
|
|
|
|
for _, file := range plugin.Files {
|
|
|
|
var buf bytes.Buffer
|
|
|
|
var methods authMethods
|
|
for _, service := range file.Services {
|
|
methods.ServiceName = service.GoName
|
|
methods.GoPackageName = string(file.GoPackageName)
|
|
methods.ProtoPackageName = *file.Proto.Package
|
|
for _, method := range service.Methods {
|
|
if options := method.Desc.Options().(*descriptorpb.MethodOptions); options != nil {
|
|
ext := proto.GetExtension(options, authoption.E_AuthOption).(*authoption.AuthOption)
|
|
if ext != nil {
|
|
methods.AuthOptions = append(methods.AuthOptions, authOption{Name: string(method.Desc.Name()), Permission: ext.Permission, CheckFieldName: ext.CheckFieldName})
|
|
}
|
|
}
|
|
}
|
|
}
|
|
if len(methods.AuthOptions) > 0 {
|
|
authTemp.Execute(&buf, &methods)
|
|
|
|
filename := file.GeneratedFilenamePrefix + ".pb.authoptions.go"
|
|
file := plugin.NewGeneratedFile(filename, ".")
|
|
|
|
file.Write(buf.Bytes())
|
|
}
|
|
}
|
|
|
|
// Generate a response from our plugin and marshall as protobuf
|
|
stdout := plugin.Response()
|
|
out, err := proto.Marshal(stdout)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
// Write the response to stdout, to be picked up by protoc
|
|
fmt.Fprintf(os.Stdout, string(out))
|
|
}
|
|
|
|
func loadTemplate(templateData []byte) *template.Template {
|
|
return template.Must(template.New("").
|
|
Parse(string(templateData)))
|
|
}
|