swagger-api/service.go

package swagger_api

import (
	"context"
	"fmt"
	//"git.ikuban.com/server/swagger-api/protoc-gen-openapiv2/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
	serviceInfoMap map[string]grpc.ServiceInfo
}

// New service
func New(srv *grpc.Server) *Service {
	return &Service{
		ser: metadata.NewServer(srv),
	}
}

// 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)
	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()
	if err != nil {
		return "", err
	}
	return string(content), nil
}