package certs

import (
	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
	"github.com/iwind/TeaGo/actions"
	"github.com/iwind/TeaGo/maps"
	"github.com/iwind/TeaGo/types"
	"net/http"
)

type Helper struct {
}

func NewHelper() *Helper {
	return &Helper{}
}

func (this *Helper) BeforeAction(actionWrapper actions.ActionWrapper) {
	var action = actionWrapper.Object()
	if action.Request.Method != http.MethodGet {
		return
	}

	action.Data["teaMenu"] = "servers"

	var countOCSP int64 = 0
	parentAction, ok := actionWrapper.(actionutils.ActionInterface)
	if ok {
		countOCSPResp, err := parentAction.RPC().SSLCertRPC().CountAllSSLCertsWithOCSPError(parentAction.AdminContext(), &pb.CountAllSSLCertsWithOCSPErrorRequest{})
		if err == nil {
			countOCSP = countOCSPResp.Count
		}
	}

	var ocspMenuName = "OCSP日志"
	if countOCSP > 0 {
		ocspMenuName += "(" + types.String(countOCSP) + ")"
	}

	var menu = []maps.Map{
		{
			"name":     "证书",
			"url":      "/servers/certs",
			"isActive": action.Data.GetString("leftMenuItem") == "cert",
		},
		{
			"name":     "申请证书",
			"url":      "/servers/certs/acme",
			"isActive": action.Data.GetString("leftMenuItem") == "acme",
		},
		{
			"name":     ocspMenuName,
			"url":      "/servers/certs/ocsp",
			"isActive": action.Data.GetString("leftMenuItem") == "ocsp",
		},
	}
	action.Data["leftMenuItems"] = menu
}