digital-psychology/apps/api/internal/handler/admin_ask_ops.go

package handler

import (
	"errors"
	"net/http"
	"strconv"

	"github.com/gin-gonic/gin"
	"github.com/google/uuid"

	"github.com/yuxingu/digital-psychology/apps/api/internal/middleware"
	"github.com/yuxingu/digital-psychology/apps/api/internal/service/admin"
	"github.com/yuxingu/digital-psychology/apps/api/pkg/response"
)

func (h *AdminHandler) registerAskOps(authed *gin.RouterGroup) {
	authed.GET("/ask/threads", middleware.RequireAdminPermission(h.Svc, admin.PermAskRead), h.ListAskThreads)
	authed.GET("/ask/threads/:id", middleware.RequireAdminPermission(h.Svc, admin.PermAskRead), h.GetAskThread)
	authed.GET("/ask/threads/:id/messages", middleware.RequireAdminPermission(h.Svc, admin.PermAskTranscriptRead), h.GetAskThreadMessages)
}

func (h *AdminHandler) ListAskThreads(c *gin.Context) {
	limit, _ := strconv.Atoi(c.DefaultQuery("limit", "20"))
	offset, _ := strconv.Atoi(c.DefaultQuery("offset", "0"))
	var userID *uuid.UUID
	if q := c.Query("user_id"); q != "" {
		id, err := uuid.Parse(q)
		if err != nil {
			response.Fail(c, http.StatusBadRequest, 40002, "invalid user_id")
			return
		}
		userID = &id
	}
	items, err := h.Svc.ListAskSessions(c.Request.Context(), userID, limit, offset)
	if err != nil {
		response.Fail(c, http.StatusInternalServerError, 50019, "list ask threads failed")
		return
	}
	response.OK(c, gin.H{"items": items})
}

func (h *AdminHandler) GetAskThread(c *gin.Context) {
	id, err := uuid.Parse(c.Param("id"))
	if err != nil {
		response.Fail(c, http.StatusBadRequest, 40002, "invalid thread id")
		return
	}
	meta, err := h.Svc.GetAskSessionMeta(c.Request.Context(), id)
	if errors.Is(err, admin.ErrAskThreadNotFound) {
		response.Fail(c, http.StatusNotFound, 40402, "ask thread not found")
		return
	}
	if err != nil {
		response.Fail(c, http.StatusInternalServerError, 50020, "get ask thread failed")
		return
	}
	response.OK(c, meta)
}

func (h *AdminHandler) GetAskThreadMessages(c *gin.Context) {
	adminID, ok := middleware.AdminIDFromContext(c)
	if !ok {
		response.Fail(c, http.StatusUnauthorized, 40102, "admin session invalid")
		return
	}
	id, err := uuid.Parse(c.Param("id"))
	if err != nil {
		response.Fail(c, http.StatusBadRequest, 40002, "invalid thread id")
		return
	}
	detail, err := h.Svc.GetAskSessionTranscript(c.Request.Context(), adminID, id)
	if errors.Is(err, admin.ErrAskThreadNotFound) {
		response.Fail(c, http.StatusNotFound, 40402, "ask thread not found")
		return
	}
	if err != nil {
		response.Fail(c, http.StatusInternalServerError, 50021, "get ask transcript failed")
		return
	}
	response.OK(c, detail)
}