rogee/gochat
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(captain): align document response actions

Browse Source
This commit is contained in:
rogee
2026-06-05 12:24:16 +08:00
parent fc2cb12e63
commit afccb5f9ae
16 changed files with 877 additions and 160 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/handler/api/v1/captain_assistant_response_handler.go
+69 -36
View File
@@ -2,9 +2,9 @@ package v1
import (
"net/http"
"strconv"
"github.com/gin-gonic/gin"
"github.com/gochat/gochat/internal/model"
"github.com/gochat/gochat/internal/service"
applogger "github.com/gochat/gochat/pkg/logger"
"github.com/gochat/gochat/pkg/response"
@@ -23,8 +23,8 @@ func NewCaptainAssistantResponseHandler(svc *service.CaptainAssistantResponseSer
// ProcessResponse generates and optionally stores an assistant response.
// POST /api/v1/accounts/:id/captain/assistant_responses
func (h *CaptainAssistantResponseHandler) ProcessResponse(c *gin.Context) {
accountID, err := strconv.ParseUint(c.Param("id"), 10, 64)
if err != nil {
accountID := parseAccountIDParam(c)
if accountID == 0 {
response.AbortWithStatusError(c, http.StatusBadRequest, response.ErrBadRequest, "invalid account_id")
return
}
@@ -35,7 +35,7 @@ func (h *CaptainAssistantResponseHandler) ProcessResponse(c *gin.Context) {
return
}
result, err := h.svc.ProcessResponse(c.Request.Context(), uint(accountID), &req)
result, err := h.svc.ProcessResponse(c.Request.Context(), accountID, &req)
if err != nil {
applogger.L().Errorf("Process assistant response: %v", err)
response.AbortWithStatusError(c, http.StatusUnprocessableEntity, response.ErrInternal, "failed to process assistant response")
@@ -49,60 +49,64 @@ func (h *CaptainAssistantResponseHandler) ProcessResponse(c *gin.Context) {
// GET /api/v1/accounts/:id/captain/assistant_responses
// Reference: Chatwoot Captain::AssistantResponsesController#index
func (h *CaptainAssistantResponseHandler) List(c *gin.Context) {
accountID, err := strconv.ParseUint(c.Param("id"), 10, 64)
if err != nil {
accountID := parseAccountIDParam(c)
if accountID == 0 {
response.AbortWithStatusError(c, http.StatusBadRequest, response.ErrBadRequest, "invalid account_id")
return
}
page, _ := strconv.Atoi(c.DefaultQuery("page", "1"))
pageSize, _ := strconv.Atoi(c.DefaultQuery("per_page", "25"))
assistantID, _ := strconv.ParseUint(c.Query("assistant_id"), 10, 64)
documentID, _ := strconv.ParseUint(c.Query("document_id"), 10, 64)
page, _ := parseIntQueryDefault(c, "page", 1)
pageSize := 25
assistantID, _ := parseOptionalUintQueryParam(c, "assistant_id")
documentID, _ := parseOptionalUintQueryParam(c, "document_id")
status := c.Query("status")
search := c.Query("search")
responses, total, err := h.svc.List(c.Request.Context(), uint(accountID), uint(assistantID), uint(documentID), status, search, page, pageSize)
responses, total, err := h.svc.List(c.Request.Context(), accountID, assistantID, documentID, status, search, page, pageSize)
if err != nil {
applogger.L().Errorf("List assistant responses: %v", err)
response.AbortWithStatusError(c, http.StatusInternalServerError, response.ErrInternal, "failed to list responses")
return
}
response.OKWithMeta(c, responses, page, pageSize, total)
payload := make([]gin.H, 0, len(responses))
for i := range responses {
payload = append(payload, captainAssistantResponsePayload(&responses[i]))
}
c.JSON(http.StatusOK, gin.H{"payload": payload, "meta": gin.H{"total_count": total, "page": page}})
}
// Get returns a single assistant response.
// GET /api/v1/accounts/:id/captain/assistant_responses/:response_id
// Reference: Chatwoot Captain::AssistantResponsesController#show
func (h *CaptainAssistantResponseHandler) Get(c *gin.Context) {
accountID, err := strconv.ParseUint(c.Param("id"), 10, 64)
if err != nil {
accountID := parseAccountIDParam(c)
if accountID == 0 {
response.AbortWithStatusError(c, http.StatusBadRequest, response.ErrBadRequest, "invalid account_id")
return
}
responseID, err := strconv.ParseUint(c.Param("response_id"), 10, 64)
responseID, err := parseUintParam(c, "response_id")
if err != nil {
response.AbortWithStatusError(c, http.StatusBadRequest, response.ErrBadRequest, "invalid response_id")
return
}
resp, err := h.svc.Get(c.Request.Context(), uint(accountID), uint(responseID))
resp, err := h.svc.Get(c.Request.Context(), accountID, responseID)
if err != nil {
applogger.L().Errorf("Get assistant response: %v", err)
response.AbortWithStatusError(c, http.StatusNotFound, response.ErrNotFound, "response not found")
return
}
response.OK(c, resp)
c.JSON(http.StatusOK, captainAssistantResponsePayload(resp))
}
// Update modifies an assistant response.
// PUT /api/v1/accounts/:id/captain/assistant_responses/:response_id
// Reference: Chatwoot Captain::AssistantResponsesController#update
func (h *CaptainAssistantResponseHandler) Update(c *gin.Context) {
accountID, err := strconv.ParseUint(c.Param("id"), 10, 64)
if err != nil {
accountID := parseAccountIDParam(c)
if accountID == 0 {
response.AbortWithStatusError(c, http.StatusBadRequest, response.ErrBadRequest, "invalid account_id")
return
}
responseID, err := strconv.ParseUint(c.Param("response_id"), 10, 64)
responseID, err := parseUintParam(c, "response_id")
if err != nil {
response.AbortWithStatusError(c, http.StatusBadRequest, response.ErrBadRequest, "invalid response_id")
return
@@ -112,34 +116,34 @@ func (h *CaptainAssistantResponseHandler) Update(c *gin.Context) {
Answer string `json:"answer"`
Status string `json:"status"`
}
if err := c.ShouldBindJSON(&req); err != nil {
if err := bindNestedJSONPayload(c, "assistant_response", &req); err != nil {
response.AbortWithStatusError(c, http.StatusBadRequest, response.ErrValidation, err.Error())
return
}
resp, err := h.svc.Update(c.Request.Context(), uint(accountID), uint(responseID), req.Question, req.Answer, req.Status)
resp, err := h.svc.Update(c.Request.Context(), accountID, responseID, req.Question, req.Answer, req.Status)
if err != nil {
applogger.L().Errorf("Update assistant response: %v", err)
response.AbortWithStatusError(c, http.StatusInternalServerError, response.ErrInternal, "failed to update response")
return
}
response.OK(c, resp)
c.JSON(http.StatusOK, captainAssistantResponsePayload(resp))
}
// Delete removes an assistant response.
// DELETE /api/v1/accounts/:id/captain/assistant_responses/:response_id
// Reference: Chatwoot Captain::AssistantResponsesController#destroy (head :no_content)
func (h *CaptainAssistantResponseHandler) Delete(c *gin.Context) {
accountID, err := strconv.ParseUint(c.Param("id"), 10, 64)
if err != nil {
accountID := parseAccountIDParam(c)
if accountID == 0 {
response.AbortWithStatusError(c, http.StatusBadRequest, response.ErrBadRequest, "invalid account_id")
return
}
responseID, err := strconv.ParseUint(c.Param("response_id"), 10, 64)
responseID, err := parseUintParam(c, "response_id")
if err != nil {
response.AbortWithStatusError(c, http.StatusBadRequest, response.ErrBadRequest, "invalid response_id")
return
}
if err := h.svc.Delete(c.Request.Context(), uint(accountID), uint(responseID)); err != nil {
if err := h.svc.Delete(c.Request.Context(), accountID, responseID); err != nil {
applogger.L().Errorf("Delete assistant response: %v", err)
response.AbortWithStatusError(c, http.StatusInternalServerError, response.ErrInternal, "failed to delete response")
return
@@ -152,16 +156,12 @@ func (h *CaptainAssistantResponseHandler) Delete(c *gin.Context) {
// POST /api/v1/accounts/:id/captain/assistant_responses
// Reference: Chatwoot Captain::AssistantResponsesController#create
func (h *CaptainAssistantResponseHandler) Create(c *gin.Context) {
accountID, err := strconv.ParseUint(c.Param("id"), 10, 64)
if err != nil {
accountID := parseAccountIDParam(c)
if accountID == 0 {
response.AbortWithStatusError(c, http.StatusBadRequest, response.ErrBadRequest, "invalid account_id")
return
}
userID := getUserID(c)
if userID == 0 {
response.AbortWithStatusError(c, http.StatusUnauthorized, response.ErrUnauthorized, "user not authenticated")
return
}
var req struct {
Question string `json:"question" validate:"required"`
@@ -169,16 +169,49 @@ func (h *CaptainAssistantResponseHandler) Create(c *gin.Context) {
AssistantID uint `json:"assistant_id" validate:"required"`
Status string `json:"status"`
}
if err := c.ShouldBindJSON(&req); err != nil {
if err := bindNestedJSONPayload(c, "assistant_response", &req); err != nil {
response.AbortWithStatusError(c, http.StatusBadRequest, response.ErrValidation, err.Error())
return
}
resp, err := h.svc.Create(c.Request.Context(), uint(accountID), userID, req.AssistantID, req.Question, req.Answer, req.Status)
resp, err := h.svc.Create(c.Request.Context(), accountID, userID, req.AssistantID, req.Question, req.Answer, req.Status)
if err != nil {
applogger.L().Errorf("Create assistant response: %v", err)
response.AbortWithStatusError(c, http.StatusInternalServerError, response.ErrInternal, "failed to create response")
return
}
response.OK(c, resp)
c.JSON(http.StatusOK, captainAssistantResponsePayload(resp))
}
func captainAssistantResponsePayload(resp *model.CaptainAssistantResponse) gin.H {
payload := gin.H{
"account_id": resp.AccountID,
"answer": resp.Answer,
"assistant": captainResponseAssistantPayload(resp),
"created_at": resp.CreatedAt.Unix(),
"id": resp.ID,
"question": resp.Question,
"updated_at": resp.UpdatedAt.Unix(),
"status": resp.Status,
"edited": resp.Edited,
}
if resp.DocumentableID != nil {
payload["documentable"] = captainResponseDocumentablePayload(resp)
}
return payload
}
func captainResponseAssistantPayload(resp *model.CaptainAssistantResponse) gin.H {
if resp.Assistant.ID == 0 {
return gin.H{"id": resp.AssistantID}
}
return captainAssistantPayload(&resp.Assistant)
}
func captainResponseDocumentablePayload(resp *model.CaptainAssistantResponse) gin.H {
payload := gin.H{"type": resp.DocumentableType, "id": *resp.DocumentableID}
if resp.DocumentableType == "Conversation" {
payload["display_id"] = *resp.DocumentableID
}
return payload
}