swagger-api/service.go

package swagger_api

import (
	"context"
	"fmt"
	//"git.ikuban.com/server/swagger-api/protoc-gen-openapiv2/generator"
	generatorJson "git.ikuban.com/server/gnostic/protoc-gen-jsonschema/generator"
	"git.ikuban.com/server/gnostic/protoc-gen-openapi/generator"
	"github.com/go-kratos/kratos/v2/api/metadata"
	"github.com/xmkuban/utils/utils"
	"google.golang.org/grpc"
	"google.golang.org/protobuf/compiler/protogen"
	"google.golang.org/protobuf/types/descriptorpb"
	"google.golang.org/protobuf/types/pluginpb"
)

// Service is service
type Service struct {
	ser     *metadata.Server
	outType string
}

// New service
func New(srv *grpc.Server, outType string) *Service {
	if outType == "" {
		outType = "yaml"
	}
	return &Service{
		ser:     metadata.NewServer(srv),
		outType: outType,
	}
}

// ListServices list services
func (s *Service) ListServices(ctx context.Context, in *metadata.ListServicesRequest) (*metadata.ListServicesReply, error) {
	return s.ser.ListServices(ctx, &metadata.ListServicesRequest{})
}

// GetServiceOpenAPI get service open api
func (s *Service) GetServiceOpenAPI(ctx context.Context, in *metadata.GetServiceDescRequest, onlyRPC bool) (string, error) {
	protoSet, err := s.ser.GetServiceDesc(ctx, in)
	if err != nil {
		return "", err
	}
	files := protoSet.FileDescSet.File
	newFiles := make([]*descriptorpb.FileDescriptorProto, 0)
	newFileMap := make(map[string]byte)
	for _, v := range files {
		if _, ok := newFileMap[v.GetName()]; !ok {
			newFiles = append(newFiles, v)
			newFileMap[v.GetName()] = 1
		}
	}
	files = newFiles
	var target string
	if len(files) == 0 {
		return "", fmt.Errorf("proto file is empty")
	}
	if files[len(files)-1].Name == nil {
		return "", fmt.Errorf("proto file's name is null")
	}
	target = *files[len(files)-1].Name

	req := new(pluginpb.CodeGeneratorRequest)
	req.FileToGenerate = []string{target}
	var para = ""
	req.Parameter = &para
	req.ProtoFile = files

	opts := protogen.Options{}
	plugin, err := opts.New(req)
	if err != nil {
		return "", err
	}
	plugin.SupportedFeatures = uint64(pluginpb.CodeGeneratorResponse_FEATURE_PROTO3_OPTIONAL)

	var content []byte
	if s.outType == "yaml" {
		gen := generator.NewOpenAPIv3Generator(plugin, generator.Configuration{
			Version:         utils.ToPointString("1.0"),
			Title:           utils.ToPointString(""),
			Description:     utils.ToPointString(""),
			Naming:          utils.ToPointString("json"),
			FQSchemaNaming:  utils.ToPointBool(true),
			EnumType:        utils.ToPointString("integer"),
			CircularDepth:   utils.ToPointInt(2),
			DefaultResponse: utils.ToPointBool(false),
			OutputMode:      utils.ToPointString("merged"),
		}, plugin.Files)
		content, err = gen.RunV2()
	} else {
		gen := generatorJson.NewJSONSchemaGenerator(plugin, generatorJson.Configuration{
			BaseURL:  utils.ToPointString(""),
			Version:  utils.ToPointString("1.0"),
			Naming:   utils.ToPointString("json"),
			EnumType: utils.ToPointString("integer"),
		})
		content, err = gen.RunV2()
	}

	if err != nil {
		return "", err
	}
	return string(content), nil
}