nginx-ui/api/certificate/acme_user.go

package certificate

import (
	"net/http"

	"github.com/0xJacky/Nginx-UI/model"
	"github.com/0xJacky/Nginx-UI/query"
	"github.com/0xJacky/Nginx-UI/settings"
	"github.com/gin-gonic/gin"
	"github.com/spf13/cast"
	"github.com/uozi-tech/cosy"
)

func GetAcmeUser(c *gin.Context) {
	u := query.AcmeUser
	id := cast.ToUint64(c.Param("id"))
	user, err := u.FirstByID(id)
	if err != nil {
		cosy.ErrHandler(c, err)
		return
	}
	c.JSON(http.StatusOK, user)
}

func CreateAcmeUser(c *gin.Context) {
	cosy.Core[model.AcmeUser](c).SetValidRules(gin.H{
		"name":                "required",
		"email":               "required,email",
		"ca_dir":              "omitempty",
		"proxy":               "omitempty",
		"register_on_startup": "omitempty",
		"eab_key_id":          "omitempty",
		"eab_hmac_key":        "omitempty",
	}).BeforeExecuteHook(func(ctx *cosy.Ctx[model.AcmeUser]) {
		if ctx.Model.CADir == "" {
			ctx.Model.CADir = settings.CertSettings.GetCADir()
		}
		err := ctx.Model.Register()
		if err != nil {
			ctx.AbortWithError(err)
			return
		}
	}).Create()
}

func ModifyAcmeUser(c *gin.Context) {
	cosy.Core[model.AcmeUser](c).SetValidRules(gin.H{
		"name":                "omitempty",
		"email":               "omitempty,email",
		"ca_dir":              "omitempty",
		"proxy":               "omitempty",
		"register_on_startup": "omitempty",
		"eab_key_id":          "omitempty",
		"eab_hmac_key":        "omitempty",
	}).BeforeExecuteHook(func(ctx *cosy.Ctx[model.AcmeUser]) {
		if ctx.Model.CADir == "" {
			ctx.Model.CADir = settings.CertSettings.GetCADir()
		}

		if ctx.OriginModel.Email != ctx.Model.Email ||
			ctx.OriginModel.CADir != ctx.Model.CADir {
			err := ctx.Model.Register()
			if err != nil {
				ctx.AbortWithError(err)
				return
			}
		}
	}).Modify()
}

func GetAcmeUserList(c *gin.Context) {
	cosy.Core[model.AcmeUser](c).
		SetFussy("name", "email").
		PagingList()
}

func DestroyAcmeUser(c *gin.Context) {
	cosy.Core[model.AcmeUser](c).Destroy()
}

func RecoverAcmeUser(c *gin.Context) {
	cosy.Core[model.AcmeUser](c).Recover()
}

func RegisterAcmeUser(c *gin.Context) {
	id := cast.ToUint64(c.Param("id"))
	u := query.AcmeUser
	user, err := u.FirstByID(id)
	if err != nil {
		cosy.ErrHandler(c, err)
		return
	}
	err = user.Register()
	if err != nil {
		cosy.ErrHandler(c, err)
		return
	}
	_, err = u.Where(u.ID.Eq(id)).Updates(user)
	if err != nil {
		cosy.ErrHandler(c, err)
		return
	}
	c.JSON(http.StatusOK, user)
}