package restful

import (
	"gitee.com/go-kade/librarys/ioc/config/http"

	"gitee.com/go-kade/librarys/ioc"
	ioc_health "gitee.com/go-kade/librarys/ioc/apps/health"
	"gitee.com/go-kade/librarys/ioc/config/log"
	"gitee.com/go-kade/librarys/http/response"
	restfulspec "github.com/emicklei/go-restful-openapi/v2"
	"github.com/emicklei/go-restful/v3"
	"github.com/rs/zerolog"
	"google.golang.org/grpc/health"
	healthgrpc "google.golang.org/grpc/health/grpc_health_v1"
)

func init() {
	ioc.Api().Registry(&HealthChecker{
		HealthCheck: ioc_health.HealthCheck{
			Path: ioc_health.DEFAUL_HEALTH_PATH,
		},
	})
}

func (h *HealthChecker) Name() string {
	return ioc_health.AppName
}

func (i *HealthChecker) Priority() int {
	return 100
}

func (h *HealthChecker) Init() error {
	if h.Service == nil {
		h.Service = health.NewServer()
	}

	h.log = log.Sub("health_check")
	return nil
}

func (h *HealthChecker) Meta() ioc.ObjectMeta {
	meta := ioc.DefaultObjectMeta()
	meta.CustomPathPrefix = h.Path
	return meta
}

type HealthChecker struct {
	ioc.ObjectImpl
	ioc_health.HealthCheck
	log *zerolog.Logger

	Service healthgrpc.HealthServer `ioc:"autowire=true;namespace=controllers"`
}

func (h *HealthChecker) Registry(ws *restful.WebService) {
	tags := []string{"健康检查"}
	ws.Route(ws.GET("/").To(h.HealthHandleFunc).
		Doc("查询服务当前状态").
		Metadata(restfulspec.KeyOpenAPITags, tags).
		Returns(200, "OK", ioc_health.HealthCheckResponse{}))

	h.log.Info().Msgf("Get the Health using http://%s%s", http.Get().Addr(), h.Path)
}

func (h *HealthChecker) HealthHandleFunc(r *restful.Request, w *restful.Response) {
	req := ioc_health.NewHealthCheckRequest()
	resp, err := h.Service.Check(
		r.Request.Context(),
		req,
	)
	if err != nil {
		response.Failed(w, err)
		return
	}
	//如果健康检查成功，则将响应转换为 JSON 格式，并通过 w 写入到响应中。
	err = w.WriteAsJson(ioc_health.NewHealth(resp))
	if err != nil {
		h.log.Error().Msgf("send success response error, %s", err)
	}
}