APIPark/service/api-doc/service.go

package api_doc

import (
	"context"
	"fmt"
	"github.com/APIParkLab/APIPark/service/universally/commit"
	"github.com/eolinker/go-common/autowire"
	"github.com/getkin/kin-openapi/openapi3"
	"reflect"
)

type IAPIDocService interface {
	// UpdateDoc 更新文档
	UpdateDoc(ctx context.Context, serviceId string, input *UpdateDoc) error
	// GetDoc 获取文档
	GetDoc(ctx context.Context, serviceId string) (*Doc, error)
	// LatestDocCommit 获取最新文档
	LatestDocCommit(ctx context.Context, serviceId string) (*commit.Commit[DocCommit], error)
	CommitDoc(ctx context.Context, serviceId string, data *DocCommit) error
}

func init() {
	autowire.Auto[IAPIDocService](func() reflect.Value {
		return reflect.ValueOf(new(imlAPIDocService))
	})
	commit.InitCommitWithKeyService[DocCommit]("service", "api_doc")
}

var (
	loader = openapi3.NewLoader()
)

type DocLoader struct {
	doc *openapi3.T
}

func NewDocLoader(content string) (*DocLoader, error) {
	doc, err := loader.LoadFromData([]byte(content))
	if err != nil {
		return nil, fmt.Errorf("load doc error:%v", err)
	}
	return &DocLoader{doc: doc}, nil
}

func (d *DocLoader) Valid() error {
	if d.doc == nil {
		return fmt.Errorf("doc is nil")
	}

	return d.doc.Validate(loader.Context)
}

func (d *DocLoader) APICount() int64 {
	if d.doc == nil || d.doc.Paths == nil {
		return 0
	}
	var count int64
	for _, item := range d.doc.Paths.Map() {
		if item.Get != nil {
			count++
		}
		if item.Post != nil {
			count++
		}
		if item.Put != nil {
			count++
		}
		if item.Patch != nil {
			count++
		}
		if item.Delete != nil {
			count++
		}
		if item.Head != nil {
			count++
		}
		if item.Options != nil {
			count++
		}
		if item.Trace != nil {
			count++
		}
	}
	return count
}