geo/server/internal/tenant/transport/kol_manage_handler.go

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
	assistSvc  *app.KolAssistService
}

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),
		assistSvc:  app.NewKolAssistService(profileSvc, a.KolAssists, a.LLM, a.RabbitMQ),
	}
}

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 (h *KolManageHandler) AssistSubmit(c *gin.Context) {
	var req app.AssistRequest
	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
	}

	taskID, err := h.assistSvc.Submit(c.Request.Context(), actor, req)
	if err != nil {
		response.Error(c, err)
		return
	}

	response.Success(c, app.KolAssistSubmitResponse{TaskID: taskID})
}

func (h *KolManageHandler) AssistGet(c *gin.Context) {
	taskID := c.Param("id")
	if taskID == "" {
		response.Error(c, response.ErrBadRequest(40001, "invalid_id", "task id must not be empty"))
		return
	}

	actor, ok := actorFromRequest(c)
	if !ok {
		return
	}

	task, err := h.assistSvc.Get(c.Request.Context(), actor, taskID)
	if err != nil {
		response.Error(c, err)
		return
	}

	data, err := app.NewKolAssistTaskResponse(task)
	if err != nil {
		response.Error(c, err)
		return
	}

	response.Success(c, data)
}

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
}