sundynix-agentix/sundynix-gateway/internal/handler/report.go

package handler

import (
	"crypto/rand"
	"encoding/hex"
	"encoding/json"
	"net/http"
	"strings"

	"github.com/gin-gonic/gin"

	"github.com/sundynix/sundynix-shared/contract"
)

// GenerateReport: POST /api/v1/reports —— 触发报告生成。
// 组装一个 intent=report 的任务发到 NATS，Dispatcher 走专用编排（规划→分章并行→渲染 docx）。
// 返回 task_id；客户端用 GET /tasks/:id/stream 看实时进度，完成后用 /reports/:id/download 取 Word。
func (h *Handler) GenerateReport(c *gin.Context) {
	var body struct {
		Topic string `json:"topic"`
		KB    string `json:"kb"`
	}
	if err := c.ShouldBindJSON(&body); err != nil || body.Topic == "" {
		c.JSON(http.StatusBadRequest, gin.H{"error": "topic required"})
		return
	}
	// 报告和普通任务一样烧钱，必须过同一道关卡（预算/计费租户/积分硬拦截）。
	// 此前这里直接 PublishTask，绕过了全部三项。
	billingTenant, ok := h.preflight(c)
	if !ok {
		return
	}
	id := newReportID()
	graph, _ := json.Marshal(map[string]any{"topic": body.Topic}) // 占位 DSL，报告编排实际读 Meta
	task := &contract.Task{
		ID:    id,
		Graph: graph,
		Meta: map[string]any{
			contract.MetaIntent:    contract.IntentReport,
			contract.MetaTopic:     body.Topic,
			contract.MetaKB:        body.KB,
			contract.MetaUserID:    userID(c),
			contract.MetaTenantID:  billingTenant, // 用量按计费租户扣，此前报告完全没记 → 漏账
			contract.MetaSessionID: sessionID(c),
		},
	}
	// launch 而非裸 PublishTask：报告也是一次「执行」，要落库（→ 进运行历史、可复盘）
	// 并开录像（→ SSE 可回放，切走再回来不丢）。
	if err := h.launch(c, task); err != nil {
		c.JSON(http.StatusBadGateway, gin.H{"error": err.Error()})
		return
	}
	c.JSON(http.StatusAccepted, gin.H{"task_id": id})
}

// ExportReport: GET /api/v1/reports/:id/export?format=docx|md —— 按需把报告源渲染为指定格式并下载。
// 生成阶段只存源；此处经 mcp-go report_export 现渲染（"导出时再处理"）。PDF 由前端打印预览生成。
func (h *Handler) ExportReport(c *gin.Context) {
	id := c.Param("id")
	if !h.requireTaskOwner(c, id) { // 报告按 task_id 寻址：仅提交者可导出
		return
	}
	format := c.DefaultQuery("format", "docx")
	res, err := h.bus.CallTool(c.Request.Context(), contract.ToolSubjectGo("report_export"),
		&contract.ToolCall{Tool: "report_export", Args: map[string]any{"task_id": id, "format": format}})
	if err != nil || res == nil || !res.OK {
		msg := "报告尚未生成或已过期"
		if res != nil && res.Error != "" {
			msg = res.Error
		}
		c.JSON(http.StatusNotFound, gin.H{"error": msg})
		return
	}
	switch format {
	case "md", "markdown":
		c.Header("Content-Disposition", `attachment; filename="`+id+`.md"`)
		c.Header("Content-Type", "text/markdown; charset=utf-8")
		c.String(http.StatusOK, res.Content)
	default: // docx：res.Content 为 minio://<key>（对象存储，跨机可取）或本地路径（单机降级）
		const docxMime = "application/vnd.openxmlformats-officedocument.wordprocessingml.document"
		c.Header("Content-Disposition", `attachment; filename="`+id+`.docx"`)
		c.Header("Content-Type", docxMime)
		if key, ok := strings.CutPrefix(res.Content, contract.BlobScheme); ok {
			// 对象存储：从 MinIO 流式取回，不依赖 gateway 与 mcp-go 共享本地盘。
			if h.blob == nil || !h.blob.Ready() {
				c.JSON(http.StatusBadGateway, gin.H{"error": "对象存储不可用，无法下载报告"})
				return
			}
			data, gerr := h.blob.GetBytes(c.Request.Context(), key)
			if gerr != nil {
				c.JSON(http.StatusNotFound, gin.H{"error": "报告产物已过期或不存在"})
				return
			}
			c.Data(http.StatusOK, docxMime, data)
			return
		}
		c.File(res.Content) // 本地路径：单机/共享卷降级
	}
}

func newReportID() string {
	var b [8]byte
	_, _ = rand.Read(b[:])
	return "report_" + hex.EncodeToString(b[:])
}