feat(kol): KOL workspace HTTP handlers + route registration

2026-04-17 09:19:37 +08:00
parent 6029b0b169
commit a6ded27c1d
3 changed files with 367 additions and 0 deletions
@@ -0,0 +1,340 @@
+package transport
+
+import (
+	"strconv"
+
+	"github.com/gin-gonic/gin"
+
+	"github.com/geo-platform/tenant-api/internal/bootstrap"
+	"github.com/geo-platform/tenant-api/internal/shared/auth"
+	"github.com/geo-platform/tenant-api/internal/shared/response"
+	"github.com/geo-platform/tenant-api/internal/tenant/app"
+)
+
+type KolManageHandler struct {
+	profileSvc *app.KolProfileService
+	pkgSvc     *app.KolPackageService
+	promptSvc  *app.KolPromptService
+}
+
+func NewKolManageHandler(a *bootstrap.App) *KolManageHandler {
+	profileSvc := app.NewKolProfileService(a.KolProfiles)
+
+	return &KolManageHandler{
+		profileSvc: profileSvc,
+		pkgSvc:     app.NewKolPackageService(profileSvc, a.KolPackages, a.KolPrompts),
+		promptSvc:  app.NewKolPromptService(a.DB, profileSvc, a.KolPackages, a.KolPrompts, a.KolPromptAsset),
+	}
+}
+
+func (h *KolManageHandler) GetProfile(c *gin.Context) {
+	actor, ok := actorFromRequest(c)
+	if !ok {
+		return
+	}
+
+	profile, err := h.profileSvc.RequireActiveKol(c.Request.Context(), actor)
+	if err != nil {
+		response.Error(c, err)
+		return
+	}
+
+	response.Success(c, app.NewKolProfileResponse(profile))
+}
+
+func (h *KolManageHandler) ListPackages(c *gin.Context) {
+	actor, ok := actorFromRequest(c)
+	if !ok {
+		return
+	}
+
+	data, err := h.pkgSvc.List(c.Request.Context(), actor)
+	if err != nil {
+		response.Error(c, err)
+		return
+	}
+
+	response.Success(c, data)
+}
+
+func (h *KolManageHandler) CreatePackage(c *gin.Context) {
+	var req app.CreateKolPackageInput
+	if err := c.ShouldBindJSON(&req); err != nil {
+		response.Error(c, response.ErrBadRequest(40001, "invalid_params", err.Error()))
+		return
+	}
+
+	actor, ok := actorFromRequest(c)
+	if !ok {
+		return
+	}
+
+	data, err := h.pkgSvc.Create(c.Request.Context(), actor, req)
+	if err != nil {
+		response.Error(c, err)
+		return
+	}
+
+	response.Success(c, data)
+}
+
+func (h *KolManageHandler) UpdatePackage(c *gin.Context) {
+	id, ok := parseInt64Param(c, "id", "package id")
+	if !ok {
+		return
+	}
+
+	var req app.UpdateKolPackageInput
+	if err := c.ShouldBindJSON(&req); err != nil {
+		response.Error(c, response.ErrBadRequest(40001, "invalid_params", err.Error()))
+		return
+	}
+
+	actor, ok := actorFromRequest(c)
+	if !ok {
+		return
+	}
+
+	data, err := h.pkgSvc.Update(c.Request.Context(), actor, id, req)
+	if err != nil {
+		response.Error(c, err)
+		return
+	}
+
+	response.Success(c, data)
+}
+
+func (h *KolManageHandler) DeletePackage(c *gin.Context) {
+	id, ok := parseInt64Param(c, "id", "package id")
+	if !ok {
+		return
+	}
+
+	actor, ok := actorFromRequest(c)
+	if !ok {
+		return
+	}
+
+	if err := h.pkgSvc.Delete(c.Request.Context(), actor, id); err != nil {
+		response.Error(c, err)
+		return
+	}
+
+	response.Success(c, nil)
+}
+
+func (h *KolManageHandler) PublishPackage(c *gin.Context) {
+	id, ok := parseInt64Param(c, "id", "package id")
+	if !ok {
+		return
+	}
+
+	actor, ok := actorFromRequest(c)
+	if !ok {
+		return
+	}
+
+	data, err := h.pkgSvc.Publish(c.Request.Context(), actor, id)
+	if err != nil {
+		response.Error(c, err)
+		return
+	}
+
+	response.Success(c, data)
+}
+
+func (h *KolManageHandler) ArchivePackage(c *gin.Context) {
+	id, ok := parseInt64Param(c, "id", "package id")
+	if !ok {
+		return
+	}
+
+	actor, ok := actorFromRequest(c)
+	if !ok {
+		return
+	}
+
+	data, err := h.pkgSvc.Archive(c.Request.Context(), actor, id)
+	if err != nil {
+		response.Error(c, err)
+		return
+	}
+
+	response.Success(c, data)
+}
+
+func (h *KolManageHandler) ListPrompts(c *gin.Context) {
+	packageID, ok := parseInt64Param(c, "id", "package id")
+	if !ok {
+		return
+	}
+
+	actor, ok := actorFromRequest(c)
+	if !ok {
+		return
+	}
+
+	data, err := h.promptSvc.ListByPackage(c.Request.Context(), actor, packageID)
+	if err != nil {
+		response.Error(c, err)
+		return
+	}
+
+	response.Success(c, data)
+}
+
+func (h *KolManageHandler) CreatePrompt(c *gin.Context) {
+	packageID, ok := parseInt64Param(c, "id", "package id")
+	if !ok {
+		return
+	}
+
+	var req app.CreatePromptInput
+	if err := c.ShouldBindJSON(&req); err != nil {
+		response.Error(c, response.ErrBadRequest(40001, "invalid_params", err.Error()))
+		return
+	}
+	req.PackageID = packageID
+
+	actor, ok := actorFromRequest(c)
+	if !ok {
+		return
+	}
+
+	data, err := h.promptSvc.CreatePrompt(c.Request.Context(), actor, req)
+	if err != nil {
+		response.Error(c, err)
+		return
+	}
+
+	response.Success(c, data)
+}
+
+func (h *KolManageHandler) GetPromptLatest(c *gin.Context) {
+	promptID, ok := parseInt64Param(c, "id", "prompt id")
+	if !ok {
+		return
+	}
+
+	actor, ok := actorFromRequest(c)
+	if !ok {
+		return
+	}
+
+	data, err := h.promptSvc.GetForOwner(c.Request.Context(), actor, promptID)
+	if err != nil {
+		response.Error(c, err)
+		return
+	}
+
+	response.Success(c, data)
+}
+
+func (h *KolManageHandler) UpdatePromptMetadata(c *gin.Context) {
+	promptID, ok := parseInt64Param(c, "id", "prompt id")
+	if !ok {
+		return
+	}
+
+	var req app.UpdateKolPromptInput
+	if err := c.ShouldBindJSON(&req); err != nil {
+		response.Error(c, response.ErrBadRequest(40001, "invalid_params", err.Error()))
+		return
+	}
+
+	actor, ok := actorFromRequest(c)
+	if !ok {
+		return
+	}
+
+	data, err := h.promptSvc.UpdateMetadata(c.Request.Context(), actor, promptID, req)
+	if err != nil {
+		response.Error(c, err)
+		return
+	}
+
+	response.Success(c, data)
+}
+
+func (h *KolManageHandler) DeletePrompt(c *gin.Context) {
+	promptID, ok := parseInt64Param(c, "id", "prompt id")
+	if !ok {
+		return
+	}
+
+	actor, ok := actorFromRequest(c)
+	if !ok {
+		return
+	}
+
+	if err := h.promptSvc.Delete(c.Request.Context(), actor, promptID); err != nil {
+		response.Error(c, err)
+		return
+	}
+
+	response.Success(c, nil)
+}
+
+func (h *KolManageHandler) SaveDraft(c *gin.Context) {
+	promptID, ok := parseInt64Param(c, "id", "prompt id")
+	if !ok {
+		return
+	}
+
+	var req app.SaveDraftInput
+	if err := c.ShouldBindJSON(&req); err != nil {
+		response.Error(c, response.ErrBadRequest(40001, "invalid_params", err.Error()))
+		return
+	}
+	req.PromptID = promptID
+
+	actor, ok := actorFromRequest(c)
+	if !ok {
+		return
+	}
+
+	data, err := h.promptSvc.SaveDraft(c.Request.Context(), actor, req)
+	if err != nil {
+		response.Error(c, err)
+		return
+	}
+
+	response.Success(c, data)
+}
+
+func (h *KolManageHandler) PublishPrompt(c *gin.Context) {
+	promptID, ok := parseInt64Param(c, "id", "prompt id")
+	if !ok {
+		return
+	}
+
+	actor, ok := actorFromRequest(c)
+	if !ok {
+		return
+	}
+
+	if err := h.promptSvc.Publish(c.Request.Context(), actor, promptID); err != nil {
+		response.Error(c, err)
+		return
+	}
+
+	response.Success(c, gin.H{"ok": true})
+}
+
+func actorFromRequest(c *gin.Context) (auth.Actor, bool) {
+	actor, ok := auth.ActorFromCtx(c.Request.Context())
+	if !ok {
+		response.Error(c, response.ErrUnauthorized(40100, "not_authenticated", "not authenticated"))
+		return auth.Actor{}, false
+	}
+	return actor, true
+}
+
+func parseInt64Param(c *gin.Context, key, label string) (int64, bool) {
+	id, err := strconv.ParseInt(c.Param(key), 10, 64)
+	if err != nil {
+		response.Error(c, response.ErrBadRequest(40001, "invalid_id", label+" must be a number"))
+		return 0, false
+	}
+	return id, true
+}