package controller

import (
	"fmt"
	"net/http"

	"gitee.com/linqwen/momo/sys/job/model"
	"gitee.com/linqwen/momo/sys/job/schema"
	"gitee.com/linqwen/momo/sys/job/service"

	"gitee.com/linqwen/momo/app"

	"github.com/gin-gonic/gin"
)

type scheduleController struct {
	app.BaseController[model.ScheduleEntity, schema.ScheduleDTO, schema.ScheduleDTO, schema.ScheduleVO]
}

func newScheduleController(s app.IBaseService[model.ScheduleEntity]) *scheduleController {
	return &scheduleController{BaseController: *app.NewBaseController[model.ScheduleEntity, schema.ScheduleDTO, schema.ScheduleDTO, schema.ScheduleVO](s)}
}

func ScheduleRouter(r *gin.RouterGroup) {
	var bc app.IBaseController = newScheduleController(service.NewScheduleService())
	c := bc.(*scheduleController)
	router := r.Group("schedule")
	{
		router.GET("/", c.PageByDTO)
		router.POST("/", c.Create)
		router.DELETE("/:id", c.Delete)
		router.PUT("/:id", c.UpdateByMap)
		router.GET("/:id", c.GetById)
		router.POST("/deletes", c.BulkDelete)
		router.POST("/run/:id", c.runTask)
		router.POST("/end/:id", c.endTask)
	}
}

var scheduleService = service.NewScheduleService()

func (bc *scheduleController) runTask(c *gin.Context) {
	id := c.Param("id")
	if err := scheduleService.RunTask(id); err != nil {
		c.JSON(http.StatusInternalServerError, gin.H{"error": fmt.Sprintf("Failed to run task: %v", err)})
		return
	}

	c.JSON(http.StatusOK, gin.H{"message": "Task executed successfully"})
}

func (bc *scheduleController) endTask(c *gin.Context) {
	id := c.Param("id")
	if err := scheduleService.EndTask(id); err != nil {
		c.JSON(http.StatusInternalServerError, gin.H{"error": fmt.Sprintf("Failed to end task: %v", err)})
		return
	}

	c.JSON(http.StatusOK, gin.H{"message": "Task end successfully"})
}