Reorganize repo: backend/, deploy/, docs/ layout + AGENTS.md

Restructure the monorepo into clear top-level directories: - backend/: Go module root (cmd, internal, pkg, configs, migrations, docs/swagger, scripts, tests, go.mod, Makefile, .air.toml) - deploy/: Docker (Dockerfile, docker-compose*), quickstart, fluentd - docs/: project documentation + reports/ (moved from repo root) - AGENTS.md: new AI coding-agent guide at repo root Update all references to the new layout: - Dockerfile: COPY backend/go.mod, COPY backend/ (context = repo root) - docker-compose files: context ../.., dockerfile deploy/docker/Dockerfile, env_file ../../.env, volume mounts ../../backend:/app - deploy/quickstart/compose.yaml: dockerfile deploy/docker/Dockerfile - CI: working-directory: backend for go commands, file deploy/docker/Dockerfile, coverage path backend/coverage.out, health_check backend/scripts/ - backend/Makefile: docker target uses -f ../deploy/docker/Dockerfile ../ - README: architecture tree, quickstart, config paths updated Move root stray scripts (rename_models.*, run_m11_tests.sh, verify_build.sh, gorm_bool_main.go) to backend/scripts/legacy/. All moves via git mv to preserve history. Build, vet, SQLite tests, and docker compose config verified.
2026-07-07 14:44:12 +08:00
parent d4ef996f49
commit aeddedf2a3
1348 changed files with 176 additions and 57 deletions
@@ -0,0 +1,305 @@
+package v1
+
+import (
+	"fmt"
+	"io"
+	"net/http"
+	"strconv"
+	"strings"
+
+	"github.com/gin-gonic/gin"
+	"github.com/gochat/gochat/internal/llm"
+	"github.com/gochat/gochat/internal/service"
+	applogger "github.com/gochat/gochat/pkg/logger"
+	"github.com/gochat/gochat/pkg/response"
+)
+
+// CaptainTaskHandler handles Captain standalone AI task endpoints.
+// Reference: Chatwoot enterprise/app/controllers/api/v1/accounts/captain/tasks_controller.rb
+// These endpoints provide one-shot AI tasks (reply_suggestion, summarize, rewrite)
+// independent of the CopilotThread flow.
+type CaptainTaskHandler struct {
+	svc *service.CaptainTaskService
+}
+
+// NewCaptainTaskHandler creates a new CaptainTaskHandler.
+func NewCaptainTaskHandler(svc *service.CaptainTaskService) *CaptainTaskHandler {
+	return &CaptainTaskHandler{svc: svc}
+}
+
+// ReplySuggestion generates AI-powered reply suggestions for a conversation.
+// POST /api/v1/accounts/:account_id/captain/tasks/reply_suggestion
+func (h *CaptainTaskHandler) ReplySuggestion(c *gin.Context) {
+	accountID, err := strconv.ParseUint(c.Param("id"), 10, 64)
+	if err != nil {
+		response.AbortWithStatusError(c, http.StatusBadRequest, response.ErrBadRequest, "invalid account_id")
+		return
+	}
+
+	var req service.TaskReplySuggestionRequest
+	if err := c.ShouldBindJSON(&req); err != nil {
+		response.AbortWithStatusError(c, http.StatusBadRequest, response.ErrBadRequest, "invalid request body: "+err.Error())
+		return
+	}
+
+	result, err := h.svc.ReplySuggestion(c.Request.Context(), uint(accountID), &req)
+	if err != nil {
+		applogger.L().Errorf("ReplySuggestion: %v", err)
+		renderCaptainTaskError(c, err)
+		return
+	}
+
+	renderCaptainTaskPayload(c, result.Message, result.FollowUpContext)
+}
+
+// Summarize generates a concise summary of a conversation.
+// POST /api/v1/accounts/:account_id/captain/tasks/summarize
+func (h *CaptainTaskHandler) Summarize(c *gin.Context) {
+	accountID, err := strconv.ParseUint(c.Param("id"), 10, 64)
+	if err != nil {
+		response.AbortWithStatusError(c, http.StatusBadRequest, response.ErrBadRequest, "invalid account_id")
+		return
+	}
+
+	var req service.TaskSummarizeRequest
+	if err := c.ShouldBindJSON(&req); err != nil {
+		response.AbortWithStatusError(c, http.StatusBadRequest, response.ErrBadRequest, "invalid request body: "+err.Error())
+		return
+	}
+
+	result, err := h.svc.Summarize(c.Request.Context(), uint(accountID), &req)
+	if err != nil {
+		applogger.L().Errorf("Summarize: %v", err)
+		renderCaptainTaskError(c, err)
+		return
+	}
+
+	renderCaptainTaskPayload(c, result.Message, result.FollowUpContext)
+}
+
+// Rewrite rewrites a draft message to improve tone, clarity, or language.
+// POST /api/v1/accounts/:account_id/captain/tasks/rewrite
+func (h *CaptainTaskHandler) Rewrite(c *gin.Context) {
+	accountID, err := strconv.ParseUint(c.Param("id"), 10, 64)
+	if err != nil {
+		response.AbortWithStatusError(c, http.StatusBadRequest, response.ErrBadRequest, "invalid account_id")
+		return
+	}
+
+	var req service.TaskRewriteRequest
+	if err := c.ShouldBindJSON(&req); err != nil {
+		response.AbortWithStatusError(c, http.StatusBadRequest, response.ErrBadRequest, "invalid request body: "+err.Error())
+		return
+	}
+
+	result, err := h.svc.Rewrite(c.Request.Context(), uint(accountID), &req)
+	if err != nil {
+		applogger.L().Errorf("Rewrite: %v", err)
+		renderCaptainTaskError(c, err)
+		return
+	}
+
+	renderCaptainTaskPayload(c, result.Message, result.FollowUpContext)
+}
+
+// --- SSE Streaming Endpoints (M12) ---
+// These endpoints stream Captain task results via Server-Sent Events,
+// following the same SSE pattern as SSEStreamHandler.StreamCopilotMessage.
+
+// StreamReplySuggestion streams AI-powered reply suggestions via SSE.
+// POST /api/v1/accounts/:id/captain/tasks/reply_suggestion/stream
+func (h *CaptainTaskHandler) StreamReplySuggestion(c *gin.Context) {
+	accountID, err := strconv.ParseUint(c.Param("id"), 10, 64)
+	if err != nil {
+		captainWriteSSEError(c, http.StatusBadRequest, "invalid account_id")
+		return
+	}
+
+	var req service.TaskReplySuggestionRequest
+	if err := c.ShouldBindJSON(&req); err != nil {
+		captainWriteSSEError(c, http.StatusBadRequest, "invalid request body: "+err.Error())
+		return
+	}
+
+	captainSetSSEHeaders(c)
+
+	err = h.svc.ReplySuggestionStream(c.Request.Context(), uint(accountID), &req, func(chunk llm.StreamChunk) error {
+		if len(chunk.Choices) > 0 {
+			content := chunk.Choices[0].Delta.Content
+			if content != "" {
+				escaped := captainEscapeJSONString(content)
+				captainWriteSSEMessage(c, "message", fmt.Sprintf(`{"content": "%s", "done": false}`, escaped))
+				c.Writer.Flush()
+			}
+
+			finishReason := chunk.Choices[0].FinishReason
+			if finishReason != "" && finishReason != "null" {
+				captainWriteSSEMessage(c, "done", `{"done": true, "content": ""}`)
+				c.Writer.Flush()
+				return io.EOF
+			}
+		}
+		return nil
+	})
+
+	if err != nil && err != io.EOF {
+		applogger.L().Errorf("StreamReplySuggestion: %v", err)
+		captainWriteSSETaskError(c, err)
+	}
+}
+
+// StreamSummarize streams a conversation summary via SSE.
+// POST /api/v1/accounts/:id/captain/tasks/summarize/stream
+func (h *CaptainTaskHandler) StreamSummarize(c *gin.Context) {
+	accountID, err := strconv.ParseUint(c.Param("id"), 10, 64)
+	if err != nil {
+		captainWriteSSEError(c, http.StatusBadRequest, "invalid account_id")
+		return
+	}
+
+	var req service.TaskSummarizeRequest
+	if err := c.ShouldBindJSON(&req); err != nil {
+		captainWriteSSEError(c, http.StatusBadRequest, "invalid request body: "+err.Error())
+		return
+	}
+
+	captainSetSSEHeaders(c)
+
+	err = h.svc.SummarizeStream(c.Request.Context(), uint(accountID), &req, func(chunk llm.StreamChunk) error {
+		if len(chunk.Choices) > 0 {
+			content := chunk.Choices[0].Delta.Content
+			if content != "" {
+				escaped := captainEscapeJSONString(content)
+				captainWriteSSEMessage(c, "message", fmt.Sprintf(`{"content": "%s", "done": false}`, escaped))
+				c.Writer.Flush()
+			}
+
+			finishReason := chunk.Choices[0].FinishReason
+			if finishReason != "" && finishReason != "null" {
+				captainWriteSSEMessage(c, "done", `{"done": true, "content": ""}`)
+				c.Writer.Flush()
+				return io.EOF
+			}
+		}
+		return nil
+	})
+
+	if err != nil && err != io.EOF {
+		applogger.L().Errorf("StreamSummarize: %v", err)
+		captainWriteSSETaskError(c, err)
+	}
+}
+
+// StreamRewrite streams a rewritten message via SSE.
+// POST /api/v1/accounts/:id/captain/tasks/rewrite/stream
+func (h *CaptainTaskHandler) StreamRewrite(c *gin.Context) {
+	accountID, err := strconv.ParseUint(c.Param("id"), 10, 64)
+	if err != nil {
+		captainWriteSSEError(c, http.StatusBadRequest, "invalid account_id")
+		return
+	}
+
+	var req service.TaskRewriteRequest
+	if err := c.ShouldBindJSON(&req); err != nil {
+		captainWriteSSEError(c, http.StatusBadRequest, "invalid request body: "+err.Error())
+		return
+	}
+
+	captainSetSSEHeaders(c)
+
+	err = h.svc.RewriteStream(c.Request.Context(), uint(accountID), &req, func(chunk llm.StreamChunk) error {
+		if len(chunk.Choices) > 0 {
+			content := chunk.Choices[0].Delta.Content
+			if content != "" {
+				escaped := captainEscapeJSONString(content)
+				captainWriteSSEMessage(c, "message", fmt.Sprintf(`{"content": "%s", "done": false}`, escaped))
+				c.Writer.Flush()
+			}
+
+			finishReason := chunk.Choices[0].FinishReason
+			if finishReason != "" && finishReason != "null" {
+				captainWriteSSEMessage(c, "done", `{"done": true, "content": ""}`)
+				c.Writer.Flush()
+				return io.EOF
+			}
+		}
+		return nil
+	})
+
+	if err != nil && err != io.EOF {
+		applogger.L().Errorf("StreamRewrite: %v", err)
+		captainWriteSSETaskError(c, err)
+	}
+}
+
+// --- Captain SSE Helper Functions ---
+// These are Captain-specific SSE helpers to avoid naming conflicts
+// with SSEStreamHandler helpers in the same v1 package.
+
+// captainSetSSEHeaders sets standard SSE response headers.
+func captainSetSSEHeaders(c *gin.Context) {
+	c.Header("Content-Type", "text/event-stream")
+	c.Header("Cache-Control", "no-cache")
+	c.Header("Connection", "keep-alive")
+	c.Header("X-Accel-Buffering", "no")
+}
+
+// captainWriteSSEMessage writes a single SSE event to the Gin response writer.
+func captainWriteSSEMessage(c *gin.Context, event string, data string) {
+	c.Writer.WriteString(fmt.Sprintf("event: %s\ndata: %s\n\n", event, data))
+}
+
+// captainWriteSSEError writes an error SSE event followed by a done event.
+func captainWriteSSEError(c *gin.Context, status int, errMsg string) {
+	captainSetSSEHeaders(c)
+	captainWriteSSEMessage(c, "error", fmt.Sprintf(`{"error": "%s", "status": %d, "done": true}`, captainEscapeJSONString(errMsg), status))
+	captainWriteSSEMessage(c, "done", `{"done": true}`)
+}
+
+func captainWriteSSETaskError(c *gin.Context, err error) {
+	status, message, ok := service.CaptainTaskErrorStatus(err)
+	if !ok {
+		status = http.StatusUnprocessableEntity
+		message = err.Error()
+	}
+	captainWriteSSEError(c, status, message)
+}
+
+// captainEscapeJSONString escapes special characters for safe JSON embedding.
+func captainEscapeJSONString(s string) string {
+	var result strings.Builder
+	for _, ch := range s {
+		switch ch {
+		case '"':
+			result.WriteString(`\"`)
+		case '\\':
+			result.WriteString(`\\`)
+		case '\n':
+			result.WriteString(`\n`)
+		case '\r':
+			result.WriteString(`\r`)
+		case '\t':
+			result.WriteString(`\t`)
+		default:
+			result.WriteRune(ch)
+		}
+	}
+	return result.String()
+}
+
+func renderCaptainTaskPayload(c *gin.Context, message string, followUpContext map[string]interface{}) {
+	payload := gin.H{"message": message}
+	if followUpContext != nil {
+		payload["follow_up_context"] = followUpContext
+	}
+	c.JSON(http.StatusOK, payload)
+}
+
+func renderCaptainTaskError(c *gin.Context, err error) {
+	status, message, ok := service.CaptainTaskErrorStatus(err)
+	if !ok {
+		status = http.StatusUnprocessableEntity
+		message = err.Error()
+	}
+	c.JSON(status, gin.H{"error": message})
+}