github.com/unionj-cloud/go-doudou@v1.3.8-0.20221011095552-0088008e5b31/cmd/internal/svc/codegen/svcimpl.go

package codegen

import (
	"bufio"
	"bytes"
	"github.com/iancoleman/strcase"
	"github.com/sirupsen/logrus"
	"github.com/unionj-cloud/go-doudou/cmd/internal/astutils"
	v3 "github.com/unionj-cloud/go-doudou/cmd/internal/protobuf/v3"
	"github.com/unionj-cloud/go-doudou/toolkit/copier"
	"github.com/unionj-cloud/go-doudou/version"
	"go/ast"
	"go/parser"
	"go/token"
	"io/ioutil"
	"os"
	"path/filepath"
	"strings"
	"text/template"
)

var svcimportTmpl = `
	"context"
	"{{.ConfigPackage}}"
	"{{.VoPackage}}"
	"github.com/jmoiron/sqlx"
	"github.com/brianvoe/gofakeit/v6"
`

var appendPart = `{{- range $m := .Meta.Methods }}
	func (receiver *{{$.Meta.Name}}Impl) {{$m.Name}}({{- range $i, $p := $m.Params}}
    {{- if $i}},{{end}}
    {{- $p.Name}} {{$p.Type}}
    {{- end }}) ({{- range $i, $r := $m.Results}}
                     {{- if $i}},{{end}}
                     {{- $r.Name}} {{$r.Type}}
                     {{- end }}) {
    	var _result struct{
			{{- range $r := $m.Results }}
			{{- if ne $r.Type "error" }}
			{{ $r.Name | toCamel }} {{ $r.Type }}
			{{- end }}
			{{- end }}
		}
		_ = gofakeit.Struct(&_result)
		return {{range $i, $r := $m.Results }}{{- if $i}},{{end}}{{ if eq $r.Type "error" }}nil{{else}}_result.{{ $r.Name | toCamel }}{{end}}{{- end }}
    }
{{- end }}`

var svcimplTmpl = `/**
* Generated by go-doudou {{.Version}}.
* You can edit it as your need.
*/
package {{.SvcPackage}}

import ()

var _ {{.Meta.Name}} = (*{{.Meta.Name}}Impl)(nil)

type {{.Meta.Name}}Impl struct {
	conf *config.Config
}

` + appendPart + `

func New{{.Meta.Name}}(conf *config.Config) *{{.Meta.Name}}Impl {
	return &{{.Meta.Name}}Impl{
		conf: conf,
	}
}
`

var svcimportTmplGrpc = `
	"context"
	"{{.ConfigPackage}}"
	"{{.VoPackage}}"
	pb "{{.PbPackage}}"
`

var appendPartGrpc = `{{- range $m := .GrpcSvc.Rpcs }}
    {{- if eq $m.StreamType 0 }}
	func (receiver *{{$.Meta.Name}}Impl) {{$m.Name}}(ctx context.Context, request *pb.{{$m.Request}}) (*pb.{{$m.Response}}, error) {
    	//TODO implement me
		panic("implement me")
    }
    {{- end }}
    {{- if eq $m.StreamType 1 }}
	func (receiver *{{$.Meta.Name}}Impl) {{$m.Name}}(server pb.{{$.GrpcSvc.Name}}_{{$m.Name}}Server) error {
		//TODO implement me
		panic("implement me")
	}
    {{- end }}
    {{- if eq $m.StreamType 2 }}
	func (receiver *{{$.Meta.Name}}Impl) {{$m.Name}}(server pb.{{$.GrpcSvc.Name}}_{{$m.Name}}Server) error {
		//TODO implement me
		panic("implement me")
	}
    {{- end }}
    {{- if eq $m.StreamType 3 }}
	func (receiver *{{$.Meta.Name}}Impl) {{$m.Name}}(request *pb.{{$m.Request}}, server pb.{{$.GrpcSvc.Name}}_{{$m.Name}}Server) error {
		//TODO implement me
		panic("implement me")
	}
    {{- end }}
{{- end }}`

/**
type Service struct {
	Name      string
	Package   string
	GoPackage string
	Syntax    string
	// go-doudou version
	Version  string
	ProtoVer string
	Rpcs     []Rpc
	Messages []Message
	Enums    []Enum
	Comments []string
	Imports  []string
}
*/
var svcimplTmplGrpc = `/**
* Generated by go-doudou {{.Version}}.
* You can edit it as your need.
*/
package {{.SvcPackage}}

import ()

var _ pb.{{.GrpcSvc.Name}}Server = (*{{.Meta.Name}}Impl)(nil)

type {{.Meta.Name}}Impl struct {
    pb.Unimplemented{{.GrpcSvc.Name}}Server
	conf *config.Config
}

` + appendPartGrpc + `

func New{{.Meta.Name}}(conf *config.Config) *{{.Meta.Name}}Impl {
	return &{{.Meta.Name}}Impl{
		conf: conf,
	}
}
`

// GenSvcImpl generates service implementation
func GenSvcImpl(dir string, ic astutils.InterfaceCollector) {
	var (
		err         error
		modfile     string
		modName     string
		svcimplfile string
		firstLine   string
		f           *os.File
		tpl         *template.Template
		buf         bytes.Buffer
		meta        astutils.InterfaceMeta
		tmpl        string
		importBuf   bytes.Buffer
	)
	svcimplfile = filepath.Join(dir, "svcimpl.go")
	err = copier.DeepCopy(ic.Interfaces[0], &meta)
	if err != nil {
		panic(err)
	}
	modfile = filepath.Join(dir, "go.mod")
	if f, err = os.Open(modfile); err != nil {
		panic(err)
	}
	reader := bufio.NewReader(f)
	firstLine, _ = reader.ReadString('\n')
	modName = strings.TrimSpace(strings.TrimPrefix(firstLine, "module"))
	if _, err = os.Stat(svcimplfile); os.IsNotExist(err) {
		if f, err = os.Create(svcimplfile); err != nil {
			panic(err)
		}
		defer f.Close()
		tmpl = svcimplTmpl
	} else {
		logrus.Warningln("New content will be append to file svcimpl.go")
		if f, err = os.OpenFile(svcimplfile, os.O_APPEND, os.ModePerm); err != nil {
			panic(err)
		}
		defer f.Close()
		tmpl = appendPart

		fset := token.NewFileSet()
		root, err := parser.ParseFile(fset, svcimplfile, nil, parser.ParseComments)
		if err != nil {
			panic(err)
		}
		sc := astutils.NewStructCollector(astutils.ExprString)
		ast.Walk(sc, root)
		if implementations, exists := sc.Methods[meta.Name+"Impl"]; exists {
			var notimplemented []astutils.MethodMeta
			for _, item := range meta.Methods {
				for _, implemented := range implementations {
					if item.Name == implemented.Name {
						goto L
					}
				}
				notimplemented = append(notimplemented, item)

			L:
			}

			meta.Methods = notimplemented
		}
	}

	funcMap := make(map[string]interface{})
	funcMap["toCamel"] = strcase.ToCamel
	if tpl, err = template.New("svcimpl.go.tmpl").Funcs(funcMap).Parse(tmpl); err != nil {
		panic(err)
	}
	if err = tpl.Execute(&buf, struct {
		ConfigPackage string
		VoPackage     string
		SvcPackage    string
		Meta          astutils.InterfaceMeta
		Version       string
	}{
		VoPackage:     modName + "/vo",
		ConfigPackage: modName + "/config",
		SvcPackage:    ic.Package.Name,
		Meta:          meta,
		Version:       version.Release,
	}); err != nil {
		panic(err)
	}

	original, err := ioutil.ReadAll(f)
	if err != nil {
		panic(err)
	}

	original = append(original, buf.Bytes()...)
	if tpl, err = template.New("simportimpl.go.tmpl").Parse(svcimportTmpl); err != nil {
		panic(err)
	}
	if err = tpl.Execute(&importBuf, struct {
		ConfigPackage string
		VoPackage     string
	}{
		VoPackage:     modName + "/vo",
		ConfigPackage: modName + "/config",
	}); err != nil {
		panic(err)
	}
	original = astutils.AppendImportStatements(original, importBuf.Bytes())
	original = astutils.RestRelatedModify(original, meta.Name)
	//fmt.Println(string(original))
	astutils.FixImport(original, svcimplfile)
}

// GenSvcImplGrpc generates service implementation for grpc
func GenSvcImplGrpc(dir string, ic astutils.InterfaceCollector, grpcSvc v3.Service) {
	var (
		err         error
		modfile     string
		modName     string
		svcimplfile string
		firstLine   string
		f           *os.File
		tpl         *template.Template
		buf         bytes.Buffer
		meta        astutils.InterfaceMeta
		tmpl        string
		importBuf   bytes.Buffer
	)
	svcimplfile = filepath.Join(dir, "svcimpl.go")
	err = copier.DeepCopy(ic.Interfaces[0], &meta)
	if err != nil {
		panic(err)
	}
	modfile = filepath.Join(dir, "go.mod")
	if f, err = os.Open(modfile); err != nil {
		panic(err)
	}
	reader := bufio.NewReader(f)
	firstLine, _ = reader.ReadString('\n')
	modName = strings.TrimSpace(strings.TrimPrefix(firstLine, "module"))
	if _, err = os.Stat(svcimplfile); os.IsNotExist(err) {
		if f, err = os.Create(svcimplfile); err != nil {
			panic(err)
		}
		defer f.Close()
		tmpl = svcimplTmplGrpc
	} else {
		logrus.Warningln("New content will be append to file svcimpl.go")
		if f, err = os.OpenFile(svcimplfile, os.O_APPEND, os.ModePerm); err != nil {
			panic(err)
		}
		defer f.Close()
		tmpl = appendPartGrpc

		fset := token.NewFileSet()
		root, err := parser.ParseFile(fset, svcimplfile, nil, parser.ParseComments)
		if err != nil {
			panic(err)
		}
		sc := astutils.NewStructCollector(astutils.ExprString)
		ast.Walk(sc, root)
		if implementations, exists := sc.Methods[meta.Name+"Impl"]; exists {
			var notimplemented []v3.Rpc
			for _, item := range grpcSvc.Rpcs {
				for _, implemented := range implementations {
					if item.Name == implemented.Name {
						goto L
					}
				}
				notimplemented = append(notimplemented, item)

			L:
			}

			grpcSvc.Rpcs = notimplemented
		}
	}

	funcMap := make(map[string]interface{})
	funcMap["toCamel"] = strcase.ToCamel
	if tpl, err = template.New("svcimpl.go.tmpl").Funcs(funcMap).Parse(tmpl); err != nil {
		panic(err)
	}
	if err = tpl.Execute(&buf, struct {
		ConfigPackage string
		VoPackage     string
		PbPackage     string
		SvcPackage    string
		Meta          astutils.InterfaceMeta
		GrpcSvc       v3.Service
		Version       string
	}{
		VoPackage:     modName + "/vo",
		ConfigPackage: modName + "/config",
		PbPackage:     modName + "/transport/grpc",
		SvcPackage:    ic.Package.Name,
		Meta:          meta,
		GrpcSvc:       grpcSvc,
		Version:       version.Release,
	}); err != nil {
		panic(err)
	}

	original, err := ioutil.ReadAll(f)
	if err != nil {
		panic(err)
	}

	original = append(original, buf.Bytes()...)
	if tpl, err = template.New("simportimpl.go.tmpl").Parse(svcimportTmplGrpc); err != nil {
		panic(err)
	}
	if err = tpl.Execute(&importBuf, struct {
		ConfigPackage string
		VoPackage     string
		PbPackage     string
	}{
		VoPackage:     modName + "/vo",
		ConfigPackage: modName + "/config",
		PbPackage:     modName + "/transport/grpc",
	}); err != nil {
		panic(err)
	}
	original = astutils.AppendImportStatements(original, importBuf.Bytes())
	original = astutils.GrpcRelatedModify(original, meta.Name, grpcSvc.Name)
	//fmt.Println(string(original))
	astutils.FixImport(original, svcimplfile)
}