package handlers

import (
	"beacon/services"
	"github.com/gin-gonic/gin"
)

type WorkflowHandler struct {
	service *services.WorkflowService
}

func NewWorkflowHandler() *WorkflowHandler {
	return &WorkflowHandler{
		service: &services.WorkflowService{},
	}
}

// StartWorkflow POST /api/workflow/start
func (h *WorkflowHandler) StartWorkflow(c *gin.Context) {

	err := h.service.Start("13")
	if err != nil {
		c.JSON(500, gin.H{
			"code":    -1,
			"message": "Failed to start workflow",
		})
		return
	}
	c.JSON(200, gin.H{
		"message": "Workflow started successfully",
	})
}

// StopWorkflow POST /api/workflow/stop
func (h *WorkflowHandler) StopWorkflow(c *gin.Context) {
	// 停止工作流的逻辑
	// 这里可以调用 Temporal 的 API 来停止指定的工作流
	// 例如，使用 WorkflowID 或 RunID 来停止工作流
	c.JSON(200, gin.H{
		"message": "Workflow stopped successfully",
	})
}

// GetWorkflowStatus GET /api/workflow/status/{workflowID}
func (h *WorkflowHandler) GetWorkflowStatus(c *gin.Context) {
	workflowID := c.Param("workflowID")
	if workflowID == "" {
		c.JSON(400, gin.H{"error": "Workflow ID is required"})
		return
	}

	status, err := h.service.GetWorkflowStatus(workflowID)
	if err != nil {
		c.JSON(500, gin.H{"error": "Failed to get workflow status", "details": err.Error()})
		return
	}

	c.JSON(200, gin.H{"status": status})
}

// GetWorkflowResults GET /api/workflow/results/{workflowID}
func (h *WorkflowHandler) GetWorkflowResults(c *gin.Context) {
	workflowID := c.Param("workflowID")
	if workflowID == "" {
		c.JSON(400, gin.H{"error": "Workflow ID is required"})
		return
	}

	results, err := h.service.GetWorkflowResults(workflowID)
	if err != nil {
		c.JSON(500, gin.H{"error": "Failed to get workflow results", "details": err.Error()})
		return
	}

	c.JSON(200, gin.H{"results": results})
}